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

Restore original return logic for force_accurate condition

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

8
pr_agent/algo/token_handler.py
View File

@ -166,8 +166,8 @@ class TokenHandler:
"""
encoder_estimate = len(self.encoder.encode(patch, disallowed_special=()))
if force_accurate:
return self.get_token_count_by_model_type(patch, encoder_estimate)
# If an estimate is enough (for example, in cases where the maximal allowed tokens is way below the known limits), return it.
return encoder_estimate
if not force_accurate:
return encoder_estimate
return self.get_token_count_by_model_type(patch, encoder_estimate)