Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ def retriever_from_chroma(docs, search_type, k):
|
|
34 |
os.makedirs(vectorstore_path)
|
35 |
vectorstore = Chroma.from_documents(
|
36 |
documents=docs, embedding=embeddings, persist_directory="docs/chroma/")
|
37 |
-
retriever =
|
38 |
return retriever
|
39 |
|
40 |
|
|
|
34 |
os.makedirs(vectorstore_path)
|
35 |
vectorstore = Chroma.from_documents(
|
36 |
documents=docs, embedding=embeddings, persist_directory="docs/chroma/")
|
37 |
+
retriever = vectorstore.as_retriever(search_type=search_type, search_kwargs={"k": k})
|
38 |
return retriever
|
39 |
|
40 |
|