From 4cfa1984ec85ea1a10b0943ef2ee3bdff597be78 Mon Sep 17 00:00:00 2001
From: pablonyx <pablo@danswer.ai>
Date: Thu, 17 Apr 2025 15:08:04 -0700
Subject: [PATCH] add o3 + o4 mini

---
 backend/onyx/llm/llm_provider_options.py | 2 ++
 backend/requirements/default.txt         | 2 +-
 backend/requirements/model_server.txt    | 2 +-
 web/src/lib/hooks.ts                     | 4 ++++
 4 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/backend/onyx/llm/llm_provider_options.py b/backend/onyx/llm/llm_provider_options.py
index 0dd66a289f..ae3f8a7fa7 100644
--- a/backend/onyx/llm/llm_provider_options.py
+++ b/backend/onyx/llm/llm_provider_options.py
@@ -41,8 +41,10 @@ class WellKnownLLMProviderDescriptor(BaseModel):
 
 OPENAI_PROVIDER_NAME = "openai"
 OPEN_AI_MODEL_NAMES = [
+    "o4-mini",
     "o3-mini",
     "o1-mini",
+    "o3",
     "o1",
     "gpt-4",
     "gpt-4.1",
diff --git a/backend/requirements/default.txt b/backend/requirements/default.txt
index 780a0839c3..1e1ccc678d 100644
--- a/backend/requirements/default.txt
+++ b/backend/requirements/default.txt
@@ -39,7 +39,7 @@ langchainhub==0.1.21
 langgraph==0.2.72
 langgraph-checkpoint==2.0.13
 langgraph-sdk==0.1.44
-litellm==1.66.1
+litellm==1.66.3
 lxml==5.3.0
 lxml_html_clean==0.2.2
 llama-index==0.12.28
diff --git a/backend/requirements/model_server.txt b/backend/requirements/model_server.txt
index e9390f3ba6..8bb9aa1e11 100644
--- a/backend/requirements/model_server.txt
+++ b/backend/requirements/model_server.txt
@@ -14,7 +14,7 @@ torch==2.6.0
 transformers==4.49.0
 uvicorn==0.21.1
 voyageai==0.2.3
-litellm==1.66.1
+litellm==1.66.3
 sentry-sdk[fastapi,celery,starlette]==2.14.0
 aioboto3==14.0.0
 prometheus_fastapi_instrumentator==7.1.0
diff --git a/web/src/lib/hooks.ts b/web/src/lib/hooks.ts
index c2974d11d4..c9a6ff7d03 100644
--- a/web/src/lib/hooks.ts
+++ b/web/src/lib/hooks.ts
@@ -676,6 +676,8 @@ const MODEL_DISPLAY_NAMES: { [key: string]: string } = {
   "gpt-4.1": "GPT 4.1",
   "gpt-4": "GPT 4",
   "gpt-4o": "GPT 4o",
+  "o4-mini": "o4 Mini",
+  o3: "o3",
   "gpt-4o-2024-08-06": "GPT 4o (Structured Outputs)",
   "gpt-4o-mini": "GPT 4o Mini",
   "gpt-4-0314": "GPT 4 (March 2023)",
@@ -822,6 +824,8 @@ export const defaultModelsByProvider: { [name: string]: string[] } = {
     "o3-mini",
     "o1-mini",
     "o1",
+    "o4-mini",
+    "o3",
   ],
   bedrock: [
     "meta.llama3-1-70b-instruct-v1:0",