feat: track llm provider traces and tracking steps in async agent loop (#2219)

This commit is contained in:
Andy Li
2025-05-19 15:50:56 -07:00
committed by GitHub
parent 969f0d65c8
commit a78abc610e
28 changed files with 920 additions and 82 deletions

View File

@@ -74,6 +74,7 @@ class AnthropicStreamingInterface:
# usage trackers
self.input_tokens = 0
self.output_tokens = 0
self.model = None
# reasoning object trackers
self.reasoning_messages = []
@@ -311,6 +312,7 @@ class AnthropicStreamingInterface:
self.message_id = event.message.id
self.input_tokens += event.message.usage.input_tokens
self.output_tokens += event.message.usage.output_tokens
self.model = event.message.model
elif isinstance(event, BetaRawMessageDeltaEvent):
self.output_tokens += event.usage.output_tokens
elif isinstance(event, BetaRawMessageStopEvent):