Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,6 +5,7 @@ import streamlit as st
 from pathlib import Path
 from typing import List, Annotated, Any
 import operator
+import tempfile
 import pandas as pd
 from tqdm import tqdm
 from pydantic import BaseModel
@@ -23,8 +24,7 @@ COHERE_API_KEY = os.environ["COHERE_API_KEY"]
 co = cohere.Client(COHERE_API_KEY)
 
 documents_path = Path(__file__).parent / "documents"
-
-persist_dir = "chroma_store"
+persist_dir = tempfile.mkdtemp()
 
 def prepare_vectorstore():
     loader = DirectoryLoader(str(documents_path), glob="**/*.txt", loader_cls=TextLoader)
@@ -33,7 +33,6 @@ def prepare_vectorstore():
     docs = splitter.split_documents(documents)
     embedding = CohereEmbeddings(
         model="embed-multilingual-light-v3.0",
-        user_agent="langchain-app/1.0",
         cohere_api_key=""
     )
     vectorstore = Chroma.from_documents(
@@ -44,8 +43,7 @@ def prepare_vectorstore():
     vectorstore.persist()
     return vectorstore
 
-if not os.path.exists(persist_dir):
-    prepare_vectorstore()
+prepare_vectorstore()
 
 class State(BaseModel):
     state: List[str] = []