wave-agent-sdk 0.0.1 → 0.0.3

package/dist/managers/subagentManager.d.ts

@@ -0,0 +1,96 @@
+import type { SubagentConfiguration } from "../utils/subagentParser.js";
+import type { Message, Logger, GatewayConfig, ModelConfig } from "../types.js";
+import { AIManager } from "./aiManager.js";
+import { MessageManager } from "./messageManager.js";
+import { ToolManager } from "./toolManager.js";
+export interface SubagentInstance {
+    subagentId: string;
+    configuration: SubagentConfiguration;
+    aiManager: AIManager;
+    messageManager: MessageManager;
+    toolManager: ToolManager;
+    status: "initializing" | "active" | "completed" | "error" | "aborted";
+    taskDescription: string;
+    messages: Message[];
+}
+export interface SubagentManagerOptions {
+    workdir: string;
+    parentToolManager: ToolManager;
+    parentMessageManager: MessageManager;
+    logger?: Logger;
+    gatewayConfig: GatewayConfig;
+    modelConfig: ModelConfig;
+    tokenLimit: number;
+}
+export declare class SubagentManager {
+    private instances;
+    private cachedConfigurations;
+    private workdir;
+    private parentToolManager;
+    private parentMessageManager;
+    private logger?;
+    private gatewayConfig;
+    private modelConfig;
+    private tokenLimit;
+    constructor(options: SubagentManagerOptions);
+    /**
+     * Initialize the SubagentManager by loading and caching configurations
+     */
+    initialize(): Promise<void>;
+    /**
+     * Load all available subagent configurations and cache them
+     */
+    loadConfigurations(): Promise<SubagentConfiguration[]>;
+    /**
+     * Get cached configurations synchronously (must call loadConfigurations first)
+     */
+    getConfigurations(): SubagentConfiguration[];
+    /**
+     * Find subagent by exact name match
+     */
+    findSubagent(name: string): Promise<SubagentConfiguration | null>;
+    /**
+     * Create a new subagent instance with isolated managers
+     */
+    createInstance(configuration: SubagentConfiguration, taskDescription: string): Promise<SubagentInstance>;
+    /**
+     * Execute task using subagent instance
+     *
+     * IMPORTANT: This method automatically filters out the Task tool from allowedTools
+     * to prevent subagents from spawning other subagents (infinite recursion protection)
+     */
+    executeTask(instance: SubagentInstance, prompt: string): Promise<string>;
+    /**
+     * Get instance by subagent ID
+     */
+    getInstance(subagentId: string): SubagentInstance | null;
+    /**
+     * Update instance status
+     */
+    updateInstanceStatus(subagentId: string, status: SubagentInstance["status"]): void;
+    /**
+     * Add message to instance
+     */
+    addMessageToInstance(subagentId: string, message: Message): void;
+    /**
+     * Abort a running subagent instance
+     */
+    abortInstance(subagentId: string): boolean;
+    /**
+     * Abort all active subagent instances
+     */
+    abortAllInstances(): void;
+    /**
+     * Clean up completed, errored, or aborted instances
+     */
+    cleanupInstance(subagentId: string): void;
+    /**
+     * Get all active instances
+     */
+    getActiveInstances(): SubagentInstance[];
+    /**
+     * Clean up all instances (for session end)
+     */
+    cleanup(): void;
+}
+//# sourceMappingURL=subagentManager.d.ts.map

package/dist/managers/subagentManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"subagentManager.d.ts","sourceRoot":"","sources":["../../src/managers/subagentManager.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACxE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC/E,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EACL,cAAc,EAEf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAE/C,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,qBAAqB,CAAC;IACrC,SAAS,EAAE,SAAS,CAAC;IACrB,cAAc,EAAE,cAAc,CAAC;IAC/B,WAAW,EAAE,WAAW,CAAC;IACzB,MAAM,EAAE,cAAc,GAAG,QAAQ,GAAG,WAAW,GAAG,OAAO,GAAG,SAAS,CAAC;IACtE,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,OAAO,EAAE,CAAC;CACrB;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE,MAAM,CAAC;IAChB,iBAAiB,EAAE,WAAW,CAAC;IAC/B,oBAAoB,EAAE,cAAc,CAAC;IACrC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,aAAa,EAAE,aAAa,CAAC;IAC7B,WAAW,EAAE,WAAW,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,SAAS,CAAuC;IACxD,OAAO,CAAC,oBAAoB,CAAwC;IAEpE,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,iBAAiB,CAAc;IACvC,OAAO,CAAC,oBAAoB,CAAiB;IAC7C,OAAO,CAAC,MAAM,CAAC,CAAS;IACxB,OAAO,CAAC,aAAa,CAAgB;IACrC,OAAO,CAAC,WAAW,CAAc;IACjC,OAAO,CAAC,UAAU,CAAS;gBAEf,OAAO,EAAE,sBAAsB;IAU3C;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAIjC;;OAEG;IACG,kBAAkB,IAAI,OAAO,CAAC,qBAAqB,EAAE,CAAC;IAY5D;;OAEG;IACH,iBAAiB,IAAI,qBAAqB,EAAE;IAS5C;;OAEG;IACG,YAAY,CAAC,IAAI,EAAE,MAAM;IAK/B;;OAEG;IACG,cAAc,CAClB,aAAa,EAAE,qBAAqB,EACpC,eAAe,EAAE,MAAM,GACtB,OAAO,CAAC,gBAAgB,CAAC;IAmF5B;;;;;OAKG;IACG,WAAW,CACf,QAAQ,EAAE,gBAAgB,EAC1B,MAAM,EAAE,MAAM,GACb,OAAO,CAAC,MAAM,CAAC;IAiElB;;OAEG;IACH,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,gBAAgB,GAAG,IAAI;IAIxD;;OAEG;IACH,oBAAoB,CAClB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC,GACjC,IAAI;IAOP;;OAEG;IACH,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAOhE;;OAEG;IACH,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;IAiC1C;;OAEG;IACH,iBAAiB,IAAI,IAAI;IAOzB;;OAEG;IACH,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAYzC;;OAEG;IACH,kBAAkB,IAAI,gBAAgB,EAAE;IAOxC;;OAEG;IACH,OAAO,IAAI,IAAI;CAKhB"}

package/dist/managers/subagentManager.js

@@ -0,0 +1,261 @@
+import { randomUUID } from "crypto";
+import { AIManager } from "./aiManager.js";
+import { MessageManager, } from "./messageManager.js";
+export class SubagentManager {
+    constructor(options) {
+        this.instances = new Map();
+        this.cachedConfigurations = null;
+        this.workdir = options.workdir;
+        this.parentToolManager = options.parentToolManager;
+        this.parentMessageManager = options.parentMessageManager;
+        this.logger = options.logger;
+        this.gatewayConfig = options.gatewayConfig;
+        this.modelConfig = options.modelConfig;
+        this.tokenLimit = options.tokenLimit;
+    }
+    /**
+     * Initialize the SubagentManager by loading and caching configurations
+     */
+    async initialize() {
+        await this.loadConfigurations();
+    }
+    /**
+     * Load all available subagent configurations and cache them
+     */
+    async loadConfigurations() {
+        if (this.cachedConfigurations === null) {
+            const { loadSubagentConfigurations } = await import("../utils/subagentParser.js");
+            this.cachedConfigurations = await loadSubagentConfigurations(this.workdir);
+        }
+        return this.cachedConfigurations;
+    }
+    /**
+     * Get cached configurations synchronously (must call loadConfigurations first)
+     */
+    getConfigurations() {
+        if (this.cachedConfigurations === null) {
+            throw new Error("SubagentManager not initialized. Call loadConfigurations() first.");
+        }
+        return this.cachedConfigurations;
+    }
+    /**
+     * Find subagent by exact name match
+     */
+    async findSubagent(name) {
+        const { findSubagentByName } = await import("../utils/subagentParser.js");
+        return findSubagentByName(name, this.workdir);
+    }
+    /**
+     * Create a new subagent instance with isolated managers
+     */
+    async createInstance(configuration, taskDescription) {
+        if (!this.parentToolManager ||
+            !this.gatewayConfig ||
+            !this.modelConfig ||
+            !this.tokenLimit) {
+            throw new Error("SubagentManager not properly initialized - call initialize() first");
+        }
+        const subagentId = randomUUID();
+        // Create isolated MessageManager for the subagent
+        const subagentCallbacks = {
+            // These callbacks will be handled by the parent agent
+            onMessagesChange: (messages) => {
+                const instance = this.instances.get(subagentId);
+                if (instance) {
+                    instance.messages = messages;
+                    // Update parent's subagent block with latest messages
+                    this.parentMessageManager.updateSubagentBlock(subagentId, {
+                        messages: messages,
+                    });
+                }
+            },
+        };
+        const messageManager = new MessageManager({
+            callbacks: subagentCallbacks,
+            workdir: this.workdir,
+            logger: this.logger,
+        });
+        // Use the parent tool manager directly - tool restrictions will be handled by allowedTools parameter
+        const toolManager = this.parentToolManager;
+        // Determine model to use
+        const modelToUse = configuration.model && configuration.model !== "inherit"
+            ? configuration.model
+            : this.modelConfig.agentModel;
+        // Create isolated AIManager for the subagent
+        const aiManager = new AIManager({
+            messageManager,
+            toolManager,
+            logger: this.logger,
+            workdir: this.workdir,
+            systemPrompt: configuration.systemPrompt,
+            gatewayConfig: this.gatewayConfig,
+            modelConfig: {
+                ...this.modelConfig,
+                agentModel: modelToUse,
+            },
+            tokenLimit: this.tokenLimit,
+        });
+        const instance = {
+            subagentId,
+            configuration,
+            aiManager,
+            messageManager,
+            toolManager,
+            status: "initializing",
+            taskDescription,
+            messages: [],
+        };
+        this.instances.set(subagentId, instance);
+        // Create subagent block in parent message manager
+        this.parentMessageManager.addSubagentBlock(subagentId, configuration.name, "active", []);
+        return instance;
+    }
+    /**
+     * Execute task using subagent instance
+     *
+     * IMPORTANT: This method automatically filters out the Task tool from allowedTools
+     * to prevent subagents from spawning other subagents (infinite recursion protection)
+     */
+    async executeTask(instance, prompt) {
+        try {
+            // Set status to active and update parent
+            this.updateInstanceStatus(instance.subagentId, "active");
+            this.parentMessageManager.updateSubagentBlock(instance.subagentId, {
+                status: "active",
+            });
+            // Add the user's prompt as a message
+            instance.messageManager.addUserMessage(prompt);
+            // Create allowed tools list - always exclude Task tool to prevent subagent recursion
+            let allowedTools = instance.configuration.tools;
+            // Always filter out the Task tool to prevent subagents from creating sub-subagents
+            if (allowedTools) {
+                allowedTools = allowedTools.filter((tool) => tool !== "Task");
+            }
+            else {
+                // If no tools specified, get all tools except Task
+                const allTools = instance.toolManager.list().map((tool) => tool.name);
+                allowedTools = allTools.filter((tool) => tool !== "Task");
+            }
+            // Execute the AI request with tool restrictions
+            await instance.aiManager.sendAIMessage({
+                allowedTools,
+                model: instance.configuration.model !== "inherit"
+                    ? instance.configuration.model
+                    : undefined,
+            });
+            // Get the latest messages to extract the response
+            const messages = instance.messageManager.getMessages();
+            const lastAssistantMessage = messages
+                .filter((msg) => msg.role === "assistant")
+                .pop();
+            if (!lastAssistantMessage) {
+                throw new Error("No response from subagent");
+            }
+            // Extract text content from the last assistant message
+            const textBlocks = lastAssistantMessage.blocks.filter((block) => block.type === "text");
+            const response = textBlocks.map((block) => block.content).join("\n");
+            // Update status to completed and update parent with final messages
+            this.updateInstanceStatus(instance.subagentId, "completed");
+            this.parentMessageManager.updateSubagentBlock(instance.subagentId, {
+                status: "completed",
+                messages: messages,
+            });
+            return response || "Task completed with no text response";
+        }
+        catch (error) {
+            this.updateInstanceStatus(instance.subagentId, "error");
+            this.parentMessageManager.updateSubagentBlock(instance.subagentId, {
+                status: "error",
+            });
+            throw error;
+        }
+    }
+    /**
+     * Get instance by subagent ID
+     */
+    getInstance(subagentId) {
+        return this.instances.get(subagentId) || null;
+    }
+    /**
+     * Update instance status
+     */
+    updateInstanceStatus(subagentId, status) {
+        const instance = this.instances.get(subagentId);
+        if (instance) {
+            instance.status = status;
+        }
+    }
+    /**
+     * Add message to instance
+     */
+    addMessageToInstance(subagentId, message) {
+        const instance = this.instances.get(subagentId);
+        if (instance) {
+            instance.messages.push(message);
+        }
+    }
+    /**
+     * Abort a running subagent instance
+     */
+    abortInstance(subagentId) {
+        const instance = this.instances.get(subagentId);
+        if (!instance) {
+            return false;
+        }
+        // Only abort active or initializing instances
+        if (instance.status !== "active" && instance.status !== "initializing") {
+            return false;
+        }
+        try {
+            // Abort the AI manager operations
+            instance.aiManager.abortAIMessage();
+            // Update status
+            this.updateInstanceStatus(subagentId, "aborted");
+            this.parentMessageManager.updateSubagentBlock(subagentId, {
+                status: "aborted",
+                messages: instance.messages,
+            });
+            this.logger?.info(`Aborted subagent instance: ${subagentId}`);
+            return true;
+        }
+        catch (error) {
+            this.logger?.error(`Failed to abort subagent instance ${subagentId}:`, error);
+            return false;
+        }
+    }
+    /**
+     * Abort all active subagent instances
+     */
+    abortAllInstances() {
+        const activeInstances = this.getActiveInstances();
+        for (const instance of activeInstances) {
+            this.abortInstance(instance.subagentId);
+        }
+    }
+    /**
+     * Clean up completed, errored, or aborted instances
+     */
+    cleanupInstance(subagentId) {
+        const instance = this.instances.get(subagentId);
+        if (instance &&
+            (instance.status === "completed" ||
+                instance.status === "error" ||
+                instance.status === "aborted")) {
+            this.instances.delete(subagentId);
+        }
+    }
+    /**
+     * Get all active instances
+     */
+    getActiveInstances() {
+        return Array.from(this.instances.values()).filter((instance) => instance.status === "active" || instance.status === "initializing");
+    }
+    /**
+     * Clean up all instances (for session end)
+     */
+    cleanup() {
+        // Abort all active instances before cleanup
+        this.abortAllInstances();
+        this.instances.clear();
+    }
+}

package/dist/managers/toolManager.d.ts

@@ -2,6 +2,8 @@ import type { ToolContext, ToolPlugin, ToolResult } from "../tools/types.js";
 import { McpManager } from "./mcpManager.js";
 import { ChatCompletionFunctionTool } from "openai/resources.js";
 import type { Logger } from "../types.js";
+import type { SubagentManager } from "./subagentManager.js";
+import type { SkillManager } from "./skillManager.js";
 export interface ToolManagerOptions {
     mcpManager: McpManager;
     logger?: Logger;
@@ -14,7 +16,37 @@ declare class ToolManager {
     private mcpManager;
     private logger?;
     constructor(options: ToolManagerOptions);
-    private initializeBuiltInTools;
+    /**
+     * Register a new tool
+     */
+    register(tool: ToolPlugin): void;
+    /**
+     * Initialize built-in tools. Can be called with dependencies for tools that require them.
+     *
+     * This method can be called multiple times safely. When called without dependencies,
+     * it registers basic tools (Bash, Read, Write, TodoWrite, etc.). When called with
+     * dependencies, it also registers tools that require managers (Task, Skill).
+     *
+     * @param deps Optional dependencies for advanced tools
+     * @param deps.subagentManager SubagentManager instance for Task tool
+     * @param deps.skillManager SkillManager instance for Skill tool
+     *
+     * @example
+     * ```typescript
+     * // Initialize basic tools only
+     * toolManager.initializeBuiltInTools();
+     *
+     * // Initialize all tools including those requiring dependencies
+     * toolManager.initializeBuiltInTools({
+     *   subagentManager: mySubagentManager,
+     *   skillManager: mySkillManager
+     * });
+     * ```
+     */
+    initializeBuiltInTools(deps?: {
+        subagentManager?: SubagentManager;
+        skillManager?: SkillManager;
+    }): void;
     execute(name: string, args: Record<string, unknown>, context: ToolContext): Promise<ToolResult>;
     list(): ToolPlugin[];
     getToolsConfig(): ChatCompletionFunctionTool[];

package/dist/managers/toolManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"toolManager.d.ts","sourceRoot":"","sources":["../../src/managers/toolManager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAY7E,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAE1C,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,UAAU,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,cAAM,WAAW;IACf,OAAO,CAAC,KAAK,CAAiC;IAC9C,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,OAAO,EAAE,kBAAkB;IAQvC,OAAO,CAAC,sBAAsB;IAqBxB,OAAO,CACX,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,WAAW,GACnB,OAAO,CAAC,UAAU,CAAC;IA2BtB,IAAI,IAAI,UAAU,EAAE;IAMpB,cAAc,IAAI,0BAA0B,EAAE;CAO/C;AAGD,OAAO,EAAE,WAAW,EAAE,CAAC"}
+{"version":3,"file":"toolManager.d.ts","sourceRoot":"","sources":["../../src/managers/toolManager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAc7E,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,0BAA0B,EAAE,MAAM,qBAAqB,CAAC;AACjE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEtD,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,UAAU,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,cAAM,WAAW;IACf,OAAO,CAAC,KAAK,CAAiC;IAC9C,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,OAAO,EAAE,kBAAkB;IAKvC;;OAEG;IACI,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,IAAI;IAIvC;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACI,sBAAsB,CAAC,IAAI,CAAC,EAAE;QACnC,eAAe,CAAC,EAAE,eAAe,CAAC;QAClC,YAAY,CAAC,EAAE,YAAY,CAAC;KAC7B,GAAG,IAAI;IAgCF,OAAO,CACX,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC7B,OAAO,EAAE,WAAW,GACnB,OAAO,CAAC,UAAU,CAAC;IA2BtB,IAAI,IAAI,UAAU,EAAE;IAMpB,cAAc,IAAI,0BAA0B,EAAE;CAO/C;AAGD,OAAO,EAAE,WAAW,EAAE,CAAC"}

package/dist/managers/toolManager.js

@@ -8,7 +8,9 @@ import { globTool } from "../tools/globTool.js";
 import { grepTool } from "../tools/grepTool.js";
 import { lsTool } from "../tools/lsTool.js";
 import { readTool } from "../tools/readTool.js";
-import { SkillManager } from "./skillManager.js";
+import { todoWriteTool } from "../tools/todoWriteTool.js";
+import { createTaskTool } from "../tools/taskTool.js";
+import { createSkillTool } from "../tools/skillTool.js";
 /**
  * Tool Manager
  */
@@ -17,10 +19,37 @@ class ToolManager {
         this.tools = new Map();
         this.mcpManager = options.mcpManager;
         this.logger = options.logger;
-        // Initialize built-in tools
-        this.initializeBuiltInTools();
     }
-    initializeBuiltInTools() {
+    /**
+     * Register a new tool
+     */
+    register(tool) {
+        this.tools.set(tool.name, tool);
+    }
+    /**
+     * Initialize built-in tools. Can be called with dependencies for tools that require them.
+     *
+     * This method can be called multiple times safely. When called without dependencies,
+     * it registers basic tools (Bash, Read, Write, TodoWrite, etc.). When called with
+     * dependencies, it also registers tools that require managers (Task, Skill).
+     *
+     * @param deps Optional dependencies for advanced tools
+     * @param deps.subagentManager SubagentManager instance for Task tool
+     * @param deps.skillManager SkillManager instance for Skill tool
+     *
+     * @example
+     * ```typescript
+     * // Initialize basic tools only
+     * toolManager.initializeBuiltInTools();
+     *
+     * // Initialize all tools including those requiring dependencies
+     * toolManager.initializeBuiltInTools({
+     *   subagentManager: mySubagentManager,
+     *   skillManager: mySkillManager
+     * });
+     * ```
+     */
+    initializeBuiltInTools(deps) {
         const builtInTools = [
             bashTool,
             bashOutputTool,
@@ -33,11 +62,20 @@ class ToolManager {
             grepTool,
             lsTool,
             readTool,
-            new SkillManager({ logger: this.logger }).createTool(),
+            todoWriteTool,
         ];
         for (const tool of builtInTools) {
             this.tools.set(tool.name, tool);
         }
+        // Register tools that require dependencies
+        if (deps?.subagentManager) {
+            const taskTool = createTaskTool(deps.subagentManager);
+            this.tools.set(taskTool.name, taskTool);
+        }
+        if (deps?.skillManager) {
+            const skillTool = createSkillTool(deps.skillManager);
+            this.tools.set(skillTool.name, skillTool);
+        }
     }
     async execute(name, args, context) {
         // Check if it's an MCP tool first

package/dist/services/aiService.d.ts

@@ -1,6 +1,9 @@
 import { ChatCompletionMessageToolCall } from "openai/resources";
 import { ChatCompletionMessageParam, ChatCompletionFunctionTool } from "openai/resources.js";
+import type { GatewayConfig, ModelConfig } from "../types.js";
 export interface CallAgentOptions {
+    gatewayConfig: GatewayConfig;
+    modelConfig: ModelConfig;
     messages: ChatCompletionMessageParam[];
     sessionId?: string;
     abortSignal?: AbortSignal;
@@ -21,6 +24,8 @@ export interface CallAgentResult {
 }
 export declare function callAgent(options: CallAgentOptions): Promise<CallAgentResult>;
 export interface CompressMessagesOptions {
+    gatewayConfig: GatewayConfig;
+    modelConfig: ModelConfig;
     messages: ChatCompletionMessageParam[];
     abortSignal?: AbortSignal;
 }

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;AAuC7B,MAAM,WAAW,gBAAgB;IAC/B,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,CA2F1B;AAED,MAAM,WAAW,uBAAuB;IACtC,QAAQ,EAAE,0BAA0B,EAAE,CAAC;IACvC,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AAED,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,MAAM,CAAC,CAmFjB"}
+{"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,aAAa,CAAC;AA4D9D,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,CA6G1B;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,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,MAAM,CAAC,CAyFjB"}

package/dist/services/aiService.js

@@ -1,5 +1,29 @@
 import OpenAI from "openai";
-import { FAST_MODEL_ID, AGENT_MODEL_ID } from "../utils/constants.js";
+import * as os from "os";
+import * as fs from "fs";
+import * as path from "path";
+/**
+ * Check if a directory is a git repository
+ * @param dirPath Directory path to check
+ * @returns "Yes" if it's a git repo, "No" otherwise
+ */
+function isGitRepository(dirPath) {
+    try {
+        // Check if .git directory exists in current directory or any parent directory
+        let currentPath = path.resolve(dirPath);
+        while (currentPath !== path.dirname(currentPath)) {
+            const gitPath = path.join(currentPath, ".git");
+            if (fs.existsSync(gitPath)) {
+                return "Yes";
+            }
+            currentPath = path.dirname(currentPath);
+        }
+        return "No";
+    }
+    catch {
+        return "No";
+    }
+}
 /**
  * Get specific configuration parameters based on model name
  * @param modelName Model name
@@ -19,31 +43,32 @@ function getModelConfig(modelName, baseConfig = {}) {
     }
     return config;
 }
-// Initialize OpenAI client with environment variables
-const openai = new OpenAI({
-    apiKey: process.env.AIGW_TOKEN,
-    baseURL: process.env.AIGW_URL,
-});
 export async function callAgent(options) {
-    const { messages, abortSignal, memory, workdir, tools, model, systemPrompt } = options;
+    const { gatewayConfig, modelConfig, messages, abortSignal, memory, workdir, tools, model, systemPrompt, } = options;
     try {
+        // Create OpenAI client with injected configuration
+        const openai = new OpenAI({
+            apiKey: gatewayConfig.apiKey,
+            baseURL: gatewayConfig.baseURL,
+        });
         // Build system prompt content
-        let systemContent;
-        if (systemPrompt) {
-            // Use custom system prompt if provided
-            systemContent = systemPrompt;
-        }
-        else {
-            // Use default system prompt
-            systemContent = `You are an interactive CLI tool that helps users with software engineering tasks. Use the instructions below and the tools available to you to assist the user.
+        let systemContent = systemPrompt ||
+            `You are an interactive CLI tool that helps users with software engineering tasks. Use the instructions below and the tools available to you to assist the user.`;
+        // Always add environment information
+        systemContent += `
 
-## Current Working Directory
-${workdir}
+Here is useful information about the environment you are running in:
+<env>
+Working directory: ${workdir}
+Is directory a git repo: ${isGitRepository(workdir)}
+Platform: ${os.platform()}
+OS Version: ${os.type()} ${os.release()}
+Today's date: ${new Date().toISOString().split("T")[0]}
+</env>
 `;
-            // If there is memory content, add it to the system prompt
-            if (memory && memory.trim()) {
-                systemContent += `\n\n## Memory Context\n\nThe following is important context and memory from previous interactions:\n\n${memory}`;
-            }
+        // If there is memory content, add it to the system prompt
+        if (memory && memory.trim()) {
+            systemContent += `\n## Memory Context\n\nThe following is important context and memory from previous interactions:\n\n${memory}`;
         }
         // Add system prompt
         const systemMessage = {
@@ -52,14 +77,14 @@ ${workdir}
         };
         // ChatCompletionMessageParam[] is already in OpenAI format, add system prompt to the beginning
         const openaiMessages = [systemMessage, ...messages];
-        // Get model configuration
-        const modelConfig = getModelConfig(model || AGENT_MODEL_ID, {
+        // Get model configuration - use injected modelConfig with optional override
+        const openaiModelConfig = getModelConfig(model || modelConfig.agentModel, {
             temperature: 0,
             max_completion_tokens: 32768,
         });
         // Prepare API call parameters
         const createParams = {
-            ...modelConfig,
+            ...openaiModelConfig,
             messages: openaiMessages,
         };
         // Only add tools if they exist
@@ -102,15 +127,20 @@ ${workdir}
     }
 }
 export async function compressMessages(options) {
-    const { messages, abortSignal } = options;
-    // Get model configuration
-    const modelConfig = getModelConfig(FAST_MODEL_ID, {
+    const { gatewayConfig, modelConfig, messages, abortSignal } = options;
+    // Create OpenAI client with injected configuration
+    const openai = new OpenAI({
+        apiKey: gatewayConfig.apiKey,
+        baseURL: gatewayConfig.baseURL,
+    });
+    // Get model configuration - use injected fast model
+    const openaiModelConfig = getModelConfig(modelConfig.fastModel, {
         temperature: 0.1,
         max_tokens: 1500,
     });
     try {
         const response = await openai.chat.completions.create({
-            ...modelConfig,
+            ...openaiModelConfig,
             messages: [
                 {
                     role: "system",

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

@@ -1 +1 @@
-{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/services/session.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAE3C,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,QAAQ,EAAE;QACR,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,CAAC;QAClB,YAAY,EAAE,MAAM,CAAC;QACrB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;CAC3B;AAkBD;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAG5D;AAcD;;GAEG;AACH,wBAAsB,WAAW,CAC/B,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE,MAAM,EACf,iBAAiB,GAAE,MAAU,EAC7B,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC,CA+Bf;AAED;;GAEG;AACH,wBAAsB,WAAW,CAC/B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAmB7B;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAa7B;AAED;;GAEG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,kBAAkB,UAAQ,GACzB,OAAO,CAAC,eAAe,EAAE,CAAC,CA2C5B;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAYvE;AAED;;GAEG;AACH,wBAAsB,sBAAsB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CA4B7E;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CASvE"}
+{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/services/session.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAE3C,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,EAAE,OAAO,EAAE,CAAC;IACpB,QAAQ,EAAE;QACR,OAAO,EAAE,MAAM,CAAC;QAChB,SAAS,EAAE,MAAM,CAAC;QAClB,YAAY,EAAE,MAAM,CAAC;QACrB,iBAAiB,EAAE,MAAM,CAAC;KAC3B,CAAC;CACH;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,CAAC;CAC3B;AAkBD;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAG5D;AAcD;;GAEG;AACH,wBAAsB,WAAW,CAC/B,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE,MAAM,EACf,iBAAiB,GAAE,MAAU,EAC7B,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC,CAoCf;AAED;;GAEG;AACH,wBAAsB,WAAW,CAC/B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAmB7B;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,MAAM,GACd,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAa7B;AAED;;GAEG;AACH,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,kBAAkB,UAAQ,GACzB,OAAO,CAAC,eAAe,EAAE,CAAC,CA2C5B;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAYvE;AAED;;GAEG;AACH,wBAAsB,sBAAsB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CA4B7E;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CASvE"}

package/dist/services/session.js

@@ -42,6 +42,10 @@ export async function saveSession(sessionId, messages, workdir, latestTotalToken
     if (process.env.NODE_ENV === "test") {
         return;
     }
+    // Do not save if there are no messages
+    if (messages.length === 0) {
+        return;
+    }
     await ensureSessionDir();
     // Filter out diff blocks before saving
     const filteredMessages = filterDiffBlocks(messages);

package/dist/tools/grepTool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"grepTool.d.ts","sourceRoot":"","sources":["../../src/tools/grepTool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAA2B,MAAM,YAAY,CAAC;AAMtE;;GAEG;AACH,eAAO,MAAM,QAAQ,EAAE,UA0QtB,CAAC"}
+{"version":3,"file":"grepTool.d.ts","sourceRoot":"","sources":["../../src/tools/grepTool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAA2B,MAAM,YAAY,CAAC;AAMtE;;GAEG;AACH,eAAO,MAAM,QAAQ,EAAE,UA6QtB,CAAC"}