@@ -96,23 +96,15 @@ def _set_agent_data(span: "sentry_sdk.tracing.Span", agent: "agents.Agent") -> N
9696
9797def _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