File tree Expand file tree Collapse file tree 1 file changed +11
-1
lines changed
src/cai/sdk/agents/models Expand file tree Collapse file tree 1 file changed +11
-1
lines changed Original file line number Diff line number Diff line change @@ -760,7 +760,17 @@ async def get_response(
760760 and response .usage .completion_tokens_details
761761 and hasattr (response .usage .completion_tokens_details , "reasoning_tokens" )
762762 ):
763- reasoning_tokens = response .usage .completion_tokens_details .reasoning_tokens
763+ # Guard against None or unexpected types for reasoning_tokens
764+ try :
765+ reasoning_tokens = response .usage .completion_tokens_details .reasoning_tokens
766+ if reasoning_tokens is None :
767+ reasoning_tokens = 0
768+ else :
769+ # coerce numeric-like values to int
770+ reasoning_tokens = int (reasoning_tokens )
771+ except Exception :
772+ reasoning_tokens = 0
773+
764774 self .total_reasoning_tokens += reasoning_tokens
765775
766776 # Process costs for non-streaming mode
You can’t perform that action at this time.
0 commit comments