Cleanup

Author: antonpirker

sentry_sdk/integrations/openai_agents.py

@@ -28,26 +28,21 @@ def _capture_exception(exc):
 
 
 class SentryRunHooks(RunHooks):
-    def __init__(self):
-        self.event_counter = 0
-
     def _usage_to_str(self, usage: Usage) -> str:
         return f"{usage.requests} requests, {usage.input_tokens} input tokens, {usage.output_tokens} output tokens, {usage.total_tokens} total tokens"
 
     async def on_agent_start(self, context: RunContextWrapper, agent: Agent) -> None:
-        self.event_counter += 1
         print(
-            f"### {self.event_counter}: Agent {agent.name} started. Usage: {self._usage_to_str(context.usage)}"
+            f"### Agent {agent.name} started. Usage: {self._usage_to_str(context.usage)}"
         )
         span = sentry_sdk.start_span(op="gen_ai.agent_start", description=agent.name)
         span.__enter__()
 
     async def on_agent_end(
         self, context: RunContextWrapper, agent: Agent, output: Any
     ) -> None:
-        self.event_counter += 1
         print(
-            f"### {self.event_counter}: Agent '{agent.name}' ended with output {output}. Usage: {self._usage_to_str(context.usage)}"
+            f"### Agent '{agent.name}' ended with output {output}. Usage: {self._usage_to_str(context.usage)}"
         )
         current_span = sentry_sdk.get_current_span()
         if current_span:
@@ -56,19 +51,17 @@ async def on_agent_end(
     async def on_tool_start(
         self, context: RunContextWrapper, agent: Agent, tool: Tool
     ) -> None:
-        self.event_counter += 1
         print(
-            f"### {self.event_counter}: Tool {tool.name} started. Usage: {self._usage_to_str(context.usage)}"
+            f"### Tool {tool.name} started. Usage: {self._usage_to_str(context.usage)}"
         )
         span = sentry_sdk.start_span(op="gen_ai.tool_start", description=tool.name)
         span.__enter__()
 
     async def on_tool_end(
         self, context: RunContextWrapper, agent: Agent, tool: Tool, result: str
     ) -> None:
-        self.event_counter += 1
         print(
-            f"### {self.event_counter}: Tool {tool.name} ended with result {result}. Usage: {self._usage_to_str(context.usage)}"
+            f"### Tool {tool.name} ended with result {result}. Usage: {self._usage_to_str(context.usage)}"
         )
         current_span = sentry_sdk.get_current_span()
         if current_span:
@@ -77,9 +70,8 @@ async def on_tool_end(
     async def on_handoff(
         self, context: RunContextWrapper, from_agent: Agent, to_agent: Agent
     ) -> None:
-        self.event_counter += 1
         print(
-            f"### {self.event_counter}: Handoff from '{from_agent.name}' to '{to_agent.name}'. Usage: {self._usage_to_str(context.usage)}"
+            f"### Handoff from '{from_agent.name}' to '{to_agent.name}'. Usage: {self._usage_to_str(context.usage)}"
         )
         current_span = sentry_sdk.get_current_span()
         if current_span: