Skip to content

Commit 09900ee

Browse files
committed
fixup! 5760a1b
1 parent 5760a1b commit 09900ee

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/git_draft/bots/claude_code.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,16 +58,17 @@ async def act(
5858
_notify(feedback, content)
5959
case sdk.AssistantMessage(content, _):
6060
_notify(feedback, content)
61-
case sdk.SystemMessage:
62-
pass
6361
case sdk.ResultMessage() as message:
6462
# This message's result appears to be identical to
6563
# the last assistant message's content, so we do
6664
# not need to show it.
6765
summary.turn_count = message.num_turns
68-
summary.token_count = _token_count(message.usage)
66+
if usage := message.usage:
67+
summary.token_count = _token_count(usage)
6968
summary.usage_details = message.usage
7069
summary.cost = message.total_cost_usd
70+
case sdk.SystemMessage():
71+
pass # TODO: Notify on tool usage?
7172
return summary
7273

7374

0 commit comments

Comments
 (0)