Fixed counting tokens

Author: antonpirker

sentry_sdk/integrations/anthropic.py

@@ -111,6 +111,7 @@ def _collect_ai_data(event, input_tokens, output_tokens, content_blocks):
             elif event.type == "message_delta":
                 output_tokens += event.usage.output_tokens
 
+    return input_tokens, output_tokens, content_blocks                
 
 def _add_ai_data_to_span(
     span, integration, input_tokens, output_tokens, content_blocks
@@ -187,7 +188,7 @@ def new_iterator():
                 content_blocks = []  # type: list[str]
 
                 for event in old_iterator:
-                    _collect_ai_data(event, input_tokens, output_tokens, content_blocks)
+                    input_tokens, output_tokens, content_blocks = _collect_ai_data(event, input_tokens, output_tokens, content_blocks)
                     if event.type != "message_stop":
                         yield event
 
@@ -203,7 +204,7 @@ async def new_iterator_async():
                 content_blocks = []  # type: list[str]
 
                 async for event in old_iterator:
-                    _collect_ai_data(event, input_tokens, output_tokens, content_blocks)
+                    input_tokens, output_tokens, content_blocks = _collect_ai_data(event, input_tokens, output_tokens, content_blocks)
                     if event.type != "message_stop":
                         yield event