adapt source in context

startmunich · May 11, 2024 · da69eb1 · da69eb1
1 parent 2d358d5
commit da69eb1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/apps/slackbot/ai_fcts.py b/apps/slackbot/ai_fcts.py
@@ -80,7 +80,10 @@
 def format_docs(docs):
     context = ""
     for index, doc in enumerate(docs):
-        context += f"Document Rank {index + 1}. Source: https://www.notion.so/{doc.metadata['page_id']}. Content: {doc.page_content}\n\n"
+        if doc.metadata['type'] == 'notion:':
+            context += f"Document Rank {index + 1}. Source: https://www.notion.so/{doc.metadata['page_id']}. Content: {doc.page_content}\n\n"
+        else:
+            context += f"Document Rank {index + 1}. Source: {doc.metadata['type']}. Content: {doc.page_content}\n\n"
     return context
 
 # create function to invoke the retrievalQA