LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 08:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update app.py

Browse Source
This commit is contained in:
Alex
2023-03-03 18:05:00 +00:00
parent 17047b6201
commit 02aca04efe
1 changed files with 1 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
application/app.py
View File

@@ -1,7 +1,6 @@
import os
import json
import traceback
import pprint
import dotenv
import requests
@@ -137,15 +136,10 @@ def api_answer():
qa_chain = load_qa_chain(llm=llm, chain_type="map_reduce",
combine_prompt=c_prompt, question_prompt=q_prompt)
chain = VectorDBQA(combine_documents_chain=qa_chain, vectorstore=docsearch, k=25, return_source_documents=True)
chain = VectorDBQA(combine_documents_chain=qa_chain, vectorstore=docsearch, k=10)
# fetch the answer
result = chain({"query": question})
# pprint.pprint(result)
# docs = docsearch.similarity_search(question, k=8)
for i in result['source_documents']:
print(i.page_content)
# some formatting for the frontend
result['answer'] = result['result']
@@ -154,7 +148,6 @@ def api_answer():
result['answer'] = result['answer'].split("SOURCES:")[0]
except:
pass
del result['source_documents']
# mock result
# result = {