mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-07-06 22:00:40 +08:00
Restore original return logic for force_accurate condition
This commit is contained in:
@ -166,8 +166,8 @@ class TokenHandler:
|
|||||||
"""
|
"""
|
||||||
encoder_estimate = len(self.encoder.encode(patch, disallowed_special=()))
|
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.
|
# If an estimate is enough (for example, in cases where the maximal allowed tokens is way below the known limits), return it.
|
||||||
|
if not force_accurate:
|
||||||
return encoder_estimate
|
return encoder_estimate
|
||||||
|
|
||||||
|
return self.get_token_count_by_model_type(patch, encoder_estimate)
|
||||||
|
Reference in New Issue
Block a user