Limit source documents sent via websockets to 5

chat/src/helpers/response.py

@@ -21,7 +21,7 @@ def debug_response_passthrough(self):
     def original_question_passthrough(self):
         def get_and_send_original_question(docs):
             source_documents = []
-            for doc in docs["context"]:
+            for doc in docs["context"][:5]:
                 doc.metadata = {key: extract_prompt_value(doc.metadata.get(key)) for key in self.config.attributes if key in doc.metadata}
                 source_document = doc.metadata.copy()
                 source_document["content"] = doc.page_content