From d47cb9b87effe0814813167e8a1d4e17d83755ce Mon Sep 17 00:00:00 2001
From: hs_junxiang <jimmy871117@gmail.com>
Date: Wed, 23 Aug 2023 15:27:47 +0800
Subject: [PATCH] Fix:
 https://discuss.streamlit.io/t/issues-with-chroma-and-sqlite/47950. Chroma to
 FAISS

---
 app.py           | 12 ++++++------
 docGPT/docGPT.py |  4 ++--
 requirements.txt |  6 +++---
 3 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/app.py b/app.py
index 32e5771..c8822b7 100644
--- a/app.py
+++ b/app.py
@@ -8,6 +8,7 @@
 import langchain
 import streamlit as st
 from langchain.cache import InMemoryCache
+from streamlit import logger
 from streamlit_chat import message
 
 from agent import AgentHelper
@@ -22,11 +23,11 @@
 
 st.session_state.openai_api_key = None
 st.session_state.serpapi_api_key = None
+app_logger = logger.get_logger(__name__)
 
 
 def theme():
     st.set_page_config(page_title="DocGPT")
-
     icon, title = st.columns([3, 20])
     with icon:
         st.image('./img/chatbot.png')
@@ -39,7 +40,6 @@ def theme():
 
 def load_api_key() -> None:
     with st.sidebar:
-
         if st.session_state.openai_api_key:
             OPENAI_API_KEY = st.session_state.openai_api_key
             st.sidebar.success('API key loaded form previous input')
@@ -101,12 +101,12 @@ def load_api_key() -> None:
             llm_tool = agent_.create_llm_chain()
 
         except Exception as e:
-            print(e)
+            app_logger.info(e)
 
         try:
             search_tool = agent_.get_searp_chain
         except Exception as e:
-            print(e)
+            app_logger.info(e)
 
         try:
             tools = [
@@ -117,7 +117,7 @@ def load_api_key() -> None:
             ]
             agent_.initialize(tools)
         except Exception as e:
-            pass
+            app_logger.info(e)
 
 
 if not st.session_state['openai_api_key']:
@@ -142,7 +142,7 @@ def get_response(query: str):
             response = agent_.query(query)
             return response
     except Exception as e:
-        print(e)
+        app_logger.info(e)
 
 query = st.text_input(
     "#### Question:",
diff --git a/docGPT/docGPT.py b/docGPT/docGPT.py
index d1f0e27..220d8b2 100644
--- a/docGPT/docGPT.py
+++ b/docGPT/docGPT.py
@@ -8,7 +8,7 @@
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.memory import ConversationBufferMemory
 from langchain.prompts import PromptTemplate
-from langchain.vectorstores import Chroma
+from langchain.vectorstores import FAISS
 
 openai.api_key = os.getenv('OPENAI_API_KEY')
 
@@ -116,7 +116,7 @@ def _helper_prompt(self, chain_type: str) -> None:
     def _embeddings(self):
         embeddings = OpenAIEmbeddings()
 
-        db = Chroma.from_documents(
+        db = FAISS.from_documents(
             documents=self.docs,
             embedding= embeddings
         )
diff --git a/requirements.txt b/requirements.txt
index 25b9311..3950e28 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,8 +1,8 @@
-langchain==0.0.228
+langchain==0.0.218
 openai==0.27.8
-streamlit==1.24.1
+streamlit==1.22.0
 streamlit_chat==0.1.1
 pymupdf==1.22.5
-chromadb==0.3.26
+faiss-cpu==1.7.4
 tiktoken==0.4.0
 google-search-results==2.4.2