Spaces:
Runtime error
Runtime error
Commit
·
d4b40f3
1
Parent(s):
e6376d0
correct type of k error
Browse files
app.py
CHANGED
@@ -106,7 +106,7 @@ def QAQuery_p(question: str):
|
|
106 |
global vectordb_p
|
107 |
# vectordb = Chroma(persist_directory='db', embedding_function=embeddings)
|
108 |
retriever = vectordb_p.as_retriever()
|
109 |
-
retriever.search_kwargs['k'] = os.getenv("search_kwargs_k")
|
110 |
# retriever.search_kwargs['fetch_k'] = 100
|
111 |
|
112 |
qa = RetrievalQA.from_chain_type(llm=chat, chain_type="stuff", retriever=retriever, return_source_documents = True)
|
|
|
106 |
global vectordb_p
|
107 |
# vectordb = Chroma(persist_directory='db', embedding_function=embeddings)
|
108 |
retriever = vectordb_p.as_retriever()
|
109 |
+
retriever.search_kwargs['k'] = int(os.getenv("search_kwargs_k"))
|
110 |
# retriever.search_kwargs['fetch_k'] = 100
|
111 |
|
112 |
qa = RetrievalQA.from_chain_type(llm=chat, chain_type="stuff", retriever=retriever, return_source_documents = True)
|