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

code

Browse Source
This commit is contained in:
mrT23
2023-11-07 14:38:37 +02:00
parent a35a75eb34
commit 094f641fb5
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pr_agent/algo/utils.py
View File

@ -344,10 +344,11 @@ def get_user_labels(current_labels):
get_logger().info(f"Keeping user labels: {user_labels}") get_logger().info(f"Keeping user labels: {user_labels}")
return user_labels return user_labels
def get_max_tokens(model): def get_max_tokens(model):
settings = get_settings()
max_tokens_model = MAX_TOKENS[model] max_tokens_model = MAX_TOKENS[model]
if get_settings().config.max_model_tokens: if settings.config.max_model_tokens and max_tokens_model > settings.config.max_model_tokens:
if max_tokens_model > get_settings().config.max_model_tokens: max_tokens_model = settings.config.max_model_tokens
max_tokens_model = get_settings().config.max_model_tokens
# get_logger().debug(f"limiting max tokens to {max_tokens_model}") # get_logger().debug(f"limiting max tokens to {max_tokens_model}")
return max_tokens_model return max_tokens_model

2
pr_agent/settings/configuration.toml
View File

@ -10,7 +10,7 @@ use_repo_settings_file=true
ai_timeout=180 ai_timeout=180
max_description_tokens = 500 max_description_tokens = 500
max_commits_tokens = 500 max_commits_tokens = 500
max_model_tokens = 32000 # even if a model supports more tokens, quality may degrade. Hence, enable to limit the number of tokens. max_model_tokens = 32000 # Limits the maximum number of tokens that can be used by any model, regardless of the model's default capabilities.
patch_extra_lines = 3 patch_extra_lines = 3
secret_provider="google_cloud_storage" secret_provider="google_cloud_storage"
cli_mode=false cli_mode=false