@workermill/agent 0.6.1 → 0.7.0

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
 // ============================================================================
 /**
@@ -183,8 +205,8 @@ export function runCriticCli(claudePath, model, prompt, env) {
         });
         const timeout = setTimeout(() => {
             proc.kill("SIGTERM");
-            reject(new Error("Critic CLI timed out after 10 minutes"));
-        }, 600_000);
+            reject(new Error("Critic CLI timed out after 20 minutes"));
+        }, 1_200_000);
         proc.on("exit", (code) => {
             clearTimeout(timeout);
             if (code !== 0) {
@@ -262,7 +284,7 @@ export async function runCriticValidation(claudePath, model, prd, plan, env, tas
             if (!providerApiKey) {
                 throw new Error(`No API key for critic provider "${effectiveProvider}"`);
             }
-            rawCriticOutput = await generateText(effectiveProvider, model, criticPrompt, providerApiKey, { maxTokens: 4096, temperature: 0.3, timeoutMs: 600_000 });
+            rawCriticOutput = await generateText(effectiveProvider, model, criticPrompt, providerApiKey, { maxTokens: 4096, temperature: 0.3, timeoutMs: 1_200_000 });
         }
         const result = parseCriticResponse(rawCriticOutput);
         const statusIcon = result.score >= AUTO_APPROVAL_THRESHOLD

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;
@@ -546,7 +546,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 +578,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 +659,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/dist/poller.js

@@ -250,7 +250,11 @@ export function startPolling(config) {
  */
 export function startHeartbeat(config) {
     setInterval(async () => {
-        const activeTaskIds = getActiveTaskIds();
+        // Include BOTH running containers AND tasks being planned/managed
+        const containerTaskIds = getActiveTaskIds();
+        const planningTaskIds = Array.from(planningInProgress);
+        const managerTaskIds = Array.from(managerInProgress);
+        const activeTaskIds = [...containerTaskIds, ...planningTaskIds, ...managerTaskIds];
         try {
             const response = await api.post("/api/agent/heartbeat", {
                 agentId: config.agentId,

package/package.json

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