remove source from out of context answer

ibizabroker · Aug 26, 2023 · d564d4e · d564d4e
1 parent 9773060
commit d564d4e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/llm/llm.py b/llm/llm.py
@@ -43,7 +43,10 @@ def get_response(input):
       response = message.content
       source_doc = chain_result['source_documents'][0].metadata['source']
       source_doc_page = chain_result['source_documents'][0].metadata['page']
-
+
+  if response.startswith("I'm sorry"):
+    return response
+
   output = f"""{response} \n \n*Source*: {source_doc} \n*Page number*: {source_doc_page}"""
 
   return output