xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-06 22:00:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Don't commment on Github, only eyes reaction

Browse Source
This commit is contained in:
Ori Kotek
2023-08-07 18:09:39 +03:00
parent b1148e5f7a
commit ebbe655c40
4 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pr_agent/agent/pr_agent.py
View File

@ -37,7 +37,7 @@ class PRAgent:
def __init__(self): def __init__(self):
pass pass
async def handle_request(self, pr_url, request) -> bool: async def handle_request(self, pr_url, request, notify=None) -> bool:
# First, apply repo specific settings if exists # First, apply repo specific settings if exists
if get_settings().config.use_repo_settings_file: if get_settings().config.use_repo_settings_file:
repo_settings_file = None repo_settings_file = None
@ -67,8 +67,12 @@ class PRAgent:
if action == "reflect_and_review" and not get_settings().pr_reviewer.ask_and_reflect: if action == "reflect_and_review" and not get_settings().pr_reviewer.ask_and_reflect:
action = "review" action = "review"
if action == "answer": if action == "answer":
if notify:
notify()
await PRReviewer(pr_url, is_answer=True, args=args).run() await PRReviewer(pr_url, is_answer=True, args=args).run()
elif action in command2class: elif action in command2class:
if notify:
notify()
await command2class[action](pr_url, args=args).run() await command2class[action](pr_url, args=args).run()
else: else:
return False return False

4
pr_agent/servers/github_action_runner.py
View File

@ -17,6 +17,7 @@ async def run_action():
GITHUB_TOKEN = os.environ.get('GITHUB_TOKEN') GITHUB_TOKEN = os.environ.get('GITHUB_TOKEN')
get_settings().set("CONFIG.PUBLISH_OUTPUT_PROGRESS", False) get_settings().set("CONFIG.PUBLISH_OUTPUT_PROGRESS", False)
# Check if required environment variables are set # Check if required environment variables are set
if not GITHUB_EVENT_NAME: if not GITHUB_EVENT_NAME:
print("GITHUB_EVENT_NAME not set") print("GITHUB_EVENT_NAME not set")
@ -65,8 +66,7 @@ async def run_action():
body = comment_body.strip().lower() body = comment_body.strip().lower()
comment_id = event_payload.get("comment", {}).get("id") comment_id = event_payload.get("comment", {}).get("id")
provider = get_git_provider()(pr_url=pr_url) provider = get_git_provider()(pr_url=pr_url)
provider.add_eyes_reaction(comment_id) await PRAgent().handle_request(pr_url, body, notify=lambda: provider.add_eyes_reaction(comment_id))
await PRAgent().handle_request(pr_url, body)
if __name__ == '__main__': if __name__ == '__main__':

3
pr_agent/servers/github_app.py
View File

@ -83,8 +83,7 @@ async def handle_request(body: Dict[str, Any]):
api_url = pull_request.get("url") api_url = pull_request.get("url")
comment_id = body.get("comment", {}).get("id") comment_id = body.get("comment", {}).get("id")
provider = get_git_provider()(pr_url=api_url) provider = get_git_provider()(pr_url=api_url)
provider.add_eyes_reaction(comment_id) await agent.handle_request(api_url, comment_body, notify=lambda: provider.add_eyes_reaction(comment_id))
await agent.handle_request(api_url, comment_body)
elif action == "opened" or 'reopened' in action: elif action == "opened" or 'reopened' in action:

4
pr_agent/servers/github_polling.py
View File

@ -101,8 +101,8 @@ async def polling_loop():
rest_of_comment = comment_body.split(user_tag)[1].strip() rest_of_comment = comment_body.split(user_tag)[1].strip()
comment_id = comment['id'] comment_id = comment['id']
git_provider.set_pr(pr_url) git_provider.set_pr(pr_url)
git_provider.add_eyes_reaction(comment_id) success = await agent.handle_request(pr_url, rest_of_comment,
success = await agent.handle_request(pr_url, rest_of_comment) notify=lambda: git_provider.add_eyes_reaction(comment_id)) # noqa E501
if not success: if not success:
git_provider.set_pr(pr_url) git_provider.set_pr(pr_url)
git_provider.publish_comment("### How to use PR-Agent\n" + git_provider.publish_comment("### How to use PR-Agent\n" +