update type of temperature in ModelConfig

jeffkayne · jeffkayne · commit 7baf93c134da · 2023-09-01T15:57:31.000+02:00
diff --git a/trubrics/platform/prompts.py b/trubrics/platform/prompts.py
@@ -7,7 +7,7 @@
 class ModelConfig(BaseModel):
     model: str
     prompt_template: str = "{prompt}"
-    temperature: Optional[str] = None
+    temperature: Optional[float] = None
 
 
 class Prompt(BaseModel):
