fix logic

Author: suzinyou

core_backend/app/llm_call/llm_prompts.py

@@ -7,7 +7,7 @@
 from enum import Enum
 from typing import ClassVar, Literal
 
-from pydantic import BaseModel, ConfigDict, Field, field_validator
+from pydantic import BaseModel, ConfigDict, Field
 
 from .utils import format_prompt, remove_json_markdown
 
@@ -428,22 +428,8 @@ def _missing_(cls, value: str) -> IdentifiedScript:  # type: ignore[override]
 class LanguageIdentificationResponse(BaseModel):
     """Pydantic model for the language identification response."""
 
-    language: str
-    script: str
-
-    @field_validator("language")
-    def validate_language(cls, value: str) -> str:
-        """Make sure language input is a valid IdentifiedLanguage"""
-        if value not in IdentifiedLanguage._member_names_:
-            raise ValueError(f"Invalid language: {value}")
-        return value
-
-    @field_validator("script")
-    def validate_script(cls, value: str) -> str:
-        """Make sure script input is a valid IdentifiedScript"""
-        if value not in IdentifiedScript._member_names_:
-            raise ValueError(f"Invalid script: {value}")
-        return value
+    language: IdentifiedLanguage
+    script: IdentifiedScript
 
     model_config = ConfigDict(strict=True)
 

core_backend/app/llm_call/process_input.py

@@ -167,12 +167,13 @@ def _process_identified_language_response(
 
     supported_languages_list = IdentifiedLanguage.get_supported_languages()
     supported_scripts_list = IdentifiedScript.get_supported_scripts()
-    supported_languages_str = ", ".join(supported_languages_list)
-    suported_scripts_str = ", ".join(supported_scripts_list)
 
     language_ok = identified_language in supported_languages_list
     script_ok = identified_script in supported_scripts_list
 
+    supported_languages_str = ", ".join(supported_languages_list)
+    suported_scripts_str = ", ".join(supported_scripts_list)
+
     if language_ok and script_ok:
         return response
     elif language_ok and not script_ok:
@@ -209,7 +210,7 @@ def _process_identified_language_response(
 
     logger.info(
         f"LANGUAGE IDENTIFICATION FAILED due to {error_message} "
-        f"language on query id: {str(response.query_id)}"
+        f"on query id: {str(response.query_id)}"
     )
 
     return error_response