From 84f2f4fe3d8b601f55c39de1e5e03988703990d9 Mon Sep 17 00:00:00 2001 From: kkan9ma Date: Sun, 25 May 2025 18:00:38 +0900 Subject: [PATCH] Fix: use ModelTypeValidator static methods directly --- pr_agent/algo/token_handler.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pr_agent/algo/token_handler.py b/pr_agent/algo/token_handler.py index 998391b1..cb313f02 100644 --- a/pr_agent/algo/token_handler.py +++ b/pr_agent/algo/token_handler.py @@ -67,7 +67,6 @@ class TokenHandler: - user: The user string. """ self.encoder = TokenEncoder.get_token_encoder() - self.model_validator = ModelTypeValidator() if pr is not None: self.prompt_tokens = self._get_system_user_tokens(pr, self.encoder, vars, system, user) @@ -144,10 +143,10 @@ class TokenHandler: """ model_name = get_settings().config.model.lower() - if self.model_validator.is_openai_model(model_name) and get_settings(use_context=False).get('openai.key'): + if ModelTypeValidator.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'): + + if ModelTypeValidator.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)