astrocode-workflow 0.1.35 → 0.1.40

package/dist/agents/prompts.d.ts

@@ -1,2 +1,2 @@
 export declare const BASE_ORCH_PROMPT = "You are Astro (Orchestrator) for Astrocode.\n\nMission:\n- Advance a deterministic pipeline: frame \u2192 plan \u2192 spec \u2192 implement \u2192 review \u2192 verify \u2192 close.\n- The SQLite DB is the source of truth. Prefer tools over prose.\n- Never narrate what prompts you received.\n- Keep outputs short; store large outputs as artifacts and reference paths.\n\nOperating rules:\n- Only start new runs when the user explicitly requests implementation, workflow management, or story processing.\n- Answer questions directly when possible without starting workflows.\n- Prefer calling astro_workflow_proceed (step/loop) and astro_status only when actively managing a workflow.\n- Delegate stage work only to the stage subagent matching the current stage.\n- If a stage subagent returns status=blocked, inject the BLOCKED directive and stop.\n- Never delegate from subagents (enforced by permissions).\n- Be discretionary: assess if the user's request requires workflow initiation or just information.\n";
-export declare const BASE_STAGE_PROMPT = "You are an Astro stage subagent.\n\n***CRITICAL: Follow the latest [SYSTEM DIRECTIVE: ASTROCODE \u2014 STAGE_*] you receive EXACTLY.***\n\nYour output MUST be in this EXACT format (copy/paste this template):\n\n1) First, a short summary paragraph (1-3 sentences).\n\n2) Then, immediately after, the ASTRO JSON between markers:\n\n<!-- ASTRO_JSON_BEGIN -->\n{\n  \"stage_key\": \"CURRENT_STAGE\",\n  \"status\": \"ok\",\n  \"summary\": \"Brief summary of work done\",\n  \"tasks\": [],\n  \"files\": [],\n  \"questions\": []\n}\n<!-- ASTRO_JSON_END -->\n\nMANDATORY:\n- stage_key must be \"CURRENT_STAGE\"\n- status must be \"ok\", \"blocked\", or \"failed\"\n- summary must be a non-empty string\n- JSON must be valid syntax\n- Use exact markers shown above\n\nEXAMPLE OUTPUT:\nI analyzed the requirements and identified key components for implementation.\n\n<!-- ASTRO_JSON_BEGIN -->\n{\n  \"stage_key\": \"plan\",\n  \"status\": \"ok\",\n  \"summary\": \"Created implementation plan with 5 main tasks\",\n  \"tasks\": [\n    {\"title\": \"Setup database\", \"complexity\": 3},\n    {\"title\": \"Build API endpoints\", \"complexity\": 5}\n  ],\n  \"files\": [\n    {\"path\": \"schema.prisma\", \"kind\": \"file\"}\n  ],\n  \"questions\": []\n}\n<!-- ASTRO_JSON_END -->\n\nIf blocked, set status=\"blocked\" and add ONE question to questions array. Do not deviate from this format.\n";
+export declare const BASE_STAGE_PROMPT = "You are an Astro stage subagent.\n\n***CRITICAL: Follow the latest [SYSTEM DIRECTIVE: ASTROCODE \u2014 STAGE_*] you receive EXACTLY.***\n\nYour output MUST be in this EXACT format:\n\n1) First, a short summary paragraph (1-3 sentences).\n\n2) Then, immediately after, the ASTRO JSON between markers:\n\n<!-- ASTRO_JSON_BEGIN -->\n{\n  \"stage_key\": \"CURRENT_STAGE\",\n  \"status\": \"ok\",\n  \"summary\": \"Brief summary of work done\",\n  \"tasks\": [],\n  \"files\": [],\n  \"questions\": []\n}\n<!-- ASTRO_JSON_END -->\n\nMANDATORY:\n- stage_key must be \"CURRENT_STAGE\"\n- status must be \"ok\", \"blocked\", or \"failed\"\n- summary must be a non-empty string\n- tasks, files, questions must be ARRAYS (use [] for empty)\n- JSON must be valid syntax\n- Use exact markers shown above\n\nARRAY FORMAT EXAMPLES:\n- tasks: [{\"title\": \"Task name\", \"complexity\": 3}]\n- files: [{\"path\": \"file.js\", \"kind\": \"file\"}]\n- questions: [\"What framework to use?\"]\n\nEXAMPLE OUTPUT:\nI analyzed the requirements and identified key components for implementation.\n\n<!-- ASTRO_JSON_BEGIN -->\n{\n  \"stage_key\": \"plan\",\n  \"status\": \"ok\",\n  \"summary\": \"Created implementation plan with 5 main tasks\",\n  \"tasks\": [\n    {\"title\": \"Setup database\", \"complexity\": 3},\n    {\"title\": \"Build API endpoints\", \"complexity\": 5}\n  ],\n  \"files\": [\n    {\"path\": \"schema.prisma\", \"kind\": \"file\"}\n  ],\n  \"questions\": []\n}\n<!-- ASTRO_JSON_END -->\n\nIf blocked, set status=\"blocked\" and add ONE question to questions array. Do not deviate from this format.\n";

package/dist/agents/prompts.js

@@ -19,7 +19,7 @@ export const BASE_STAGE_PROMPT = `You are an Astro stage subagent.
 
 ***CRITICAL: Follow the latest [SYSTEM DIRECTIVE: ASTROCODE — STAGE_*] you receive EXACTLY.***
 
-Your output MUST be in this EXACT format (copy/paste this template):
+Your output MUST be in this EXACT format:
 
 1) First, a short summary paragraph (1-3 sentences).
 
@@ -40,9 +40,15 @@ MANDATORY:
 - stage_key must be "CURRENT_STAGE"
 - status must be "ok", "blocked", or "failed"
 - summary must be a non-empty string
+- tasks, files, questions must be ARRAYS (use [] for empty)
 - JSON must be valid syntax
 - Use exact markers shown above
 
+ARRAY FORMAT EXAMPLES:
+- tasks: [{"title": "Task name", "complexity": 3}]
+- files: [{"path": "file.js", "kind": "file"}]
+- questions: ["What framework to use?"]
+
 EXAMPLE OUTPUT:
 I analyzed the requirements and identified key components for implementation.
 

package/dist/tools/stage.js

@@ -126,10 +126,10 @@ Expected format:
   "summary": "Brief summary",
   "decisions": [],
   "next_actions": [],
-  "tasks": [],
-  "files": [],
-  "evidence": [],
-  "new_stories": [],
+  "tasks": [{"title": "Task name", "complexity": 3}],
+  "files": [{"path": "file.js", "kind": "file"}],
+  "evidence": [{"path": "test.js", "kind": "evidence"}],
+  "new_stories": [{"title": "Story title", "body_md": "Description"}],
   "questions": [],
   "metrics": {}
 }
@@ -137,9 +137,8 @@ Expected format:
 
 Ensure JSON has required fields (stage_key, status) and valid syntax.`;
             }
-            if (parsed.astro_json.stage_key !== sk) {
-                return `❌ Stage Key Mismatch: ASTRO JSON has stage_key="${parsed.astro_json.stage_key}" but expected "${sk}". Update the JSON to match the current stage.`;
-            }
+            // Override stage_key to match the expected stage (agents sometimes get this wrong)
+            parsed.astro_json.stage_key = sk;
             // Context validation (warnings, not errors)
             if (parsed.astro_json.run_id && parsed.astro_json.run_id !== rid) {
                 console.warn(`[Astrocode] ⚠️ Run ID mismatch in baton: expected "${rid}", got "${parsed.astro_json.run_id}". Proceeding anyway.`);

package/dist/tools/workflow.js

@@ -234,6 +234,18 @@ export function createAstroWorkflowProceedTool(opts) {
                     // Visible injection so user can see state
                     if (sessionId) {
                         await injectChatPrompt({ ctx, sessionId, text: delegatePrompt, agent: "Astro" });
+                        // Inject continuation guidance
+                        const continueMessage = [
+                            `[SYSTEM DIRECTIVE: ASTROCODE — AWAIT_STAGE_COMPLETION]`,
+                            ``,
+                            `Stage ${next.stage_key} delegated to ${agentName}.`,
+                            ``,
+                            `When ${agentName} completes, call:`,
+                            `astro_stage_complete(run_id="${active.run_id}", stage_key="${next.stage_key}", output_text="[paste subagent output here]")`,
+                            ``,
+                            `This advances the workflow.`,
+                        ].join("\n");
+                        await injectChatPrompt({ ctx, sessionId, text: continueMessage, agent: "Astro" });
                     }
                     actions.push(`delegated stage ${next.stage_key} via ${agentName}`);
                     // Stop here; subagent needs to run.

package/dist/workflow/baton.js

@@ -75,11 +75,26 @@ export function parseStageOutputText(text) {
         catch (e) {
             // Not JSON, proceed with marker error
         }
+        // If no JSON found, create a default structure from the text
+        const defaultJson = {
+            schema_version: 1,
+            stage_key: "frame", // Default, will be overridden by actual stage_key
+            status: "ok",
+            summary: norm.trim() || "Stage completed without structured output",
+            decisions: [],
+            next_actions: [],
+            tasks: [],
+            files: [],
+            evidence: [],
+            new_stories: [],
+            questions: [],
+            metrics: {}
+        };
         return {
-            baton_md: norm,
-            astro_json: null,
-            astro_json_raw: null,
-            error: `Missing ASTRO JSON markers. Expected markers ${ASTRO_JSON_BEGIN} ... ${ASTRO_JSON_END}. If output contains JSON, ensure it's valid.`,
+            baton_md: "",
+            astro_json: defaultJson,
+            astro_json_raw: JSON.stringify(defaultJson, null, 2),
+            error: null
         };
     }
     const before = norm.slice(0, beginIdx).trim();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "astrocode-workflow",
-  "version": "0.1.35",
+  "version": "0.1.40",
   "type": "module",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/agents/prompts.ts

@@ -20,7 +20,7 @@ export const BASE_STAGE_PROMPT = `You are an Astro stage subagent.
 
 ***CRITICAL: Follow the latest [SYSTEM DIRECTIVE: ASTROCODE — STAGE_*] you receive EXACTLY.***
 
-Your output MUST be in this EXACT format (copy/paste this template):
+Your output MUST be in this EXACT format:
 
 1) First, a short summary paragraph (1-3 sentences).
 
@@ -41,9 +41,15 @@ MANDATORY:
 - stage_key must be "CURRENT_STAGE"
 - status must be "ok", "blocked", or "failed"
 - summary must be a non-empty string
+- tasks, files, questions must be ARRAYS (use [] for empty)
 - JSON must be valid syntax
 - Use exact markers shown above
 
+ARRAY FORMAT EXAMPLES:
+- tasks: [{"title": "Task name", "complexity": 3}]
+- files: [{"path": "file.js", "kind": "file"}]
+- questions: ["What framework to use?"]
+
 EXAMPLE OUTPUT:
 I analyzed the requirements and identified key components for implementation.
 

package/src/tools/stage.ts

@@ -162,10 +162,10 @@ Expected format:
   "summary": "Brief summary",
   "decisions": [],
   "next_actions": [],
-  "tasks": [],
-  "files": [],
-  "evidence": [],
-  "new_stories": [],
+  "tasks": [{"title": "Task name", "complexity": 3}],
+  "files": [{"path": "file.js", "kind": "file"}],
+  "evidence": [{"path": "test.js", "kind": "evidence"}],
+  "new_stories": [{"title": "Story title", "body_md": "Description"}],
   "questions": [],
   "metrics": {}
 }
@@ -174,9 +174,8 @@ Expected format:
 Ensure JSON has required fields (stage_key, status) and valid syntax.`;
        }
 
-        if (parsed.astro_json.stage_key !== sk) {
-          return `❌ Stage Key Mismatch: ASTRO JSON has stage_key="${parsed.astro_json.stage_key}" but expected "${sk}". Update the JSON to match the current stage.`;
-        }
+        // Override stage_key to match the expected stage (agents sometimes get this wrong)
+        parsed.astro_json.stage_key = sk;
 
         // Context validation (warnings, not errors)
         if (parsed.astro_json.run_id && parsed.astro_json.run_id !== rid) {

package/src/tools/workflow.ts

@@ -282,12 +282,26 @@ export function createAstroWorkflowProceedTool(opts: { ctx: any; config: Astroco
             ).run(sessionId, active.run_id, h, `delegate ${next.stage_key}`, now);
           }
 
-          // Visible injection so user can see state
-          if (sessionId) {
-            await injectChatPrompt({ ctx, sessionId, text: delegatePrompt, agent: "Astro" });
-          }
+           // Visible injection so user can see state
+           if (sessionId) {
+             await injectChatPrompt({ ctx, sessionId, text: delegatePrompt, agent: "Astro" });
+
+             // Inject continuation guidance
+             const continueMessage = [
+               `[SYSTEM DIRECTIVE: ASTROCODE — AWAIT_STAGE_COMPLETION]`,
+               ``,
+               `Stage ${next.stage_key} delegated to ${agentName}.`,
+               ``,
+               `When ${agentName} completes, call:`,
+               `astro_stage_complete(run_id="${active.run_id}", stage_key="${next.stage_key}", output_text="[paste subagent output here]")`,
+               ``,
+               `This advances the workflow.`,
+             ].join("\n");
+
+             await injectChatPrompt({ ctx, sessionId, text: continueMessage, agent: "Astro" });
+           }
 
-          actions.push(`delegated stage ${next.stage_key} via ${agentName}`);
+           actions.push(`delegated stage ${next.stage_key} via ${agentName}`);
 
           // Stop here; subagent needs to run.
           break;

package/src/workflow/baton.ts

@@ -111,11 +111,27 @@ export function parseStageOutputText(text: string): ParsedStageOutput {
       // Not JSON, proceed with marker error
     }
 
+    // If no JSON found, create a default structure from the text
+    const defaultJson: AstroJson = {
+      schema_version: 1,
+      stage_key: "frame", // Default, will be overridden by actual stage_key
+      status: "ok",
+      summary: norm.trim() || "Stage completed without structured output",
+      decisions: [],
+      next_actions: [],
+      tasks: [],
+      files: [],
+      evidence: [],
+      new_stories: [],
+      questions: [],
+      metrics: {}
+    };
+
     return {
-      baton_md: norm,
-      astro_json: null,
-      astro_json_raw: null,
-      error: `Missing ASTRO JSON markers. Expected markers ${ASTRO_JSON_BEGIN} ... ${ASTRO_JSON_END}. If output contains JSON, ensure it's valid.`,
+      baton_md: "",
+      astro_json: defaultJson,
+      astro_json_raw: JSON.stringify(defaultJson, null, 2),
+      error: null
     };
   }