xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-05 05:10:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace get_settings() with self.settings

Browse Source
This commit is contained in:
kkan9ma
2025-05-20 13:50:30 +09:00
parent 81fa22e4df
commit e72bb28c4e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pr_agent/algo/token_handler.py
View File

@ -129,12 +129,12 @@ class TokenHandler:
return ceil(factor * default_estimate) return ceil(factor * default_estimate)
def get_token_count_by_model_type(self, patch: str, default_estimate: int) -> int: 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) 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 default_estimate
return self.apply_estimation_factor(model_name, default_estimate) return self.apply_estimation_factor(model_name, default_estimate)