workermill 0.3.2 → 0.3.3

package/dist/index.js

@@ -743,7 +743,8 @@ function useAgent(options) {
     setTrustAll,
     setPlanMode,
     addSystemMessage,
-    addUserMessage
+    addUserMessage,
+    setCost
   };
 }
 
@@ -797,7 +798,7 @@ var PERSONA_EMOJIS = {
 function getEmoji(persona) {
   return PERSONA_EMOJIS[persona] || "\u{1F916}";
 }
-function useOrchestrator(addMessage2) {
+function useOrchestrator(addMessage2, setCost) {
   const [running, setRunning] = useState2(false);
   const [statusMessage, setStatusMessage] = useState2("");
   const [confirmRequest, setConfirmRequest] = useState2(null);
@@ -816,7 +817,7 @@ function useOrchestrator(addMessage2) {
             setRunning(false);
             return;
           }
-          const { classifyComplexity, runOrchestration } = await import("./orchestrator-A2CDVJAS.js");
+          const { classifyComplexity, runOrchestration } = await import("./orchestrator-VRGIOUHT.js");
           const output = {
             log(persona, message) {
               const emoji = getEmoji(persona);
@@ -880,6 +881,9 @@ function useOrchestrator(addMessage2) {
               addMessage2(
                 `[${emoji} ${persona}] \u2193 ${toolName}${detail ? " " + detail : ""}`
               );
+            },
+            updateCost(cost) {
+              setCost?.(cost);
             }
           };
           addMessage2("Analyzing task complexity\u2026");
@@ -1368,11 +1372,6 @@ function PermissionPrompt({ request }) {
 init_esm_shims();
 import { Box as Box4, Text as Text4, useStdout } from "ink";
 import { jsx as jsx4, jsxs as jsxs4 } from "react/jsx-runtime";
-function formatTokens(n) {
-  if (n >= 1e6) return `${(n / 1e6).toFixed(1)}M`;
-  if (n >= 1e3) return `${(n / 1e3).toFixed(1)}k`;
-  return String(n);
-}
 function formatCost(c) {
   if (c < 0.01) return "$0.00";
   return `$${c.toFixed(2)}`;
@@ -1399,7 +1398,8 @@ function StatusBar(props) {
   }
   const bgColor = theme.subtleDark;
   const modelStr = ` ${props.provider}/${props.model} `;
-  const tokenStr = ` ${formatTokens(props.tokens)} `;
+  const pct = props.maxContext > 0 ? Math.round(usage * 100) : 0;
+  const tokenStr = ` ${pct}% `;
   const costStr = formatCost(props.cost);
   const branchStr = props.gitBranch ? ` git:(${props.gitBranch})` : "";
   const cwdStr = props.cwd ? ` ${props.cwd}` : "";
@@ -1679,7 +1679,7 @@ function Root(props) {
     },
     [agent]
   );
-  const orchestrator = useOrchestrator(addOrchestratorMessage);
+  const orchestrator = useOrchestrator(addOrchestratorMessage, agent.setCost);
   const buildStarted = useRef3(false);
   useEffect2(() => {
     if (props.initialBuildTask && !buildStarted.current) {
@@ -1997,7 +1997,7 @@ function printWelcome(provider, model, workingDir) {
   console.log(dim("  Type ") + white("/help") + dim(" for all commands."));
   console.log();
 }
-var VERSION = "0.3.2";
+var VERSION = "0.3.3";
 function addSharedOptions(cmd) {
   return cmd.option("--provider <provider>", "Override default provider").option("--model <model>", "Override model").option("--trust", "Skip all tool permission prompts").option("--full-disk", "Allow tools to access files outside working directory");
 }

package/dist/{orchestrator-A2CDVJAS.js → orchestrator-VRGIOUHT.js}

@@ -401,6 +401,7 @@ Available personas: backend_developer, frontend_developer, devops_engineer, qa_e
   if (finalText && finalText.length > planText.length) {
     planText = finalText;
   }
+  const planUsage = await planStream.totalUsage;
   let stories = parseStoriesFromText(planText, output);
   if (stories.length === 0) {
     output.log("system", "Planner didn't produce structured stories, falling back to single story");
@@ -411,7 +412,13 @@ Available personas: backend_developer, frontend_developer, devops_engineer, qa_e
       description: userTask
     }];
   }
-  return stories;
+  return {
+    stories,
+    provider: pProvider,
+    model: pModel,
+    inputTokens: planUsage?.inputTokens || 0,
+    outputTokens: planUsage?.outputTokens || 0
+  };
 }
 function parseStoriesFromText(text, output) {
   const codeBlocks = [...text.matchAll(/```(?:json)?\s*\n?([\s\S]*?)```/g)];
@@ -542,7 +549,10 @@ async function runOrchestration(config, userTask, trustAll, sandboxed, output) {
   };
   const sessionAllow = /* @__PURE__ */ new Set();
   const workingDir = process.cwd();
-  const plannerStories = await planStories(config, userTask, workingDir, sandboxed, output);
+  const planResult = await planStories(config, userTask, workingDir, sandboxed, output);
+  const plannerStories = planResult.stories;
+  costTracker.addUsage("Planner", planResult.provider, planResult.model, planResult.inputTokens, planResult.outputTokens);
+  output.updateCost?.(costTracker.getTotalCost());
   output.log("planner", `Plan generated: ${plannerStories.length} stories`);
   plannerStories.forEach((s, i) => {
     output.log("planner", `Step ${i + 1}: [${s.persona}] ${s.title}${s.dependsOn?.length ? ` (after: ${s.dependsOn.join(", ")})` : ""}`);
@@ -753,6 +763,7 @@ ${revisionFeedback}` : ""}`;
         const inTokens = usage?.inputTokens || 0;
         const outTokens = usage?.outputTokens || 0;
         costTracker.addUsage(persona.name, provider, modelName, inTokens, outTokens);
+        output.updateCost?.(costTracker.getTotalCost());
         output.log(story.persona, `${story.title} \u2014 completed! (${i + 1}/${sorted.length})`);
         output.log("system", "");
         break;
@@ -804,7 +815,7 @@ ${revisionFeedback}` : ""}`;
     for (let reviewRound = 0; reviewRound <= maxRevisions; reviewRound++) {
       const isRevision = reviewRound > 0;
       output.coordinatorLog(isRevision ? `Starting Tech Lead review (revision ${reviewRound}/${maxRevisions})...` : "Starting Tech Lead review...");
-      output.log("tech_lead", "Starting agent execution");
+      output.log("tech_lead", `Starting agent execution (model: ${revModel})`);
       output.status(isRevision ? "Reviewer -- Re-checking after revisions" : "Reviewer -- Checking code quality");
       try {
         const previousFeedbackSection = isRevision && previousReviewFeedback ? `## Previous Review Feedback (Review ${reviewRound}/${maxRevisions})
@@ -908,6 +919,7 @@ AFFECTED_REASONS: {"2": "Missing error handling in auth controller", "3": "Front
           reviewUsage?.inputTokens || 0,
           reviewUsage?.outputTokens || 0
         );
+        output.updateCost?.(costTracker.getTotalCost());
         if (approved) break;
         if (reviewRound >= maxRevisions) {
           output.log("system", `Max review revisions (${maxRevisions}) reached`);
@@ -965,7 +977,7 @@ AFFECTED_REASONS: {"2": "Missing error handling in auth controller", "3": "Front
             }
           }
           output.coordinatorLog(`Revision pass for story ${i + 1}/${sorted.length}`);
-          output.log(story.persona, `Starting revision: ${story.title}`);
+          output.log(story.persona, `Starting revision: ${story.title} (model: ${sModel})`);
           output.status("");
           const storyModel = createModel(sProvider, sModel, sHost, sCtx);
           const storyAllTools = createToolDefinitions(workingDir, storyModel, sandboxed);
@@ -1038,6 +1050,7 @@ ${story.description}`,
               revUsage?.inputTokens || 0,
               revUsage?.outputTokens || 0
             );
+            output.updateCost?.(costTracker.getTotalCost());
             output.log(story.persona, `${story.title} \u2014 revision complete!`);
           } catch (err) {
             output.statusDone();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "workermill",
-  "version": "0.3.2",
+  "version": "0.3.3",
   "description": "AI coding agent with multi-expert orchestration. Works with any LLM provider.",
   "type": "module",
   "bin": {