From a49eb9b26297e8be97fef67c1dce5cf7e01de319 Mon Sep 17 00:00:00 2001
From: "J.C. Zhong" <jianchengzhong@pinterest.com>
Date: Tue, 14 May 2024 14:27:23 -0700
Subject: [PATCH] chore: add openai new model gpt-4o (#1451)

* chore: add openai new model gpt-4o

* upgrade langchain
---
 .../server/lib/ai_assistant/assistants/openai_assistant.py    | 1 +
 requirements/ai/langchain.txt                                 | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

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