feat: add more logging

This commit is contained in:
hh
2026-05-22 22:48:12 +02:00
parent b8ac545a51
commit 33eece331a
3 changed files with 89 additions and 3 deletions
+33 -1
View File
@@ -27,6 +27,7 @@ agent's ``system_prompt`` is the canonical identity of the agent.
from __future__ import annotations
import json
import logging
import uuid
from dataclasses import dataclass, field
from typing import TYPE_CHECKING, Any, Self
@@ -67,6 +68,9 @@ if TYPE_CHECKING:
from beaver_gateway.core.events import MessageStreamEvent
_log = logging.getLogger("beaver_gateway.backends.claude_code")
__all__ = ["ClaudeCodeBackendAdapter", "TurnCapture"]
@@ -257,12 +261,24 @@ class ClaudeCodeBackendAdapter:
)
raise ValueError(msg)
msgs_list: list[Mapping[str, Any]] = list(messages)
_log.info(
"complete: agent=%s n_messages=%d capture=%s live_sessions=%d",
agent.name,
len(msgs_list),
capture is not None,
self._backend.live_session_count,
)
message_id = f"msg_{uuid.uuid4().hex}"
yield build_message_start(message_id=message_id, model=agent.model)
next_index = 0
stop_reason: str | None = None
usage: Mapping[str, Any] | None = None
n_text = 0
n_thinking = 0
n_tool_use = 0
# We keep raw events so we can hand them to
# ``synthesize_turn_messages`` after the stream closes — the
# markdown frontend stores the result in its conversation
@@ -271,10 +287,16 @@ class ClaudeCodeBackendAdapter:
# are silently discarded from the wire but kept here.
raw_events: list[Any] = []
async for event in self._backend.complete(list(messages)):
async for event in self._backend.complete(msgs_list):
raw_events.append(event)
if isinstance(event, AssistantMessage):
for block in event.content:
if isinstance(block, TextBlock):
n_text += 1
elif isinstance(block, ThinkingBlock):
n_thinking += 1
elif isinstance(block, ToolUseBlock):
n_tool_use += 1
for ev in _emit_block(block, next_index):
yield ev
next_index += 1
@@ -298,6 +320,16 @@ class ClaudeCodeBackendAdapter:
if capture is not None:
capture.synthesized_messages = synthesize_turn_messages(raw_events)
_log.info(
"complete: agent=%s DONE text=%d thinking=%d tool_use=%d stop=%s synth=%d",
agent.name,
n_text,
n_thinking,
n_tool_use,
stop_reason,
len(capture.synthesized_messages) if capture is not None else 0,
)
yield build_message_delta(
stop_reason=_map_stop_reason(stop_reason), usage=_normalize_usage(usage)
)