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

Don't add "How to use" when running from the command line

Browse Source
This commit is contained in:
Ori Kotek
2023-07-06 17:52:12 +03:00
parent 6648c04799
commit d04c0f490c
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/cli.py
View File

@ -15,11 +15,11 @@ def run():
logging.basicConfig(level=os.environ.get("LOGLEVEL", "INFO")) logging.basicConfig(level=os.environ.get("LOGLEVEL", "INFO"))
if args.question: if args.question:
print(f"Question: {args.question} about PR {args.pr_url}") 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()) asyncio.run(reviewer.answer())
else: else:
print(f"Reviewing PR: {args.pr_url}") 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()) asyncio.run(reviewer.review())

7
pr_agent/tools/pr_reviewer.py
View File

@ -14,7 +14,7 @@ from pr_agent.git_providers import get_git_provider
class PRReviewer: 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.git_provider = get_git_provider()(pr_url, installation_id)
self.main_language = self.git_provider.get_main_pr_language() self.main_language = self.git_provider.get_main_pr_language()
@ -22,6 +22,7 @@ class PRReviewer:
self.ai_handler = AiHandler() self.ai_handler = AiHandler()
self.patches_diff = None self.patches_diff = None
self.prediction = None self.prediction = None
self.cli_mode = cli_mode
self.vars = { self.vars = {
"title": self.git_provider.pr.title, "title": self.git_provider.pr.title,
"branch": self.git_provider.get_pr_branch(), "branch": self.git_provider.get_pr_branch(),
@ -92,7 +93,9 @@ class PRReviewer:
markdown_text = convert_to_markdown(data) markdown_text = convert_to_markdown(data)
user = self.git_provider.get_user_id() user = self.git_provider.get_user_id()
markdown_text += "\n### How to use\n" 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 += 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, " \ markdown_text += "> You can also tag me and ask any question, " \
f"for example '@{user} is the PR ready for merge?'" f"for example '@{user} is the PR ready for merge?'"