Update utils/database.py

utils/database.py

@@ -107,21 +107,8 @@ def format_chat_history(messages: list[BaseMessage]) -> list[dict]:
             formatted.append({"role": "system", "content": msg.content})
     return formatted
 
-# utils/database.py
-from langchain.memory import ConversationBufferWindowMemory
-from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
-from langchain.chains import ConversationalRetrievalChain
-from langchain.chat_models import ChatOpenAI
-from langchain.agents import AgentExecutor, Tool, create_openai_tools_agent
-from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
-from langchain.agents.format_scratchpad.tools import format_to_tool_messages
-from langchain.agents.output_parsers.openai_tools import OpenAIToolsAgentOutputParser
-from langchain_core.runnables import RunnablePassthrough
-import os
-import streamlit as st
-
 def initialize_qa_system(vector_store):
-    """Initialize QA system with proper chat handling"""
+
     try:
         llm = ChatOpenAI(
             temperature=0.5,
@@ -156,7 +143,7 @@ def initialize_qa_system(vector_store):
                 "agent_scratchpad": lambda x: format_to_tool_messages(x["intermediate_steps"])
             }
             | prompt
-            | llm.bind(stop=["\nHuman:"])
+            | llm.bind(stop=["\\nHuman:"])
             | OpenAIToolsAgentOutputParser()
         )
 
@@ -183,7 +170,7 @@ def initialize_qa_system(vector_store):
         return None
         
 def initialize_faiss(embeddings, documents, document_names):
-    """Initialize FAISS vector store"""
+
     try:
         from langchain.vectorstores import FAISS
 
@@ -200,7 +187,7 @@ def initialize_faiss(embeddings, documents, document_names):
 
 @st.cache_resource
 def get_embeddings_model():
-    """Get the embeddings model"""
+    
     try:
         from langchain.embeddings import HuggingFaceEmbeddings
 
@@ -209,4 +196,11 @@ def get_embeddings_model():
         return embeddings
     except Exception as e:
         st.error(f"Error loading embeddings model: {e}")
-        return None
+        return None
+
+# Writing the updated contents back to the original files
+with open(components_chat_path, 'w') as chat_file:
+    chat_file.write(updated_components_chat_content)
+
+with open(utils_database_path, 'w') as database_file:
+    database_file.write(updated_utils_database_content)