@workermill/agent 0.6.2 → 0.7.1

package/dist/config.d.ts

@@ -17,6 +17,7 @@ export interface AgentConfig {
     gitlabToken: string;
     workerImage: string;
     teamPlanningEnabled: boolean;
+    analystModel: string;
 }
 export interface FileConfig {
     apiUrl: string;
@@ -32,6 +33,7 @@ export interface FileConfig {
     };
     workerImage: string;
     teamPlanningEnabled?: boolean;
+    analystModel?: string;
     setupCompletedAt: string;
 }
 export declare function getConfigDir(): string;

package/dist/config.js

@@ -76,6 +76,7 @@ export function loadConfigFromFile() {
         gitlabToken: fc.tokens?.gitlab || "",
         workerImage,
         teamPlanningEnabled: fc.teamPlanningEnabled ?? true,
+        analystModel: fc.analystModel || "sonnet",
     };
 }
 /**
@@ -121,6 +122,7 @@ export function loadConfig() {
         gitlabToken: process.env.GITLAB_TOKEN || "",
         workerImage: process.env.WORKER_IMAGE || "workermill-worker:local",
         teamPlanningEnabled: process.env.TEAM_PLANNING_ENABLED !== "false",
+        analystModel: process.env.ANALYST_MODEL || "sonnet",
     };
 }
 /**

package/dist/plan-validator.d.ts

@@ -53,6 +53,14 @@ export declare function applyFileCap(plan: ExecutionPlan): {
     truncatedCount: number;
     details: string[];
 };
+/**
+ * Apply story cap to the plan. Truncates stories beyond maxStories.
+ * Returns details about dropped stories for logging.
+ */
+export declare function applyStoryCap(plan: ExecutionPlan, maxStories: number): {
+    droppedCount: number;
+    details: string[];
+};
 /**
  * Re-serialize plan as a JSON code block for posting to the API.
  * The server-side parseExecutionPlan() expects ```json ... ``` blocks.

package/dist/plan-validator.js

@@ -59,6 +59,28 @@ export function applyFileCap(plan) {
     return { truncatedCount, details };
 }
 // ============================================================================
+// STORY CAP
+// ============================================================================
+/**
+ * Apply story cap to the plan. Truncates stories beyond maxStories.
+ * Returns details about dropped stories for logging.
+ */
+export function applyStoryCap(plan, maxStories) {
+    if (plan.stories.length <= maxStories) {
+        return { droppedCount: 0, details: [] };
+    }
+    const droppedCount = plan.stories.length - maxStories;
+    const dropped = plan.stories.slice(maxStories);
+    const details = dropped.map((s) => `${s.id}: "${s.title}" (${s.persona})`);
+    plan.stories = plan.stories.slice(0, maxStories);
+    // Fix dependencies that reference dropped stories
+    const validIds = new Set(plan.stories.map((s) => s.id));
+    for (const story of plan.stories) {
+        story.dependencies = story.dependencies.filter((dep) => validIds.has(dep));
+    }
+    return { droppedCount, details };
+}
+// ============================================================================
 // PLAN SERIALIZATION
 // ============================================================================
 /**

package/dist/planner.js

@@ -18,7 +18,7 @@ import chalk from "chalk";
 import { spawn, execSync } from "child_process";
 import { findClaudePath } from "./config.js";
 import { api } from "./api.js";
-import { parseExecutionPlan, applyFileCap, serializePlan, runCriticValidation, formatCriticFeedback, AUTO_APPROVAL_THRESHOLD, } from "./plan-validator.js";
+import { parseExecutionPlan, applyFileCap, applyStoryCap, serializePlan, runCriticValidation, formatCriticFeedback, AUTO_APPROVAL_THRESHOLD, } from "./plan-validator.js";
 import { generateText } from "./providers.js";
 /** Max Planner-Critic iterations before giving up */
 const MAX_ITERATIONS = 3;
@@ -154,10 +154,45 @@ function runClaudeCli(claudePath, model, prompt, env, taskId, startTime) {
                     continue;
                 try {
                     const event = JSON.parse(trimmed);
-                    if (event.type === "content_block_delta" && event.delta?.text) {
+                    // Claude CLI stream-json wraps content in assistant message events
+                    if (event.type === "assistant" && event.message?.content) {
+                        const content = event.message.content;
+                        if (Array.isArray(content)) {
+                            for (const block of content) {
+                                if (block.type === "text" && block.text) {
+                                    fullText += block.text;
+                                    charsReceived += block.text.length;
+                                    if (!firstTextSeen) {
+                                        firstTextSeen = true;
+                                        if (toolCallCount > 0 && !milestoneSent.analyzing) {
+                                            transitionPhase("analyzing");
+                                            milestoneSent.analyzing = true;
+                                        }
+                                    }
+                                    if (charsReceived > 500 && !milestoneSent.generating) {
+                                        transitionPhase("generating_plan");
+                                        milestoneSent.generating = true;
+                                        lastProgressLogAt = Math.round((Date.now() - startTime) / 1000);
+                                    }
+                                }
+                                else if (block.type === "tool_use") {
+                                    toolCallCount++;
+                                    if (!milestoneSent.reading) {
+                                        transitionPhase("reading_repo");
+                                        milestoneSent.reading = true;
+                                    }
+                                }
+                            }
+                        }
+                        else if (typeof content === "string" && content) {
+                            fullText += content;
+                            charsReceived += content.length;
+                        }
+                    }
+                    else if (event.type === "content_block_delta" && event.delta?.text) {
+                        // Fallback: raw API streaming format
                         fullText += event.delta.text;
                         charsReceived += event.delta.text.length;
-                        // Phase: first text after tool calls → analyzing
                         if (!firstTextSeen) {
                             firstTextSeen = true;
                             if (toolCallCount > 0 && !milestoneSent.analyzing) {
@@ -165,7 +200,6 @@ function runClaudeCli(claudePath, model, prompt, env, taskId, startTime) {
                                 milestoneSent.analyzing = true;
                             }
                         }
-                        // Phase: substantial text → generating_plan
                         if (charsReceived > 500 && !milestoneSent.generating) {
                             transitionPhase("generating_plan");
                             milestoneSent.generating = true;
@@ -179,13 +213,6 @@ function runClaudeCli(claudePath, model, prompt, env, taskId, startTime) {
                             milestoneSent.reading = true;
                         }
                     }
-                    else if (event.type === "assistant" && event.message?.content) {
-                        const text = typeof event.message.content === "string" ? event.message.content : "";
-                        if (text) {
-                            fullText += text;
-                            charsReceived += text.length;
-                        }
-                    }
                     else if (event.type === "result" && event.result) {
                         resultText = typeof event.result === "string" ? event.result : "";
                     }
@@ -340,7 +367,27 @@ function runAnalyst(name, claudePath, model, prompt, repoPath, env, timeoutMs =
                     continue;
                 try {
                     const event = JSON.parse(trimmed);
-                    if (event.type === "content_block_delta" && event.delta?.text) {
+                    // Claude CLI stream-json wraps content in assistant message events
+                    if (event.type === "assistant" && event.message?.content) {
+                        const content = event.message.content;
+                        if (Array.isArray(content)) {
+                            for (const block of content) {
+                                if (block.type === "text" && block.text) {
+                                    fullText += block.text;
+                                }
+                                else if (block.type === "tool_use") {
+                                    toolCalls++;
+                                    const toolName = block.name || "unknown";
+                                    console.log(`${ts()} ${label} ${chalk.dim(`Tool: ${toolName}`)} (${toolCalls} total)`);
+                                }
+                            }
+                        }
+                        else if (typeof content === "string") {
+                            fullText += content;
+                        }
+                    }
+                    else if (event.type === "content_block_delta" && event.delta?.text) {
+                        // Fallback: raw API streaming format (may appear in some CLI versions)
                         fullText += event.delta.text;
                     }
                     else if (event.type === "content_block_start" && event.content_block?.type === "tool_use") {
@@ -546,7 +593,8 @@ export async function planTask(task, config, credentials) {
     const promptResponse = await api.get("/api/agent/planning-prompt", {
         params: { taskId: task.id },
     });
-    const { prompt: basePrompt, model, provider: planningProvider } = promptResponse.data;
+    const { prompt: basePrompt, model, provider: planningProvider, maxStories: apiMaxStories } = promptResponse.data;
+    const maxStories = typeof apiMaxStories === "number" ? apiMaxStories : 8;
     const cliModel = model || "sonnet";
     const provider = (planningProvider || "anthropic");
     const isAnthropicPlanning = provider === "anthropic";
@@ -577,7 +625,9 @@ export async function planTask(task, config, credentials) {
             console.log(`${ts()} ${taskLabel} ${chalk.yellow("⚠")} No SCM token for ${scmProvider}, skipping team planning`);
         }
         if (repoPath) {
-            const analysisResult = await runTeamAnalysis(task, basePrompt, claudePath, cliModel, cleanEnv, repoPath, task.id, startTime);
+            const analystModel = config.analystModel || "sonnet";
+            console.log(`${ts()} ${taskLabel} Analysts using model: ${chalk.yellow(analystModel)} (planner: ${chalk.yellow(cliModel)})`);
+            const analysisResult = await runTeamAnalysis(task, basePrompt, claudePath, analystModel, cleanEnv, repoPath, task.id, startTime);
             if (analysisResult) {
                 enhancedBasePrompt = analysisResult;
             }
@@ -656,7 +706,17 @@ export async function planTask(task, config, credentials) {
                     console.log(`${ts()} ${taskLabel}   ${chalk.dim(detail)}`);
                 }
             }
-            console.log(`${ts()} ${taskLabel} Plan: ${chalk.bold(plan.stories.length)} stories`);
+            // 2c2. Apply story cap (max stories from org calibration)
+            const { droppedCount: storyDropCount, details: storyDropDetails } = applyStoryCap(plan, maxStories);
+            if (storyDropCount > 0) {
+                const msg = `${PREFIX} Story cap applied: ${storyDropCount} stories dropped (max ${maxStories})`;
+                console.log(`${ts()} ${taskLabel} ${chalk.yellow("⚠")} ${msg}`);
+                await postLog(task.id, msg);
+                for (const detail of storyDropDetails) {
+                    console.log(`${ts()} ${taskLabel}   ${chalk.dim(detail)}`);
+                }
+            }
+            console.log(`${ts()} ${taskLabel} Plan: ${chalk.bold(plan.stories.length)} stories (max ${maxStories})`);
             await postLog(task.id, `${PREFIX} Plan generated: ${plan.stories.length} stories (${formatElapsed(elapsed)}). Running critic validation...`);
             // 2d. Run critic validation
             const criticResult = await runCriticValidation(claudePath, cliModel, prd, plan, cleanEnv, taskLabel, provider, providerApiKey);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@workermill/agent",
-  "version": "0.6.2",
+  "version": "0.7.1",
   "description": "WorkerMill Remote Agent - Run AI workers locally with your Claude Max subscription",
   "type": "module",
   "main": "./dist/index.js",