From 34096059ff746cccddf3a42a58b9960a786dec11 Mon Sep 17 00:00:00 2001 From: idavidov Date: Tue, 25 Jul 2023 13:05:56 +0300 Subject: [PATCH] quick and dirty response for github API ratelimit, until some smart solution will be implemented --- pr_agent/algo/pr_processing.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pr_agent/algo/pr_processing.py b/pr_agent/algo/pr_processing.py index 20933d51..bfe72387 100644 --- a/pr_agent/algo/pr_processing.py +++ b/pr_agent/algo/pr_processing.py @@ -10,6 +10,7 @@ from pr_agent.algo.token_handler import TokenHandler from pr_agent.algo.utils import load_large_diff from pr_agent.config_loader import settings from pr_agent.git_providers.git_provider import GitProvider +from github import GithubException DELETED_FILES_ = "Deleted files:\n" @@ -40,7 +41,10 @@ def get_pr_diff(git_provider: GitProvider, token_handler: TokenHandler, model: s global PATCH_EXTRA_LINES PATCH_EXTRA_LINES = 0 - diff_files = list(git_provider.get_diff_files()) + try: + diff_files = list(git_provider.get_diff_files()) + except GithubException.RateLimitExceededException as e: + logging.warning("Rate limit exceeded for GitHub API.") # get pr languages pr_languages = sort_files_by_main_languages(git_provider.get_languages(), diff_files)