From af08fae4942d7d151cd6482b98d44ffd8c5255bc Mon Sep 17 00:00:00 2001
From: BeatrixCohere <128378696+BeatrixCohere@users.noreply.github.com>
Date: Fri, 12 Jul 2024 10:22:28 +0100
Subject: [PATCH] Change

---
 src/backend/chat/custom/custom.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/backend/chat/custom/custom.py b/src/backend/chat/custom/custom.py
index b33a383979..494eb312bb 100644
--- a/src/backend/chat/custom/custom.py
+++ b/src/backend/chat/custom/custom.py
@@ -54,6 +54,13 @@ async def chat(self, chat_request: CohereChatRequest, **kwargs: Any) -> Any:
             stream = self.call_chat(self.chat_request, deployment_model, **kwargs)
 
             async for event in stream:
+                send_log_message(
+                        logger,
+                        f"Stream event: {event}",
+                        level="info",
+                        conversation_id=kwargs.get("conversation_id"),
+                        user_id=kwargs.get("user_id"),
+                    )
                 result = self.handle_event(event, chat_request)
 
                 if result: