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

Rename method

Browse Source
This commit is contained in:
kkan9ma
2025-05-21 17:51:03 +09:00
parent 97f2b6f736
commit 648829b770
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/algo/token_handler.py
View File

@ -15,7 +15,7 @@ class ModelTypeValidator:
return 'gpt' in model_name or re.match(r"^o[1-9](-mini|-preview)?$", model_name) return 'gpt' in model_name or re.match(r"^o[1-9](-mini|-preview)?$", model_name)
@staticmethod @staticmethod
def is_claude_model(model_name: str) -> bool: def is_anthropic_model(model_name: str) -> bool:
return 'claude' in model_name return 'claude' in model_name
@ -145,7 +145,7 @@ class TokenHandler:
""" """
model_name = self.settings.config.model.lower() model_name = self.settings.config.model.lower()
if self.model_validator.is_claude_model(model_name) and self.settings.get('anthropic.key'): if self.model_validator.is_anthropic_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 self.settings.get('openai.key'): if self.model_validator.is_openai_model(model_name) and self.settings.get('openai.key'):