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

Remove unused filter_bad_extensions function and rename diff_files_original to diff_files

Browse Source
This commit is contained in:
mrT23
2025-03-11 16:56:41 +02:00
parent 4713175fcf
commit 0f33750035
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pr_agent/algo/language_handler.py
View File

@ -45,8 +45,7 @@ def sort_files_by_main_languages(languages: Dict, files: list):
else:
main_extensions.append([])
# filter out files bad extensions
files_filtered = filter_bad_extensions(files)
# sort files by their extension, put the files that are in the main extension first
# and the rest files after, map languages_sorted to their respective files
files_sorted = []
@ -54,7 +53,7 @@ def sort_files_by_main_languages(languages: Dict, files: list):
# if no languages detected, put all files in the "Other" category
if not languages:
files_sorted = [({"language": "Other", "files": list(files_filtered)})]
files_sorted = [({"language": "Other", "files": list(files)})]
return files_sorted
main_extensions_flat = []
@ -63,7 +62,7 @@ def sort_files_by_main_languages(languages: Dict, files: list):
for extensions, lang in zip(main_extensions, languages_sorted_list): # noqa: B905
tmp = []
for file in files_filtered:
for file in files:
extension_str = f".{file.filename.split('.')[-1]}"
if extension_str in extensions:
tmp.append(file)

4
pr_agent/algo/pr_processing.py
View File

@ -51,7 +51,7 @@ def get_pr_diff(git_provider: GitProvider, token_handler: TokenHandler,
PATCH_EXTRA_LINES_AFTER = cap_and_log_extra_lines(PATCH_EXTRA_LINES_AFTER, "after")
try:
diff_files_original = git_provider.get_diff_files()
diff_files = git_provider.get_diff_files()
except RateLimitExceededException as e:
get_logger().error(f"Rate limit exceeded for git provider API. original message {e}")
raise
@ -145,7 +145,7 @@ def get_pr_diff(git_provider: GitProvider, token_handler: TokenHandler,
def get_pr_diff_multiple_patchs(git_provider: GitProvider, token_handler: TokenHandler, model: str,
add_line_numbers_to_hunks: bool = False, disable_extra_lines: bool = False):
try:
diff_files_original = git_provider.get_diff_files()
diff_files = git_provider.get_diff_files()
except RateLimitExceededException as e:
get_logger().error(f"Rate limit exceeded for git provider API. original message {e}")
raise