fix: workflow dispatch with pr

Author: zakahan

.github/workflows/agentkit-build.yaml

@@ -1,22 +1,14 @@
 name: agentkit-build
 
 on:
-  pull_request:
-    types:
-      - closed
   workflow_dispatch:
     inputs:
-      base_sha:
-        description: "Base commit SHA (optional)"
-        required: false
-      head_sha:
-        description: "Head commit SHA (optional)"
-        required: false
+      pr_number:
+        description: "Pull request number"
+        required: true
 
 jobs:
   agentkit-build:
-    if: >
-      ${{ (github.event_name == 'pull_request' && github.event.pull_request.merged == true) || github.event_name == 'workflow_dispatch' }}
     runs-on: ubuntu-latest
 
     env:
@@ -28,6 +20,41 @@ jobs:
         with:
           fetch-depth: 0
 
+      - name: Resolve BASE/HEAD from PR number
+        if: github.event.inputs.pr_number != ''
+        env:
+          PR_NUMBER: ${{ github.event.inputs.pr_number }}
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          python - << 'PY'
+          import json
+          import os
+          import urllib.request
+
+          repo = os.environ["GITHUB_REPOSITORY"]
+          pr_number = os.environ["PR_NUMBER"]
+          token = os.environ["GITHUB_TOKEN"]
+
+          url = f"https://api.github.com/repos/{repo}/pulls/{pr_number}"
+          req = urllib.request.Request(
+              url,
+              headers={
+                  "Authorization": f"Bearer {token}",
+                  "Accept": "application/vnd.github+json",
+              },
+          )
+          with urllib.request.urlopen(req) as resp:
+              data = json.load(resp)
+
+          base_sha = data["base"]["sha"]
+          head_sha = data["head"]["sha"]
+
+          github_env = os.environ["GITHUB_ENV"]
+          with open(github_env, "a", encoding="utf-8") as f:
+              f.write(f"BASE_SHA={base_sha}\n")
+              f.write(f"HEAD_SHA={head_sha}\n")
+          PY
+
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
@@ -45,8 +72,8 @@ jobs:
 
       - name: Run main.py in changed use-case directories (build)
         env:
-          BASE_SHA: ${{ github.event_name == 'pull_request' && github.event.pull_request.base.sha || github.event.inputs.base_sha }}
-          HEAD_SHA: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.event.inputs.head_sha }}
+          BASE_SHA: ${{ env.BASE_SHA }}
+          HEAD_SHA: ${{ env.HEAD_SHA }}
           AGENTKIT_COMMAND: build
         run: |
           python -m workflow_utils.check_usecases

.github/workflows/agentkit-check.yaml

@@ -1,15 +1,11 @@
 name: agentkit-check
 
 on:
-  pull_request:
   workflow_dispatch:
     inputs:
-      base_sha:
-        description: "Base commit SHA (optional)"
-        required: false
-      head_sha:
-        description: "Head commit SHA (optional)"
-        required: false
+      pr_number:
+        description: "Pull request number"
+        required: true
 
 jobs:
   agentkit-launch:
@@ -39,9 +35,44 @@ jobs:
           cat > .env << 'EOF'
           EOF
 
+      - name: Resolve BASE/HEAD from PR number
+        if: github.event.inputs.pr_number != ''
+        env:
+          PR_NUMBER: ${{ github.event.inputs.pr_number }}
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          python - << 'PY'
+          import json
+          import os
+          import urllib.request
+
+          repo = os.environ["GITHUB_REPOSITORY"]
+          pr_number = os.environ["PR_NUMBER"]
+          token = os.environ["GITHUB_TOKEN"]
+
+          url = f"https://api.github.com/repos/{repo}/pulls/{pr_number}"
+          req = urllib.request.Request(
+              url,
+              headers={
+                  "Authorization": f"Bearer {token}",
+                  "Accept": "application/vnd.github+json",
+              },
+          )
+          with urllib.request.urlopen(req) as resp:
+              data = json.load(resp)
+
+          base_sha = data["base"]["sha"]
+          head_sha = data["head"]["sha"]
+
+          github_env = os.environ["GITHUB_ENV"]
+          with open(github_env, "a", encoding="utf-8") as f:
+              f.write(f"BASE_SHA={base_sha}\n")
+              f.write(f"HEAD_SHA={head_sha}\n")
+          PY
+
       - name: Run main.py in changed use-case directories
         env:
-          BASE_SHA: ${{ github.event_name == 'pull_request' && github.event.pull_request.base.sha || github.event.inputs.base_sha }}
-          HEAD_SHA: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.event.inputs.head_sha }}
+          BASE_SHA: ${{ env.BASE_SHA }}
+          HEAD_SHA: ${{ env.HEAD_SHA }}
         run: |
           python -m workflow_utils.check_usecases