upgrade to gpt4o and mini

app.py

@@ -34,13 +34,13 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 IS_LOAD_FROM_VECTOR_STORE = True 
 VDB_COLLECTION_NAME = "demo-v6"
-MODEL_NAME = ChatbotVersion.CHATGPT_4.value
+MODEL_NAME = ChatbotVersion.CHATGPT_4O.value
 
 CHUNK_SIZE = 8191
 LLM, EMBED_MODEL = get_service_provider_config(
     service_provider=ServiceProvider.OPENAI, model_name=MODEL_NAME)
 
-LLM = Ollama(model="llama3.1:latest", request_timeout=60.0, context_window=10000)
+# LLM = Ollama(model="llama3.1:latest", request_timeout=60.0, context_window=10000)
 
 Settings.embed_model = EMBED_MODEL
 Settings.llm = LLM

requirements.txt

@@ -1,12 +1,15 @@
-gradio==4.7.1
-llama_index==0.9.6.post2
+gradio==4.36.1
+llama-index==0.10.58
 nltk==3.8.1
-openai==1.3.5
-phoenix==0.9.1
-python-dotenv==1.0.0
-qdrant_client==1.9.1
-unstructured==0.11.0
-arize-phoenix==1.2.1
-numpy
-pytest
-# unstructure io
+openai==1.37.0
+python-dotenv==1.0.1
+qdrant-client==1.10.1
+arize-phoenix==4.14.1
+fastapi>=0.80
+llama-index-vector-stores-qdrant==0.2.14
+llama-index-agent-openai==0.2.9
+llama-index-embeddings-azure-openai==0.1.11
+llama-index-embeddings-openai==0.1.11
+llama-index-llms-openai==0.1.27
+llama-index-callbacks-arize-phoenix==0.1.6
+llama-index-llms-ollama==0.2.0

schemas.py

@@ -5,6 +5,8 @@ class ChatbotVersion(str, Enum):
     # make a enum of chatbot type and string
     CHATGPT_35 = "gpt-3.5-turbo-1106"
     CHATGPT_4 = "gpt-4-1106-preview"
+    CHATGPT_4O = "gpt-4o"
+    CHATGPT_4O_MINI = "gpt-4o-mini"
 
 
 class ServiceProvider(str, Enum):