Update models.py

models.py

@@ -8,6 +8,7 @@ from llama_index.core import Settings
 from groq import Groq
 from mem0 import MemoryClient
 from config import config
+from langchain_openai import ChatOpenAI
 
 # Initialize embedding function for Chroma
 embedding_function = chromadb.utils.embedding_functions.OpenAIEmbeddingFunction(
@@ -24,7 +25,7 @@ embedding_model = OpenAIEmbeddings(
 )
 
 # Initialize Chat OpenAI model
-llm = AzureChatOpenAI(
+llm =ChatOpenAI(
     openai_api_base=config.OPENAI_API_BASE,
     openai_api_key=config.API_KEY,
     model=config.CHAT_MODEL,