astrocode-workflow 0.1.23 → 0.1.27

package/dist/hooks/continuation-enforcer.js

@@ -117,12 +117,12 @@ export function createContinuationEnforcer(opts) {
         await recordContinuation(sessionId, active.run_id, directive, reason);
         // Injection mode
         if (config.continuation.injection_mode === "visible") {
-            await injectChatPrompt({ ctx, sessionId, text: directive.body });
+            await injectChatPrompt({ ctx, sessionId, text: directive.body, agent: "Astro" });
         }
         else {
             // Silent mode is TODO: requires experimental.chat.messages.transform.
             // For v2-alpha, we fall back to visible injection but mark it.
-            await injectChatPrompt({ ctx, sessionId, text: directive.body + "\n\n(Injected in silent mode fallback)" });
+            await injectChatPrompt({ ctx, sessionId, text: directive.body + "\n\n(Injected in silent mode fallback)", agent: "Astro" });
         }
         if (config.ui.toasts.enabled && config.ui.toasts.show_auto_continue) {
             await toasts.show({ title: "Astrocode", message: "Continue directive injected", variant: "info" });

package/dist/tools/stage.js

@@ -27,7 +27,6 @@ function splitTasksIntoStories(db, tasks, run, now, newStoryKeys, relationReason
         const subtasks = task.subtasks ?? [];
         if (subtasks.length > 0) {
             // Split into subtasks
-            console.log(`[Astrocode] Splitting task "${task.title}" into ${subtasks.length} subtasks`);
             for (const subtask of subtasks) {
                 const key = insertStory(db, {
                     title: `${task.title}: ${subtask}`,
@@ -37,13 +36,11 @@ function splitTasksIntoStories(db, tasks, run, now, newStoryKeys, relationReason
                     epic_key: run.story_key
                 });
                 newStoryKeys.push(key);
-                console.log(`[Astrocode] Created story ${key} for subtask`);
                 db.prepare("INSERT INTO story_relations (parent_story_key, child_story_key, relation_type, reason, created_at) VALUES (?, ?, ?, ?, ?)").run(run.story_key, key, "split", relationReason, now);
             }
         }
         else if (complexity > 6) {
             // Split complex tasks
-            console.log(`[Astrocode] Splitting complex task "${task.title}" (complexity ${complexity})`);
             const key = insertStory(db, {
                 title: task.title,
                 body_md: task.description ?? "",
@@ -52,7 +49,6 @@ function splitTasksIntoStories(db, tasks, run, now, newStoryKeys, relationReason
                 epic_key: run.story_key
             });
             newStoryKeys.push(key);
-            console.log(`[Astrocode] Created story ${key} for complex task`);
             db.prepare("INSERT INTO story_relations (parent_story_key, child_story_key, relation_type, reason, created_at) VALUES (?, ?, ?, ?, ?)").run(run.story_key, key, "split", relationReason, now);
         }
     }
@@ -233,7 +229,6 @@ Ensure JSON has required fields (stage_key, status) and valid syntax.`;
                 }
                 // Automatic story splitting for complex tasks
                 if (allow_new_stories && parsed.astro_json.tasks?.length) {
-                    console.log(`[Astrocode] Splitting ${parsed.astro_json.tasks.length} tasks into stories`);
                     splitTasksIntoStories(db, parsed.astro_json.tasks, run, now, newStoryKeys, "split from stage output");
                 }
                 // Skip spec stage if spec already exists

package/dist/tools/story.js

@@ -13,10 +13,8 @@ export function createAstroStoryQueueTool(opts) {
             priority: tool.schema.number().int().default(0),
         },
         execute: async ({ title, body_md, epic_key, priority }) => {
-            console.log(`[Astrocode] Creating story: ${title}`);
             const story_key = withTx(db, () => {
                 const key = insertStory(db, { title, body_md, epic_key: epic_key ?? null, priority: priority ?? 0, state: 'queued' });
-                console.log(`[Astrocode] Inserted story ${key} into DB`);
                 return key;
             });
             return `✅ Queued story ${story_key}: ${title}`;

package/dist/tools/workflow.js

@@ -122,6 +122,26 @@ export function createAstroWorkflowProceedTool(opts) {
                     if (config.ui.toasts.enabled && config.ui.toasts.show_run_completed) {
                         await toasts.show({ title: "Astrocode", message: `Run completed (${next.run_id})`, variant: "success" });
                     }
+                    // Check for next approved story to start
+                    const nextStory = db.prepare("SELECT story_key, title FROM stories WHERE state = 'approved' ORDER BY priority DESC, created_at ASC LIMIT 1").get();
+                    if (nextStory && sessionId) {
+                        const nextDirective = [
+                            `[SYSTEM DIRECTIVE: ASTROCODE — START_NEXT_STORY]`,
+                            ``,
+                            `The previous run completed successfully. Start the next approved story.`,
+                            ``,
+                            `Next Story: ${nextStory.story_key} — ${nextStory.title}`,
+                            ``,
+                            `Action: Call astro_story_approve with story_key="${nextStory.story_key}" to start it, or select a different story.`,
+                        ].join("\n");
+                        await injectChatPrompt({
+                            ctx,
+                            sessionId,
+                            text: nextDirective,
+                            agent: "Astro"
+                        });
+                        actions.push(`injected directive to start next story ${nextStory.story_key}`);
+                    }
                     if (mode === "step")
                         break;
                     continue;
@@ -154,16 +174,12 @@ export function createAstroWorkflowProceedTool(opts) {
                         return false;
                     };
                     if (!agentExists(agentName)) {
-                        console.warn(`[Astrocode] Agent ${agentName} not found in config. Falling back to General.`);
+                        console.warn(`[Astrocode] Agent ${agentName} not found in config. Falling back to orchestrator instead of General.`);
                         console.warn(`[Astrocode] Agent check: config.agent exists: ${!!systemConfig.agent}, agentName in config: ${systemConfig.agent ? agentName in systemConfig.agent : 'N/A'}`);
-                        agentName = "General";
-                        // Second fallback to orchestrator if General also unavailable
+                        // Skip General fallback for stage agents to avoid malformed output
+                        agentName = config.agents?.orchestrator_name || "Astro";
                         if (!agentExists(agentName)) {
-                            console.warn(`[Astrocode] General agent also unavailable. Falling back to orchestrator.`);
-                            agentName = config.agents?.orchestrator_name || "Astro";
-                            if (!agentExists(agentName)) {
-                                throw new Error(`Critical: No agents available for delegation. Primary: ${resolveAgentName(next.stage_key, config)}, General, Orchestrator: ${agentName}`);
-                            }
+                            throw new Error(`Critical: No agents available for delegation. Primary: ${resolveAgentName(next.stage_key, config)}, Orchestrator: ${agentName}`);
                         }
                     }
                     console.log(`[Astrocode] Delegating stage ${next.stage_key} to agent: ${agentName}`);
@@ -199,7 +215,7 @@ export function createAstroWorkflowProceedTool(opts) {
                     }
                     // Visible injection so user can see state
                     if (sessionId) {
-                        await injectChatPrompt({ ctx, sessionId, text: delegatePrompt });
+                        await injectChatPrompt({ ctx, sessionId, text: delegatePrompt, agent: "Astro" });
                     }
                     actions.push(`delegated stage ${next.stage_key} via ${agentName}`);
                     // Stop here; subagent needs to run.
@@ -222,7 +238,7 @@ export function createAstroWorkflowProceedTool(opts) {
                         const h = directiveHash(prompt);
                         const now = nowISO();
                         db.prepare("INSERT INTO continuations (session_id, run_id, directive_hash, kind, reason, created_at) VALUES (?, ?, ?, 'continue', ?, ?)").run(sessionId, next.run_id, h, `await ${next.stage_key}`, now);
-                        await injectChatPrompt({ ctx, sessionId, text: prompt });
+                        await injectChatPrompt({ ctx, sessionId, text: prompt, agent: "Astro" });
                     }
                     break;
                 }

package/dist/ui/inject.d.ts

@@ -2,4 +2,5 @@ export declare function injectChatPrompt(opts: {
     ctx: any;
     sessionId: string;
     text: string;
+    agent?: string;
 }): Promise<void>;

package/dist/ui/inject.js

@@ -1,12 +1,12 @@
 let isInjecting = false;
 let queuedInjection = null;
 export async function injectChatPrompt(opts) {
-    const { ctx, sessionId, text } = opts;
+    const { ctx, sessionId, text, agent = "Astro" } = opts;
     // Prefix to clearly indicate source as Astro agent
-    const prefixedText = `[Astro Agent]\n\n${text}`;
+    const prefixedText = `[${agent}]\n\n${text}`;
     if (isInjecting) {
         // Replace any existing queued injection (keep only latest)
-        queuedInjection = { ...opts, text: prefixedText };
+        queuedInjection = { ...opts, text: prefixedText, agent };
         return;
     }
     isInjecting = true;
@@ -15,6 +15,8 @@ export async function injectChatPrompt(opts) {
             path: { id: sessionId },
             body: {
                 parts: [{ type: "text", text: prefixedText }],
+                // Pass agent context for systems that support it
+                agent: agent,
             },
         });
     }

package/dist/workflow/state-machine.js

@@ -52,7 +52,6 @@ function isInitialStory(db, storyKey) {
     return relations.count === 0;
 }
 export function createRunForStory(db, config, storyKey) {
-    console.log(`[Astrocode] Creating run for story: ${storyKey}`);
     const story = getStory(db, storyKey);
     if (!story)
         throw new Error(`Story not found: ${storyKey}`);
@@ -61,21 +60,17 @@ export function createRunForStory(db, config, storyKey) {
     const run_id = newRunId();
     const now = nowISO();
     const pipeline = config.workflow.pipeline;
-    console.log(`[Astrocode] Generated run ID: ${run_id}`);
     // Convert to genesis planning story if needed
     const isGenesisCandidate = storyKey === 'S-0001' || isInitialStory(db, storyKey) ||
         (story.body_md && story.body_md.length > 100 &&
             (story.title.toLowerCase().includes('implement') || story.body_md.toLowerCase().includes('implement')));
     if (isGenesisCandidate) {
-        console.log(`[Astrocode] Converting story ${storyKey} to genesis planning story`);
         const planningTitle = `Plan and decompose: ${story.title}`;
         const planningBody = `Analyze the requirements and break down "${story.title}" into 50-200 detailed, granular implementation stories. Each story should be focused on a specific, implementable task with clear acceptance criteria.\n\nOriginal request: ${story.body_md || ''}`;
         db.prepare("UPDATE stories SET title=?, body_md=? WHERE story_key=?").run(planningTitle, planningBody, storyKey);
     }
     // Lock story
-    console.log(`[Astrocode] Locking story ${storyKey} for run ${run_id}`);
     db.prepare("UPDATE stories SET state='in_progress', in_progress=1, locked_by_run_id=?, locked_at=?, updated_at=? WHERE story_key=?").run(run_id, now, now, storyKey);
-    console.log(`[Astrocode] Inserting run ${run_id} into DB`);
     db.prepare("INSERT INTO runs (run_id, story_key, status, pipeline_stages_json, current_stage_key, created_at, started_at, updated_at) VALUES (?, ?, 'running', ?, ?, ?, ?, ?)").run(run_id, storyKey, JSON.stringify(pipeline), pipeline[0] ?? null, now, now, now);
     // Stage runs
     const insertStage = db.prepare("INSERT INTO stage_runs (stage_run_id, run_id, stage_key, stage_index, status, updated_at) VALUES (?, ?, ?, ?, 'pending', ?)");

package/package.json

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

package/src/hooks/continuation-enforcer.ts

@@ -166,11 +166,11 @@ export function createContinuationEnforcer(opts: {
 
     // Injection mode
     if (config.continuation.injection_mode === "visible") {
-      await injectChatPrompt({ ctx, sessionId, text: directive.body });
+      await injectChatPrompt({ ctx, sessionId, text: directive.body, agent: "Astro" });
     } else {
       // Silent mode is TODO: requires experimental.chat.messages.transform.
       // For v2-alpha, we fall back to visible injection but mark it.
-      await injectChatPrompt({ ctx, sessionId, text: directive.body + "\n\n(Injected in silent mode fallback)" });
+      await injectChatPrompt({ ctx, sessionId, text: directive.body + "\n\n(Injected in silent mode fallback)", agent: "Astro" });
     }
 
     if (config.ui.toasts.enabled && config.ui.toasts.show_auto_continue) {

package/src/tools/stage.ts

@@ -39,7 +39,6 @@ function splitTasksIntoStories(
     const subtasks = task.subtasks ?? [];
     if (subtasks.length > 0) {
       // Split into subtasks
-      console.log(`[Astrocode] Splitting task "${task.title}" into ${subtasks.length} subtasks`);
       for (const subtask of subtasks) {
         const key = insertStory(db, {
           title: `${task.title}: ${subtask}`,
@@ -49,7 +48,6 @@ function splitTasksIntoStories(
           epic_key: run.story_key
         });
         newStoryKeys.push(key);
-        console.log(`[Astrocode] Created story ${key} for subtask`);
         db.prepare(
           "INSERT INTO story_relations (parent_story_key, child_story_key, relation_type, reason, created_at) VALUES (?, ?, ?, ?, ?)"
         ).run(
@@ -62,7 +60,6 @@ function splitTasksIntoStories(
       }
     } else if (complexity > 6) {
       // Split complex tasks
-      console.log(`[Astrocode] Splitting complex task "${task.title}" (complexity ${complexity})`);
       const key = insertStory(db, {
         title: task.title,
         body_md: task.description ?? "",
@@ -71,7 +68,6 @@ function splitTasksIntoStories(
         epic_key: run.story_key
       });
       newStoryKeys.push(key);
-      console.log(`[Astrocode] Created story ${key} for complex task`);
       db.prepare(
         "INSERT INTO story_relations (parent_story_key, child_story_key, relation_type, reason, created_at) VALUES (?, ?, ?, ?, ?)"
       ).run(
@@ -313,7 +309,6 @@ Ensure JSON has required fields (stage_key, status) and valid syntax.`;
 
         // Automatic story splitting for complex tasks
         if (allow_new_stories && parsed.astro_json.tasks?.length) {
-          console.log(`[Astrocode] Splitting ${parsed.astro_json.tasks.length} tasks into stories`);
           splitTasksIntoStories(db, parsed.astro_json.tasks, run, now, newStoryKeys, "split from stage output");
         }
 

package/src/tools/story.ts

@@ -20,11 +20,8 @@ export function createAstroStoryQueueTool(opts: { ctx: any; config: AstrocodeCon
       priority: tool.schema.number().int().default(0),
     },
     execute: async ({ title, body_md, epic_key, priority }) => {
-      console.log(`[Astrocode] Creating story: ${title}`);
-
       const story_key = withTx(db, () => {
         const key = insertStory(db, { title, body_md, epic_key: epic_key ?? null, priority: priority ?? 0, state: 'queued' });
-        console.log(`[Astrocode] Inserted story ${key} into DB`);
         return key;
       });
 

package/src/tools/workflow.ts

@@ -153,6 +153,32 @@ export function createAstroWorkflowProceedTool(opts: { ctx: any; config: Astroco
             await toasts.show({ title: "Astrocode", message: `Run completed (${next.run_id})`, variant: "success" });
           }
 
+          // Check for next approved story to start
+          const nextStory = db.prepare(
+            "SELECT story_key, title FROM stories WHERE state = 'approved' ORDER BY priority DESC, created_at ASC LIMIT 1"
+          ).get() as { story_key: string; title: string } | undefined;
+
+          if (nextStory && sessionId) {
+            const nextDirective = [
+              `[SYSTEM DIRECTIVE: ASTROCODE — START_NEXT_STORY]`,
+              ``,
+              `The previous run completed successfully. Start the next approved story.`,
+              ``,
+              `Next Story: ${nextStory.story_key} — ${nextStory.title}`,
+              ``,
+              `Action: Call astro_story_approve with story_key="${nextStory.story_key}" to start it, or select a different story.`,
+            ].join("\n");
+
+            await injectChatPrompt({
+              ctx,
+              sessionId,
+              text: nextDirective,
+              agent: "Astro"
+            });
+
+            actions.push(`injected directive to start next story ${nextStory.story_key}`);
+          }
+
           if (mode === "step") break;
           continue;
         }
@@ -187,19 +213,15 @@ export function createAstroWorkflowProceedTool(opts: { ctx: any; config: Astroco
              return false;
           };
 
-          if (!agentExists(agentName)) {
-            console.warn(`[Astrocode] Agent ${agentName} not found in config. Falling back to General.`);
-            console.warn(`[Astrocode] Agent check: config.agent exists: ${!!systemConfig.agent}, agentName in config: ${systemConfig.agent ? agentName in systemConfig.agent : 'N/A'}`);
-            agentName = "General";
-            // Second fallback to orchestrator if General also unavailable
-            if (!agentExists(agentName)) {
-              console.warn(`[Astrocode] General agent also unavailable. Falling back to orchestrator.`);
-              agentName = config.agents?.orchestrator_name || "Astro";
-              if (!agentExists(agentName)) {
-                throw new Error(`Critical: No agents available for delegation. Primary: ${resolveAgentName(next.stage_key, config)}, General, Orchestrator: ${agentName}`);
-              }
-            }
-          }
+           if (!agentExists(agentName)) {
+             console.warn(`[Astrocode] Agent ${agentName} not found in config. Falling back to orchestrator instead of General.`);
+             console.warn(`[Astrocode] Agent check: config.agent exists: ${!!systemConfig.agent}, agentName in config: ${systemConfig.agent ? agentName in systemConfig.agent : 'N/A'}`);
+             // Skip General fallback for stage agents to avoid malformed output
+             agentName = config.agents?.orchestrator_name || "Astro";
+             if (!agentExists(agentName)) {
+               throw new Error(`Critical: No agents available for delegation. Primary: ${resolveAgentName(next.stage_key, config)}, Orchestrator: ${agentName}`);
+             }
+           }
 
           console.log(`[Astrocode] Delegating stage ${next.stage_key} to agent: ${agentName}`);
 
@@ -243,7 +265,7 @@ export function createAstroWorkflowProceedTool(opts: { ctx: any; config: Astroco
 
           // Visible injection so user can see state
           if (sessionId) {
-            await injectChatPrompt({ ctx, sessionId, text: delegatePrompt });
+            await injectChatPrompt({ ctx, sessionId, text: delegatePrompt, agent: "Astro" });
           }
 
           actions.push(`delegated stage ${next.stage_key} via ${agentName}`);
@@ -272,7 +294,7 @@ export function createAstroWorkflowProceedTool(opts: { ctx: any; config: Astroco
               "INSERT INTO continuations (session_id, run_id, directive_hash, kind, reason, created_at) VALUES (?, ?, ?, 'continue', ?, ?)"
             ).run(sessionId, next.run_id, h, `await ${next.stage_key}`, now);
 
-            await injectChatPrompt({ ctx, sessionId, text: prompt });
+            await injectChatPrompt({ ctx, sessionId, text: prompt, agent: "Astro" });
           }
           break;
         }

package/src/ui/inject.ts

@@ -1,19 +1,20 @@
 let isInjecting = false;
-let queuedInjection: { ctx: any; sessionId: string; text: string } | null = null;
+let queuedInjection: { ctx: any; sessionId: string; text: string; agent?: string } | null = null;
 
 export async function injectChatPrompt(opts: {
   ctx: any;
   sessionId: string;
   text: string;
+  agent?: string;
 }) {
-  const { ctx, sessionId, text } = opts;
+  const { ctx, sessionId, text, agent = "Astro" } = opts;
 
   // Prefix to clearly indicate source as Astro agent
-  const prefixedText = `[Astro Agent]\n\n${text}`;
+  const prefixedText = `[${agent}]\n\n${text}`;
 
   if (isInjecting) {
     // Replace any existing queued injection (keep only latest)
-    queuedInjection = { ...opts, text: prefixedText };
+    queuedInjection = { ...opts, text: prefixedText, agent };
     return;
   }
 
@@ -24,6 +25,8 @@ export async function injectChatPrompt(opts: {
       path: { id: sessionId },
       body: {
         parts: [{ type: "text", text: prefixedText }],
+        // Pass agent context for systems that support it
+        agent: agent,
       },
     });
   } finally {

package/src/workflow/state-machine.ts

@@ -81,7 +81,6 @@ function isInitialStory(db: SqliteDb, storyKey: string): boolean {
 }
 
 export function createRunForStory(db: SqliteDb, config: AstrocodeConfig, storyKey: string): { run_id: string } {
-  console.log(`[Astrocode] Creating run for story: ${storyKey}`);
   const story = getStory(db, storyKey);
   if (!story) throw new Error(`Story not found: ${storyKey}`);
   if (story.state !== "approved") throw new Error(`Story must be approved to run: ${storyKey} (state=${story.state})`);
@@ -89,7 +88,6 @@ export function createRunForStory(db: SqliteDb, config: AstrocodeConfig, storyKe
   const run_id = newRunId();
   const now = nowISO();
   const pipeline = config.workflow.pipeline;
-  console.log(`[Astrocode] Generated run ID: ${run_id}`);
 
   // Convert to genesis planning story if needed
   const isGenesisCandidate = storyKey === 'S-0001' || isInitialStory(db, storyKey) ||
@@ -97,19 +95,16 @@ export function createRunForStory(db: SqliteDb, config: AstrocodeConfig, storyKe
       (story.title.toLowerCase().includes('implement') || story.body_md.toLowerCase().includes('implement')));
 
   if (isGenesisCandidate) {
-    console.log(`[Astrocode] Converting story ${storyKey} to genesis planning story`);
     const planningTitle = `Plan and decompose: ${story.title}`;
     const planningBody = `Analyze the requirements and break down "${story.title}" into 50-200 detailed, granular implementation stories. Each story should be focused on a specific, implementable task with clear acceptance criteria.\n\nOriginal request: ${story.body_md || ''}`;
     db.prepare("UPDATE stories SET title=?, body_md=? WHERE story_key=?").run(planningTitle, planningBody, storyKey);
   }
 
   // Lock story
-  console.log(`[Astrocode] Locking story ${storyKey} for run ${run_id}`);
   db.prepare(
     "UPDATE stories SET state='in_progress', in_progress=1, locked_by_run_id=?, locked_at=?, updated_at=? WHERE story_key=?"
   ).run(run_id, now, now, storyKey);
 
-  console.log(`[Astrocode] Inserting run ${run_id} into DB`);
   db.prepare(
     "INSERT INTO runs (run_id, story_key, status, pipeline_stages_json, current_stage_key, created_at, started_at, updated_at) VALUES (?, ?, 'running', ?, ?, ?, ?, ?)"
   ).run(run_id, storyKey, JSON.stringify(pipeline), pipeline[0] ?? null, now, now, now);