@workermill/agent 0.6.2 → 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
 // ============================================================================
 /**

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/package.json

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