@getpochi/cli 0.5.75 → 0.5.77

package/dist/cli.js

@@ -404610,7 +404610,7 @@ var {
 // package.json
 var package_default = {
   name: "@getpochi/cli",
-  version: "0.5.75",
+  version: "0.5.77",
   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 {
@@ -431007,20 +431015,23 @@ async function initializeMcp(program5) {
   const spinner = ora("Initializing MCP connections...").start();
   let attempts = 0;
   const maxAttempts = 15;
-  while (attempts < maxAttempts) {
+  while (true) {
     const status3 = mcpHub.status.value;
     const connections = Object.values(status3.connections);
     const readyConnections = connections.filter((conn) => conn.status === "ready").length;
     const errorConnections = connections.filter((conn) => conn.status === "error").length;
-    if (errorConnections > 0) {
-      spinner.fail(`Failed to initialize MCP connections after ${attempts} attempts.`);
-      return program5.error("MCP initialization failed");
-    }
     if (readyConnections >= connections.length) {
       break;
     }
+    if (errorConnections > 0) {
+      if (attempts < maxAttempts) {
+        attempts++;
+      } else {
+        spinner.fail(`Failed to initialize MCP connections after ${attempts} attempts.`);
+        return program5.error("MCP initialization failed");
+      }
+    }
     await new Promise((resolve11) => setTimeout(resolve11, 500));
-    attempts++;
   }
   spinner.stop();
   return mcpHub;
@@ -444897,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>";
@@ -444938,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
       };
     }
@@ -445441,21 +445461,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) {
@@ -445699,7 +445725,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({
@@ -446870,7 +446896,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
@@ -446934,7 +446960,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 });

package/package.json

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