OuroborosM commited on
Commit
43408d7
·
1 Parent(s): ba8c5f0

update QAQuery

Browse files
Files changed (1) hide show
  1. app.py +9 -4
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
- qa = RetrievalQA.from_chain_type(llm=chat, chain_type="stuff",
1383
- retriever=retriever, return_source_documents = True,
1384
- verbose = True)
 
 
 
 
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})