Update conversation.py

conversation.py

@@ -4,21 +4,28 @@ from langchain.vectorstores import Pinecone
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.memory import ConversationBufferMemory
 import pinecone
+import os
+from langchain.vectorstores import Chroma
+from dotenv import load_dotenv
+load_dotenv()
 
+openai_api_key=os.getenv('OPENAI_API_KEY')
 
-def create_conversation(query: str, chat_history: list, pinecone_api_key: str, pinecone_environment: str, pinecone_index_name: str, openai_api_key: str) -> tuple:
+def create_conversation(query: str, chat_history: list, collection_name: str) -> tuple:
     try:
-        pinecone.init(
-            api_key=pinecone_api_key,
-            environment=pinecone_environment,
-        )
+
         embeddings = OpenAIEmbeddings(
             openai_api_key=openai_api_key
         )
-        db = Pinecone.from_existing_index(
-            index_name=pinecone_index_name,
-            embedding=embeddings
+        
+        persist_directory = './db_metadata'
+        
+        db = Chroma(
+            collection_name=collection_name,
+            persist_directory=persist_directory,
+            embedding_function=embeddings
         )
+        
         memory = ConversationBufferMemory(
             memory_key='chat_history',
             return_messages=False
@@ -26,13 +33,22 @@ def create_conversation(query: str, chat_history: list, pinecone_api_key: str, p
         cqa = ConversationalRetrievalChain.from_llm(
             llm=ChatOpenAI(temperature=0.0,
                            openai_api_key=openai_api_key),
+            chain_type='stuff',
             retriever=db.as_retriever(),
             memory=memory,
             get_chat_history=lambda h: h,
+            verbose=True,
+            return_source_documents=True,
         )
         result = cqa({'question': query, 'chat_history': chat_history})
         chat_history.append((query, result['answer']))
         return '', chat_history
+    # except Exception as e:
+    #     chat_history.append((query, e))
+    #     return '', chat_history
+    except pinecone.exceptions.PineconeException as pe:
+        chat_history.append((query, f"Pinecone Error: {pe}"))
+        return '', chat_history
     except Exception as e:
-        chat_history.append((query, e))
+        chat_history.append((query, f"Unexpected Error: {e}"))
         return '', chat_history