From cc686ef26d961bd515b731c447f98cd23a240633 Mon Sep 17 00:00:00 2001 From: kkan9ma Date: Thu, 22 May 2025 13:12:04 +0900 Subject: [PATCH] Reorder model check: OpenAI before Anthropic OpenAI is the default in most cases, so checking it first skips unnecessary Anthropic logic. --- pr_agent/algo/token_handler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pr_agent/algo/token_handler.py b/pr_agent/algo/token_handler.py index e45d611f..67a6eed9 100644 --- a/pr_agent/algo/token_handler.py +++ b/pr_agent/algo/token_handler.py @@ -144,11 +144,11 @@ class TokenHandler: """ model_name = get_settings().config.model.lower() - if self.model_validator.is_anthropic_model(model_name) and get_settings(use_context=False).get('anthropic.key'): - return self._calc_claude_tokens(patch) - if self.model_validator.is_openai_model(model_name) and get_settings(use_context=False).get('openai.key'): return default_estimate + + if self.model_validator.is_anthropic_model(model_name) and get_settings(use_context=False).get('anthropic.key'): + return self._calc_claude_tokens(patch) return self._apply_estimation_factor(model_name, default_estimate)