ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 02:45:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix condition

Browse source
This commit is contained in:
mrT23 2024-07-28 08:57:39 +03:00
parent 86d9612882
commit 6ba7b3eea2
No known key found for this signature in database
GPG key ID: D350490E39D5F5AD
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pr_agent/algo/utils.py
View file

@ -705,7 +705,7 @@ def get_max_tokens(model):
settings = get_settings() settings = get_settings()
if model in MAX_TOKENS: if model in MAX_TOKENS:
max_tokens_model = MAX_TOKENS[model] max_tokens_model = MAX_TOKENS[model]
elif model in settings.config.custom_model_max_tokens > 0: elif settings.config.custom_model_max_tokens > 0:
max_tokens_model = settings.config.custom_model_max_tokens max_tokens_model = settings.config.custom_model_max_tokens
else: else:
raise Exception(f"MAX_TOKENS must be set for model {model} in ./pr_agent/algo/__init__.py, or set config.custom_model_max_tokens") raise Exception(f"MAX_TOKENS must be set for model {model} in ./pr_agent/algo/__init__.py, or set config.custom_model_max_tokens")