feat: DIA-1886: Add o3-mini OpenAI model (#325)

Co-authored-by: hakan458 <[email protected]>
HumanSignal · Feb 3, 2025 · 8acdba7 · 8acdba7
1 parent c07891e
commit 8acdba7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/adala/runtimes/_litellm.py b/adala/runtimes/_litellm.py
@@ -261,8 +261,8 @@ class LiteLLMChatRuntime(InstructorClientMixin, Runtime):
     """
 
     model: str = "gpt-4o-mini"
-    max_tokens: int = 1000
-    temperature: float = 0.0
+    max_tokens: Optional[int] = 1000
+    temperature: Optional[float] = 0.0
     seed: Optional[int] = 47
 
     model_config = ConfigDict(extra="allow")
@@ -406,8 +406,8 @@ class AsyncLiteLLMChatRuntime(InstructorAsyncClientMixin, AsyncRuntime):
     """
 
     model: str = "gpt-4o-mini"
-    max_tokens: int = 1000
-    temperature: float = 0.0
+    max_tokens: Optional[int] = 1000
+    temperature: Optional[float] = 0.0
     seed: Optional[int] = 47
 
     model_config = ConfigDict(extra="allow")