wave-agent-sdk 0.16.9 → 0.16.12

package/dist/utils/containerSetup.js

@@ -29,6 +29,7 @@ import { USER_MEMORY_FILE } from "./constants.js";
 import { getGitMainRepoRoot } from "./gitUtils.js";
 import { logger } from "./globalLogger.js";
 import { authService } from "../services/authService.js";
+import { remoteSettingsService } from "../services/remoteSettingsService.js";
 export function setupAgentContainer(setupOptions) {
     const { options, workdir, configurationService, systemPrompt, stream, onBackgroundTasksChange, onTasksChange, onPermissionModeChange, handlePlanModeTransition, setPermissionMode, addPermissionRule, addUsage, } = setupOptions;
     const callbacks = options.callbacks || {};
@@ -89,24 +90,21 @@ export function setupAgentContainer(setupOptions) {
         workdir,
     });
     container.register("BackgroundTaskManager", backgroundTaskManager);
-    const ssoToken = authService.getSSOToken();
-    const serverUrl = options.serverUrl || process.env.WAVE_SERVER_URL;
     if (options.serverUrl) {
         authService.setServerUrl(options.serverUrl);
     }
     const mcpManager = new McpManager(container, {
         callbacks,
         mcpServers: options.mcpServers,
-        serverUrl,
-        ssoToken,
     });
     container.register("McpManager", mcpManager);
-    // Wire up auth change callback to reconnect MCP servers after SSO login
+    // Wire up auth change callback to refresh/clear remote settings
     authService.onAuthChange((event) => {
         if (event === "login") {
-            const newServerUrl = options.serverUrl || process.env.WAVE_SERVER_URL;
-            const newToken = authService.getSSOToken();
-            mcpManager.refreshCredentials(newServerUrl, newToken);
+            remoteSettingsService.refresh();
+        }
+        else if (event === "logout") {
+            remoteSettingsService.clear();
         }
     });
     const lspManager = options.lspManager || new LspManager(container);
@@ -128,6 +126,9 @@ export function setupAgentContainer(setupOptions) {
     const planManager = new PlanManager(container);
     container.register("PlanManager", planManager);
     const hookManager = new HookManager(container, workdir);
+    if (options.hooks) {
+        hookManager.loadConfiguration(options.hooks);
+    }
     container.register("HookManager", hookManager);
     const skillManager = new SkillManager(container, {
         workdir,

package/dist/utils/gitUtils.d.ts

@@ -23,7 +23,24 @@ export declare function getGitCommonDir(cwd: string): string;
  */
 export declare function getGitMainRepoRoot(cwd: string): string;
 /**
- * Get the default remote branch (e.g., origin/main)
+ * Resolve the git directory for a repository by walking up from `cwd`.
+ * Handles both normal repos (.git is a directory) and worktrees
+ * (.git is a file pointing to the main repo's worktree git dir).
+ * For worktrees, reads the `commondir` file to find the common git dir.
+ * @param cwd Working directory to start searching from
+ * @returns Absolute path to the git directory, or null if not found
+ */
+export declare function resolveGitDir(cwd: string): string | null;
+/**
+ * Get the default remote branch (e.g., origin/main) using filesystem reads.
+ * No subprocess calls — matches Claude Code's approach.
+ *
+ * Priority:
+ * 1. Read refs/remotes/origin/HEAD symref → extract branch name (verify it exists)
+ * 2. Check if refs/remotes/origin/main ref exists
+ * 3. Check if refs/remotes/origin/master ref exists
+ * 4. Hardcoded "main" fallback
+ *
  * @param cwd Working directory
  * @returns Default remote branch name
  */

package/dist/utils/gitUtils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"gitUtils.d.ts","sourceRoot":"","sources":["../../src/utils/gitUtils.ts"],"names":[],"mappings":"AAIA;;;;GAIG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAevD;AAED;;;;GAIG;AACH,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAUlD;AAED;;;;GAIG;AACH,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAWnD;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAetD;AAED;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CA2D1D;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAW1D;AAED;;;;;GAKG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAYvE"}
+{"version":3,"file":"gitUtils.d.ts","sourceRoot":"","sources":["../../src/utils/gitUtils.ts"],"names":[],"mappings":"AAIA;;;;GAIG;AACH,wBAAgB,eAAe,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAevD;AAED;;;;GAIG;AACH,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAUlD;AAED;;;;GAIG;AACH,wBAAgB,eAAe,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAWnD;AAED;;;;GAIG;AACH,wBAAgB,kBAAkB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAetD;AAED;;;;;;;GAOG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAkCxD;AAwDD;;;;;;;;;;;;GAYG;AACH,wBAAgB,sBAAsB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CA4B1D;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAW1D;AAED;;;;;GAKG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,GAAG,OAAO,CAYvE"}

package/dist/utils/gitUtils.js

@@ -81,69 +81,140 @@ export function getGitMainRepoRoot(cwd) {
     }
 }
 /**
- * Get the default remote branch (e.g., origin/main)
- * @param cwd Working directory
- * @returns Default remote branch name
+ * Resolve the git directory for a repository by walking up from `cwd`.
+ * Handles both normal repos (.git is a directory) and worktrees
+ * (.git is a file pointing to the main repo's worktree git dir).
+ * For worktrees, reads the `commondir` file to find the common git dir.
+ * @param cwd Working directory to start searching from
+ * @returns Absolute path to the git directory, or null if not found
  */
-export function getDefaultRemoteBranch(cwd) {
-    // 1. Try git symbolic-ref refs/remotes/origin/HEAD
-    try {
-        const head = execSync("git symbolic-ref refs/remotes/origin/HEAD", {
-            cwd,
-            encoding: "utf8",
-            stdio: ["ignore", "pipe", "ignore"],
-        }).trim();
-        return head.replace("refs/remotes/", "");
-    }
-    catch {
-        // Ignore error and proceed to next step
+export function resolveGitDir(cwd) {
+    let currentPath = path.resolve(cwd);
+    while (currentPath !== path.dirname(currentPath)) {
+        const gitPath = path.join(currentPath, ".git");
+        if (fsSync.existsSync(gitPath)) {
+            const stat = fsSync.statSync(gitPath);
+            if (stat.isDirectory()) {
+                // Normal repo: .git is a directory
+                return gitPath;
+            }
+            // Worktree: .git is a file containing "gitdir: <path>"
+            try {
+                const content = fsSync.readFileSync(gitPath, "utf8").trim();
+                const prefix = "gitdir: ";
+                if (content.startsWith(prefix)) {
+                    const worktreeGitDir = content.substring(prefix.length);
+                    // Read commondir to find the common git dir
+                    const commondirPath = path.join(worktreeGitDir, "commondir");
+                    if (fsSync.existsSync(commondirPath)) {
+                        const commondirRel = fsSync
+                            .readFileSync(commondirPath, "utf8")
+                            .trim();
+                        return path.resolve(worktreeGitDir, commondirRel);
+                    }
+                    // Fallback: resolve ../.. from the worktree git dir
+                    return path.resolve(worktreeGitDir, "..", "..");
+                }
+            }
+            catch {
+                return null;
+            }
+        }
+        currentPath = path.dirname(currentPath);
     }
-    // 2. Check if origin/main exists
+    return null;
+}
+/**
+ * Read a symbolic ref file in the git directory.
+ * If the file content starts with "ref: ", returns the target ref path.
+ * Symbolic refs are never packed in packed-refs, so only loose refs are checked.
+ * @param gitDir Absolute path to the git directory
+ * @param refPath Relative path to the ref file (e.g., "refs/remotes/origin/HEAD")
+ * @returns The symbolic ref target, or null if not a symbolic ref or on error
+ */
+function readSymref(gitDir, refPath) {
     try {
-        execSync("git rev-parse --verify origin/main", {
-            cwd,
-            stdio: "ignore",
-        });
-        return "origin/main";
+        const fullPath = path.join(gitDir, refPath);
+        const content = fsSync.readFileSync(fullPath, "utf8").trim();
+        if (content.startsWith("ref: ")) {
+            return content.substring("ref: ".length);
+        }
+        return null;
     }
     catch {
-        // Ignore error and proceed to next step
+        return null;
     }
-    // 3. Check if origin/master exists
+}
+/**
+ * Check if a git ref exists, checking both loose refs and packed-refs.
+ * @param gitDir Absolute path to the git directory
+ * @param refPath Relative path to the ref (e.g., "refs/remotes/origin/main")
+ * @returns True if the ref exists, false otherwise
+ */
+function refExists(gitDir, refPath) {
+    // 1. Check loose ref
+    const loosePath = path.join(gitDir, refPath);
+    if (fsSync.existsSync(loosePath)) {
+        return true;
+    }
+    // 2. Check packed-refs
     try {
-        execSync("git rev-parse --verify origin/master", {
-            cwd,
-            stdio: "ignore",
-        });
-        return "origin/master";
+        const packedRefsPath = path.join(gitDir, "packed-refs");
+        const content = fsSync.readFileSync(packedRefsPath, "utf8");
+        for (const line of content.split("\n")) {
+            // Skip comments and peeled lines
+            if (line.startsWith("#") || line.startsWith("^")) {
+                continue;
+            }
+            // Format: <sha> <refPath>
+            const parts = line.split(" ");
+            if (parts.length >= 2 && parts[1] === refPath) {
+                return true;
+            }
+        }
     }
     catch {
-        // Ignore error and proceed to next step
+        // packed-refs doesn't exist or can't be read
     }
-    // 4. Try to get the current branch's upstream
-    try {
-        return execSync("git rev-parse --abbrev-ref --symbolic-full-name @{u}", {
-            cwd,
-            encoding: "utf8",
-            stdio: ["ignore", "pipe", "ignore"],
-        }).trim();
+    return false;
+}
+/**
+ * Get the default remote branch (e.g., origin/main) using filesystem reads.
+ * No subprocess calls — matches Claude Code's approach.
+ *
+ * Priority:
+ * 1. Read refs/remotes/origin/HEAD symref → extract branch name (verify it exists)
+ * 2. Check if refs/remotes/origin/main ref exists
+ * 3. Check if refs/remotes/origin/master ref exists
+ * 4. Hardcoded "main" fallback
+ *
+ * @param cwd Working directory
+ * @returns Default remote branch name
+ */
+export function getDefaultRemoteBranch(cwd) {
+    const gitDir = resolveGitDir(cwd);
+    if (!gitDir) {
+        return "main";
     }
-    catch {
-        // Ignore error and proceed to next step
+    // 1. Try reading origin/HEAD symref
+    const symref = readSymref(gitDir, "refs/remotes/origin/HEAD");
+    if (symref) {
+        const branch = symref.replace("refs/remotes/", "");
+        // Verify the resolved branch actually exists (origin/HEAD can be stale)
+        if (refExists(gitDir, symref)) {
+            return branch;
+        }
     }
-    // 5. Try to get the current branch name
-    try {
-        return execSync("git rev-parse --abbrev-ref HEAD", {
-            cwd,
-            encoding: "utf8",
-            stdio: ["ignore", "pipe", "ignore"],
-        }).trim();
+    // 2. Check if origin/main exists
+    if (refExists(gitDir, "refs/remotes/origin/main")) {
+        return "origin/main";
     }
-    catch {
-        // Ignore error and proceed to next step
+    // 3. Check if origin/master exists
+    if (refExists(gitDir, "refs/remotes/origin/master")) {
+        return "origin/master";
     }
-    // 6. Fallback to origin/main
-    return "origin/main";
+    // 4. Hardcoded fallback
+    return "main";
 }
 /**
  * Check if there are uncommitted changes in the working directory

package/dist/utils/mcpUtils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcpUtils.d.ts","sourceRoot":"","sources":["../../src/utils/mcpUtils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,KAAK,EAAE,UAAU,EAAc,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAC7E,OAAO,KAAK,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AA8ClE;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,OAAO,EAChB,UAAU,EAAE,MAAM,GACjB,0BAA0B,CAgB5B;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,OAAO,EAChB,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,CACX,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,CAAC,EAAE,WAAW,KAClB,OAAO,CAAC;IACX,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACtD,CAAC,GACD,UAAU,CAwBZ;AAED;;GAEG;AACH,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,eAAe,EAAE,GACzB,eAAe,GAAG,SAAS,CAK7B"}
+{"version":3,"file":"mcpUtils.d.ts","sourceRoot":"","sources":["../../src/utils/mcpUtils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,KAAK,EAAE,UAAU,EAAc,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAC7E,OAAO,KAAK,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAgDlE;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,OAAO,EAChB,UAAU,EAAE,MAAM,GACjB,0BAA0B,CAgB5B;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,OAAO,EAChB,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,CACX,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,CAAC,EAAE,WAAW,KAClB,OAAO,CAAC;IACX,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACtD,CAAC,GACD,UAAU,CAiCZ;AAED;;GAEG;AACH,wBAAgB,cAAc,CAC5B,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,eAAe,EAAE,GACzB,eAAe,GAAG,SAAS,CAK7B"}

package/dist/utils/mcpUtils.js

@@ -1,3 +1,5 @@
+import { processToolResult } from "./toolResultStorage.js";
+import { DEFAULT_MAX_RESULT_SIZE_CHARS } from "../constants/toolLimits.js";
 /**
  * Recursively clean schema to remove unsupported fields
  */
@@ -60,9 +62,12 @@ export function createMcpToolPlugin(mcpTool, serverName, executeTool) {
         async execute(args, context) {
             try {
                 const result = await executeTool(prefixedName, args, context);
+                // Process content for size limits — only text content, not images
+                const processedContent = processToolResult(result.content || `Executed ${mcpTool.name}`, DEFAULT_MAX_RESULT_SIZE_CHARS, `mcp_${serverName}_${mcpTool.name}`);
                 return {
                     success: true,
-                    content: result.content || `Executed ${mcpTool.name}`,
+                    content: processedContent,
+                    images: result.images,
                 };
             }
             catch (error) {

package/dist/utils/openaiClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"openaiClient.d.ts","sourceRoot":"","sources":["../../src/utils/openaiClient.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sCAAsC,EACtC,mCAAmC,EACnC,mBAAmB,EACnB,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAGnD,KAAK,YAAY,GACb,sCAAsC,GACtC,mCAAmC,CAAC;AAExC,UAAU,WAAW,CAAC,CAAC;IACrB,IAAI,EAAE,CAAC,CAAC;IACR,QAAQ,EAAE,QAAQ,CAAC;CACpB;AAED,UAAU,UAAU,CAAC,CAAC,CAAE,SAAQ,OAAO,CAAC,CAAC,CAAC;IACxC,YAAY,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;CACzC;AAED,qBAAa,YAAY;IACX,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,aAAa;IAEzC,IAAI,IAAI;;qBAGO,CAAC,SAAS,YAAY,UACrB,CAAC,YACC;gBAAE,MAAM,CAAC,EAAE,WAAW,CAAA;aAAE,KACjC,UAAU,CACX,CAAC,SAAS,mCAAmC,GACzC,aAAa,CAAC,mBAAmB,CAAC,GAClC,cAAc,CACnB;;MA2BN;YAEa,OAAO;YAqIN,oBAAoB;CAqCpC"}
+{"version":3,"file":"openaiClient.d.ts","sourceRoot":"","sources":["../../src/utils/openaiClient.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sCAAsC,EACtC,mCAAmC,EACnC,mBAAmB,EACnB,cAAc,EACf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAGnD,KAAK,YAAY,GACb,sCAAsC,GACtC,mCAAmC,CAAC;AAExC,UAAU,WAAW,CAAC,CAAC;IACrB,IAAI,EAAE,CAAC,CAAC;IACR,QAAQ,EAAE,QAAQ,CAAC;CACpB;AAED,UAAU,UAAU,CAAC,CAAC,CAAE,SAAQ,OAAO,CAAC,CAAC,CAAC;IACxC,YAAY,IAAI,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;CACzC;AAED,qBAAa,YAAY;IACX,OAAO,CAAC,MAAM;gBAAN,MAAM,EAAE,aAAa;IAEzC,IAAI,IAAI;;qBAGO,CAAC,SAAS,YAAY,UACrB,CAAC,YACC;gBAAE,MAAM,CAAC,EAAE,WAAW,CAAA;aAAE,KACjC,UAAU,CACX,CAAC,SAAS,mCAAmC,GACzC,aAAa,CAAC,mBAAmB,CAAC,GAClC,cAAc,CACnB;;MA2BN;YAEa,OAAO;YAwIN,oBAAoB;CAqCpC"}

package/dist/utils/openaiClient.js

@@ -107,8 +107,10 @@ export class OpenAIClient {
                     : response.statusText);
             error.status = response.status;
             error.body = errorBody;
-            if (response.status === 429 && attempt < maxRetries) {
-                logger.warn("OpenAI API 429 Too Many Requests, retrying...", {
+            const retryableStatus = response.status === 429 ||
+                (response.status >= 500 && response.status !== 501);
+            if (retryableStatus && attempt < maxRetries) {
+                logger.warn("OpenAI API error, retrying...", {
                     attempt: attempt + 1,
                     status: response.status,
                 });

package/dist/utils/toolResultStorage.d.ts

@@ -0,0 +1,46 @@
+/**
+ * Shared tool result persistence and truncation logic.
+ *
+ * When a tool result exceeds a size threshold, the full content is saved to a
+ * file in /tmp/wave-tool-results/ and the model receives a <persisted-output>
+ * preview with the file path so it can use the Read tool to access the full output.
+ */
+/**
+ * Get (and create if needed) the tool-results directory.
+ * Uses /tmp/wave-tool-results/ for simplicity and automatic OS cleanup.
+ */
+export declare function getToolResultsDir(): string;
+/**
+ * Persist full tool output to a file in the tool-results directory.
+ * Returns the file path on success, or undefined on failure.
+ */
+export declare function persistToolResult(content: string, prefix?: string): string | undefined;
+/**
+ * Generate a preview from content: first `previewSize` characters with ellipsis.
+ */
+export declare function generatePreview(content: string, previewSize?: number): string;
+/**
+ * Build the <persisted-output> wrapper message that the model sees.
+ *
+ * Example output:
+ * <persisted-output>
+ * Output too large (150,000 characters). Full output saved to: /tmp/wave-tool-results/mcp_server_tool_12345.txt
+ * Preview (first 2,048 characters):
+ * {preview content}
+ * ...
+ * </persisted-output>
+ */
+export declare function buildPersistedOutputMessage(totalChars: number, filePath: string, preview: string): string;
+/**
+ * Process tool result: if content exceeds maxChars, persist to file and return
+ * truncated content with <persisted-output> wrapper. Otherwise return unchanged.
+ *
+ * This is the main entry point for both MCP and bash tools.
+ *
+ * @param content - The tool result content
+ * @param maxChars - Maximum characters before persistence kicks in (defaults to DEFAULT_MAX_RESULT_SIZE_CHARS)
+ * @param prefix - File name prefix for the persisted file (e.g. "bash", "mcp")
+ * @returns The content to send to the model (either original or persisted-output wrapper)
+ */
+export declare function processToolResult(content: string, maxChars?: number, prefix?: string): string;
+//# sourceMappingURL=toolResultStorage.d.ts.map

package/dist/utils/toolResultStorage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolResultStorage.d.ts","sourceRoot":"","sources":["../../src/utils/toolResultStorage.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAaH;;;GAGG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,CAG1C;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,MAAM,EACf,MAAM,GAAE,MAAe,GACtB,MAAM,GAAG,SAAS,CAWpB;AAED;;GAEG;AACH,wBAAgB,eAAe,CAC7B,OAAO,EAAE,MAAM,EACf,WAAW,GAAE,MAA2B,GACvC,MAAM,CAGR;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,2BAA2B,CACzC,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,MAAM,EAChB,OAAO,EAAE,MAAM,GACd,MAAM,CAQR;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,MAAM,EACf,QAAQ,GAAE,MAAsC,EAChD,MAAM,GAAE,MAAe,GACtB,MAAM,CAiBR"}

package/dist/utils/toolResultStorage.js

@@ -0,0 +1,90 @@
+/**
+ * Shared tool result persistence and truncation logic.
+ *
+ * When a tool result exceeds a size threshold, the full content is saved to a
+ * file in /tmp/wave-tool-results/ and the model receives a <persisted-output>
+ * preview with the file path so it can use the Read tool to access the full output.
+ */
+import * as fs from "fs";
+import * as path from "path";
+import * as os from "os";
+import { DEFAULT_MAX_RESULT_SIZE_CHARS, PREVIEW_SIZE_BYTES, } from "../constants/toolLimits.js";
+import { logger } from "./globalLogger.js";
+const TOOL_RESULTS_DIR = path.join(os.tmpdir(), "wave-tool-results");
+/**
+ * Get (and create if needed) the tool-results directory.
+ * Uses /tmp/wave-tool-results/ for simplicity and automatic OS cleanup.
+ */
+export function getToolResultsDir() {
+    fs.mkdirSync(TOOL_RESULTS_DIR, { recursive: true });
+    return TOOL_RESULTS_DIR;
+}
+/**
+ * Persist full tool output to a file in the tool-results directory.
+ * Returns the file path on success, or undefined on failure.
+ */
+export function persistToolResult(content, prefix = "tool") {
+    try {
+        const dir = getToolResultsDir();
+        const id = `${prefix}_${Date.now()}_${Math.random().toString(36).substring(2, 9)}`;
+        const filePath = path.join(dir, `${id}.txt`);
+        fs.writeFileSync(filePath, content, "utf8");
+        return filePath;
+    }
+    catch (error) {
+        logger?.error("Failed to persist tool result:", error);
+        return undefined;
+    }
+}
+/**
+ * Generate a preview from content: first `previewSize` characters with ellipsis.
+ */
+export function generatePreview(content, previewSize = PREVIEW_SIZE_BYTES) {
+    if (content.length <= previewSize)
+        return content;
+    return content.substring(0, previewSize) + "\n...";
+}
+/**
+ * Build the <persisted-output> wrapper message that the model sees.
+ *
+ * Example output:
+ * <persisted-output>
+ * Output too large (150,000 characters). Full output saved to: /tmp/wave-tool-results/mcp_server_tool_12345.txt
+ * Preview (first 2,048 characters):
+ * {preview content}
+ * ...
+ * </persisted-output>
+ */
+export function buildPersistedOutputMessage(totalChars, filePath, preview) {
+    return [
+        "<persisted-output>",
+        `Output too large (${totalChars.toLocaleString()} characters). Full output saved to: ${filePath}`,
+        `Preview (first ${PREVIEW_SIZE_BYTES.toLocaleString()} characters):`,
+        preview,
+        "</persisted-output>",
+    ].join("\n");
+}
+/**
+ * Process tool result: if content exceeds maxChars, persist to file and return
+ * truncated content with <persisted-output> wrapper. Otherwise return unchanged.
+ *
+ * This is the main entry point for both MCP and bash tools.
+ *
+ * @param content - The tool result content
+ * @param maxChars - Maximum characters before persistence kicks in (defaults to DEFAULT_MAX_RESULT_SIZE_CHARS)
+ * @param prefix - File name prefix for the persisted file (e.g. "bash", "mcp")
+ * @returns The content to send to the model (either original or persisted-output wrapper)
+ */
+export function processToolResult(content, maxChars = DEFAULT_MAX_RESULT_SIZE_CHARS, prefix = "tool") {
+    if (content.length <= maxChars) {
+        return content;
+    }
+    const filePath = persistToolResult(content, prefix);
+    if (filePath) {
+        const preview = generatePreview(content);
+        return buildPersistedOutputMessage(content.length, filePath, preview);
+    }
+    // Fallback: truncation only (persistence failed)
+    return (content.substring(0, maxChars) +
+        "\n\n... (output truncated, failed to persist full output)");
+}

package/dist/utils/worktreeUtils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"worktreeUtils.d.ts","sourceRoot":"","sources":["../../src/utils/worktreeUtils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAQH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;IACf,mFAAmF;IACnF,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAmBvD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,MAAM,CA6B7C;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAKjD;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,YAAY,CAwFtE;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CA+DvD;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,YAAY,EAAE,MAAM,EACpB,kBAAkB,EAAE,MAAM,GAAG,SAAS,GACrC;IAAE,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CA8BlD"}
+{"version":3,"file":"worktreeUtils.d.ts","sourceRoot":"","sources":["../../src/utils/worktreeUtils.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAQH,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;IACf,mFAAmF;IACnF,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAmBvD;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,MAAM,CA6B7C;AAED;;GAEG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAKjD;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,YAAY,CAuJtE;AAED;;GAEG;AACH,wBAAgB,cAAc,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI,CA+DvD;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAClC,YAAY,EAAE,MAAM,EACpB,kBAAkB,EAAE,MAAM,GAAG,SAAS,GACrC;IAAE,YAAY,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CA8BlD"}

package/dist/utils/worktreeUtils.js

@@ -142,6 +142,64 @@ export function createWorktree(name, cwd) {
                 throw new Error(`Failed to add worktree: ${innerError.message}`);
             }
         }
+        if (stderr.includes("not a valid object name") ||
+            stderr.includes("unknown revision")) {
+            // Base branch not fetched yet — try fetching then retrying
+            const branchNameOnly = baseBranch.split("/").pop();
+            try {
+                execSync(`git fetch origin ${branchNameOnly}`, {
+                    cwd: repoRoot,
+                    stdio: ["ignore", "pipe", "pipe"],
+                    env: {
+                        ...process.env,
+                        GIT_TERMINAL_PROMPT: "0",
+                        GIT_ASKPASS: "",
+                    },
+                });
+                execSync(`git worktree add -b ${branchName} "${worktreePath}" ${baseBranch}`, {
+                    cwd: repoRoot,
+                    stdio: ["ignore", "pipe", "pipe"],
+                    env: {
+                        ...process.env,
+                        GIT_TERMINAL_PROMPT: "0",
+                        GIT_ASKPASS: "",
+                    },
+                });
+                return {
+                    name,
+                    path: worktreePath,
+                    branch: branchName,
+                    repoRoot,
+                    isNew: true,
+                    originalHeadCommit,
+                };
+            }
+            catch {
+                // Fetch or retry failed — fall back to HEAD
+                try {
+                    execSync(`git worktree add -b ${branchName} "${worktreePath}" HEAD`, {
+                        cwd: repoRoot,
+                        stdio: ["ignore", "pipe", "pipe"],
+                        env: {
+                            ...process.env,
+                            GIT_TERMINAL_PROMPT: "0",
+                            GIT_ASKPASS: "",
+                        },
+                    });
+                    return {
+                        name,
+                        path: worktreePath,
+                        branch: branchName,
+                        repoRoot,
+                        isNew: true,
+                        originalHeadCommit,
+                    };
+                }
+                catch {
+                    throw new Error(`Failed to create worktree: ${error.message}\n${stderr}`);
+                }
+            }
+        }
         throw new Error(`Failed to create worktree: ${error.message}\n${stderr}`);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-agent-sdk",
-  "version": "0.16.9",
+  "version": "0.16.12",
   "description": "SDK for building AI-powered development tools and agents",
   "keywords": [
     "ai",
@@ -48,11 +48,11 @@
   },
   "devDependencies": {
     "@types/turndown": "^5.0.6",
-    "@vitest/coverage-v8": "^4.0.18",
+    "@vitest/coverage-v8": "^4.1.7",
     "rimraf": "^6.1.2",
     "tsc-alias": "^1.8.16",
     "tsx": "^4.20.4",
-    "vitest": "^4.0.18"
+    "vitest": "^4.1.7"
   },
   "engines": {
     "node": ">=22.0.0"

package/src/agent.ts

@@ -51,6 +51,7 @@ import {
   shutdownTelemetry,
 } from "./telemetry/instrumentation.js";
 import { logOTelEvent } from "./telemetry/events.js";
+import { remoteSettingsService } from "./services/remoteSettingsService.js";
 
 export class Agent {
   private messageManager: MessageManager;
@@ -220,6 +221,13 @@ export class Agent {
       }
     };
 
+    // Wire up CWD change callback from AIManager to sync Agent's workdir
+    this.aiManager.setOnCwdChange((newCwd) => {
+      this.workdir = newCwd;
+      this.container.register("Workdir", newCwd);
+      this.options.callbacks?.onWorkdirChange?.(newCwd);
+    });
+
     // Wire up message queue to process when agent becomes idle
     this.messageQueue.onMessageEnqueued = () => {
       // If the AI is NOT loading and command is not running, trigger dequeue
@@ -282,6 +290,16 @@ export class Agent {
     return this.workdir;
   }
 
+  /**
+   * Set the working directory
+   * @param newCwd - The new working directory
+   */
+  public setWorkdir(newCwd: string): void {
+    this.workdir = newCwd;
+    this.container.register("Workdir", newCwd);
+    this.options.callbacks?.onWorkdirChange?.(newCwd);
+  }
+
   /** Get project memory content */
   public get projectMemory(): string {
     const memoryService =
@@ -720,6 +738,8 @@ export class Agent {
         error,
       );
     }
+    // Cleanup remote settings polling
+    remoteSettingsService.shutdown();
     // Cleanup memory store
   }
 

package/src/constants/toolLimits.ts

@@ -5,6 +5,9 @@
 /** System-wide default max result size in characters. */
 export const DEFAULT_MAX_RESULT_SIZE_CHARS = 50_000;
 
+/** Per-command cap for bash tool output before persistence. */
+export const BASH_MAX_OUTPUT_CHARS = 30_000;
+
 /** Per-command cap for skill bash substitution (inline/block). */
 export const SKILL_BASH_MAX_OUTPUT_CHARS = 30_000;