From c996c7117fcc0add9fe998a3b7ba2c54c0f055db Mon Sep 17 00:00:00 2001 From: Benedict Lee Date: Thu, 8 Aug 2024 11:32:10 +0900 Subject: [PATCH] Fix function to return an empty list instead of a single joined string when patches_extended is empty. --- pr_agent/algo/pr_processing.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pr_agent/algo/pr_processing.py b/pr_agent/algo/pr_processing.py index b7c21a75..83972096 100644 --- a/pr_agent/algo/pr_processing.py +++ b/pr_agent/algo/pr_processing.py @@ -70,7 +70,10 @@ def get_pr_diff(git_provider: GitProvider, token_handler: TokenHandler, if total_tokens + OUTPUT_BUFFER_TOKENS_SOFT_THRESHOLD < get_max_tokens(model): get_logger().info(f"Tokens: {total_tokens}, total tokens under limit: {get_max_tokens(model)}, " f"returning full diff.") - return "\n".join(patches_extended) + if patches_extended : + return ["\n".join(patches_extended)] + else: + return [] # if we are over the limit, start pruning get_logger().info(f"Tokens: {total_tokens}, total tokens over limit: {get_max_tokens(model)}, "