Update utills.py

utills.py

@@ -72,21 +72,12 @@ def load_uploaded_documents(uploaded_files):
         documents.append({"content": content, "filename": uploaded_file.name})
     return documents
 
-@st.cache_resource
-def chroma_db(docs, _embeddings):
-    try:
-        vectordb = Chroma.from_documents(
-            documents=docs, embedding=embeddings, persist_directory="docs/chroma/"
-        )
-        return vectordb
-    except Exception as e:
-        print(f"Error creating Chroma vector database: {e}")
-        return None  # or handle the error in an appropriate manner
-
-@st.cache_resource
-def retriever_from_chroma(vectordb, _search_type, k):
+def retriever_from_chroma(docs, embeddings, search_type, k):
+    vectordb = Chroma.from_documents(
+        documents=docs, embedding=embeddings, persist_directory="docs/chroma/"
+    )
     retriever = vectordb.as_retriever(search_type=search_type, search_kwargs={"k": k})
-    return retriever    
+    return retriever
 
 @st.cache_resource
 def history_aware_retriever(llm, retriever, contextualize_q_system_prompt):