mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-07-02 03:40:38 +08:00
28 lines
925 B
Python
28 lines
925 B
Python
import argparse
|
|
import asyncio
|
|
import logging
|
|
import os
|
|
|
|
from pr_agent.tools.pr_questions import PRQuestions
|
|
from pr_agent.tools.pr_reviewer import PRReviewer
|
|
|
|
|
|
def run():
|
|
parser = argparse.ArgumentParser(description='AI based pull request analyzer')
|
|
parser.add_argument('--pr_url', type=str, help='The URL of the PR to review', required=True)
|
|
parser.add_argument('--question', type=str, help='Optional question to ask', required=False)
|
|
args = parser.parse_args()
|
|
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)
|
|
asyncio.run(reviewer.answer())
|
|
else:
|
|
print(f"Reviewing PR: {args.pr_url}")
|
|
reviewer = PRReviewer(args.pr_url, cli_mode=True)
|
|
asyncio.run(reviewer.review())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run()
|