Update utils/database.py

utils/database.py

@@ -1,6 +1,6 @@
 # utils/database.py
 # Update the imports first
-#from langchain_community.chat_models import ChatOpenAI
+from langchain_community.chat_models import ChatOpenAI
 from langchain_core.messages import (
     HumanMessage,
     AIMessage,
@@ -10,8 +10,7 @@ from langchain_core.messages import (
 from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.runnables import RunnablePassthrough
 from langchain.chains import ConversationalRetrievalChain
-#from langchain.chat_models import ChatOpenAI
-from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain.chat_models import ChatOpenAI
 from langchain.agents import AgentExecutor, Tool, create_openai_tools_agent
 from langchain.agents.format_scratchpad.tools import format_to_tool_messages
 from langchain.agents.output_parsers.openai_tools import OpenAIToolsAgentOutputParser
@@ -336,11 +335,12 @@ def display_vector_store_info():
 def initialize_qa_system(vector_store):
     """Initialize QA system with proper chat handling."""
     try:
-        llm = ChatGoogleGenerativeAI(
+        llm = ChatOpenAI(
             temperature=0.5,
-            model="gemini-1.5-pro", 
-            api_key=os.environ.get("GEMINI_API_KEY")
+            model_name="gpt-4",
+            api_key=os.environ.get("OPENAI_API_KEY")
         )
+
         # Create retriever function
         retriever = vector_store.as_retriever(search_kwargs={"k": 2})