@getpochi/cli 0.5.95 → 0.5.96

package/dist/cli.js

@@ -113843,14 +113843,14 @@ var init_ShardManager = __esm(() => {
       unassignedShards.unsafeUpdate(stats.unassigned, []);
     }
     updateShardMetrics();
-    function withRetry(effect4) {
+    function withRetry2(effect4) {
       return effect4.pipe(retry2({
         schedule: spaced2(config5.persistRetryCount),
         times: config5.persistRetryCount
       }), ignore2);
     }
-    const persistRunners = unsafeMakeSemaphore2(1).withPermits(1)(withRetry(suspend3(() => storage.saveRunners(map4(state3.allRunners, ([address, runner]) => [address, runner.runner])))));
-    const persistAssignments = unsafeMakeSemaphore2(1).withPermits(1)(withRetry(suspend3(() => storage.saveAssignments(state3.assignments))));
+    const persistRunners = unsafeMakeSemaphore2(1).withPermits(1)(withRetry2(suspend3(() => storage.saveRunners(map4(state3.allRunners, ([address, runner]) => [address, runner.runner])))));
+    const persistAssignments = unsafeMakeSemaphore2(1).withPermits(1)(withRetry2(suspend3(() => storage.saveAssignments(state3.assignments))));
     const notifyUnhealthyRunner = fnUntraced2(function* (address) {
       if (!has6(state3.allRunners, address))
         return;
@@ -391690,6 +391690,28 @@ var VendorId = "pochi";
 
 // ../vendor-pochi/src/vendor.ts
 var logger3 = getLogger("PochiVendor");
+async function withRetry(fn, options = {}) {
+  const {
+    maxRetries = 3,
+    initialDelay = 1000,
+    maxDelay = 1e4,
+    delayMultiplier = 2
+  } = options;
+  let lastError;
+  let delay2 = initialDelay;
+  for (let attempt = 0;attempt <= maxRetries; attempt++) {
+    try {
+      return await fn();
+    } catch (error40) {
+      lastError = error40 instanceof Error ? error40 : new Error(String(error40));
+      if (attempt < maxRetries) {
+        await new Promise((resolve2) => setTimeout(resolve2, delay2));
+        delay2 = Math.min(delay2 * delayMultiplier, maxDelay);
+      }
+    }
+  }
+  throw lastError;
+}
 
 class Pochi extends VendorBase {
   cachedModels;
@@ -391702,7 +391724,14 @@ class Pochi extends VendorBase {
   async fetchModels() {
     if (!this.cachedModels) {
       const apiClient = hc(getServerBaseUrl());
-      const data = await apiClient.api.models.$get().then((x) => x.json()).catch(() => {
+      const data = await withRetry(async () => {
+        const response = await apiClient.api.models.$get();
+        return response.json();
+      }, {
+        maxRetries: 3,
+        initialDelay: 1000
+      }).catch((error40) => {
+        logger3.error(`Failed to fetch models: ${error40.message}`);
         return [];
       });
       this.cachedModels = Object.fromEntries(data.map((x) => [
@@ -405887,7 +405916,7 @@ var {
 // package.json
 var package_default = {
   name: "@getpochi/cli",
-  version: "0.5.95",
+  version: "0.5.96",
   type: "module",
   bin: {
     pochi: "src/cli.ts"
@@ -419134,20 +419163,43 @@ class JsonRenderer {
   state;
   outputMessageIds = new Set;
   lastMessageCount = 0;
-  constructor(store, state3) {
+  mode;
+  constructor(store, state3, options4 = { mode: "full" }) {
     this.store = store;
     this.state = state3;
-    this.state.signal.messages.subscribe((messages) => {
-      if (messages.length > this.lastMessageCount) {
-        this.outputMessages(messages.slice(0, -1));
-        this.lastMessageCount = messages.length;
-      }
-    });
+    this.mode = options4.mode;
+    if (this.mode === "full") {
+      this.state.signal.messages.subscribe((messages) => {
+        if (messages.length > this.lastMessageCount) {
+          this.outputMessages(messages.slice(0, -1));
+          this.lastMessageCount = messages.length;
+        }
+      });
+    }
   }
   shutdown() {
-    this.outputMessages(this.state.signal.messages.value);
+    if (this.mode === "result-only") {
+      this.outputResult();
+    } else {
+      this.outputMessages(this.state.signal.messages.value);
+    }
   }
   renderSubTask(_task) {}
+  outputResult() {
+    const messages = this.state.signal.messages.value;
+    const lastMessage = messages.at(-1);
+    if (lastMessage?.role === "assistant") {
+      for (const part of lastMessage.parts || []) {
+        if (isToolUIPart(part) && part.type === "tool-attemptCompletion") {
+          if (part.input) {
+            const result2 = part.input.result || "";
+            console.log(result2);
+          }
+          return;
+        }
+      }
+    }
+  }
   outputMessages(messages) {
     for (const message of messages) {
       if (!this.outputMessageIds.has(message.id)) {
@@ -448492,6 +448544,9 @@ class LiveChatKit {
   store;
   chat;
   transport;
+  onStreamStart;
+  onStreamFinish;
+  onStreamFailed;
   spawn;
   constructor({
     taskId,
@@ -448504,10 +448559,16 @@ class LiveChatKit {
     isCli,
     customAgent,
     outputSchema: outputSchema2,
+    onStreamStart,
+    onStreamFinish,
+    onStreamFailed,
     ...chatInit
   }) {
     this.taskId = taskId;
     this.store = store;
+    this.onStreamStart = onStreamStart;
+    this.onStreamFinish = onStreamFinish;
+    this.onStreamFailed = onStreamFailed;
     this.transport = new FlexibleChatTransport({
       store,
       onStart: this.onStart,
@@ -448661,6 +448722,7 @@ class LiveChatKit {
         updatedAt: new Date,
         modelId: llm.id
       }));
+      this.onStreamStart?.();
     }
   };
   onFinish = ({
@@ -448679,13 +448741,20 @@ class LiveChatKit {
     if (message.metadata?.kind !== "assistant") {
       return this.onError(abortError);
     }
+    const status3 = toTaskStatus(message, message.metadata?.finishReason);
     store.commit(events.chatStreamFinished({
       id: this.taskId,
-      status: toTaskStatus(message, message.metadata?.finishReason),
+      status: status3,
       data: message,
       totalTokens: message.metadata.totalTokens,
       updatedAt: new Date
     }));
+    this.onStreamFinish?.({
+      id: this.taskId,
+      cwd: this.task?.cwd ?? null,
+      status: status3,
+      messages: [...this.chat.messages]
+    });
   };
   onError = (error46) => {
     logger23.error("onError", error46);
@@ -448696,6 +448765,11 @@ class LiveChatKit {
       data: lastMessage,
       updatedAt: new Date
     }));
+    this.onStreamFailed?.({
+      cwd: this.task?.cwd ?? null,
+      error: error46,
+      messages: [...this.chat.messages]
+    });
   };
 }
 // src/lib/read-environment.ts
@@ -449918,7 +449992,7 @@ var parsePositiveInt = (input2) => {
   }
   return result2;
 };
-var program5 = new Command().name("pochi").description(`${source_default.bold("Pochi")} v${package_default.version} - A powerful CLI tool for AI-driven development.`).optionsGroup("Prompt:").option("-p, --prompt <prompt>", "Create a new task with a given prompt. Input can also be piped. For example: `cat my-prompt.md | pochi`. Workflows can be triggered with `/workflow-name`, like `pochi -p /create-pr`.").option("-a, --attach <path...>", "Attach one or more files to the prompt, e.g images").optionsGroup("Options:").option("--stream-json", "Stream the output in JSON format. This is useful for parsing the output in scripts.").option("--max-steps <number>", "Set the maximum number of steps for a task. The task will stop if it exceeds this limit.", parsePositiveInt, 24).option("--max-retries <number>", "Set the maximum number of retries for a single step in a task.", parsePositiveInt, 3).addOption(new Option("--experimental-output-schema <schema>", "Specify a JSON schema for the output of the task. The task will be validated against this schema.").hideHelp()).optionsGroup("Model:").option("-m, --model <model>", "Specify the model to be used for the task.", "qwen/qwen3-coder").optionsGroup("MCP:").option("--no-mcp", "Disable MCP (Model Context Protocol) integration completely.").action(async (options6) => {
+var program5 = new Command().name("pochi").description(`${source_default.bold("Pochi")} v${package_default.version} - A powerful CLI tool for AI-driven development.`).optionsGroup("Prompt:").option("-p, --prompt <prompt>", "Create a new task with a given prompt. Input can also be piped. For example: `cat my-prompt.md | pochi`. Workflows can be triggered with `/workflow-name`, like `pochi -p /create-pr`.").option("-a, --attach <path...>", "Attach one or more files to the prompt, e.g images").optionsGroup("Options:").option("--stream-json", "Stream the output in JSON format. This is useful for parsing the output in scripts.").option("-x, --output-result", "Output the result from attemptCompletion to stdout. This is useful for scripts that need to capture the final result.").option("--max-steps <number>", "Set the maximum number of steps for a task. The task will stop if it exceeds this limit.", parsePositiveInt, 24).option("--max-retries <number>", "Set the maximum number of retries for a single step in a task.", parsePositiveInt, 3).addOption(new Option("--experimental-output-schema <schema>", "Specify a JSON schema for the output of the task. The task will be validated against this schema.").hideHelp()).optionsGroup("Model:").option("-m, --model <model>", "Specify the model to be used for the task.", "google/gemini-2.5-flash").optionsGroup("MCP:").option("--no-mcp", "Disable MCP (Model Context Protocol) integration completely.").action(async (options6) => {
   const customAgents = await loadAgents(process.cwd());
   const workflows = await loadWorkflows(process.cwd());
   const { uid, prompt: prompt3, attachments } = await parseTaskInput(options6, program5, {
@@ -449995,7 +450069,11 @@ ${error46}`);
   const renderer = new OutputRenderer(runner.state);
   let jsonRenderer;
   if (options6.streamJson) {
-    jsonRenderer = new JsonRenderer(store, runner.state);
+    jsonRenderer = new JsonRenderer(store, runner.state, { mode: "full" });
+  } else if (options6.outputResult) {
+    jsonRenderer = new JsonRenderer(store, runner.state, {
+      mode: "result-only"
+    });
   }
   await runner.run();
   renderer.shutdown();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@getpochi/cli",
-  "version": "0.5.95",
+  "version": "0.5.96",
   "type": "module",
   "bin": {
     "pochi": "dist/cli.js"