From e72bb28c4e9cad740f9240b25a5adb7339943044 Mon Sep 17 00:00:00 2001 From: kkan9ma Date: Tue, 20 May 2025 13:50:30 +0900 Subject: [PATCH] Replace get_settings() with self.settings --- 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 2781fc5c..99dbc635 100644 --- a/pr_agent/algo/token_handler.py +++ b/pr_agent/algo/token_handler.py @@ -129,12 +129,12 @@ class TokenHandler: return ceil(factor * default_estimate) def get_token_count_by_model_type(self, patch: str, default_estimate: int) -> int: - model_name = get_settings().config.model.lower() + model_name = self.settings.config.model.lower() - if self.model_validator.is_claude_model(model_name) and get_settings(use_context=False).get('anthropic.key'): + if self.model_validator.is_claude_model(model_name) and self.settings.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'): + if self.model_validator.is_openai_model(model_name) and self.settings.get('openai.key'): return default_estimate return self.apply_estimation_factor(model_name, default_estimate)