Skip to content

Commit 315d853

Browse files
committed
revert openai-agents changes
1 parent 9a3ce88 commit 315d853

File tree

1 file changed

+8
-16
lines changed
  • sentry_sdk/integrations/openai_agents

1 file changed

+8
-16
lines changed

sentry_sdk/integrations/openai_agents/utils.py

Lines changed: 8 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -96,23 +96,15 @@ def _set_agent_data(span: "sentry_sdk.tracing.Span", agent: "agents.Agent") -> N
9696

9797
def _set_usage_data(span: "sentry_sdk.tracing.Span", usage: "Usage") -> None:
9898
span.set_data(SPANDATA.GEN_AI_USAGE_INPUT_TOKENS, usage.input_tokens)
99-
100-
if hasattr(usage, "input_tokens_details") and usage.input_tokens_details:
101-
if hasattr(usage.input_tokens_details, "cached_tokens"):
102-
span.set_data(
103-
SPANDATA.GEN_AI_USAGE_INPUT_TOKENS_CACHED,
104-
usage.input_tokens_details.cached_tokens,
105-
)
106-
99+
span.set_data(
100+
SPANDATA.GEN_AI_USAGE_INPUT_TOKENS_CACHED,
101+
usage.input_tokens_details.cached_tokens,
102+
)
107103
span.set_data(SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS, usage.output_tokens)
108-
109-
if hasattr(usage, "output_tokens_details") and usage.output_tokens_details:
110-
if hasattr(usage.output_tokens_details, "reasoning_tokens"):
111-
span.set_data(
112-
SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS_REASONING,
113-
usage.output_tokens_details.reasoning_tokens,
114-
)
115-
104+
span.set_data(
105+
SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS_REASONING,
106+
usage.output_tokens_details.reasoning_tokens,
107+
)
116108
span.set_data(SPANDATA.GEN_AI_USAGE_TOTAL_TOKENS, usage.total_tokens)
117109

118110

0 commit comments

Comments
 (0)