Update models.py for OpenRouter models

models.py

@@ -3,10 +3,13 @@ from typing import Optional
 from enum import Enum
 
 
-class ZaiModel(str, Enum):
-    GLM_5 = "glm-5"
-    GLM_47 = "glm-4.7"
-    GLM_45_AIR = "glm-4.5-air"
+class OpenRouterModel(str, Enum):
+    CLAUDE_35_SONNET = "claude-3.5-sonnet"
+    CLAUDE_3_OPUS = "claude-3-opus"
+    GPT_4O = "gpt-4o"
+    GPT_4O_MINI = "gpt-4o-mini"
+    LLAMA_31_70B = "llama-3.1-70b"
+    LLAMA_31_8B = "llama-3.1-8b"
 
 
 class SamplerType(str, Enum):
@@ -25,14 +28,14 @@ class ColumnConfig(BaseModel):
 
 class GenerateRequest(BaseModel):
     num_records: int = Field(default=10, ge=1, le=1000, description="Number of records to generate")
-    model: ZaiModel = Field(default=ZaiModel.GLM_47, description="z.ai model to use")
+    model: OpenRouterModel = Field(default=OpenRouterModel.GPT_4O_MINI, description="OpenRouter model to use")
     columns: list[ColumnConfig] = Field(..., description="Column configurations")
     temperature: float = Field(default=0.7, ge=0.0, le=2.0)
     max_tokens: int = Field(default=512, ge=64, le=4096)
 
 
 class PreviewRequest(BaseModel):
-    model: ZaiModel = Field(default=ZaiModel.GLM_47)
+    model: OpenRouterModel = Field(default=OpenRouterModel.GPT_4O_MINI)
     columns: list[ColumnConfig] = Field(..., description="Column configurations")
     temperature: float = Field(default=0.7)
     max_tokens: int = Field(default=512)