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

Fix: use ModelTypeValidator static methods directly

Browse Source
This commit is contained in:
kkan9ma
2025-05-25 18:00:38 +09:00
parent facfb5f46b
commit 84f2f4fe3d
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pr_agent/algo/token_handler.py
View File

@ -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)