Spaces:
Runtime error
Runtime error
Commit
·
43408d7
1
Parent(s):
ba8c5f0
update QAQuery
Browse files
app.py
CHANGED
@@ -1374,14 +1374,19 @@ def CreatDb_P():
|
|
1374 |
|
1375 |
def QAQuery_p(question: str):
|
1376 |
global vectordb_p
|
|
|
1377 |
# vectordb = Chroma(persist_directory='db', embedding_function=embeddings)
|
1378 |
retriever = vectordb_p.as_retriever()
|
1379 |
retriever.search_kwargs['k'] = int(os.environ["search_kwargs_k"])
|
1380 |
# retriever.search_kwargs['fetch_k'] = 100
|
1381 |
-
|
1382 |
-
|
1383 |
-
|
1384 |
-
|
|
|
|
|
|
|
|
|
1385 |
# qa = VectorDBQA.from_chain_type(llm=chat, chain_type="stuff", vectorstore=vectordb, return_source_documents=True)
|
1386 |
# res = qa.run(question)
|
1387 |
res = qa({"query": question})
|
|
|
1374 |
|
1375 |
def QAQuery_p(question: str):
|
1376 |
global vectordb_p
|
1377 |
+
global agent
|
1378 |
# vectordb = Chroma(persist_directory='db', embedding_function=embeddings)
|
1379 |
retriever = vectordb_p.as_retriever()
|
1380 |
retriever.search_kwargs['k'] = int(os.environ["search_kwargs_k"])
|
1381 |
# retriever.search_kwargs['fetch_k'] = 100
|
1382 |
+
if agent == agent_ZEROSHOT_REACT_2:
|
1383 |
+
qa = RetrievalQA.from_chain_type(llm=GPTfake, chain_type="stuff",
|
1384 |
+
retriever=retriever, return_source_documents = True,
|
1385 |
+
verbose = True)
|
1386 |
+
else:
|
1387 |
+
qa = RetrievalQA.from_chain_type(llm=chat, chain_type="stuff",
|
1388 |
+
retriever=retriever, return_source_documents = True,
|
1389 |
+
verbose = True)
|
1390 |
# qa = VectorDBQA.from_chain_type(llm=chat, chain_type="stuff", vectorstore=vectordb, return_source_documents=True)
|
1391 |
# res = qa.run(question)
|
1392 |
res = qa({"query": question})
|