feat: implemented extra log level for LLM token stream

2025-12-16 11:26:35 +00:00
parent 2366255b92
commit 78abad55d3
4 changed files with 23 additions and 6 deletions
--- a/src/control_backend/agents/llm/llm_agent.py
+++ b/src/control_backend/agents/llm/llm_agent.py
@@ -125,7 +125,7 @@ class LLMAgent(BaseAgent):
                full_message += token
                current_chunk += token

-                self.logger.info(
+                self.logger.llm(
                    "Received token: %s",
                    full_message,
                    extra={"reference": message_id},  # Used in the UI to update old logs