fix: Use self instead of model in get_raw_thinking_tokens

This commit is contained in:
Paul Gauthier (aider) 2025-03-28 20:09:05 -10:00
parent 96aa77288b
commit 14d1742869

View file

@ -695,16 +695,16 @@ class Model(ModelSettings):
if self.extra_params: if self.extra_params:
# Check for OpenRouter reasoning format # Check for OpenRouter reasoning format
if ( if (
"reasoning" in model.extra_params "reasoning" in self.extra_params
and "max_tokens" in model.extra_params["reasoning"] and "max_tokens" in self.extra_params["reasoning"]
): ):
budget = model.extra_params["reasoning"]["max_tokens"] budget = self.extra_params["reasoning"]["max_tokens"]
# Check for standard thinking format # Check for standard thinking format
elif ( elif (
"thinking" in model.extra_params "thinking" in self.extra_params
and "budget_tokens" in model.extra_params["thinking"] and "budget_tokens" in self.extra_params["thinking"]
): ):
budget = model.extra_params["thinking"]["budget_tokens"] budget = self.extra_params["thinking"]["budget_tokens"]
return budget return budget