From f87051e35b48aff52df12b23150527483cf5fd71 Mon Sep 17 00:00:00 2001
From: collhoun <2904yr@mail.ru>
Date: Sat, 2 May 2026 12:19:28 +0300
Subject: [PATCH] =?UTF-8?q?=D0=9E=D0=B1=D0=BD=D0=BE=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=D0=BE=20=D0=BB=D0=BE=D0=B3=D0=B8=D1=80=D0=BE=D0=B2=D0=B0?=
 =?UTF-8?q?=D0=BD=D0=B8=D0=B5,=20=D0=B4=D0=BE=D0=B1=D0=B0=D0=B2=D0=B8?=
 =?UTF-8?q?=D0=BB=D0=B8=20=D0=BF=D1=80=D0=BE=D0=B7=D1=80=D0=B0=D1=87=D0=BD?=
 =?UTF-8?q?=D0=BE=D1=81=D1=82=D1=8C=20=D0=B7=D0=B0=D0=BF=D1=80=D0=BE=D1=81?=
 =?UTF-8?q?=D0=BE=D0=B2=20=D0=B8=20=D0=BE=D1=82=D0=B2=D0=B5=D1=82=D0=BE?=
 =?UTF-8?q?=D0=B2=20=D0=B0=D0=B3=D0=B5=D0=BD=D1=82=D0=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/agent/service.py |  4 ++--
 src/api/external.py  | 14 +++++++++-----
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/agent/service.py b/src/agent/service.py
index 9548334..99c8db5 100644
--- a/src/agent/service.py
+++ b/src/agent/service.py
@@ -110,7 +110,7 @@ class AgentService:
 
         new_message = text
         if attachments:
-            logger.debug(f"Processing {len(attachments)} attachments for chat {chat_id}")
+            logger.debug(f"Processing {attachments} attachments for chat {chat_id}")
             attachments_description = await self.__describe_attachments(attachments)
             new_message += "\n" + attachments_description
 
@@ -130,7 +130,7 @@ class AgentService:
 
                 # обычный текст
                 if chunk.content:
-                    logger.trace(f"Yielding text chunk for chat {chat_id}")
+                    logger.trace(f"Yielding text chunk {chunk.content} for chat {chat_id}")
                     yield MsgEventTextChunk(text=chunk.content)
 
                 # если вернулся tool_call
diff --git a/src/api/external.py b/src/api/external.py
index 6721bf6..4f8217a 100644
--- a/src/api/external.py
+++ b/src/api/external.py
@@ -49,8 +49,8 @@ async def websocket_endpoint(
             # Генерируем ID для каждого сообщения
             message_id = generate_message_id()
             set_message_id(message_id)
-            
-            logger.trace(f"Received raw message: {len(raw)} characters for chat_id: {chat_id}")
+            # тут должен быть trace или дебаг в будущем, но пока что info для прозрачности
+            logger.info(f"Received raw message: {raw} for chat_id: {chat_id}")
             try:
                 msg = ClientMessage.validate_json(raw)
             except ValidationError as e:
@@ -74,9 +74,13 @@ async def websocket_endpoint(
 async def process_message(ws: WebSocket, chat: AgentChat, msg):
     match msg:
         case MsgUserMessage():
-            logger.debug(f"Processing user message for chat {chat.chat_id} (text length: {len(msg.text)}, attachments: {len(msg.attachments) if msg.attachments else 0})")
+            logger.debug(f"Processing user message for chat {chat.chat_id} (text length: {len(msg.text)}, attachments: {msg.attachments if msg.attachments else None})")
+            chunks = []
             async for chunk in chat.astream(msg.text, msg.attachments):
-                logger.trace(f"Sending stream chunk to chat {chat.chat_id}: {chunk.__class__.__name__}")
+                logger.trace(f"Sending stream chunk to chat {chat.chat_id}: {chunk}")
                 await ws.send_text(chunk.model_dump_json())
-            logger.debug(f"Finished processing user message for chat {chat.chat_id}")
+                chunks.append(chunk.text if hasattr(chunk, 'text') else '')
+            chunks = ''.join(chunks)
+            logger.debug(f"Finished processing user message for chat {chat.chat_id} with {MsgEventEnd(tokens_used=0)}")
+            logger.info(f"Processed user message for chat {chat.chat_id}: \n {chunks}")
             await ws.send_text(MsgEventEnd(tokens_used=0).model_dump_json())  # TODO: подставить реальное потребление токенов