diff --git a/app.py b/app.py index 36775d0..16bc39b 100644 --- a/app.py +++ b/app.py @@ -10,7 +10,6 @@ import openai from dotenv import load_dotenv from openai import OpenAI -from langchain.vectorstores import Chroma load_dotenv() @@ -95,12 +94,6 @@ chroma_client.delete_collection(collection.name) except AttributeError: st.error("Collection erased.") - -if chroma_client.get_collection(collection.name) is not None: - langchain_agent = Chroma(client=chroma_client, - collection_name=collection.name, - embedding_function=embedding_function - ) diff --git a/wk_flow_requirements.txt b/wk_flow_requirements.txt index fd84690..b54a825 100644 --- a/wk_flow_requirements.txt +++ b/wk_flow_requirements.txt @@ -2,5 +2,6 @@ streamlit pymupdf openai tiktoken -pylint +pylint} +langchain chromadb>='0.4.18'