@getpochi/cli 0.5.76 → 0.5.78

package/dist/cli.js

@@ -404610,7 +404610,7 @@ var {
 // package.json
 var package_default = {
   name: "@getpochi/cli",
-  version: "0.5.76",
+  version: "0.5.78",
   type: "module",
   bin: {
     pochi: "src/cli.ts"
@@ -416482,18 +416482,26 @@ async function loadAgents(workingDirectory2, includeSystemAgents = true) {
 
 // src/lib/workflow-loader.ts
 import * as fs6 from "node:fs/promises";
+import { homedir as homedir4 } from "node:os";
 import * as path11 from "node:path";
 function getWorkflowPath(id4) {
   const workflowsDir = path11.join(...exports_constants.WorkspaceWorkflowPathSegments);
   return path11.join(workflowsDir, `${id4}.md`);
 }
-async function loadWorkflow(id4, cwd) {
-  try {
-    const content3 = await fs6.readFile(path11.join(cwd, getWorkflowPath(id4)), "utf-8");
-    return content3;
-  } catch (error42) {
-    return null;
+async function loadWorkflow(id4, cwd, includeGlobalWorkflow = true) {
+  const workflowFilePaths = [path11.join(cwd, getWorkflowPath(id4))];
+  if (includeGlobalWorkflow) {
+    workflowFilePaths.push(path11.join(homedir4(), getWorkflowPath(id4)));
+  }
+  for (const filePath of workflowFilePaths) {
+    try {
+      const content3 = await fs6.readFile(filePath, "utf-8");
+      if (content3) {
+        return content3;
+      }
+    } catch (error42) {}
   }
+  return null;
 }
 function containsWorkflowReference(prompt) {
   return /\/\w+[\w-]*/.test(prompt);
@@ -430452,7 +430460,7 @@ function isToolEnabledChanged(oldConfig, newConfig) {
 
 // ../common/src/mcp-utils/mcp-connection.ts
 var AbortedError = "AbortedError";
-var AutoReconnectDelay = 20000;
+var AutoReconnectDelay = 1000;
 var AutoReconnectMaxAttempts = 20;
 
 class McpConnection {
@@ -431006,8 +431014,8 @@ async function initializeMcp(program5) {
   }
   const spinner = ora("Initializing MCP connections...").start();
   let attempts = 0;
-  const maxAttempts = 3;
-  while (attempts < maxAttempts) {
+  const maxAttempts = 15;
+  while (true) {
     const status3 = mcpHub.status.value;
     const connections = Object.values(status3.connections);
     const readyConnections = connections.filter((conn) => conn.status === "ready").length;
@@ -444900,7 +444908,7 @@ function createNewTaskMiddleware(store, cwd2, parentTaskId, customAgents) {
   };
 }
 // ../livekit/src/chat/middlewares/tool-call-middleware.ts
-function createToolCallMiddleware() {
+function createToolCallMiddleware(useStopWordStream) {
   const toolCallEndTag = "</api-request>";
   const toolResponseTagTemplate = (name17) => `<api-response name="${name17}">`;
   const toolResponseEndTag = "</api-response>";
@@ -444941,16 +444949,25 @@ ${processedPrompt[0].content}`
         },
         ...processedPrompt
       ];
+      let stopSequences = params3.stopSequences;
+      if (!useStopWordStream) {
+        stopSequences = [...stopSequences || [], "</api-section>"];
+      }
       return {
         ...params3,
+        stopSequences,
         tools: undefined,
         prompt: promptWithTools
       };
     },
     wrapStream: async ({ doStream }) => {
       const { stream: stream12, ...rest } = await doStream();
+      let newStream = stream12;
+      if (useStopWordStream) {
+        newStream = stream12.pipeThrough(createStopWordStream(toolSectionEndTag));
+      }
       return {
-        stream: stream12.pipeThrough(createStopWordStream(toolSectionEndTag)).pipeThrough(createToolCallStream(toolCallStartRegex, toolCallStartPrefix, toolCallEndTag, toolSectionStartTag)),
+        stream: newStream.pipeThrough(createToolCallStream(toolCallStartRegex, toolCallStartPrefix, toolCallEndTag, toolSectionStartTag)),
         ...rest
       };
     }
@@ -445157,7 +445174,7 @@ function processToolResult(tool2, toolResponseTagTemplate, toolResponseEndTag) {
     } else {
       content3.push({
         type: "text",
-        text: `${toolResponseTagTemplate(x11.toolName)}${JSON.stringify(x11.output)}${toolResponseEndTag}`
+        text: `${toolResponseTagTemplate(x11.toolName)}${convertToolResultOutput(x11.output)}${toolResponseEndTag}`
       });
     }
   }
@@ -445305,6 +445322,23 @@ function createStopWordStream(stop3) {
     }
   });
 }
+function convertToolResultOutput(x11) {
+  if (x11.type === "json") {
+    return JSON.stringify(x11.value);
+  }
+  if (x11.type === "text") {
+    return x11.value;
+  }
+  if (x11.type === "error-text" || x11.type === "error-json") {
+    return JSON.stringify({
+      error: x11.value
+    });
+  }
+  assertUnreachable(x11);
+}
+function assertUnreachable(x11) {
+  throw new Error(`Unreachable case: ${x11}`);
+}
 // ../livekit/src/chat/middlewares/output-schema-middleware.ts
 function createOutputSchemaMiddleware(taskId, model2, outputSchema2) {
   return {
@@ -445444,21 +445478,27 @@ function createPatchedFetchForFinetune(accessToken) {
       ...requestInit,
       headers
     };
+    let finalUrl;
     if (requestInfo instanceof URL) {
-      const patchedUrl = new URL(requestInfo);
-      patchedUrl.pathname = patchString(patchedUrl.pathname);
-      return fetch(patchedUrl, patchedRequestInit);
-    }
-    if (requestInfo instanceof Request) {
+      finalUrl = new URL(requestInfo);
+      finalUrl.pathname = patchString(finalUrl.pathname);
+    } else if (requestInfo instanceof Request) {
       const patchedUrl = patchString(requestInfo.url);
-      const patchedRequest = new Request(patchedUrl, requestInfo);
-      return fetch(patchedRequest, patchedRequestInit);
-    }
-    if (typeof requestInfo === "string") {
+      finalUrl = new URL(patchedUrl);
+    } else if (typeof requestInfo === "string") {
       const patchedUrl = patchString(requestInfo);
-      return fetch(patchedUrl, patchedRequestInit);
+      finalUrl = new URL(patchedUrl);
+    } else {
+      throw new Error(`Unexpected requestInfo type: ${typeof requestInfo}`);
+    }
+    if (globalThis.POCHI_CORS_PROXY_PORT) {
+      const origin = finalUrl.origin;
+      finalUrl.protocol = "http:";
+      finalUrl.host = "localhost";
+      finalUrl.port = globalThis.POCHI_CORS_PROXY_PORT;
+      patchedRequestInit.headers.set("x-proxy-origin", origin);
     }
-    throw new Error(`Unexpected requestInfo type: ${typeof requestInfo}`);
+    return fetch(finalUrl, patchedRequestInit);
   };
 }
 function createVertexModel(vertex2, modelId) {
@@ -445651,9 +445691,9 @@ function createModel2({ llm }) {
   if (llm.type === "openai-responses") {
     return createOpenAIResponsesModel(llm);
   }
-  assertUnreachable(llm);
+  assertUnreachable2(llm);
 }
-function assertUnreachable(_x) {
+function assertUnreachable2(_x) {
   throw new Error("Didn't expect to get here");
 }
 
@@ -445702,7 +445742,7 @@ class FlexibleChatTransport {
       middlewares.push(createOutputSchemaMiddleware(chatId, model2, this.outputSchema));
     }
     if (llm.useToolCallMiddleware) {
-      middlewares.push(createToolCallMiddleware());
+      middlewares.push(createToolCallMiddleware(llm.type !== "google-vertex-tuning"));
     }
     const mcpTools = mcpInfo?.toolset && parseMcpToolSet(this.store, mcpInfo.toolset);
     const tools = d2({
@@ -446873,7 +446913,7 @@ import {
   rmSync,
   statSync as statSync3
 } from "node:fs";
-import { homedir as homedir5, tmpdir as tmpdir3 } from "node:os";
+import { homedir as homedir6, tmpdir as tmpdir3 } from "node:os";
 import { extname as extname5, join as join21 } from "node:path";
 
 // src/upgrade/platform-utils.ts
@@ -446937,7 +446977,7 @@ function extractVersionFromTag(tag9) {
 
 // src/upgrade/binary-installer.ts
 function getPochiDir() {
-  const pochiDir = join21(homedir5(), ".pochi");
+  const pochiDir = join21(homedir6(), ".pochi");
   const binDir = join21(pochiDir, "bin");
   if (!existsSync7(pochiDir)) {
     mkdirSync(pochiDir, { recursive: true });
@@ -447356,7 +447396,7 @@ async function createLLMConfigWithProviders(program6, model2) {
       useToolCallMiddleware: modelSetting.useToolCallMiddleware
     };
   }
-  assertUnreachable2(modelProvider.kind);
+  assertUnreachable3(modelProvider.kind);
 }
 async function waitForSync(inputStore, timeoutDuration = "1 second") {
   if (!process.env.POCHI_LIVEKIT_SYNC_ON) {
@@ -447376,7 +447416,7 @@ async function waitForSync(inputStore, timeoutDuration = "1 second") {
     await store.shutdown();
   }
 }
-function assertUnreachable2(_x) {
+function assertUnreachable3(_x) {
   throw new Error("Didn't expect to get here");
 }
 async function getModelFromWorkflow(options6) {

package/package.json

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