@probelabs/probe 0.6.0-rc225 → 0.6.0-rc226

package/cjs/agent/ProbeAgent.cjs

@@ -32675,6 +32675,25 @@ var init_error_types = __esm({
 });
 
 // src/utils/path-validation.js
+function safeRealpath(inputPath) {
+  try {
+    return (0, import_fs3.realpathSync)(inputPath);
+  } catch (error2) {
+    const normalized = import_path4.default.normalize(inputPath);
+    const parts = normalized.split(import_path4.default.sep);
+    for (let i5 = parts.length - 1; i5 >= 0; i5--) {
+      const ancestorPath = parts.slice(0, i5).join(import_path4.default.sep) || import_path4.default.sep;
+      try {
+        const resolvedAncestor = (0, import_fs3.realpathSync)(ancestorPath);
+        const remainingParts = parts.slice(i5);
+        return import_path4.default.join(resolvedAncestor, ...remainingParts);
+      } catch (ancestorError) {
+        continue;
+      }
+    }
+    return normalized;
+  }
+}
 async function validateCwdPath(inputPath, defaultPath = process.cwd()) {
   const targetPath = inputPath || defaultPath;
   const normalizedPath = import_path4.default.normalize(import_path4.default.resolve(targetPath));
@@ -32707,6 +32726,53 @@ async function validateCwdPath(inputPath, defaultPath = process.cwd()) {
   }
   return normalizedPath;
 }
+function getCommonPrefix(folders) {
+  if (!folders || folders.length === 0) {
+    return process.cwd();
+  }
+  if (folders.length === 1) {
+    return safeRealpath(folders[0]);
+  }
+  const normalized = folders.map((f5) => safeRealpath(f5));
+  const segments = normalized.map((f5) => f5.split(import_path4.default.sep));
+  const minLen = Math.min(...segments.map((s5) => s5.length));
+  const commonSegments = [];
+  for (let i5 = 0; i5 < minLen; i5++) {
+    const segment = segments[0][i5];
+    if (segments.every((s5) => s5[i5] === segment)) {
+      commonSegments.push(segment);
+    } else {
+      break;
+    }
+  }
+  if (commonSegments.length === 0) {
+    return normalized[0];
+  }
+  if (commonSegments.length === 1 && /^[a-zA-Z]:$/.test(commonSegments[0])) {
+    return normalized[0];
+  }
+  if (commonSegments.length === 1 && commonSegments[0] === "") {
+    return normalized[0];
+  }
+  return commonSegments.join(import_path4.default.sep);
+}
+function toRelativePath(absolutePath, workspaceRoot) {
+  if (!absolutePath || !workspaceRoot) {
+    return absolutePath;
+  }
+  let normalized = safeRealpath(absolutePath);
+  let normalizedRoot = safeRealpath(workspaceRoot);
+  while (normalizedRoot.length > 1 && normalizedRoot.endsWith(import_path4.default.sep)) {
+    normalizedRoot = normalizedRoot.slice(0, -1);
+  }
+  if (normalized === normalizedRoot) {
+    return ".";
+  }
+  if (normalized.startsWith(normalizedRoot + import_path4.default.sep)) {
+    return import_path4.default.relative(normalizedRoot, normalized);
+  }
+  return absolutePath;
+}
 var import_path4, import_fs3;
 var init_path_validation = __esm({
   "src/utils/path-validation.js"() {
@@ -33904,6 +33970,7 @@ async function analyzeAll(options) {
     sessionId,
     debug = false,
     cwd,
+    workspaceRoot,
     allowedFolders,
     provider,
     model,
@@ -33916,10 +33983,11 @@ async function analyzeAll(options) {
   if (!question) {
     throw new Error('The "question" parameter is required.');
   }
+  const effectiveWorkspaceRoot = workspaceRoot || cwd || allowedFolders?.[0] || path9;
   const delegateOptions = {
     debug,
     sessionId,
-    path: allowedFolders?.[0] || cwd || path9,
+    path: effectiveWorkspaceRoot,
     allowedFolders,
     provider,
     model,
@@ -38132,21 +38200,24 @@ var init_zod = __esm({
 // src/tools/edit.js
 function isPathAllowed(filePath, allowedFolders) {
   if (!allowedFolders || allowedFolders.length === 0) {
-    const resolvedPath3 = (0, import_path5.resolve)(filePath);
-    const cwd = (0, import_path5.resolve)(process.cwd());
+    const resolvedPath3 = safeRealpath(filePath);
+    const cwd = safeRealpath(process.cwd());
     return resolvedPath3 === cwd || resolvedPath3.startsWith(cwd + import_path5.sep);
   }
-  const resolvedPath2 = (0, import_path5.resolve)(filePath);
+  const resolvedPath2 = safeRealpath(filePath);
   return allowedFolders.some((folder) => {
-    const allowedPath = (0, import_path5.resolve)(folder);
+    const allowedPath = safeRealpath(folder);
     return resolvedPath2 === allowedPath || resolvedPath2.startsWith(allowedPath + import_path5.sep);
   });
 }
 function parseFileToolOptions(options = {}) {
+  const allowedFolders = options.allowedFolders || [];
   return {
     debug: options.debug || false,
-    allowedFolders: options.allowedFolders || [],
-    cwd: options.cwd
+    allowedFolders,
+    cwd: options.cwd,
+    // Consistent fallback chain: workspaceRoot > cwd > allowedFolders[0] > process.cwd()
+    workspaceRoot: options.workspaceRoot || options.cwd || allowedFolders.length > 0 && allowedFolders[0] || process.cwd()
   };
 }
 var import_ai, import_fs4, import_path5, import_fs5, editTool, createTool, editSchema, createSchema, editDescription, createDescription, editToolDefinition, createToolDefinition;
@@ -38157,8 +38228,9 @@ var init_edit = __esm({
     import_fs4 = require("fs");
     import_path5 = require("path");
     import_fs5 = require("fs");
+    init_path_validation();
     editTool = (options = {}) => {
-      const { debug, allowedFolders, cwd } = parseFileToolOptions(options);
+      const { debug, allowedFolders, cwd, workspaceRoot } = parseFileToolOptions(options);
       return (0, import_ai.tool)({
         name: "edit",
         description: `Edit files using exact string replacement (Claude Code style).
@@ -38214,7 +38286,8 @@ Important:
               console.error(`[Edit] Attempting to edit file: ${resolvedPath2}`);
             }
             if (!isPathAllowed(resolvedPath2, allowedFolders)) {
-              return `Error editing file: Permission denied - ${file_path} is outside allowed directories`;
+              const relativePath = toRelativePath(resolvedPath2, workspaceRoot);
+              return `Error editing file: Permission denied - ${relativePath} is outside allowed directories`;
             }
             if (!(0, import_fs5.existsSync)(resolvedPath2)) {
               return `Error editing file: File not found - ${file_path}`;
@@ -38250,7 +38323,7 @@ Important:
       });
     };
     createTool = (options = {}) => {
-      const { debug, allowedFolders, cwd } = parseFileToolOptions(options);
+      const { debug, allowedFolders, cwd, workspaceRoot } = parseFileToolOptions(options);
       return (0, import_ai.tool)({
         name: "create",
         description: `Create new files with specified content.
@@ -38298,7 +38371,8 @@ Important:
               console.error(`[Create] Attempting to create file: ${resolvedPath2}`);
             }
             if (!isPathAllowed(resolvedPath2, allowedFolders)) {
-              return `Error creating file: Permission denied - ${file_path} is outside allowed directories`;
+              const relativePath = toRelativePath(resolvedPath2, workspaceRoot);
+              return `Error creating file: Permission denied - ${relativePath} is outside allowed directories`;
             }
             if ((0, import_fs5.existsSync)(resolvedPath2) && !overwrite) {
               return `Error creating file: File already exists - ${file_path}. Use overwrite: true to replace it.`;
@@ -39860,7 +39934,7 @@ var init_vercel = __esm({
       });
     };
     delegateTool = (options = {}) => {
-      const { debug = false, timeout = 300, cwd, allowedFolders, enableBash = false, bashConfig, architectureFileName, enableMcp = false, mcpConfig = null, mcpConfigPath = null, delegationManager = null } = options;
+      const { debug = false, timeout = 300, cwd, allowedFolders, workspaceRoot, enableBash = false, bashConfig, architectureFileName, enableMcp = false, mcpConfig = null, mcpConfigPath = null, delegationManager = null } = options;
       return (0, import_ai2.tool)({
         name: "delegate",
         description: delegateDescription,
@@ -39893,8 +39967,8 @@ var init_vercel = __esm({
           if (searchDelegate !== void 0 && typeof searchDelegate !== "boolean") {
             throw new TypeError("searchDelegate must be a boolean if provided");
           }
-          const workspaceRoot = allowedFolders && allowedFolders[0];
-          const effectivePath = path9 || workspaceRoot || cwd;
+          const effectiveWorkspaceRoot = workspaceRoot || allowedFolders && allowedFolders[0];
+          const effectivePath = path9 || effectiveWorkspaceRoot || cwd;
           if (debug) {
             console.error(`Executing delegate with task: "${task.substring(0, 100)}${task.length > 100 ? "..." : ""}"`);
             if (parentSessionId) {
@@ -39931,7 +40005,7 @@ var init_vercel = __esm({
       });
     };
     analyzeAllTool = (options = {}) => {
-      const { sessionId, debug = false, delegationManager = null } = options;
+      const { sessionId, debug = false, delegationManager = null, workspaceRoot } = options;
       return (0, import_ai2.tool)({
         name: "analyze_all",
         description: analyzeAllDescription,
@@ -39950,12 +40024,14 @@ var init_vercel = __esm({
               console.error(`[analyze_all] Question: ${question}`);
               console.error(`[analyze_all] Path: ${searchPath}`);
             }
+            const effectiveWorkspaceRoot = workspaceRoot || options.cwd || options.allowedFolders && options.allowedFolders[0];
             const result = await analyzeAll({
               question,
               path: searchPath,
               sessionId,
               debug,
               cwd: options.cwd,
+              workspaceRoot: effectiveWorkspaceRoot,
               allowedFolders: options.allowedFolders,
               provider: options.provider,
               model: options.model,
@@ -41432,14 +41508,17 @@ var init_bash = __esm({
     import_path8 = require("path");
     init_bashPermissions();
     init_bashExecutor();
+    init_path_validation();
     bashTool = (options = {}) => {
       const {
         bashConfig = {},
         debug = false,
         cwd,
         allowedFolders = [],
+        workspaceRoot: providedWorkspaceRoot,
         tracer = null
       } = options;
+      const workspaceRoot = providedWorkspaceRoot || cwd || allowedFolders.length > 0 && allowedFolders[0] || process.cwd();
       const permissionChecker = new BashPermissionChecker({
         allow: bashConfig.allow,
         deny: bashConfig.deny,
@@ -41455,6 +41534,9 @@ var init_bash = __esm({
         if (cwd) {
           return cwd;
         }
+        if (workspaceRoot) {
+          return workspaceRoot;
+        }
         if (allowedFolders && allowedFolders.length > 0) {
           return allowedFolders[0];
         }
@@ -41542,13 +41624,15 @@ For code exploration, try these safe alternatives:
             const defaultDir = getDefaultWorkingDirectory();
             const workingDir = workingDirectory ? (0, import_path8.isAbsolute)(workingDirectory) ? (0, import_path8.resolve)(workingDirectory) : (0, import_path8.resolve)(defaultDir, workingDirectory) : defaultDir;
             if (allowedFolders && allowedFolders.length > 0) {
-              const resolvedWorkingDir = (0, import_path8.resolve)(workingDir);
+              const resolvedWorkingDir = safeRealpath(workingDir);
               const isAllowed = allowedFolders.some((folder) => {
-                const resolvedFolder = (0, import_path8.resolve)(folder);
+                const resolvedFolder = safeRealpath(folder);
                 return resolvedWorkingDir === resolvedFolder || resolvedWorkingDir.startsWith(resolvedFolder + import_path8.sep);
               });
               if (!isAllowed) {
-                return `Error: Working directory "${workingDir}" is not within allowed folders: ${allowedFolders.join(", ")}`;
+                const relativeDir = toRelativePath(workingDir, workspaceRoot);
+                const relativeAllowed = allowedFolders.map((f5) => toRelativePath(f5, workspaceRoot));
+                return `Error: Working directory "${relativeDir}" is not within allowed folders: ${relativeAllowed.join(", ")}`;
               }
             }
             const executionOptions = {
@@ -45715,7 +45799,7 @@ var init_esm3 = __esm({
 });
 
 // node_modules/path-scurry/dist/esm/index.js
-var import_node_path, import_node_url, import_fs9, actualFS, import_promises, realpathSync, defaultFS, fsFromOption, uncDriveRegexp, uncToDrive, eitherSep, UNKNOWN, IFIFO, IFCHR, IFDIR, IFBLK, IFREG, IFLNK, IFSOCK, IFMT, IFMT_UNKNOWN, READDIR_CALLED, LSTAT_CALLED, ENOTDIR, ENOENT, ENOREADLINK, ENOREALPATH, ENOCHILD, TYPEMASK, entToType, normalizeCache, normalize, normalizeNocaseCache, normalizeNocase, ResolveCache, ChildrenCache, setAsCwd, PathBase, PathWin32, PathPosix, PathScurryBase, PathScurryWin32, PathScurryPosix, PathScurryDarwin, Path, PathScurry;
+var import_node_path, import_node_url, import_fs9, actualFS, import_promises, realpathSync2, defaultFS, fsFromOption, uncDriveRegexp, uncToDrive, eitherSep, UNKNOWN, IFIFO, IFCHR, IFDIR, IFBLK, IFREG, IFLNK, IFSOCK, IFMT, IFMT_UNKNOWN, READDIR_CALLED, LSTAT_CALLED, ENOTDIR, ENOENT, ENOREADLINK, ENOREALPATH, ENOCHILD, TYPEMASK, entToType, normalizeCache, normalize, normalizeNocaseCache, normalizeNocase, ResolveCache, ChildrenCache, setAsCwd, PathBase, PathWin32, PathPosix, PathScurryBase, PathScurryWin32, PathScurryPosix, PathScurryDarwin, Path, PathScurry;
 var init_esm4 = __esm({
   "node_modules/path-scurry/dist/esm/index.js"() {
     init_esm2();
@@ -45725,13 +45809,13 @@ var init_esm4 = __esm({
     actualFS = __toESM(require("node:fs"), 1);
     import_promises = require("node:fs/promises");
     init_esm3();
-    realpathSync = import_fs9.realpathSync.native;
+    realpathSync2 = import_fs9.realpathSync.native;
     defaultFS = {
       lstatSync: import_fs9.lstatSync,
       readdir: import_fs9.readdir,
       readdirSync: import_fs9.readdirSync,
       readlinkSync: import_fs9.readlinkSync,
-      realpathSync,
+      realpathSync: realpathSync2,
       promises: {
         lstat: import_promises.lstat,
         readdir: import_promises.readdir,
@@ -99126,6 +99210,7 @@ var init_ProbeAgent = __esm({
     init_FallbackManager();
     init_contextCompactor();
     init_error_types();
+    init_path_validation();
     init_outputTruncator();
     init_delegate();
     init_tasks();
@@ -99243,7 +99328,8 @@ var init_ProbeAgent = __esm({
         } else {
           this.allowedFolders = [process.cwd()];
         }
-        this.cwd = options.cwd || null;
+        this.workspaceRoot = getCommonPrefix(this.allowedFolders);
+        this.cwd = options.cwd || this.workspaceRoot;
         this.clientApiProvider = options.provider || null;
         this.clientApiModel = options.model || null;
         this.clientApiKey = null;
@@ -99255,6 +99341,8 @@ var init_ProbeAgent = __esm({
           console.log(`[DEBUG] Maximum tool iterations configured: ${MAX_TOOL_ITERATIONS}`);
           console.log(`[DEBUG] Allow Edit (implement tool): ${this.allowEdit}`);
           console.log(`[DEBUG] Search delegation enabled: ${this.searchDelegate}`);
+          console.log(`[DEBUG] Workspace root: ${this.workspaceRoot}`);
+          console.log(`[DEBUG] Working directory (cwd): ${this.cwd}`);
         }
         this.initializeTools();
         this.history = [];
@@ -99602,8 +99690,9 @@ var init_ProbeAgent = __esm({
         const configOptions = {
           sessionId: this.sessionId,
           debug: this.debug,
-          // Use explicit cwd if set, otherwise fall back to first allowed folder
-          cwd: this.cwd || (this.allowedFolders.length > 0 ? this.allowedFolders[0] : process.cwd()),
+          // Use cwd (which defaults to workspaceRoot in constructor)
+          cwd: this.cwd,
+          workspaceRoot: this.workspaceRoot,
           allowedFolders: this.allowedFolders,
           outline: this.outline,
           searchDelegate: this.searchDelegate,
@@ -100290,16 +100379,16 @@ var init_ProbeAgent = __esm({
           let absolutePath;
           let isPathAllowed2 = false;
           if ((0, import_path17.isAbsolute)(imagePath)) {
-            absolutePath = (0, import_path17.normalize)((0, import_path17.resolve)(imagePath));
+            absolutePath = safeRealpath((0, import_path17.resolve)(imagePath));
             isPathAllowed2 = allowedDirs.some((dir) => {
-              const normalizedDir = (0, import_path17.normalize)((0, import_path17.resolve)(dir));
-              return absolutePath === normalizedDir || absolutePath.startsWith(normalizedDir + import_path17.sep);
+              const resolvedDir = safeRealpath(dir);
+              return absolutePath === resolvedDir || absolutePath.startsWith(resolvedDir + import_path17.sep);
             });
           } else {
             for (const dir of allowedDirs) {
-              const normalizedDir = (0, import_path17.normalize)((0, import_path17.resolve)(dir));
-              const resolvedPath2 = (0, import_path17.normalize)((0, import_path17.resolve)(dir, imagePath));
-              if (resolvedPath2 === normalizedDir || resolvedPath2.startsWith(normalizedDir + import_path17.sep)) {
+              const resolvedDir = safeRealpath(dir);
+              const resolvedPath2 = safeRealpath((0, import_path17.resolve)(dir, imagePath));
+              if (resolvedPath2 === resolvedDir || resolvedPath2.startsWith(resolvedDir + import_path17.sep)) {
                 absolutePath = resolvedPath2;
                 isPathAllowed2 = true;
                 break;
@@ -100482,7 +100571,7 @@ var init_ProbeAgent = __esm({
         if (this._architectureContextLoaded) {
           return this.architectureContext;
         }
-        const rootDirectory = this.allowedFolders.length > 0 ? this.allowedFolders[0] : process.cwd();
+        const rootDirectory = this.workspaceRoot || (this.allowedFolders.length > 0 ? this.allowedFolders[0] : process.cwd());
         const configuredName = typeof this.architectureFileName === "string" ? this.architectureFileName.trim() : "";
         const hasConfiguredName = !!configuredName;
         let guidanceCandidates = [];
@@ -100619,6 +100708,9 @@ ${this.architectureContext.content}
 `;
       }
       _getSkillsRepoRoot() {
+        if (this.workspaceRoot) {
+          return (0, import_path17.resolve)(this.workspaceRoot);
+        }
         if (this.allowedFolders && this.allowedFolders.length > 0) {
           return (0, import_path17.resolve)(this.allowedFolders[0]);
         }
@@ -100686,7 +100778,7 @@ Workspace: ${this.allowedFolders.join(", ")}`;
 
 # Repository Structure
 `;
-          systemPrompt += `You are working with a repository located at: ${this.allowedFolders[0]}
+          systemPrompt += `You are working with a repository located at: ${this.workspaceRoot}
 
 `;
           systemPrompt += `Here's an overview of the repository structure (showing up to 100 most relevant files):
@@ -100739,7 +100831,7 @@ Workspace: ${this.allowedFolders.join(", ")}`;
 
 # Repository Structure
 `;
-          systemPrompt += `You are working with a repository located at: ${this.allowedFolders[0]}
+          systemPrompt += `You are working with a repository located at: ${this.workspaceRoot}
 
 `;
           systemPrompt += `Here's an overview of the repository structure (showing up to 100 most relevant files):
@@ -101038,21 +101130,34 @@ For MCP tools, use JSON format within the params tag, e.g.:
 `;
           }
         }
-        const searchDirectory = this.allowedFolders.length > 0 ? this.allowedFolders[0] : process.cwd();
+        const searchDirectory = this.workspaceRoot;
         if (this.debug) {
-          console.log(`[DEBUG] Generating file list for base directory: ${searchDirectory}...`);
+          console.log(`[DEBUG] Generating file list for workspace root: ${searchDirectory}...`);
+        }
+        const relativeWorkspaces = this.allowedFolders.map((f5) => {
+          const rel = toRelativePath(f5, this.workspaceRoot);
+          if (rel && rel !== "." && !rel.startsWith(".") && !rel.startsWith("/")) {
+            return "./" + rel;
+          }
+          return rel;
+        }).filter((f5) => f5 && f5 !== ".");
+        let workspaceDesc;
+        if (relativeWorkspaces.length === 0) {
+          workspaceDesc = ". (current directory)";
+        } else {
+          workspaceDesc = relativeWorkspaces.join(", ");
         }
         try {
           const files = await listFilesByLevel({
             directory: searchDirectory,
             maxFiles: 100,
             respectGitignore: !process.env.PROBE_NO_GITIGNORE || process.env.PROBE_NO_GITIGNORE === "",
-            cwd: process.cwd()
+            cwd: this.workspaceRoot
           });
           systemMessage += `
 # Repository Structure
 
-You are working with a repository located at: ${searchDirectory}
+You are working with a workspace. Available paths: ${workspaceDesc}
 
 Here's an overview of the repository structure (showing up to 100 most relevant files):
 
@@ -101068,15 +101173,22 @@ ${files}
           systemMessage += `
 # Repository Structure
 
-You are working with a repository located at: ${searchDirectory}
+You are working with a workspace. Available paths: ${workspaceDesc}
 
 `;
         }
         await this.loadArchitectureContext();
         systemMessage += this.getArchitectureSection();
         if (this.allowedFolders.length > 0) {
+          const relativeAllowed = this.allowedFolders.map((f5) => {
+            const rel = toRelativePath(f5, this.workspaceRoot);
+            if (rel && rel !== "." && !rel.startsWith(".") && !rel.startsWith("/")) {
+              return "./" + rel;
+            }
+            return rel;
+          });
           systemMessage += `
-**Important**: For security reasons, you can only search within these allowed folders: ${this.allowedFolders.join(", ")}
+**Important**: For security reasons, you can only access these paths: ${relativeAllowed.join(", ")}
 
 `;
         }
@@ -101590,6 +101702,7 @@ You are working with a repository located at: ${searchDirectory}
                       console.error(`[DEBUG] ========================================
 `);
                     }
+                    currentMessages.push({ role: "assistant", content: assistantResponseContent });
                     currentMessages.push({ role: "user", content: `<tool_result>
 ${toolResultContent}
 </tool_result>` });
@@ -101610,17 +101723,18 @@ ${toolResultContent}
 `);
                     }
                     const errorXml = formatErrorForAI(error2);
+                    currentMessages.push({ role: "assistant", content: assistantResponseContent });
                     currentMessages.push({ role: "user", content: `<tool_result>
 ${errorXml}
 </tool_result>` });
                   }
                 } else if (this.toolImplementations[toolName]) {
                   try {
-                    let resolvedWorkingDirectory = this.cwd || this.allowedFolders && this.allowedFolders[0] || process.cwd();
+                    let resolvedWorkingDirectory = this.workspaceRoot || this.cwd || this.allowedFolders && this.allowedFolders[0] || process.cwd();
                     if (params.workingDirectory) {
-                      const requestedDir = (0, import_path17.isAbsolute)(params.workingDirectory) ? (0, import_path17.resolve)(params.workingDirectory) : (0, import_path17.resolve)(resolvedWorkingDirectory, params.workingDirectory);
+                      const requestedDir = safeRealpath((0, import_path17.isAbsolute)(params.workingDirectory) ? (0, import_path17.resolve)(params.workingDirectory) : (0, import_path17.resolve)(resolvedWorkingDirectory, params.workingDirectory));
                       const isWithinAllowed = !this.allowedFolders || this.allowedFolders.length === 0 || this.allowedFolders.some((folder) => {
-                        const resolvedFolder = (0, import_path17.resolve)(folder);
+                        const resolvedFolder = safeRealpath(folder);
                         return requestedDir === resolvedFolder || requestedDir.startsWith(resolvedFolder + import_path17.sep);
                       });
                       if (isWithinAllowed) {
@@ -102089,7 +102203,7 @@ Convert your previous response content into actual JSON data that follows this s
                   }
                   const mermaidValidation = await validateAndFixMermaidResponse(finalResult, {
                     debug: this.debug,
-                    path: this.allowedFolders[0],
+                    path: this.workspaceRoot || this.allowedFolders[0],
                     provider: this.clientApiProvider,
                     model: this.model,
                     tracer: this.tracer
@@ -102162,7 +102276,7 @@ Convert your previous response content into actual JSON data that follows this s
                   }
                   const { JsonFixingAgent: JsonFixingAgent2 } = await Promise.resolve().then(() => (init_schemaUtils(), schemaUtils_exports));
                   const jsonFixer = new JsonFixingAgent2({
-                    path: this.allowedFolders[0],
+                    path: this.workspaceRoot || this.allowedFolders[0],
                     provider: this.clientApiProvider,
                     model: this.model,
                     debug: this.debug,
@@ -102231,7 +102345,7 @@ Convert your previous response content into actual JSON data that follows this s
                 }
                 const mermaidValidation = await validateAndFixMermaidResponse(finalResult, {
                   debug: this.debug,
-                  path: this.allowedFolders[0],
+                  path: this.workspaceRoot || this.allowedFolders[0],
                   provider: this.clientApiProvider,
                   model: this.model,
                   tracer: this.tracer
@@ -102363,7 +102477,7 @@ Convert your previous response content into actual JSON data that follows this s
               }
               const finalMermaidValidation = await validateAndFixMermaidResponse(finalResult, {
                 debug: this.debug,
-                path: this.allowedFolders[0],
+                path: this.workspaceRoot || this.allowedFolders[0],
                 provider: this.clientApiProvider,
                 model: this.model,
                 tracer: this.tracer
@@ -102513,8 +102627,7 @@ Convert your previous response content into actual JSON data that follows this s
           allowEdit: this.allowEdit,
           enableDelegate: this.enableDelegate,
           architectureFileName: this.architectureFileName,
-          path: this.allowedFolders[0],
-          // Use first allowed folder as primary path
+          // Pass allowedFolders which will recompute workspaceRoot correctly
           allowedFolders: [...this.allowedFolders],
           cwd: this.cwd,
           // Preserve explicit working directory