From d04c0f490c8577b2860dedee3abdec3168306dbb Mon Sep 17 00:00:00 2001
From: Ori Kotek <ori.k@codium.ai>
Date: Thu, 6 Jul 2023 17:52:12 +0300
Subject: [PATCH] Don't add "How to use" when running from the command line

---
 pr_agent/cli.py               | 4 ++--
 pr_agent/tools/pr_reviewer.py | 7 +++++--
 2 files changed, 7 insertions(+), 4 deletions(-)

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/tools/pr_reviewer.py b/pr_agent/tools/pr_reviewer.py
index a7ebc2ef..33dd0162 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(),
@@ -92,7 +93,9 @@ 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:
+        if self.cli_mode:
+            pass
+        elif 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?'"