From 02aca04efed6a970d845b91956ab7a5d8a7bbd04 Mon Sep 17 00:00:00 2001
From: Alex <a@tushynski.me>
Date: Fri, 3 Mar 2023 18:05:00 +0000
Subject: [PATCH] Update app.py

---
 application/app.py | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/application/app.py b/application/app.py
index 933db17..f9ea185 100644
--- a/application/app.py
+++ b/application/app.py
@@ -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 = {