@getpochi/cli 0.5.69 → 0.5.70

package/dist/cli.js

@@ -373774,7 +373774,7 @@ var {
 // package.json
 var package_default = {
   name: "@getpochi/cli",
-  version: "0.5.69",
+  version: "0.5.70",
   type: "module",
   bin: {
     pochi: "src/cli.ts"
@@ -385428,6 +385428,41 @@ async function parseWorkflowFrontmatter(content3) {
     model: frontmatterData.model
   };
 }
+// ../common/src/tool-utils/notebook-utils.ts
+function validateNotebookPath(filePath) {
+  if (!filePath.endsWith(".ipynb")) {
+    throw new Error("File must be a Jupyter notebook (.ipynb)");
+  }
+}
+function validateNotebookStructure(notebook) {
+  if (!notebook || typeof notebook !== "object" || !("cells" in notebook) || !Array.isArray(notebook.cells)) {
+    throw new Error("Invalid notebook format: no cells array found");
+  }
+}
+function parseNotebook(content3) {
+  const notebook = JSON.parse(content3);
+  validateNotebookStructure(notebook);
+  return notebook;
+}
+function editNotebookCell(notebook, cellId, newContent) {
+  let cellFound = false;
+  for (let i6 = 0;i6 < notebook.cells.length; i6++) {
+    const cell = notebook.cells[i6];
+    const currentCellId = cell.id;
+    if (currentCellId === cellId || i6.toString() === cellId) {
+      notebook.cells[i6].source = newContent;
+      cellFound = true;
+      break;
+    }
+  }
+  if (!cellFound) {
+    throw new Error(`Cell with ID "${cellId}" not found in notebook`);
+  }
+  return notebook;
+}
+function serializeNotebook(notebook) {
+  return JSON.stringify(notebook, null, 2);
+}
 // src/lib/load-agents.ts
 var logger15 = getLogger("loadAgents");
 async function readAgentsFromDir(dir2) {
@@ -402099,7 +402134,7 @@ async function fetchTools() {
 }
 
 // src/mcp/initializer.ts
-async function initializeMcp() {
+async function initializeMcp(program5) {
   const mcpHub = await createCliMcpHub();
   if (Object.keys(mcpHub.status.value.connections).length === 0) {
     return;
@@ -402112,10 +402147,12 @@ async function initializeMcp() {
     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 (connections.length > 0) {
-      if (readyConnections + errorConnections >= connections.length) {
-        break;
-      }
+    if (errorConnections > 0) {
+      spinner.fail(`Failed to initialize MCP connections after ${attempts} attempts.`);
+      return program5.error("MCP initialization failed");
+    }
+    if (readyConnections >= connections.length) {
+      break;
     }
     await new Promise((resolve11) => setTimeout(resolve11, 500));
     attempts++;
@@ -417483,6 +417520,23 @@ var applyDiff2 = () => async ({ path: path27, searchContent, replaceContent, exp
   return { success: true };
 };
 
+// src/tools/edit-notebook.ts
+import * as fs13 from "node:fs/promises";
+var editNotebook2 = () => async ({ path: filePath, cellId, content: content3 }, { cwd: cwd2 }) => {
+  try {
+    const absolutePath = resolvePath(filePath, cwd2);
+    validateNotebookPath(absolutePath);
+    const fileContent3 = await fs13.readFile(absolutePath, "utf-8");
+    const notebook = parseNotebook(fileContent3);
+    const updatedNotebook = editNotebookCell(notebook, cellId, content3);
+    const serialized = serializeNotebook(updatedNotebook);
+    await fs13.writeFile(absolutePath, serialized, "utf-8");
+    return { success: true };
+  } catch (error46) {
+    return { success: false };
+  }
+};
+
 // src/tools/execute-command.ts
 import {
   exec as exec8
@@ -417589,15 +417643,15 @@ var listFiles3 = () => async ({ path: dirPath, recursive }, { abortSignal, cwd:
 };
 
 // src/tools/multi-apply-diff.ts
-import * as fs13 from "node:fs/promises";
+import * as fs14 from "node:fs/promises";
 var multiApplyDiff2 = () => async ({ path: path28, edits }, { cwd: cwd2 }) => {
   const fileUri = resolvePath(path28, cwd2);
   await ensureFileDirectoryExists(fileUri);
-  const fileBuffer = await fs13.readFile(fileUri);
+  const fileBuffer = await fs14.readFile(fileUri);
   validateTextFile(fileBuffer);
   const fileContent3 = fileBuffer.toString();
   const updatedContent = await processMultipleDiffs(fileContent3, edits);
-  await fs13.writeFile(fileUri, updatedContent);
+  await fs14.writeFile(fileUri, updatedContent);
   return { success: true };
 };
 
@@ -417635,10 +417689,10 @@ var newTask = (options6) => async ({ _meta, agentType }) => {
 };
 
 // src/tools/read-file.ts
-import * as fs14 from "node:fs/promises";
-var readFile13 = () => async ({ path: path28, startLine, endLine }, { cwd: cwd2 }) => {
+import * as fs15 from "node:fs/promises";
+var readFile14 = () => async ({ path: path28, startLine, endLine }, { cwd: cwd2 }) => {
   const resolvedPath = resolvePath(path28, cwd2);
-  const fileBuffer = await fs14.readFile(resolvedPath);
+  const fileBuffer = await fs15.readFile(resolvedPath);
   validateTextFile(fileBuffer);
   const fileContent3 = fileBuffer.toString();
   const addLineNumbers = !!process.env.VSCODE_TEST_OPTIONS;
@@ -417650,11 +417704,11 @@ var readFile13 = () => async ({ path: path28, startLine, endLine }, { cwd: cwd2
 };
 
 // src/tools/search-files.ts
-import * as fs15 from "node:fs";
+import * as fs16 from "node:fs";
 var logger25 = getLogger("searchFiles");
 var searchFiles2 = (context15) => async ({ path: path28, regex: regex3, filePattern }, { abortSignal, cwd: cwd2 }) => {
   const rgPath = context15.rg;
-  if (!rgPath || !fs15.existsSync(rgPath)) {
+  if (!rgPath || !fs16.existsSync(rgPath)) {
     logger25.error("Ripgrep not found at path", rgPath);
     throw new Error(`Ripgrep not found at path: ${rgPath}`);
   }
@@ -417669,23 +417723,24 @@ var todoWrite2 = (_options) => async () => {
 };
 
 // src/tools/write-to-file.ts
-import * as fs16 from "node:fs/promises";
+import * as fs17 from "node:fs/promises";
 import * as nodePath from "node:path";
 var writeToFile2 = () => async ({ path: path28, content: content3 }, { cwd: cwd2 }) => {
   const filePath = resolvePath(path28, cwd2);
   if (!await isFileExists(filePath)) {
     const dirPath = nodePath.dirname(filePath);
-    await fs16.mkdir(dirPath, { recursive: true });
+    await fs17.mkdir(dirPath, { recursive: true });
   }
   const processedContent = fixCodeGenerationOutput(content3);
-  await fs16.writeFile(filePath, processedContent);
+  await fs17.writeFile(filePath, processedContent);
   return { success: true };
 };
 
 // src/tools/index.ts
 var ToolMap = {
-  readFile: readFile13,
+  readFile: readFile14,
   applyDiff: applyDiff2,
+  editNotebook: editNotebook2,
   globFiles: globFiles3,
   listFiles: listFiles3,
   multiApplyDiff: multiApplyDiff2,
@@ -418247,7 +418302,7 @@ var program5 = new Command().name("pochi").description(`${source_default.bold("P
     renderer.renderSubTask(runner2);
   };
   const customAgents = await loadAgents(process.cwd());
-  const mcpHub = options6.mcp ? await initializeMcp() : undefined;
+  const mcpHub = options6.mcp ? await initializeMcp(program5) : undefined;
   const runner = new TaskRunner({
     uid,
     store,
@@ -418277,7 +418332,7 @@ var program5 = new Command().name("pochi").description(`${source_default.bold("P
 var otherOptionsGroup = "Others:";
 program5.optionsGroup(otherOptionsGroup).version(package_default.version, "-V, --version", "Print the version string.").addHelpOption(new Option3("-h, --help", "Print this help message.").helpGroup(otherOptionsGroup)).configureHelp({
   styleTitle: (title) => source_default.bold(title)
-}).showHelpAfterError().showSuggestionAfterError().configureOutput({
+}).showSuggestionAfterError().configureOutput({
   outputError: (str, write5) => write5(source_default.red(str))
 });
 program5.hook("preAction", async () => {

package/package.json

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