From d50c2a3bf4812272e9be5c2c51726d554e72c12d Mon Sep 17 00:00:00 2001
From: Pigbibi <20649888+Pigbibi@users.noreply.github.com>
Date: Thu, 2 Apr 2026 03:16:44 +0800
Subject: [PATCH] fix: prepare planner payloads from ai review artifacts

---
 .../monthly_optimization_planner.yml          | 24 +++++++++++++++++--
 ...ly_optimization_planner_workflow_config.py |  6 ++++-
 2 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/monthly_optimization_planner.yml b/.github/workflows/monthly_optimization_planner.yml
index e990a3f..562ee6f 100644
--- a/.github/workflows/monthly_optimization_planner.yml
+++ b/.github/workflows/monthly_optimization_planner.yml
@@ -58,11 +58,31 @@ jobs:
             echo "downstream_dir=${DOWNSTREAM_DIR}"
           } >> "${GITHUB_OUTPUT}"
 
+      - name: Prepare upstream review payload
+        run: |
+          python3 scripts/build_ai_review_payload.py \
+            --source-repo "${GITHUB_REPOSITORY}" \
+            --review-kind upstream_selector \
+            --issue-context-file "${{ steps.artifact_paths.outputs.upstream_dir }}/issue_context.json" \
+            --secondary-review-file "${{ steps.artifact_paths.outputs.upstream_dir }}/secondary_review.json" \
+            --run-url "${GITHUB_SERVER_URL}/${GITHUB_REPOSITORY}/actions/runs/${{ inputs.upstream_run_id }}" \
+            --output-file data/output/prepared/upstream_review_payload.json
+
+      - name: Prepare downstream review payload
+        run: |
+          python3 scripts/build_ai_review_payload.py \
+            --source-repo "${{ inputs.downstream_repo }}" \
+            --review-kind execution_runtime \
+            --issue-context-file "${{ steps.artifact_paths.outputs.downstream_dir }}/issue_context.json" \
+            --secondary-review-file "${{ steps.artifact_paths.outputs.downstream_dir }}/secondary_review.json" \
+            --run-url "${GITHUB_SERVER_URL}/${{ inputs.downstream_repo }}/actions/runs/${{ inputs.downstream_run_id }}" \
+            --output-file data/output/prepared/downstream_review_payload.json
+
       - name: Build monthly optimization plan
         run: |
           python3 scripts/build_monthly_optimization_plan.py \
-            --upstream-review-file "${{ steps.artifact_paths.outputs.upstream_dir }}/final_review_payload.json" \
-            --downstream-review-file "${{ steps.artifact_paths.outputs.downstream_dir }}/final_review_payload.json" \
+            --upstream-review-file data/output/prepared/upstream_review_payload.json \
+            --downstream-review-file data/output/prepared/downstream_review_payload.json \
             --output-dir data/output/monthly_optimization
 
       - name: Append optimization summary
diff --git a/tests/test_monthly_optimization_planner_workflow_config.py b/tests/test_monthly_optimization_planner_workflow_config.py
index cf9920b..ddc1427 100644
--- a/tests/test_monthly_optimization_planner_workflow_config.py
+++ b/tests/test_monthly_optimization_planner_workflow_config.py
@@ -20,9 +20,13 @@ def test_planner_workflow_downloads_artifacts_and_posts_issue(self) -> None:
         self.assertIn("CROSS_REPO_GITHUB_TOKEN", workflow)
         self.assertIn("gh run download", workflow)
         self.assertIn("Resolve downloaded artifact paths", workflow)
+        self.assertIn("Prepare upstream review payload", workflow)
+        self.assertIn("Prepare downstream review payload", workflow)
+        self.assertIn("build_ai_review_payload.py", workflow)
         self.assertIn("build_monthly_optimization_plan.py", workflow)
         self.assertIn("post_monthly_optimization_issue.py", workflow)
-        self.assertIn("final_review_payload.json", workflow)
+        self.assertIn("upstream_review_payload.json", workflow)
+        self.assertIn("downstream_review_payload.json", workflow)
         self.assertIn("actions/upload-artifact@v7", workflow)
         self.assertIn("monthly-optimization-plan-", workflow)