mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-07-02 20:00:41 +08:00
26 lines
971 B
Python
26 lines
971 B
Python
import re
|
|
from typing import Optional
|
|
|
|
from pr_agent.tools.pr_questions import PRQuestions
|
|
from pr_agent.tools.pr_reviewer import PRReviewer
|
|
|
|
|
|
class PRAgent:
|
|
def __init__(self, installation_id: Optional[int] = None):
|
|
self.installation_id = installation_id
|
|
|
|
async def handle_request(self, pr_url, request):
|
|
if 'please review' in request.lower() or 'review' == request.lower().strip() or len(request) == 0:
|
|
reviewer = PRReviewer(pr_url, self.installation_id)
|
|
await reviewer.review()
|
|
|
|
else:
|
|
if "please answer" in request.lower():
|
|
question = re.split(r'(?i)please answer', request)[1].strip()
|
|
elif request.lower().strip().startswith("answer"):
|
|
question = re.split(r'(?i)answer', request)[1].strip()
|
|
else:
|
|
question = request
|
|
answerer = PRQuestions(pr_url, question, self.installation_id)
|
|
await answerer.answer()
|