Refactor source message handling

QuivrHQ · Apr 22, 2024 · 422db14 · 422db14
1 parent a77f29e
commit 422db14
Showing 1 changed file with 30 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -362,7 +362,21 @@ def ask_question(self, body, brain_id, thread_ts, question=None):
                             for source in sources
                         ]
                     )
-                    question_response["assistant"] += f"\n\n*Sources:*\n{source_text}"
+                    # Send sources as a separate message
+                    self.app.client.chat_postMessage(
+                        channel=body["event"]["channel"],
+                        text=f"*Sources:*\n{source_text}",
+                        blocks=[
+                            {
+                                "type": "section",
+                                "text": {
+                                    "type": "mrkdwn",
+                                    "text": f"*Sources:*\n{source_text}",
+                                },
+                            }
+                        ],
+                        thread_ts=thread_ts,
+                    )
 
             # Split the message if it's too long
             messages = [
@@ -448,7 +462,21 @@ def handle_iteractive_request(self, payload):
                             for source in sources
                         ]
                     )
-                    question_response["assistant"] += f"\n\n*Sources:*\n{source_text}"
+                    # Send sources as a separate message
+                    self.app.client.chat_postMessage(
+                        channel=body["event"]["channel"],
+                        text=f"*Sources:*\n{source_text}",
+                        blocks=[
+                            {
+                                "type": "section",
+                                "text": {
+                                    "type": "mrkdwn",
+                                    "text": f"*Sources:*\n{source_text}",
+                                },
+                            }
+                        ],
+                        thread_ts=thread_ts,
+                    )
 
             # Split the message if it's too long
             messages = [