Update utills.py
Browse files
utills.py
CHANGED
@@ -84,7 +84,7 @@ def chroma_db(docs, _embeddings):
|
|
84 |
return None # or handle the error in an appropriate manner
|
85 |
|
86 |
@st.cache_resource
|
87 |
-
def retriever_from_chroma(vectordb,
|
88 |
retriever = vectordb.as_retriever(search_type=search_type, search_kwargs={"k": k})
|
89 |
return retriever
|
90 |
|
|
|
84 |
return None # or handle the error in an appropriate manner
|
85 |
|
86 |
@st.cache_resource
|
87 |
+
def retriever_from_chroma(vectordb, _search_type, k):
|
88 |
retriever = vectordb.as_retriever(search_type=search_type, search_kwargs={"k": k})
|
89 |
return retriever
|
90 |
|