xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 11:50:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix pr_processing.get_pr_multi_diffs

Browse Source 
Fix function to return an empty list instead of a single joined string when patches_extended is empty.
This commit is contained in:
Benedict Lee
2024-08-08 11:46:26 +09:00
committed by GitHub
parent c996c7117f
commit 4c0fd37ac2
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pr_agent/algo/pr_processing.py
View File

@ -70,10 +70,7 @@ 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.")
if patches_extended :
return ["\n".join(patches_extended)]
else:
return []
return "\n".join(patches_extended)
# if we are over the limit, start pruning
get_logger().info(f"Tokens: {total_tokens}, total tokens over limit: {get_max_tokens(model)}, "
@ -413,7 +410,7 @@ def get_pr_multi_diffs(git_provider: GitProvider,
patches_extended, total_tokens, patches_extended_tokens = pr_generate_extended_diff(
pr_languages, token_handler, add_line_numbers_to_hunks=True)
if total_tokens + OUTPUT_BUFFER_TOKENS_SOFT_THRESHOLD < get_max_tokens(model):
return ["\n".join(patches_extended)]
return ["\n".join(patches_extended)] if patches_extended else []
patches = []
final_diff_list = []