Github custom action development - WIP

This commit is contained in:
Ori Kotek
2023-07-13 18:32:28 +03:00
parent 4033303c1f
commit f337d76af6
2 changed files with 25 additions and 12 deletions

View File

@ -8,6 +8,6 @@ jobs:
- name: PR Agent action step
id: pragent
uses: Codium-ai/pr-agent@feature/github_action
with:
openai-key: "TBD: openai key"
env:
OPENAI_KEY: ${{ secrets.OPENAI_KEY }}
OPENAI_ORG: ${{ secrets.OPENAI_ORG }}

View File

@ -1,6 +1,9 @@
import json
import os
from pr_agent.config_loader import settings
from pr_agent.tools.pr_reviewer import PRReviewer
def run_action():
GITHUB_EVENT_NAME = os.environ.get('GITHUB_EVENT_NAME', None)
@ -12,22 +15,32 @@ def run_action():
print("GITHUB_EVENT_PATH not set")
return
event_payload = json.load(open(GITHUB_EVENT_PATH, 'r'))
GITHUB_REPOSITORY = os.environ.get('GITHUB_REPOSITORY', None)
if not GITHUB_REPOSITORY:
print("GITHUB_REPOSITORY not set")
return
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
### DEBUG
print(event_payload)
print(GITHUB_REPOSITORY)
print(GITHUB_EVENT_NAME)
print(GITHUB_EVENT_PATH)
print(RUNNER_DEBUG)
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":
print("PR event")
action = event_payload.get("action", None)
if action in ["opened", "reopened"]:
pr_url = event_payload.get("pull_request", {}).get("url", None)
if pr_url:
PRReviewer(pr_url).review()
if __name__ == '__main__':