diff --git a/pr_agent/algo/language_handler.py b/pr_agent/algo/language_handler.py index 12d86d1c..2bd2609b 100644 --- a/pr_agent/algo/language_handler.py +++ b/pr_agent/algo/language_handler.py @@ -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) diff --git a/pr_agent/algo/pr_processing.py b/pr_agent/algo/pr_processing.py index db6f4309..17a973dc 100644 --- a/pr_agent/algo/pr_processing.py +++ b/pr_agent/algo/pr_processing.py @@ -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