astrocode-workflow 0.1.20 → 0.1.24

package/dist/shared/paths.js

@@ -42,9 +42,21 @@ export function assertInsideAstro(repoRoot, filePath) {
     const absRepo = path.resolve(repoRoot);
     const abs = path.resolve(filePath);
     const astroRoot = path.resolve(path.join(repoRoot, ".astro"));
+    // Allow writing certain files in the repo root
+    const relPath = path.relative(repoRoot, filePath);
+    const allowedOutside = [
+        "stories.md",
+        "README.md",
+        "CHANGELOG.md",
+        "CONTRIBUTING.md",
+        "LICENSE",
+        ".gitignore"
+    ];
     if (!abs.startsWith(astroRoot + path.sep) && abs !== astroRoot) {
-        const relPath = path.relative(repoRoot, filePath);
-        throw new Error(`Refusing to write outside .astro: ${filePath} (relative: ${relPath}, astroRoot: ${astroRoot})`);
+        // Check if it's an allowed file in repo root
+        if (!allowedOutside.includes(relPath)) {
+            throw new Error(`Refusing to write outside .astro: ${filePath} (relative: ${relPath}, astroRoot: ${astroRoot})`);
+        }
     }
     if (!abs.startsWith(absRepo + path.sep) && abs !== absRepo) {
         throw new Error(`Refusing to write outside repo root: ${filePath}`);

package/dist/tools/workflow.js

@@ -154,16 +154,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}`);

package/dist/ui/inject.js

@@ -2,9 +2,11 @@ let isInjecting = false;
 let queuedInjection = null;
 export async function injectChatPrompt(opts) {
     const { ctx, sessionId, text } = opts;
+    // Prefix to clearly indicate source as Astro agent
+    const prefixedText = `[Astro Agent]\n\n${text}`;
     if (isInjecting) {
         // Replace any existing queued injection (keep only latest)
-        queuedInjection = opts;
+        queuedInjection = { ...opts, text: prefixedText };
         return;
     }
     isInjecting = true;
@@ -12,7 +14,7 @@ export async function injectChatPrompt(opts) {
         await ctx.client.session.prompt({
             path: { id: sessionId },
             body: {
-                parts: [{ type: "text", text }],
+                parts: [{ type: "text", text: prefixedText }],
             },
         });
     }
@@ -23,7 +25,7 @@ export async function injectChatPrompt(opts) {
             const next = queuedInjection;
             queuedInjection = null;
             // Schedule next injection asynchronously to prevent recursion
-            setImmediate(() => injectChatPrompt(next));
+            setImmediate(() => injectChatPrompt({ ...queuedInjection, text: prefixedText }));
         }
     }
 }

package/dist/workflow/directives.js

@@ -10,6 +10,8 @@ export function buildContinueDirective(opts) {
     const body = clampChars(normalizeNewlines([
         `[SYSTEM DIRECTIVE: ASTROCODE — CONTINUE]`,
         ``,
+        `This directive is injected by the Astro agent to continue the workflow.`,
+        ``,
         `Run: \`${run_id}\`${stage_key ? `  Stage: \`${stage_key}\`` : ""}`,
         ``,
         `Next action: ${next_action}`,
@@ -34,6 +36,8 @@ export function buildBlockedDirective(opts) {
     const body = normalizeNewlines([
         `[SYSTEM DIRECTIVE: ASTROCODE — BLOCKED]`,
         ``,
+        `This directive is injected by the Astro agent indicating the workflow is blocked.`,
+        ``,
         `Run: \`${run_id}\`  Stage: \`${stage_key}\``,
         ``,
         `You are blocked. Ask the user exactly ONE question (below), then stop.`,
@@ -54,6 +58,8 @@ export function buildRepairDirective(opts) {
     const body = normalizeNewlines([
         `[SYSTEM DIRECTIVE: ASTROCODE — REPAIR]`,
         ``,
+        `This directive is injected by the Astro agent after performing a repair pass.`,
+        ``,
         `Astrocode performed a repair pass. Summarize in <= 10 lines and continue.`,
         ``,
         `Repair report:`,
@@ -75,6 +81,8 @@ export function buildStageDirective(opts) {
     const body = clampChars(normalizeNewlines([
         `[SYSTEM DIRECTIVE: ASTROCODE — STAGE_${stageKeyUpper}]`,
         ``,
+        `This directive is injected by the Astro agent to delegate the stage task.`,
+        ``,
         `You are: \`${stage_agent_name}\``,
         `Run: \`${run_id}\``,
         `Story: \`${story_key}\` — ${story_title}`,
@@ -90,7 +98,6 @@ export function buildStageDirective(opts) {
         `     "schema_version": 1,`,
         `     "stage_key": "${stage_key}",`,
         `     "status": "ok",`,
-        `     "summary": "Brief summary of work done",`,
         `     ...`,
         `   }`,
         `   ${"<!-- ASTRO_JSON_END -->"}`,

package/package.json

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

package/src/shared/paths.ts

@@ -61,10 +61,25 @@ export function assertInsideAstro(repoRoot: string, filePath: string) {
   const absRepo = path.resolve(repoRoot);
   const abs = path.resolve(filePath);
   const astroRoot = path.resolve(path.join(repoRoot, ".astro"));
+
+  // Allow writing certain files in the repo root
+  const relPath = path.relative(repoRoot, filePath);
+  const allowedOutside = [
+    "stories.md",
+    "README.md",
+    "CHANGELOG.md",
+    "CONTRIBUTING.md",
+    "LICENSE",
+    ".gitignore"
+  ];
+
   if (!abs.startsWith(astroRoot + path.sep) && abs !== astroRoot) {
-    const relPath = path.relative(repoRoot, filePath);
-    throw new Error(`Refusing to write outside .astro: ${filePath} (relative: ${relPath}, astroRoot: ${astroRoot})`);
+    // Check if it's an allowed file in repo root
+    if (!allowedOutside.includes(relPath)) {
+      throw new Error(`Refusing to write outside .astro: ${filePath} (relative: ${relPath}, astroRoot: ${astroRoot})`);
+    }
   }
+
   if (!abs.startsWith(absRepo + path.sep) && abs !== absRepo) {
     throw new Error(`Refusing to write outside repo root: ${filePath}`);
   }

package/src/tools/workflow.ts

@@ -187,19 +187,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}`);
 

package/src/ui/inject.ts

@@ -8,9 +8,12 @@ export async function injectChatPrompt(opts: {
 }) {
   const { ctx, sessionId, text } = opts;
 
+  // Prefix to clearly indicate source as Astro agent
+  const prefixedText = `[Astro Agent]\n\n${text}`;
+
   if (isInjecting) {
     // Replace any existing queued injection (keep only latest)
-    queuedInjection = opts;
+    queuedInjection = { ...opts, text: prefixedText };
     return;
   }
 
@@ -20,7 +23,7 @@ export async function injectChatPrompt(opts: {
     await ctx.client.session.prompt({
       path: { id: sessionId },
       body: {
-        parts: [{ type: "text", text }],
+        parts: [{ type: "text", text: prefixedText }],
       },
     });
   } finally {
@@ -30,7 +33,7 @@ export async function injectChatPrompt(opts: {
       const next = queuedInjection;
       queuedInjection = null;
       // Schedule next injection asynchronously to prevent recursion
-      setImmediate(() => injectChatPrompt(next));
+      setImmediate(() => injectChatPrompt({ ...queuedInjection, text: prefixedText }));
     }
   }
 }

package/src/workflow/directives.ts

@@ -32,6 +32,8 @@ export function buildContinueDirective(opts: {
       [
         `[SYSTEM DIRECTIVE: ASTROCODE — CONTINUE]`,
         ``,
+        `This directive is injected by the Astro agent to continue the workflow.`,
+        ``,
         `Run: \`${run_id}\`${stage_key ? `  Stage: \`${stage_key}\`` : ""}`,
         ``,
         `Next action: ${next_action}`,
@@ -67,6 +69,8 @@ export function buildBlockedDirective(opts: {
     [
       `[SYSTEM DIRECTIVE: ASTROCODE — BLOCKED]`,
       ``,
+      `This directive is injected by the Astro agent indicating the workflow is blocked.`,
+      ``,
       `Run: \`${run_id}\`  Stage: \`${stage_key}\``,
       ``,
       `You are blocked. Ask the user exactly ONE question (below), then stop.`,
@@ -91,6 +95,8 @@ export function buildRepairDirective(opts: { report_md: string }): BuiltDirectiv
     [
       `[SYSTEM DIRECTIVE: ASTROCODE — REPAIR]`,
       ``,
+      `This directive is injected by the Astro agent after performing a repair pass.`,
+      ``,
       `Astrocode performed a repair pass. Summarize in <= 10 lines and continue.`,
       ``,
       `Repair report:`,
@@ -130,6 +136,8 @@ export function buildStageDirective(opts: {
       [
         `[SYSTEM DIRECTIVE: ASTROCODE — STAGE_${stageKeyUpper}]`,
         ``,
+        `This directive is injected by the Astro agent to delegate the stage task.`,
+        ``,
         `You are: \`${stage_agent_name}\``,
         `Run: \`${run_id}\``,
         `Story: \`${story_key}\` — ${story_title}`,
@@ -145,7 +153,6 @@ export function buildStageDirective(opts: {
         `     "schema_version": 1,`,
         `     "stage_key": "${stage_key}",`,
         `     "status": "ok",`,
-        `     "summary": "Brief summary of work done",`,
         `     ...`,
         `   }`,
         `   ${"<!-- ASTRO_JSON_END -->"}`,