minor fix for output tokens

AgentOps-AI · Aug 10, 2024 · 27dabf5 · 27dabf5
1 parent 364c6f6
commit 27dabf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py
@@ -663,7 +663,7 @@ def handle_stream_chunk(chunk: Message):
                 elif isinstance(chunk, RawContentBlockStopEvent):
                     pass
                 elif isinstance(chunk, RawMessageDeltaEvent):
-                    self.llm_event.completion_tokens += chunk.message.usage.output_tokens
+                    self.llm_event.completion_tokens = chunk.usage.output_tokens
                 elif isinstance(chunk, RawMessageStopEvent):
                     self.llm_event.completion = accumulated_delta
                     self.llm_event.end_timestamp = get_ISO_time()