diff --git a/google/genai/_interactions/types/usage.py b/google/genai/_interactions/types/usage.py index 5bfeafd10..06b1ab9fb 100644 --- a/google/genai/_interactions/types/usage.py +++ b/google/genai/_interactions/types/usage.py @@ -93,7 +93,7 @@ class Usage(BaseModel): total_output_tokens: Optional[int] = None """Total number of tokens across all the generated responses.""" - total_reasoning_tokens: Optional[int] = None + total_thought_tokens: Optional[int] = None """Number of tokens of thoughts for thinking models.""" total_tokens: Optional[int] = None diff --git a/google/genai/_interactions/types/usage_param.py b/google/genai/_interactions/types/usage_param.py index 355c4d508..f18d07457 100644 --- a/google/genai/_interactions/types/usage_param.py +++ b/google/genai/_interactions/types/usage_param.py @@ -93,7 +93,7 @@ class UsageParam(TypedDict, total=False): total_output_tokens: int """Total number of tokens across all the generated responses.""" - total_reasoning_tokens: int + total_thought_tokens: int """Number of tokens of thoughts for thinking models.""" total_tokens: int