From 835d3109ce575b20c2d56f021a47c2cb91a79b06 Mon Sep 17 00:00:00 2001
From: Brezn <brezn@evalcode.com>
Date: Tue, 21 Apr 2026 10:26:51 -0600
Subject: [PATCH] Add Wrap-Up guidance and delivery mode handling

---
 src/index.ts | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index c362cb3..1599dbc 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -47,6 +47,12 @@ function formatAvailabilitySummary(contract: Contract | null, availability: Sche
     parts.push(`Active availability window: ${availability.activeWindow.label} (${availability.activeWindow.mode})`);
   }
 
+  if (availability.wrapUpGuidance?.active) {
+    const remaining = availability.wrapUpGuidance.remainingMinutes;
+    const timing = typeof remaining === "number" ? `${remaining}m remaining` : "active";
+    parts.push(`Wrap-Up guidance: ${timing} (${availability.wrapUpGuidance.selectedMode})`);
+  }
+
   if (availability.nextWindow) {
     parts.push(`Next availability window: ${availability.nextWindow.label} (${availability.nextWindow.mode})`);
   }
@@ -89,7 +95,8 @@ export const HeadsDownOpenCodePlugin: Plugin = async () => {
           estimated_files: tool.schema.number().int().positive().optional(),
           estimated_minutes: tool.schema.number().int().positive().optional(),
           scope_summary: tool.schema.string().min(3).optional(),
-          source_ref: tool.schema.string().min(2).optional()
+          source_ref: tool.schema.string().min(2).optional(),
+          delivery_mode: tool.schema.enum(["auto", "wrap_up", "full_depth"]).optional()
         },
         async execute(args) {
           const client = await getAuthenticatedClient();
@@ -100,7 +107,8 @@ export const HeadsDownOpenCodePlugin: Plugin = async () => {
             estimatedFiles: args.estimated_files,
             estimatedMinutes: args.estimated_minutes,
             scopeSummary: args.scope_summary,
-            sourceRef: args.source_ref
+            sourceRef: args.source_ref,
+            deliveryMode: args.delivery_mode
           });
 
           if (verdict.decision === "approved") {
@@ -132,7 +140,8 @@ export const HeadsDownOpenCodePlugin: Plugin = async () => {
               decision: verdict.decision,
               reason: verdict.reason,
               proposalId: verdict.proposalId,
-              evaluatedAt: verdict.evaluatedAt
+              evaluatedAt: verdict.evaluatedAt,
+              wrapUpGuidance: verdict.wrapUpGuidance
             },
             null,
             2