From 54f41dd603180cab03e674e31e17c31ec63e1c88 Mon Sep 17 00:00:00 2001 From: mrT23 Date: Tue, 7 Nov 2023 14:41:15 +0200 Subject: [PATCH] code --- pr_agent/algo/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py index f993bdc9..ecaf4aa8 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -348,7 +348,7 @@ def get_user_labels(current_labels): def get_max_tokens(model): settings = get_settings() max_tokens_model = MAX_TOKENS[model] - if settings.config.max_model_tokens and max_tokens_model > settings.config.max_model_tokens: - max_tokens_model = settings.config.max_model_tokens + if settings.config.max_model_tokens: + max_tokens_model = min(settings.config.max_model_tokens, max_tokens_model) # get_logger().debug(f"limiting max tokens to {max_tokens_model}") return max_tokens_model