Upload models.py with huggingface_hub

models.py

@@ -4,13 +4,9 @@ from enum import Enum
 
 
 class OpenRouterModel(str, Enum):
-    ARCEE_PRIME = "arcee-ai/prime"
-
-
-class SamplerType(str, Enum):
-    CATEGORY = "CATEGORY"
-    UNIFORM = "UNIFORM"
-    GAUSSIAN = "GAUSSIAN"
+    GLM_5 = "z-ai/glm-5"
+    GPT_4O_MINI = "openai/gpt-4o-mini"
+    CLAUDE_SONNET = "anthropic/claude-sonnet-4.6"
 
 
 class ColumnConfig(BaseModel):
@@ -21,17 +17,17 @@ class ColumnConfig(BaseModel):
 
 class GenerateRequest(BaseModel):
     num_records: int = Field(default=10, ge=1, le=100)
-    model: OpenRouterModel = Field(default=OpenRouterModel.ARCEE_PRIME)
-    columns: list[ColumnConfig] = Field(..., description="Column configurations")
+    model: OpenRouterModel = Field(default=OpenRouterModel.GLM_5)
+    columns: list[ColumnConfig] = Field(...)
     temperature: float = Field(default=0.7, ge=0.0, le=2.0)
-    max_tokens: int = Field(default=512, ge=64, le=4096)
+    max_tokens: int = Field(default=256, ge=64, le=4096)
 
 
 class PreviewRequest(BaseModel):
-    model: OpenRouterModel = Field(default=OpenRouterModel.ARCEE_PRIME)
+    model: OpenRouterModel = Field(default=OpenRouterModel.GLM_5)
     columns: list[ColumnConfig] = Field(...)
     temperature: float = Field(default=0.7)
-    max_tokens: int = Field(default=512)
+    max_tokens: int = Field(default=256)
 
 
 class GenerateResponse(BaseModel):