Merge pull request #47 from Codium-ai/feature/github_action

Github custom action development
2025-07-19 12:00:37 +08:00 · 2023-07-16 12:54:40 +03:00
parent 84bf95e9ab ea1cd7ae45
commit 5d2fe07bf7
5 changed files with 91 additions and 0 deletions
--- a/pr_agent/servers/github_action_runner.py
+++ b/pr_agent/servers/github_action_runner.py
@ -0,0 +1,58 @@
+import asyncio
+import json
+import os
+
+from pr_agent.config_loader import settings
+from pr_agent.tools.pr_questions import PRQuestions
+from pr_agent.tools.pr_reviewer import PRReviewer
+
+
+async def run_action():
+    GITHUB_EVENT_NAME = os.environ.get('GITHUB_EVENT_NAME', None)
+    if not GITHUB_EVENT_NAME:
+        print("GITHUB_EVENT_NAME not set")
+        return
+    GITHUB_EVENT_PATH = os.environ.get('GITHUB_EVENT_PATH', None)
+    if not GITHUB_EVENT_PATH:
+        print("GITHUB_EVENT_PATH not set")
+        return
+    event_payload = json.load(open(GITHUB_EVENT_PATH, 'r'))
+    RUNNER_DEBUG = os.environ.get('RUNNER_DEBUG', None)
+    if not RUNNER_DEBUG:
+        print("RUNNER_DEBUG not set")
+    OPENAI_KEY = os.environ.get('OPENAI_KEY', None)
+    if not OPENAI_KEY:
+        print("OPENAI_KEY not set")
+        return
+    OPENAI_ORG = os.environ.get('OPENAI_ORG', None)
+    GITHUB_TOKEN = os.environ.get('GITHUB_TOKEN', None)
+    if not GITHUB_TOKEN:
+        print("GITHUB_TOKEN not set")
+        return
+    settings.set("OPENAI.KEY", OPENAI_KEY)
+    if OPENAI_ORG:
+        settings.set("OPENAI.ORG", OPENAI_ORG)
+    settings.set("GITHUB.USER_TOKEN", GITHUB_TOKEN)
+    settings.set("GITHUB.DEPLOYMENT_TYPE", "user")
+    if GITHUB_EVENT_NAME == "pull_request":
+        action = event_payload.get("action", None)
+        if action in ["opened", "reopened"]:
+            pr_url = event_payload.get("pull_request", {}).get("url", None)
+            if pr_url:
+                await PRReviewer(pr_url).review()
+
+    elif GITHUB_EVENT_NAME == "issue_comment":
+        action = event_payload.get("action", None)
+        if action in ["created", "edited"]:
+            comment_body = event_payload.get("comment", {}).get("body", None)
+            if comment_body:
+                pr_url = event_payload.get("issue", {}).get("pull_request", {}).get("url", None)
+                if pr_url:
+                    if comment_body.strip().lower() == "review":
+                        await PRReviewer(pr_url).review()
+                    elif comment_body.lstrip().lower().startswith("answer"):
+                        await PRQuestions(pr_url, comment_body).answer()
+
+
+if __name__ == '__main__':
+    asyncio.run(run_action())