diff --git a/.dockerignore b/.dockerignore index eba74f4c..9c8ebe57 100644 --- a/.dockerignore +++ b/.dockerignore @@ -1 +1,2 @@ -venv/ \ No newline at end of file +venv/ +pr_agent/settings/.secrets.toml \ No newline at end of file diff --git a/pr_agent/cli.py b/pr_agent/cli.py index 44d631a1..0d32e921 100644 --- a/pr_agent/cli.py +++ b/pr_agent/cli.py @@ -15,11 +15,11 @@ def run(): logging.basicConfig(level=os.environ.get("LOGLEVEL", "INFO")) if args.question: print(f"Question: {args.question} about PR {args.pr_url}") - reviewer = PRQuestions(args.pr_url, args.question, None) + reviewer = PRQuestions(args.pr_url, args.question, installation_id=None) asyncio.run(reviewer.answer()) else: print(f"Reviewing PR: {args.pr_url}") - reviewer = PRReviewer(args.pr_url, None) + reviewer = PRReviewer(args.pr_url, installation_id=None, cli_mode=True) asyncio.run(reviewer.review()) diff --git a/pr_agent/config_loader.py b/pr_agent/config_loader.py index 394c877f..24dfdea7 100644 --- a/pr_agent/config_loader.py +++ b/pr_agent/config_loader.py @@ -9,6 +9,7 @@ settings = Dynaconf( "settings/.secrets.toml", "settings/configuration.toml", "settings/pr_reviewer_prompts.toml", - "settings/pr_questions_prompts.toml" + "settings/pr_questions_prompts.toml", + "settings_prod/.secrets.toml" ]] ) diff --git a/pr_agent/servers/github_polling.py b/pr_agent/servers/github_polling.py index e1e6fa84..46adb4f0 100644 --- a/pr_agent/servers/github_polling.py +++ b/pr_agent/servers/github_polling.py @@ -20,6 +20,7 @@ def now() -> str: async def polling_loop(): + handled_ids = set() since = [now()] last_modified = [None] git_provider = get_git_provider()() @@ -54,6 +55,9 @@ async def polling_loop(): since[0] = None notifications = await response.json() for notification in notifications: + if 'id' in notification and notification['id'] in handled_ids: + continue + handled_ids.add(notification['id']) if 'reason' in notification and notification['reason'] == 'mention': if 'subject' in notification and notification['subject']['type'] == 'PullRequest': pr_url = notification['subject']['url'] diff --git a/pr_agent/tools/pr_reviewer.py b/pr_agent/tools/pr_reviewer.py index 9ac25063..888108d2 100644 --- a/pr_agent/tools/pr_reviewer.py +++ b/pr_agent/tools/pr_reviewer.py @@ -14,7 +14,7 @@ from pr_agent.git_providers import get_git_provider class PRReviewer: - def __init__(self, pr_url: str, installation_id: Optional[int] = None): + def __init__(self, pr_url: str, installation_id: Optional[int] = None, cli_mode=False): self.git_provider = get_git_provider()(pr_url, installation_id) self.main_language = self.git_provider.get_main_pr_language() @@ -22,6 +22,7 @@ class PRReviewer: self.ai_handler = AiHandler() self.patches_diff = None self.prediction = None + self.cli_mode = cli_mode self.vars = { "title": self.git_provider.pr.title, "branch": self.git_provider.get_pr_branch(), @@ -91,16 +92,19 @@ class PRReviewer: markdown_text = convert_to_markdown(data) user = self.git_provider.get_user_id() - markdown_text += "\n### How to use\n" - if user and '[bot]' not in user: - markdown_text += f"> Tag me in a comment '@{user}' to ask for a new review after you update the PR.\n" - markdown_text += "> You can also tag me and ask any question, " \ - f"for example '@{user} is the PR ready for merge?'" - else: - markdown_text += "> Add a comment that says 'review' to ask for a new review " \ - "after you update the PR.\n" - markdown_text += "> You can also add a comment that says 'answer QUESTION', " \ - "for example 'answer is the PR ready for merge?'" + + if not self.cli_mode: + markdown_text += "\n### How to use\n" + if user and '[bot]' not in user: + markdown_text += "\n### How to use\n" + markdown_text += f"> Tag me in a comment '@{user}' to ask for a new review after you update the PR.\n" + markdown_text += "> You can also tag me and ask any question, " \ + f"for example '@{user} is the PR ready for merge?'" + else: + markdown_text += "> Add a comment that says 'review' to ask for a new review " \ + "after you update the PR.\n" + markdown_text += "> You can also add a comment that says 'answer QUESTION', " \ + "for example 'answer is the PR ready for merge?'" if settings.config.verbosity_level >= 2: logging.info(f"Markdown response:\n{markdown_text}")