Upload models.py

Type safety and enum-like behavior in Python 3.10

models.py

@@ -1,18 +1,27 @@
-from enum import StrEnum
+# from enum import StrEnum
+from enum import Enum
 from pydantic import BaseModel, ConfigDict
 
 
-class GoogleModelID(StrEnum):
+class GoogleModelID(Enum):
   GEMINI_2_0_FLASH = "gemini-2.0-flash"
   GEMINI_2_5_FLASH_PREVIEW = "gemini-2.5-flash-preview"
+  
+  # Type safety and enum-like behavior in Python 3.10
+  def __str__(self):
+    return self.value
 
-class OpenRouterModelID(StrEnum):
+class OpenRouterModelID(Enum):
   QWEN_3_14B_FREE = "openrouter/qwen/qwen3-14b:free"
   GPT_4_1_MINI = "openrouter/openai/gpt-4.1-mini"
   GPT_O4_MINI = "openrouter/openai/o4-mini"
   GROK_3_MINI_BETA = "openrouter/x-ai/grok-3-mini-beta"
   GROK_3_BETA = "openrouter/x-ai/grok-3-beta"
   
+  # Type safety and enum-like behavior in Python 3.10
+  def __str__(self):
+    return self.value
+  
 class Question(BaseModel):
     model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
     task_id: str