Update utils/database.py

utils/database.py

@@ -10,7 +10,8 @@ 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.chat_models import ChatOpenAI
+from langchain.chat_models import ChatGemini  # Import ChatGemini
 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
@@ -335,12 +336,11 @@ def display_vector_store_info():
 def initialize_qa_system(vector_store):
     """Initialize QA system with proper chat handling."""
     try:
-        llm = ChatOpenAI(
+        llm = ChatGemini(
             temperature=0.5,
-            model_name="gemini-1.5-pro",
+            model="gemini-1.5-pro", 
             api_key=os.environ.get("GEMINI_API_KEY")
-        )
-
+            
         # Create retriever function
         retriever = vector_store.as_retriever(search_kwargs={"k": 2})