wave-agent-sdk 0.6.4 → 0.7.0

package/dist/utils/container.js

@@ -0,0 +1,79 @@
+/**
+ * A simple Dependency Injection container to manage late-binding dependencies
+ * and resolve circular references.
+ */
+export class Container {
+    constructor(parent) {
+        this.services = new Map();
+        this.factories = new Map();
+        this.parent = parent;
+    }
+    /**
+     * Create a child container that inherits from this container
+     */
+    createChild() {
+        return new Container(this);
+    }
+    /**
+     * Register a singleton instance
+     */
+    register(token, instance) {
+        this.services.set(token, instance);
+    }
+    /**
+     * Register a factory function for lazy resolution
+     */
+    provide(token, factory) {
+        this.factories.set(token, factory);
+    }
+    /**
+     * Resolve a service by token
+     */
+    get(token) {
+        if (this.services.has(token)) {
+            return this.services.get(token);
+        }
+        const factory = this.factories.get(token);
+        if (factory) {
+            const instance = factory();
+            this.services.set(token, instance);
+            return instance;
+        }
+        if (this.parent) {
+            return this.parent.get(token);
+        }
+        return undefined;
+    }
+    /**
+     * Check if a service exists
+     */
+    has(token) {
+        return (this.services.has(token) ||
+            this.factories.has(token) ||
+            (this.parent ? this.parent.has(token) : false));
+    }
+}
+// Example usage for ToolManager:
+/*
+const container = new Container();
+
+// 1. Provide managers lazily
+container.provide('SubagentManager', () => new SubagentManager(...));
+container.provide('SkillManager', () => new SkillManager(...));
+
+// 2. ToolManager can now resolve them only when needed (e.g., in execute)
+class ToolManager {
+  constructor(private container: Container) {}
+
+  async execute(name: string, args: any, context: any) {
+    const enhancedContext = {
+      ...context,
+      subagentManager: this.container.has('SubagentManager')
+        ? this.container.get('SubagentManager')
+        : undefined,
+      // ...
+    };
+    // ...
+  }
+}
+*/

package/dist/utils/containerSetup.d.ts

@@ -0,0 +1,26 @@
+import { Container } from "./container.js";
+import { ConfigurationService } from "../services/configurationService.js";
+import type { AgentOptions } from "../agent.js";
+import type { PermissionMode, Usage, Task, BackgroundTask } from "../types/index.js";
+import type { GatewayConfig, ModelConfig } from "../types/config.js";
+export interface AgentContainerSetupOptions {
+    options: AgentOptions;
+    workdir: string;
+    configurationService: ConfigurationService;
+    systemPrompt?: string;
+    stream: boolean;
+    onSessionIdChange: (sessionId: string) => void;
+    onTasksChange: (tasks: BackgroundTask[]) => void;
+    onSessionTasksChange: (tasks: Task[]) => void;
+    onPermissionModeChange: (mode: PermissionMode) => void;
+    handlePlanModeTransition: (mode: PermissionMode) => void;
+    setPermissionMode: (mode: PermissionMode) => void;
+    addPermissionRule: (rule: string) => Promise<void>;
+    addUsage: (usage: Usage) => void;
+    getGatewayConfig: () => GatewayConfig;
+    getModelConfig: () => ModelConfig;
+    getMaxInputTokens: () => number;
+    getLanguage: () => string | undefined;
+}
+export declare function setupAgentContainer(setupOptions: AgentContainerSetupOptions): Container;
+//# sourceMappingURL=containerSetup.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"containerSetup.d.ts","sourceRoot":"","sources":["../../src/utils/containerSetup.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAoB3C,OAAO,EAAE,oBAAoB,EAAE,MAAM,qCAAqC,CAAC;AAE3E,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAChD,OAAO,KAAK,EACV,cAAc,EACd,KAAK,EACL,IAAI,EACJ,cAAc,EAEf,MAAM,mBAAmB,CAAC;AAC3B,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAIrE,MAAM,WAAW,0BAA0B;IACzC,OAAO,EAAE,YAAY,CAAC;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,oBAAoB,EAAE,oBAAoB,CAAC;IAC3C,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,MAAM,EAAE,OAAO,CAAC;IAGhB,iBAAiB,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,IAAI,CAAC;IAC/C,aAAa,EAAE,CAAC,KAAK,EAAE,cAAc,EAAE,KAAK,IAAI,CAAC;IACjD,oBAAoB,EAAE,CAAC,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,CAAC;IAC9C,sBAAsB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IACvD,wBAAwB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IACzD,iBAAiB,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAC;IAClD,iBAAiB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACnD,QAAQ,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,IAAI,CAAC;IAGjC,gBAAgB,EAAE,MAAM,aAAa,CAAC;IACtC,cAAc,EAAE,MAAM,WAAW,CAAC;IAClC,iBAAiB,EAAE,MAAM,MAAM,CAAC;IAChC,WAAW,EAAE,MAAM,MAAM,GAAG,SAAS,CAAC;CACvC;AAED,wBAAgB,mBAAmB,CACjC,YAAY,EAAE,0BAA0B,GACvC,SAAS,CAmMX"}

package/dist/utils/containerSetup.js

@@ -0,0 +1,165 @@
+import { Container } from "./container.js";
+import { ForegroundTaskManager } from "../managers/foregroundTaskManager.js";
+import { BackgroundTaskManager } from "../managers/backgroundTaskManager.js";
+import { TaskManager } from "../services/taskManager.js";
+import { MessageManager } from "../managers/messageManager.js";
+import { AIManager } from "../managers/aiManager.js";
+import { ToolManager } from "../managers/toolManager.js";
+import { McpManager } from "../managers/mcpManager.js";
+import { LspManager } from "../managers/lspManager.js";
+import { PermissionManager } from "../managers/permissionManager.js";
+import { PlanManager } from "../managers/planManager.js";
+import { HookManager } from "../managers/hookManager.js";
+import { SkillManager } from "../managers/skillManager.js";
+import { SlashCommandManager } from "../managers/slashCommandManager.js";
+import { PluginManager } from "../managers/pluginManager.js";
+import { BashManager } from "../managers/bashManager.js";
+import { MemoryRuleManager } from "../managers/MemoryRuleManager.js";
+import { ReversionManager } from "../managers/reversionManager.js";
+import { SubagentManager } from "../managers/subagentManager.js";
+import { LiveConfigManager } from "../managers/liveConfigManager.js";
+import { ReversionService } from "../services/reversionService.js";
+import { logger } from "./globalLogger.js";
+export function setupAgentContainer(setupOptions) {
+    const { options, workdir, configurationService, systemPrompt, stream, onSessionIdChange, onTasksChange, onSessionTasksChange, onPermissionModeChange, handlePlanModeTransition, setPermissionMode, addPermissionRule, addUsage, getGatewayConfig, getModelConfig, getMaxInputTokens, getLanguage, } = setupOptions;
+    const callbacks = options.callbacks || {};
+    const container = new Container();
+    const foregroundTaskManager = new ForegroundTaskManager(container);
+    container.register("ForegroundTaskManager", foregroundTaskManager);
+    container.register("ConfigurationService", configurationService);
+    const memoryRuleManager = new MemoryRuleManager(container, { workdir });
+    container.register("MemoryRuleManager", memoryRuleManager);
+    const messageManager = new MessageManager(container, {
+        callbacks: {
+            ...callbacks,
+            onSessionIdChange: (sessionId) => {
+                onSessionIdChange(sessionId);
+                callbacks.onSessionIdChange?.(sessionId);
+            },
+        },
+        workdir,
+    });
+    container.register("MessageManager", messageManager);
+    const resolvedTaskListId = configurationService.getEnvironmentVars().WAVE_TASK_LIST_ID ||
+        process.env.WAVE_TASK_LIST_ID ||
+        messageManager.getRootSessionId();
+    const taskManager = new TaskManager(container, resolvedTaskListId);
+    container.register("TaskManager", taskManager);
+    taskManager.on("tasksChange", async () => {
+        const tasks = await taskManager.listTasks();
+        onSessionTasksChange(tasks);
+    });
+    const backgroundTaskManager = new BackgroundTaskManager(container, {
+        callbacks: {
+            ...callbacks,
+            onTasksChange: (tasks) => {
+                onTasksChange(tasks);
+                callbacks.onTasksChange?.(tasks);
+            },
+        },
+        workdir,
+    });
+    container.register("BackgroundTaskManager", backgroundTaskManager);
+    const mcpManager = new McpManager(container, { callbacks });
+    container.register("McpManager", mcpManager);
+    const lspManager = options.lspManager || new LspManager(container);
+    container.register("LspManager", lspManager);
+    const permissionManager = new PermissionManager(container, { workdir });
+    container.register("PermissionManager", permissionManager);
+    permissionManager.setOnConfiguredDefaultModeChange((mode) => {
+        handlePlanModeTransition(mode);
+        onPermissionModeChange(mode);
+    });
+    const planManager = new PlanManager(container);
+    container.register("PlanManager", planManager);
+    const hookManager = new HookManager(container, workdir);
+    container.register("HookManager", hookManager);
+    const skillManager = new SkillManager(container, { workdir });
+    container.register("SkillManager", skillManager);
+    container.register("ReversionService", new ReversionService(messageManager.getTranscriptPath()));
+    const reversionManager = new ReversionManager(container);
+    container.register("ReversionManager", reversionManager);
+    const canUseToolWithNotification = options.canUseTool
+        ? async (context) => {
+            try {
+                const notificationMessage = `Claude needs your permission to use ${context.toolName}`;
+                await hookManager.executeHooks("Notification", {
+                    event: "Notification",
+                    projectDir: workdir,
+                    timestamp: new Date(),
+                    sessionId: messageManager.getSessionId(),
+                    transcriptPath: messageManager.getTranscriptPath(),
+                    cwd: workdir,
+                    message: notificationMessage,
+                    notificationType: "permission_prompt",
+                    env: configurationService.getEnvironmentVars(),
+                });
+            }
+            catch (error) {
+                logger.warn("Failed to execute notification hooks", {
+                    toolName: context.toolName,
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            }
+            const decision = await options.canUseTool(context);
+            if (decision.newPermissionMode) {
+                setPermissionMode(decision.newPermissionMode);
+            }
+            if (decision.newPermissionRule) {
+                await addPermissionRule(decision.newPermissionRule);
+            }
+            return decision;
+        }
+        : undefined;
+    const toolManager = new ToolManager({
+        container,
+        tools: options.tools,
+    });
+    container.register("ToolManager", toolManager);
+    container.register("PermissionMode", options.permissionMode);
+    container.register("CanUseToolCallback", canUseToolWithNotification);
+    const liveConfigManager = new LiveConfigManager(container, { workdir });
+    container.register("LiveConfigManager", liveConfigManager);
+    const subagentManager = new SubagentManager(container, {
+        workdir,
+        callbacks: {
+            onSubagentUserMessageAdded: callbacks.onSubagentUserMessageAdded,
+            onSubagentAssistantMessageAdded: callbacks.onSubagentAssistantMessageAdded,
+            onSubagentAssistantContentUpdated: callbacks.onSubagentAssistantContentUpdated,
+            onSubagentAssistantReasoningUpdated: callbacks.onSubagentAssistantReasoningUpdated,
+            onSubagentToolBlockUpdated: callbacks.onSubagentToolBlockUpdated,
+            onSubagentMessagesChange: callbacks.onSubagentMessagesChange,
+            onSubagentLatestTotalTokensChange: callbacks.onSubagentLatestTotalTokensChange,
+        },
+        getGatewayConfig,
+        getModelConfig,
+        getMaxInputTokens,
+        getLanguage,
+        getEnvironmentVars: () => configurationService.getEnvironmentVars(),
+        onUsageAdded: (usage) => addUsage(usage),
+    });
+    container.register("SubagentManager", subagentManager);
+    const aiManager = new AIManager(container, {
+        callbacks: {
+            ...callbacks,
+            onUsageAdded: (usage) => addUsage(usage),
+        },
+        workdir,
+        systemPrompt,
+        stream,
+        getGatewayConfig,
+        getModelConfig,
+        getMaxInputTokens,
+        getLanguage,
+        getEnvironmentVars: () => configurationService.getEnvironmentVars(),
+    });
+    container.register("AIManager", aiManager);
+    const slashCommandManager = new SlashCommandManager(container, { workdir });
+    container.register("SlashCommandManager", slashCommandManager);
+    slashCommandManager.initialize();
+    const pluginManager = new PluginManager(container, { workdir });
+    container.register("PluginManager", pluginManager);
+    const bashManager = new BashManager(container, { workdir });
+    container.register("BashManager", bashManager);
+    return container;
+}

package/dist/utils/editUtils.d.ts

@@ -1,6 +1,3 @@
-/**
- * Utility functions for file editing tools
- */
 /**
  * Escape regular expression special characters
  */

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

@@ -1 +1 @@
-{"version":3,"file":"editUtils.d.ts","sourceRoot":"","sources":["../../src/utils/editUtils.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAEnD;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,YAAY,EAAE,MAAM,GACnB,MAAM,CAoER"}
+{"version":3,"file":"editUtils.d.ts","sourceRoot":"","sources":["../../src/utils/editUtils.ts"],"names":[],"mappings":"AAKA;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAEnD;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,YAAY,EAAE,MAAM,GACnB,MAAM,CAoER"}

package/dist/utils/editUtils.js

@@ -1,6 +1,7 @@
 /**
  * Utility functions for file editing tools
  */
+import { formatLineNumberPrefix } from "./stringUtils.js";
 /**
  * Escape regular expression special characters
  */
@@ -57,11 +58,11 @@ export function analyzeEditMismatch(content, searchString) {
         const actualLine = contentLines[bestMatchIndex + j];
         const expectedLine = searchLines[j];
         if (actualLine === expectedLine) {
-            reportLines.push(`${lineNum.toString().padStart(4)} | ${actualLine}`);
+            reportLines.push(`${formatLineNumberPrefix(lineNum)}${actualLine}`);
         }
         else {
-            reportLines.push(`${lineNum.toString().padStart(4)} | - ${expectedLine}`);
-            reportLines.push(`${lineNum.toString().padStart(4)} | + ${actualLine}`);
+            reportLines.push(`${formatLineNumberPrefix(lineNum)}- ${expectedLine}`);
+            reportLines.push(`${formatLineNumberPrefix(lineNum)}+ ${actualLine}`);
         }
     }
     return reportLines.join("\n");

package/dist/utils/hookMatcher.d.ts

@@ -34,7 +34,7 @@ export declare class HookMatcher {
      * Get all tool names that would match this pattern from a given list
      * Useful for testing and validation
      */
-    getMatches(pattern: string, toolNames: string[]): string[];
+    getMatches(pattern: string, tools: string[]): string[];
     /**
      * Optimize pattern for repeated matching
      * Returns a compiled matcher function for performance

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

@@ -1 +1 @@
-{"version":3,"file":"hookMatcher.d.ts","sourceRoot":"","sources":["../../src/utils/hookMatcher.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,qBAAa,WAAW;IACtB;;;;;;;OAOG;IACH,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO;IAYnD;;OAEG;IACH,OAAO,CAAC,aAAa;IAmBrB;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAoBxC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAmB5B;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,GAAG,cAAc;IA6B5E;;;OAGG;IACH,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE;IAI1D;;;OAGG;IACH,OAAO,CAAC,OAAO,EAAE,MAAM,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO;CAkCxD"}
+{"version":3,"file":"hookMatcher.d.ts","sourceRoot":"","sources":["../../src/utils/hookMatcher.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,qBAAa,WAAW;IACtB;;;;;;;OAOG;IACH,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,OAAO;IAYnD;;OAEG;IACH,OAAO,CAAC,aAAa;IAmBrB;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO;IAoBxC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAmB5B;;OAEG;IACH,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,GAAG,cAAc;IA6B5E;;;OAGG;IACH,UAAU,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,EAAE;IAItD;;;OAGG;IACH,OAAO,CAAC,OAAO,EAAE,MAAM,GAAG,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO;CAkCxD"}

package/dist/utils/hookMatcher.js

@@ -112,8 +112,8 @@ export class HookMatcher {
      * Get all tool names that would match this pattern from a given list
      * Useful for testing and validation
      */
-    getMatches(pattern, toolNames) {
-        return toolNames.filter((toolName) => this.matches(pattern, toolName));
+    getMatches(pattern, tools) {
+        return tools.filter((toolName) => this.matches(pattern, toolName));
     }
     /**
      * Optimize pattern for repeated matching

package/dist/utils/openaiClient.js

@@ -29,8 +29,8 @@ export class OpenAIClient {
         };
         const fetchFn = customFetch || fetch;
         let lastError;
-        const maxRetries = 3;
-        const initialDelay = 1000;
+        const maxRetries = 5;
+        const initialDelay = 2000;
         for (let attempt = 0; attempt <= maxRetries; attempt++) {
             if (attempt > 0) {
                 const delay = initialDelay * Math.pow(2, attempt - 1);

package/dist/utils/stringUtils.d.ts

@@ -17,4 +17,10 @@ export declare function parseCustomHeaders(headersString: string): Record<string
  * @returns Plain text with color codes removed
  */
 export declare const stripAnsiColors: (text: string) => string;
+/**
+ * Format a line number prefix in cat -n style (padStart(6) + tab)
+ * @param lineNumber The line number
+ * @returns Formatted line number prefix
+ */
+export declare function formatLineNumberPrefix(lineNumber: number): string;
 //# sourceMappingURL=stringUtils.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"stringUtils.d.ts","sourceRoot":"","sources":["../../src/utils/stringUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAgC/D;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAChC,aAAa,EAAE,MAAM,GACpB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAwBxB;AAED;;;;GAIG;AACH,eAAO,MAAM,eAAe,GAAI,MAAM,MAAM,KAAG,MAK9C,CAAC"}
+{"version":3,"file":"stringUtils.d.ts","sourceRoot":"","sources":["../../src/utils/stringUtils.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAgC/D;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAChC,aAAa,EAAE,MAAM,GACpB,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAwBxB;AAED;;;;GAIG;AACH,eAAO,MAAM,eAAe,GAAI,MAAM,MAAM,KAAG,MAK9C,CAAC;AAEF;;;;GAIG;AACH,wBAAgB,sBAAsB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAEjE"}

package/dist/utils/stringUtils.js

@@ -69,3 +69,11 @@ export const stripAnsiColors = (text) => {
     const ansiEscapeRegex = new RegExp(`${escapeChar}\\[[0-9;]*[a-zA-Z]`, "g");
     return text.replace(ansiEscapeRegex, "");
 };
+/**
+ * Format a line number prefix in cat -n style (padStart(6) + tab)
+ * @param lineNumber The line number
+ * @returns Formatted line number prefix
+ */
+export function formatLineNumberPrefix(lineNumber) {
+    return `${lineNumber.toString().padStart(6)}\t`;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wave-agent-sdk",
-  "version": "0.6.4",
+  "version": "0.7.0",
   "description": "SDK for building AI-powered development tools and agents",
   "keywords": [
     "ai",