From 1d881a4622a2a2b95d3c06c1ddc2eb5b96baacf1 Mon Sep 17 00:00:00 2001
From: Eric Ma <ericmajinglong@gmail.com>
Date: Mon, 6 Nov 2023 23:51:31 -0500
Subject: [PATCH] fix(llamabot): update model names and limit openai version

- Limited the version of openai dependency to <=0.28.1 in pyproject.toml

This is a temporary change. See LangChain: https://github.com/langchain-ai/langchain/issues/12956 for more context.
tl;dr: OpenAI's update breaks a lot of LangChain.
---
 llamabot/bot/ollama_model_names.txt | 2 +-
 pyproject.toml                      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/llamabot/bot/ollama_model_names.txt b/llamabot/bot/ollama_model_names.txt
index ac811f349..4e61ed5cb 100644
--- a/llamabot/bot/ollama_model_names.txt
+++ b/llamabot/bot/ollama_model_names.txt
@@ -27,9 +27,9 @@ wizardlm
 sqlcoder
 dolphin2.1-mistral
 dolphin2.2-mistral
+yarn-mistral
 codebooga
 nexusraven
-yarn-mistral
 mistrallite
 openhermes2.5-mistral
 yarn-llama2
diff --git a/pyproject.toml b/pyproject.toml
index f16ea3a92..838ae27ee 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -64,7 +64,7 @@ namespaces = false
 name = "llamabot"
 version = "0.0.87"
 dependencies = [
-    "openai",
+    "openai<=0.28.1",
     "langchain",
     "llama_index",
     "panel>=1.1.0",