wave-agent-sdk 0.0.7 → 0.0.10

package/dist/managers/toolManager.js

@@ -9,16 +9,25 @@ import { grepTool } from "../tools/grepTool.js";
 import { lsTool } from "../tools/lsTool.js";
 import { readTool } from "../tools/readTool.js";
 import { todoWriteTool } from "../tools/todoWriteTool.js";
+import { lspTool } from "../tools/lspTool.js";
 import { createTaskTool } from "../tools/taskTool.js";
 import { createSkillTool } from "../tools/skillTool.js";
 /**
  * Tool Manager
+ *
+ * Manages tool registration and execution with optional permission system integration.
+ * Supports both built-in tools and MCP (Model Context Protocol) tools.
  */
 class ToolManager {
     constructor(options) {
         this.tools = new Map();
         this.mcpManager = options.mcpManager;
+        this.lspManager = options.lspManager;
         this.logger = options.logger;
+        this.permissionManager = options.permissionManager;
+        // Store CLI permission mode, let PermissionManager resolve effective mode
+        this.permissionMode = options.permissionMode;
+        this.canUseToolCallback = options.canUseToolCallback;
     }
     /**
      * Register a new tool
@@ -63,6 +72,7 @@ class ToolManager {
             lsTool,
             readTool,
             todoWriteTool,
+            lspTool,
         ];
         for (const tool of builtInTools) {
             this.tools.set(tool.name, tool);
@@ -77,18 +87,55 @@ class ToolManager {
             this.tools.set(skillTool.name, skillTool);
         }
     }
+    /**
+     * Execute a tool by name with the provided arguments and context
+     *
+     * Enhances the context with permission-related fields before execution:
+     * - permissionMode: The current permission mode (default or bypassPermissions)
+     * - canUseToolCallback: Custom permission callback if provided
+     * - permissionManager: The PermissionManager instance for permission checks
+     *
+     * @param name - Name of the tool to execute
+     * @param args - Arguments to pass to the tool
+     * @param context - Execution context for the tool
+     * @returns Promise resolving to the tool execution result
+     */
     async execute(name, args, context) {
+        // Resolve effective permission mode (CLI override > configuration > default)
+        const effectivePermissionMode = this.permissionManager
+            ? this.permissionManager.getCurrentEffectiveMode(this.permissionMode)
+            : this.permissionMode || "default";
+        // Enhance context with permission-related fields
+        const enhancedContext = {
+            ...context,
+            permissionMode: effectivePermissionMode,
+            canUseToolCallback: this.canUseToolCallback,
+            permissionManager: this.permissionManager,
+            mcpManager: this.mcpManager,
+            lspManager: this.lspManager,
+        };
+        this.logger?.debug("Executing tool with enhanced context", {
+            toolName: name,
+            cliPermissionMode: this.permissionMode,
+            effectivePermissionMode,
+            hasPermissionManager: !!this.permissionManager,
+            hasPermissionCallback: !!this.canUseToolCallback,
+        });
         // Check if it's an MCP tool first
         if (this.mcpManager.isMcpTool(name)) {
-            return this.mcpManager.executeMcpToolByRegistry(name, args, context);
+            return this.mcpManager.executeMcpToolByRegistry(name, args, enhancedContext);
         }
         // Check built-in tools
         const plugin = this.tools.get(name);
         if (plugin) {
             try {
-                return await plugin.execute(args, context);
+                return await plugin.execute(args, enhancedContext);
             }
             catch (error) {
+                this.logger?.error("Tool execution failed", {
+                    toolName: name,
+                    error: error instanceof Error ? error.message : String(error),
+                });
                 return {
                     success: false,
                     content: "",
@@ -96,6 +143,7 @@ class ToolManager {
                 };
             }
         }
+        this.logger?.warn("Tool not found", { toolName: name });
         return {
             success: false,
             content: "",
@@ -112,6 +160,22 @@ class ToolManager {
         const mcpToolsConfig = this.mcpManager.getMcpToolsConfig();
         return [...builtInToolsConfig, ...mcpToolsConfig];
     }
+    /**
+     * Get the current permission mode
+     */
+    getPermissionMode() {
+        if (this.permissionManager) {
+            return this.permissionManager.getCurrentEffectiveMode(this.permissionMode);
+        }
+        return this.permissionMode || "default";
+    }
+    /**
+     * Set the permission mode
+     * @param mode - The new permission mode
+     */
+    setPermissionMode(mode) {
+        this.permissionMode = mode;
+    }
 }
 // Export tool registry class and types
 export { ToolManager };

package/dist/services/aiService.d.ts

@@ -1,6 +1,7 @@
 import { ChatCompletionMessageToolCall } from "openai/resources";
 import { ChatCompletionMessageParam, ChatCompletionFunctionTool } from "openai/resources.js";
 import type { GatewayConfig, ModelConfig } from "../types/index.js";
+import { type ClaudeUsage } from "../utils/cacheControlUtils.js";
 export interface CallAgentOptions {
     gatewayConfig: GatewayConfig;
     modelConfig: ModelConfig;
@@ -12,15 +13,24 @@ export interface CallAgentOptions {
     tools?: ChatCompletionFunctionTool[];
     model?: string;
     systemPrompt?: string;
+    onContentUpdate?: (content: string) => void;
+    onToolUpdate?: (toolCall: {
+        id: string;
+        name: string;
+        parameters: string;
+        parametersChunk?: string;
+        stage?: "start" | "streaming" | "running" | "end";
+    }) => void;
+    onReasoningUpdate?: (content: string) => void;
 }
 export interface CallAgentResult {
     content?: string;
     tool_calls?: ChatCompletionMessageToolCall[];
-    usage?: {
-        prompt_tokens: number;
-        completion_tokens: number;
-        total_tokens: number;
-    };
+    reasoning_content?: string;
+    usage?: ClaudeUsage;
+    finish_reason?: "stop" | "length" | "tool_calls" | "content_filter" | "function_call" | null;
+    response_headers?: Record<string, string>;
+    additionalFields?: Record<string, unknown>;
 }
 export declare function callAgent(options: CallAgentOptions): Promise<CallAgentResult>;
 export interface CompressMessagesOptions {

package/dist/services/aiService.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"aiService.d.ts","sourceRoot":"","sources":["../../src/services/aiService.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,6BAA6B,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAEL,0BAA0B,EAC1B,0BAA0B,EAC3B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AA4DpE,MAAM,WAAW,gBAAgB;IAE/B,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACrC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,6BAA6B,EAAE,CAAC;IAC7C,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,SAAS,CAC7B,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,eAAe,CAAC,CAoH1B;AAED,MAAM,WAAW,uBAAuB;IAEtC,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,sBAAsB,CAAC,CAuGjC"}
+{"version":3,"file":"aiService.d.ts","sourceRoot":"","sources":["../../src/services/aiService.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,6BAA6B,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAGL,0BAA0B,EAC1B,0BAA0B,EAE3B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACpE,OAAO,EAKL,KAAK,WAAW,EACjB,MAAM,+BAA+B,CAAC;AAqGvC,MAAM,WAAW,gBAAgB;IAE/B,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACrC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IAGtB,eAAe,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;IAC5C,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE;QACxB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,CAAC;QACnB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,GAAG,KAAK,CAAC;KACnD,KAAK,IAAI,CAAC;IACX,iBAAiB,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;CAC/C;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,6BAA6B,EAAE,CAAC;IAC7C,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,aAAa,CAAC,EACV,MAAM,GACN,QAAQ,GACR,YAAY,GACZ,gBAAgB,GAChB,eAAe,GACf,IAAI,CAAC;IACT,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC1C,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC5C;AAED,wBAAsB,SAAS,CAC7B,OAAO,EAAE,gBAAgB,GACxB,OAAO,CAAC,eAAe,CAAC,CA2U1B;AA4OD,MAAM,WAAW,uBAAuB;IAEtC,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IAGzB,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,sBAAsB,CAAC,CA8JjC"}