From 18ba72a722a128f2458f617d2f83ca92963b7a09 Mon Sep 17 00:00:00 2001
From: Brendan Quinn <brendan-quinn@northwestern.edu>
Date: Fri, 25 Aug 2023 14:23:32 +0000
Subject: [PATCH] Update prototype to GPT-4

---
 python/src/setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/src/setup.py b/python/src/setup.py
index 4ca47399..8869f4d6 100644
--- a/python/src/setup.py
+++ b/python/src/setup.py
@@ -8,11 +8,13 @@ def openai_chat_client():
   deployment = os.getenv("AZURE_OPENAI_LLM_DEPLOYMENT_ID")
   key = os.getenv("AZURE_OPENAI_API_KEY")
   resource = os.getenv("AZURE_OPENAI_RESOURCE_NAME")
+  version = "2023-07-01-preview"
 
   return AzureChatOpenAI(deployment_name=deployment, 
                          openai_api_key=key, 
                          openai_api_base=f"https://{resource}.openai.azure.com/",
-                         openai_api_version="2023-03-15-preview")
+                         openai_api_version=version)
+                         
 
 
 def weaviate_vector_store(index_name: str, text_key: str, attributes: List[str] = []):