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

added github action support

Browse Source
This commit is contained in:
mrT23
2024-02-16 12:58:55 +02:00
parent 3eef0a4ebd
commit 40fbd55da4
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pr_agent/servers/github_action_runner.py
View File

@ -8,6 +8,7 @@ from pr_agent.config_loader import get_settings
from pr_agent.git_providers import get_git_provider
from pr_agent.git_providers.utils import apply_repo_settings
from pr_agent.log import get_logger
from pr_agent.servers.github_app import handle_line_comments
from pr_agent.tools.pr_code_suggestions import PRCodeSuggestions
from pr_agent.tools.pr_description import PRDescription
from pr_agent.tools.pr_reviewer import PRReviewer
@ -114,6 +115,13 @@ async def run_action():
is_pr = True
else:
url = event_payload.get("issue", {}).get("url")
try:
if 'subject_type' in event_payload["comment"] and event_payload["comment"]["subject_type"] == "line":
comment_body = handle_line_comments(event_payload, comment_body)
except Exception as e:
get_logger().error(f"Failed to handle line comments: {e}")
if url:
body = comment_body.strip().lower()
comment_id = event_payload.get("comment", {}).get("id")

10
pr_agent/servers/github_app.py
View File

@ -100,8 +100,11 @@ async def handle_request(body: Dict[str, Any], event: str):
api_url = body["issue"]["pull_request"]["url"]
elif "comment" in body and "pull_request_url" in body["comment"]:
api_url = body["comment"]["pull_request_url"]
if 'subject_type' in body["comment"] and body["comment"]["subject_type"] == "line":
comment_body = await handle_line_comments(action, body, comment_body, event)
try:
if 'subject_type' in body["comment"] and body["comment"]["subject_type"] == "line":
comment_body = handle_line_comments(body, comment_body)
except Exception as e:
get_logger().error(f"Failed to handle line comments: {e}")
else:
return {}
@ -193,7 +196,7 @@ async def handle_request(body: Dict[str, Any], event: str):
return {}
async def handle_line_comments(action, body, comment_body, event):
def handle_line_comments(body, comment_body):
# handle line comments
start_line = body["comment"]["start_line"]
end_line = body["comment"]["line"]
@ -205,7 +208,6 @@ async def handle_line_comments(action, body, comment_body, event):
side = body["comment"]["side"]
comment_id = body["comment"]["id"]
if '/ask' in comment_body:
get_logger().info(f"Handling line comment because of event={event} and action={action}")
comment_body = f"/ask_line --line_start={start_line} --line_end={end_line} --side={side} --file_name={path} --comment_id={comment_id} {question}"
return comment_body