diff --git a/querybook/server/lib/ai_assistant/assistants/openai_assistant.py b/querybook/server/lib/ai_assistant/assistants/openai_assistant.py
index 6d9369768..54c6e47f7 100644
--- a/querybook/server/lib/ai_assistant/assistants/openai_assistant.py
+++ b/querybook/server/lib/ai_assistant/assistants/openai_assistant.py
@@ -14,6 +14,7 @@
     "gpt-4": 8192,
     "gpt-4-32k": 32768,
     "gpt-4-turbo": 128000,
+    "gpt-4o": 128000,
 }
 DEFAULT_MODEL_NAME = "gpt-3.5-turbo"
 
diff --git a/requirements/ai/langchain.txt b/requirements/ai/langchain.txt
index 79045f2a6..f5f23a27a 100644
--- a/requirements/ai/langchain.txt
+++ b/requirements/ai/langchain.txt
@@ -1,3 +1,3 @@
-langchain==0.1.6
-langchain-openai==0.0.5
+langchain==0.1.20
+langchain-openai==0.1.6
 opensearch-py==2.3.0