Update chat.py

chat.py

@@ -18,7 +18,7 @@ class ChatHandler:
     def __init__(self, vector_db_path, open_api_key, grok_api_key,db_final):
         self.vector_db_path = vector_db_path
         self.openai_embeddings = OpenAIEmbeddings(api_key=open_api_key)
-        self.llm_openai = ChatOpenAI(model_name="gpt-4", api_key=open_api_key, max_tokens=500, temperature=0.2)
+        self.llm_openai = ChatOpenAI(model_name="gpt-4o-mini", api_key=open_api_key, max_tokens=500, temperature=0.2)
         self.grok_api_key = grok_api_key
         self.openai_api_key = open_api_key
         self.sql_db = db_final
@@ -105,8 +105,8 @@ class ChatHandler:
                 template=template_query_generation
             )
 
-            # Initialize the language modelgpt-4
-            llm = ChatOpenAI(model_name="gpt-4", api_key=self.openai_api_key, max_tokens=500, temperature=0.2)
+            # Initialize the language modelgpt-4o-mini
+            llm = ChatOpenAI(model_name="gpt-4o-mini", api_key=self.openai_api_key, max_tokens=500, temperature=0.2)
 
             # Create the runnable sequence
             chain = prompt | llm | StrOutputParser()