From 8acdba7960f48f9b23f2accd0b62fe94e5bf9bd6 Mon Sep 17 00:00:00 2001 From: Hakan Erol Date: Mon, 3 Feb 2025 07:05:48 -0800 Subject: [PATCH] feat: DIA-1886: Add o3-mini OpenAI model (#325) Co-authored-by: hakan458 --- adala/runtimes/_litellm.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/adala/runtimes/_litellm.py b/adala/runtimes/_litellm.py index 3f398e2b..0defec7d 100644 --- 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")