Update vector_store_retriever.py
Browse files
vector_store_retriever.py
CHANGED
@@ -129,7 +129,7 @@ def predict(message, temperature=0.9, max_new_tokens=512, top_p=0.6, repetition_
|
|
129 |
res_result = llm_response['result']
|
130 |
|
131 |
res_relevant_doc = [source.metadata['source'] for source in llm_response["source_documents"]]
|
132 |
-
|
133 |
print("response: =====> \n", response, "\n\n")
|
134 |
|
135 |
tokens = response.split('\n')
|
|
|
129 |
res_result = llm_response['result']
|
130 |
|
131 |
res_relevant_doc = [source.metadata['source'] for source in llm_response["source_documents"]]
|
132 |
+
response = f"{res_result}" + "\n\n" + "[Answer Source Documents (Ctrl + Click!)] :" + "\n" + f" \n {res_relevant_doc}"
|
133 |
print("response: =====> \n", response, "\n\n")
|
134 |
|
135 |
tokens = response.split('\n')
|