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})
|