klaus-agent 0.1.0

package/dist/core/agent.js

@@ -0,0 +1,335 @@
+// Agent class — the runtime object with full module integration
+import { SessionManager } from "../session/session-manager.js";
+import { CheckpointManager } from "../checkpoint/checkpoint-manager.js";
+import { InjectionManager } from "../injection/injection-manager.js";
+import { ExtensionRunner } from "../extensions/runner.js";
+import { LaborMarket } from "../multi-agent/labor-market.js";
+import { TaskExecutor } from "../multi-agent/task-executor.js";
+import { createTaskTool } from "../multi-agent/task-tool.js";
+import { MCPAdapter } from "../tools/mcp-adapter.js";
+import { discoverSkills } from "../skills/discovery.js";
+import { createInvokeSkillTool } from "../skills/skill-tool.js";
+import { LLMSummarizer, agentMessagesToCompactionInput } from "../compaction/summarizer.js";
+import { Wire } from "../wire/wire.js";
+import { BackgroundTaskManager } from "../background/task-manager.js";
+import { createBackgroundTaskTools } from "../background/tools.js";
+import { runAgentLoop } from "./agent-loop.js";
+export class Agent {
+    _state;
+    _provider;
+    _approval;
+    _listeners = new Set();
+    _steeringQueue = [];
+    _followUpQueue = [];
+    _abortController = null;
+    _config;
+    _name;
+    // Integrated modules
+    _sessionManager;
+    _checkpointManager;
+    _injectionManager;
+    _extensionRunner;
+    _laborMarket;
+    _taskExecutor;
+    _mcpAdapter;
+    _wire;
+    _backgroundTaskManager;
+    _initialized = false;
+    constructor(config) {
+        this._config = config;
+        this._provider = config.provider;
+        this._approval = config.approval;
+        this._name = config.name ?? "agent";
+        this._wire = new Wire({ bufferSize: config.wire?.bufferSize ?? 0 });
+        this._state = {
+            systemPrompt: typeof config.systemPrompt === "string" ? config.systemPrompt : "",
+            model: config.model,
+            thinkingLevel: config.thinkingLevel ?? "off",
+            tools: config.tools,
+            messages: [],
+            isRunning: false,
+        };
+        // Session
+        if (config.session) {
+            this._sessionManager = new SessionManager(config.session);
+        }
+        // Checkpoint (requires session)
+        if (config.checkpoint?.enabled !== false && this._sessionManager) {
+            this._checkpointManager = new CheckpointManager(this._sessionManager);
+        }
+        // Injection
+        if (config.injection?.length) {
+            this._injectionManager = new InjectionManager(config.injection);
+        }
+        // MCP
+        if (config.mcp?.servers.length) {
+            this._mcpAdapter = new MCPAdapter(config.mcp.servers, config.mcp.clientFactory);
+        }
+        // Multi-agent
+        if (config.subagents && Object.keys(config.subagents).length > 0) {
+            this._laborMarket = new LaborMarket();
+            this._taskExecutor = new TaskExecutor(this._laborMarket, (name, event) => {
+                this._emit({ ...event, type: event.type });
+            });
+        }
+        // Background tasks
+        if (config.backgroundTasks) {
+            this._backgroundTaskManager = new BackgroundTaskManager((taskEvent) => {
+                const { task } = taskEvent;
+                if (taskEvent.type === "task_started") {
+                    this._emit({ type: "task_started", taskId: task.id, taskName: task.name });
+                    this._extensionRunner?.emitSimple("task_started", { taskId: task.id, taskName: task.name });
+                }
+                else if (taskEvent.type === "task_completed") {
+                    this._emit({ type: "task_completed", taskId: task.id, taskName: task.name });
+                    this._extensionRunner?.emitSimple("task_completed", { taskId: task.id, taskName: task.name, result: taskEvent.result });
+                }
+                else if (taskEvent.type === "task_failed") {
+                    this._emit({ type: "task_failed", taskId: task.id, taskName: task.name, error: taskEvent.error });
+                    this._extensionRunner?.emitSimple("task_failed", { taskId: task.id, taskName: task.name, error: taskEvent.error });
+                }
+            });
+        }
+    }
+    // --- Public API ---
+    async prompt(input) {
+        if (this._state.isRunning) {
+            throw new Error("Agent is already running. Use steer() or followUp() to inject messages.");
+        }
+        await this._ensureInitialized();
+        // Resolve system prompt if dynamic
+        if (typeof this._config.systemPrompt === "function") {
+            this._state.systemPrompt = await this._config.systemPrompt();
+        }
+        // Normalize input to messages
+        const inputMessages = this._normalizeInput(input);
+        this._state.messages.push(...inputMessages);
+        // Persist input to session
+        for (const msg of inputMessages) {
+            await this._sessionManager?.appendMessage(msg);
+        }
+        return this._runLoop();
+    }
+    async continue() {
+        if (this._state.isRunning) {
+            throw new Error("Agent is already running.");
+        }
+        await this._ensureInitialized();
+        return this._runLoop();
+    }
+    steer(message) {
+        this._steeringQueue.push(message);
+    }
+    followUp(message) {
+        this._followUpQueue.push(message);
+    }
+    abort() {
+        this._abortController?.abort();
+    }
+    subscribe(fn) {
+        this._listeners.add(fn);
+        return () => this._listeners.delete(fn);
+    }
+    get state() {
+        return this._state;
+    }
+    get approval() {
+        return this._approval;
+    }
+    get session() {
+        return this._sessionManager;
+    }
+    get checkpoints() {
+        return this._checkpointManager;
+    }
+    get extensions() {
+        return this._extensionRunner;
+    }
+    get injections() {
+        return this._injectionManager;
+    }
+    get laborMarket() {
+        return this._laborMarket;
+    }
+    get mcpAdapter() {
+        return this._mcpAdapter;
+    }
+    get wire() {
+        return this._wire;
+    }
+    get backgroundTasks() {
+        return this._backgroundTaskManager;
+    }
+    setSystemPrompt(prompt) {
+        this._state.systemPrompt = prompt;
+    }
+    setModel(model) {
+        this._state.model = model;
+    }
+    setTools(tools) {
+        if (this._state.isRunning) {
+            throw new Error("Cannot change tools while agent is running.");
+        }
+        this._state.tools = tools;
+    }
+    setThinkingLevel(level) {
+        if (this._state.isRunning) {
+            throw new Error("Cannot change thinking level while agent is running.");
+        }
+        this._state.thinkingLevel = level;
+    }
+    async dispose() {
+        this.abort();
+        this._listeners.clear();
+        this._steeringQueue = [];
+        this._followUpQueue = [];
+        await this._mcpAdapter?.dispose();
+        this._backgroundTaskManager?.dispose();
+        this._wire.dispose();
+    }
+    // --- Internal ---
+    async _ensureInitialized() {
+        if (this._initialized)
+            return;
+        this._initialized = true;
+        // Init session and restore messages
+        if (this._sessionManager) {
+            await this._sessionManager.init();
+            const ctx = this._sessionManager.buildSessionContext();
+            if (ctx.messages.length > 0) {
+                this._state.messages = ctx.messages;
+            }
+        }
+        // Load extensions
+        if (this._config.extensions?.length) {
+            this._extensionRunner = new ExtensionRunner();
+            await this._extensionRunner.loadExtensions(this._config.extensions);
+        }
+        // Notify extensions of session start
+        if (this._sessionManager && this._extensionRunner) {
+            await this._extensionRunner.emitSimple("session_start", {
+                sessionId: this._sessionManager.getSessionId(),
+            });
+        }
+        // Load MCP tools in background
+        if (this._mcpAdapter) {
+            this._mcpAdapter.startLoading();
+            await this._mcpAdapter.waitForReady();
+        }
+        // Discover skills and register skill tool
+        if (this._config.skills?.length) {
+            const skills = await discoverSkills(this._config.skills);
+            if (skills.length > 0) {
+                // Add skill list to system prompt for awareness
+                const skillList = skills.map((s) => `- /${s.name}: ${s.description}`).join("\n");
+                this._state.systemPrompt += `\n\nAvailable skills (use invoke_skill tool to invoke):\n${skillList}`;
+                // Add built-in InvokeSkillTool so LLM can actually invoke skills
+                this._state.tools = [...this._state.tools, createInvokeSkillTool(skills)];
+            }
+        }
+        // Init fixed subagents and create TaskTool
+        if (this._config.subagents && this._laborMarket && this._taskExecutor) {
+            for (const [name, subConfig] of Object.entries(this._config.subagents)) {
+                const subAgent = new Agent({
+                    model: this._config.model,
+                    systemPrompt: subConfig.systemPrompt,
+                    tools: subConfig.tools ?? [],
+                    provider: this._provider,
+                    approval: this._approval.share(),
+                    name,
+                });
+                this._laborMarket.addFixed(name, subAgent, subConfig.description);
+            }
+            // Add built-in TaskTool so LLM can delegate to subagents
+            this._state.tools = [...this._state.tools, createTaskTool(this._laborMarket, this._taskExecutor)];
+        }
+        // Background task tools
+        if (this._backgroundTaskManager) {
+            const bgTools = createBackgroundTaskTools(this._backgroundTaskManager, this._config.backgroundTasks?.factories);
+            this._state.tools = [...this._state.tools, ...bgTools];
+        }
+    }
+    _normalizeInput(input) {
+        if (typeof input === "string") {
+            const msg = { role: "user", content: input };
+            return [msg];
+        }
+        if (Array.isArray(input))
+            return input;
+        return [input];
+    }
+    _emit(event) {
+        for (const fn of this._listeners) {
+            try {
+                fn(event);
+            }
+            catch {
+                // Listener errors should not break the loop
+            }
+        }
+        this._wire.publish(event.type, event);
+    }
+    async _runLoop() {
+        this._state.isRunning = true;
+        this._state.error = undefined;
+        this._abortController = new AbortController();
+        // Collect all tools: config tools + MCP tools + extension tools
+        const mcpTools = this._mcpAdapter?.getAllTools() ?? [];
+        const allConfigTools = [...this._state.tools, ...mcpTools];
+        // Build compaction config with summarizer
+        const compactionWithSummarizer = this._config.compaction ? {
+            ...this._config.compaction,
+            maxContextTokens: this._config.compaction.maxContextTokens ?? this._config.model.maxContextTokens,
+            summarize: this._config.compaction.customSummarizer
+                ? (msgs) => this._config.compaction.customSummarizer.summarize(agentMessagesToCompactionInput(msgs))
+                : (msgs) => {
+                    const summarizer = new LLMSummarizer(this._provider, this._state.model.model);
+                    return summarizer.summarize(agentMessagesToCompactionInput(msgs));
+                },
+        } : undefined;
+        try {
+            const result = await runAgentLoop(this._state.messages, {
+                provider: this._provider,
+                modelId: this._state.model.model,
+                systemPrompt: this._state.systemPrompt,
+                tools: allConfigTools,
+                approval: this._approval,
+                agentName: this._name,
+                toolExecution: this._config.toolExecution ?? "parallel",
+                maxStepsPerTurn: this._config.maxStepsPerTurn ?? 50,
+                hooks: this._config.hooks,
+                thinkingLevel: this._state.thinkingLevel,
+                capabilities: this._state.model.capabilities,
+                signal: this._abortController.signal,
+                getSteeringMessages: () => {
+                    const msgs = [...this._steeringQueue];
+                    this._steeringQueue = [];
+                    return msgs;
+                },
+                getFollowUpMessages: () => {
+                    const msgs = [...this._followUpQueue];
+                    this._followUpQueue = [];
+                    return msgs;
+                },
+                onEvent: (event) => this._emit(event),
+                // Integrated modules
+                sessionManager: this._sessionManager,
+                checkpointManager: this._checkpointManager,
+                injectionManager: this._injectionManager,
+                extensionRunner: this._extensionRunner,
+                compaction: compactionWithSummarizer,
+            });
+            this._state.messages = result;
+            return result;
+        }
+        catch (err) {
+            this._state.error = err instanceof Error ? err.message : String(err);
+            throw err;
+        }
+        finally {
+            this._state.isRunning = false;
+            this._abortController = null;
+        }
+    }
+}
+//# sourceMappingURL=agent.js.map

package/dist/core/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/core/agent.ts"],"names":[],"mappings":"AAAA,gEAAgE;AAuBhE,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AACxE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mCAAmC,CAAC;AACrE,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,gCAAgC,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAE,MAAM,iCAAiC,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,aAAa,EAAE,8BAA8B,EAAE,MAAM,6BAA6B,CAAC;AAC5F,OAAO,EAAE,IAAI,EAAE,MAAM,iBAAiB,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AACtE,OAAO,EAAE,yBAAyB,EAAE,MAAM,wBAAwB,CAAC;AACnE,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AA2B/C,MAAM,OAAO,KAAK;IACR,MAAM,CAAa;IACnB,SAAS,CAAc;IACvB,SAAS,CAAW;IACpB,UAAU,GAAG,IAAI,GAAG,EAA+B,CAAC;IACpD,cAAc,GAAmB,EAAE,CAAC;IACpC,cAAc,GAAmB,EAAE,CAAC;IACpC,gBAAgB,GAA2B,IAAI,CAAC;IAChD,OAAO,CAAc;IACrB,KAAK,CAAS;IAEtB,qBAAqB;IACb,eAAe,CAA6B;IAC5C,kBAAkB,CAAgC;IAClD,iBAAiB,CAA+B;IAChD,gBAAgB,CAA8B;IAC9C,YAAY,CAA0B;IACtC,aAAa,CAA2B;IACxC,WAAW,CAAyB;IACpC,KAAK,CAAO;IACZ,sBAAsB,CAAoC;IAC1D,YAAY,GAAG,KAAK,CAAC;IAE7B,YAAY,MAAmB;QAC7B,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,IAAI,IAAI,OAAO,CAAC;QACpC,IAAI,CAAC,KAAK,GAAG,IAAI,IAAI,CAAC,EAAE,UAAU,EAAE,MAAM,CAAC,IAAI,EAAE,UAAU,IAAI,CAAC,EAAE,CAAC,CAAC;QAEpE,IAAI,CAAC,MAAM,GAAG;YACZ,YAAY,EAAE,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE;YAChF,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,KAAK;YAC5C,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,EAAE;YACZ,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,UAAU;QACV,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,IAAI,CAAC,eAAe,GAAG,IAAI,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC5D,CAAC;QAED,gCAAgC;QAChC,IAAI,MAAM,CAAC,UAAU,EAAE,OAAO,KAAK,KAAK,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACjE,IAAI,CAAC,kBAAkB,GAAG,IAAI,iBAAiB,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QACxE,CAAC;QAED,YAAY;QACZ,IAAI,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;YAC7B,IAAI,CAAC,iBAAiB,GAAG,IAAI,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAClE,CAAC;QAED,MAAM;QACN,IAAI,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;YAC/B,IAAI,CAAC,WAAW,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAClF,CAAC;QAED,cAAc;QACd,IAAI,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACjE,IAAI,CAAC,YAAY,GAAG,IAAI,WAAW,EAAE,CAAC;YACtC,IAAI,CAAC,aAAa,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;gBACvE,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG,KAAK,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAgB,CAAC,CAAC;YAC3D,CAAC,CAAC,CAAC;QACL,CAAC;QAED,mBAAmB;QACnB,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YAC3B,IAAI,CAAC,sBAAsB,GAAG,IAAI,qBAAqB,CAAC,CAAC,SAAS,EAAE,EAAE;gBACpE,MAAM,EAAE,IAAI,EAAE,GAAG,SAAS,CAAC;gBAC3B,IAAI,SAAS,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;oBACtC,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC3E,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAC9F,CAAC;qBAAM,IAAI,SAAS,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;oBAC/C,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC7E,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC1H,CAAC;qBAAM,IAAI,SAAS,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBAC5C,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC;oBAClG,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC;gBACrH,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,qBAAqB;IAErB,KAAK,CAAC,MAAM,CAAC,KAA6C;QACxD,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,yEAAyE,CAAC,CAAC;QAC7F,CAAC;QAED,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAEhC,mCAAmC;QACnC,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,YAAY,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;QAC/D,CAAC;QAED,8BAA8B;QAC9B,MAAM,aAAa,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,aAAa,CAAC,CAAC;QAE5C,2BAA2B;QAC3B,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,MAAM,IAAI,CAAC,eAAe,EAAE,aAAa,CAAC,GAAG,CAAC,CAAC;QACjD,CAAC;QAED,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;QAC/C,CAAC;QACD,MAAM,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAChC,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,OAAqB;QACzB,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAED,QAAQ,CAAC,OAAqB;QAC5B,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,gBAAgB,EAAE,KAAK,EAAE,CAAC;IACjC,CAAC;IAED,SAAS,CAAC,EAA+B;QACvC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxB,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,eAAe,CAAC;IAC9B,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,gBAAgB,CAAC;IAC/B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,sBAAsB,CAAC;IACrC,CAAC;IAED,eAAe,CAAC,MAAc;QAC5B,IAAI,CAAC,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC;IACpC,CAAC;IAED,QAAQ,CAAC,KAAkB;QACzB,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;IAC5B,CAAC;IAED,QAAQ,CAAC,KAAkB;QACzB,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;IAC5B,CAAC;IAED,gBAAgB,CAAC,KAAoB;QACnC,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,KAAK,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,OAAO;QACX,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;QACzB,MAAM,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,CAAC;QAClC,IAAI,CAAC,sBAAsB,EAAE,OAAO,EAAE,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;IACvB,CAAC;IAED,mBAAmB;IAEX,KAAK,CAAC,kBAAkB;QAC9B,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,oCAAoC;QACpC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;YAClC,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,mBAAmB,EAAE,CAAC;YACvD,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC;YACtC,CAAC;QACH,CAAC;QAED,kBAAkB;QAClB,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;YACpC,IAAI,CAAC,gBAAgB,GAAG,IAAI,eAAe,EAAE,CAAC;YAC9C,MAAM,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACtE,CAAC;QAED,qCAAqC;QACrC,IAAI,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAClD,MAAM,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,eAAe,EAAE;gBACtD,SAAS,EAAE,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE;aAC/C,CAAC,CAAC;QACL,CAAC;QAED,+BAA+B;QAC/B,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;YAChC,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,CAAC;QACxC,CAAC;QAED,0CAA0C;QAC1C,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACzD,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACtB,gDAAgD;gBAChD,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjF,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,4DAA4D,SAAS,EAAE,CAAC;gBACpG,iEAAiE;gBACjE,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,qBAAqB,CAAC,MAAM,CAAC,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC;QAED,2CAA2C;QAC3C,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACtE,KAAK,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;gBACvE,MAAM,QAAQ,GAAG,IAAI,KAAK,CAAC;oBACzB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;oBACzB,YAAY,EAAE,SAAS,CAAC,YAAY;oBACpC,KAAK,EAAE,SAAS,CAAC,KAAK,IAAI,EAAE;oBAC5B,QAAQ,EAAE,IAAI,CAAC,SAAS;oBACxB,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE;oBAChC,IAAI;iBACL,CAAC,CAAC;gBACH,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,EAAE,SAAS,CAAC,WAAW,CAAC,CAAC;YACpE,CAAC;YACD,yDAAyD;YACzD,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,cAAc,CAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;QACpG,CAAC;QAED,wBAAwB;QACxB,IAAI,IAAI,CAAC,sBAAsB,EAAE,CAAC;YAChC,MAAM,OAAO,GAAG,yBAAyB,CAAC,IAAI,CAAC,sBAAsB,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;YAChH,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,OAAO,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,KAA6C;QACnE,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAgB,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC1D,OAAO,CAAC,GAAG,CAAC,CAAC;QACf,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QACvC,OAAO,CAAC,KAAK,CAAC,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,KAAiB;QAC7B,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,EAAE,CAAC,KAAK,CAAC,CAAC;YACZ,CAAC;YAAC,MAAM,CAAC;gBACP,4CAA4C;YAC9C,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACxC,CAAC;IAEO,KAAK,CAAC,QAAQ;QACpB,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,SAAS,CAAC;QAC9B,IAAI,CAAC,gBAAgB,GAAG,IAAI,eAAe,EAAE,CAAC;QAE9C,gEAAgE;QAChE,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;QACvD,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,GAAG,QAAQ,CAAC,CAAC;QAE3D,0CAA0C;QAC1C,MAAM,wBAAwB,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;YACzD,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU;YAC1B,gBAAgB,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,gBAAgB,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,gBAAgB;YACjG,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,gBAAgB;gBACjD,CAAC,CAAC,CAAC,IAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,UAAW,CAAC,gBAAiB,CAAC,SAAS,CAC5E,8BAA8B,CAAC,IAAI,CAAC,CACrC;gBACH,CAAC,CAAC,CAAC,IAAoB,EAAE,EAAE;oBACvB,MAAM,UAAU,GAAG,IAAI,aAAa,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC9E,OAAO,UAAU,CAAC,SAAS,CAAC,8BAA8B,CAAC,IAAI,CAAC,CAAC,CAAC;gBACpE,CAAC;SACN,CAAC,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;gBACtD,QAAQ,EAAE,IAAI,CAAC,SAAS;gBACxB,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK;gBAChC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;gBACtC,KAAK,EAAE,cAAc;gBACrB,QAAQ,EAAE,IAAI,CAAC,SAAS;gBACxB,SAAS,EAAE,IAAI,CAAC,KAAK;gBACrB,aAAa,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,IAAI,UAAU;gBACvD,eAAe,EAAE,IAAI,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE;gBACnD,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;gBACzB,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;gBACxC,YAAY,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY;gBAC5C,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,MAAM;gBACpC,mBAAmB,EAAE,GAAG,EAAE;oBACxB,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;oBACzB,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,mBAAmB,EAAE,GAAG,EAAE;oBACxB,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,IAAI,CAAC,cAAc,GAAG,EAAE,CAAC;oBACzB,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;gBAErC,qBAAqB;gBACrB,cAAc,EAAE,IAAI,CAAC,eAAe;gBACpC,iBAAiB,EAAE,IAAI,CAAC,kBAAkB;gBAC1C,gBAAgB,EAAE,IAAI,CAAC,iBAAiB;gBACxC,eAAe,EAAE,IAAI,CAAC,gBAAgB;gBACtC,UAAU,EAAE,wBAAwB;aACrC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,QAAQ,GAAG,MAAM,CAAC;YAC9B,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACrE,MAAM,GAAG,CAAC;QACZ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,KAAK,CAAC;YAC9B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC/B,CAAC;IACH,CAAC;CACF"}

package/dist/extensions/runner.d.ts

@@ -0,0 +1,19 @@
+import type { ExtensionFactory, ExtensionEventType, BeforeAgentStartEvent, BeforeAgentStartResult, BeforeProviderRequestEvent, BeforeProviderRequestResult, ToolCallEvent, ToolCallEventResult, ToolResultEvent, ToolResultEventResult, ContextEvent, ContextEventResult, BeforeCompactEvent, BeforeCompactResult, CommandHandler } from "./types.js";
+import type { AgentTool } from "../tools/types.js";
+import type { AgentMessage } from "../types.js";
+export declare class ExtensionRunner {
+    private _extensions;
+    private _pendingMessages;
+    loadExtensions(factories: ExtensionFactory[]): Promise<void>;
+    getRegisteredTools(): AgentTool[];
+    getCommand(name: string): CommandHandler | undefined;
+    drainPendingMessages(): AgentMessage[];
+    emitBeforeAgentStart(event: BeforeAgentStartEvent): Promise<BeforeAgentStartResult | void>;
+    emitBeforeProviderRequest(event: BeforeProviderRequestEvent): Promise<BeforeProviderRequestResult | void>;
+    emitToolCall(event: ToolCallEvent): Promise<ToolCallEventResult | void>;
+    emitToolResult(event: ToolResultEvent): Promise<ToolResultEventResult | void>;
+    emitContext(event: ContextEvent): Promise<ContextEventResult | void>;
+    emitBeforeCompact(event: BeforeCompactEvent): Promise<BeforeCompactResult | void>;
+    emitSimple(eventType: ExtensionEventType, payload?: any): Promise<void>;
+    private _emit;
+}

package/dist/extensions/runner.js

@@ -0,0 +1,88 @@
+// Extension runner — event dispatch, tool registration, lifecycle
+export class ExtensionRunner {
+    _extensions = [];
+    _pendingMessages = [];
+    async loadExtensions(factories) {
+        for (const factory of factories) {
+            const ext = {
+                handlers: new Map(),
+                tools: [],
+                commands: new Map(),
+            };
+            const api = {
+                on: (event, handler) => {
+                    if (!ext.handlers.has(event))
+                        ext.handlers.set(event, []);
+                    ext.handlers.get(event).push(handler);
+                },
+                registerTool: (tool) => {
+                    ext.tools.push(tool);
+                },
+                registerCommand: (name, handler) => {
+                    ext.commands.set(name, handler);
+                },
+                sendMessage: (message) => {
+                    this._pendingMessages.push(message);
+                },
+            };
+            await factory(api);
+            this._extensions.push(ext);
+        }
+    }
+    getRegisteredTools() {
+        return this._extensions.flatMap((ext) => ext.tools);
+    }
+    getCommand(name) {
+        for (const ext of this._extensions) {
+            const handler = ext.commands.get(name);
+            if (handler)
+                return handler;
+        }
+        return undefined;
+    }
+    drainPendingMessages() {
+        const msgs = [...this._pendingMessages];
+        this._pendingMessages = [];
+        return msgs;
+    }
+    // --- Interceptable event emitters (type-safe, return results) ---
+    async emitBeforeAgentStart(event) {
+        return this._emit("before_agent_start", event);
+    }
+    async emitBeforeProviderRequest(event) {
+        return this._emit("before_provider_request", event);
+    }
+    async emitToolCall(event) {
+        return this._emit("tool_call", event);
+    }
+    async emitToolResult(event) {
+        return this._emit("tool_result", event);
+    }
+    async emitContext(event) {
+        return this._emit("context", event);
+    }
+    async emitBeforeCompact(event) {
+        return this._emit("before_compact", event);
+    }
+    // --- Notification emitter (fire-and-forget) ---
+    async emitSimple(eventType, payload) {
+        await this._emit(eventType, payload);
+    }
+    // --- Internal ---
+    async _emit(eventType, payload) {
+        let lastResult;
+        for (const ext of this._extensions) {
+            const handlers = ext.handlers.get(eventType);
+            if (!handlers)
+                continue;
+            for (const handler of handlers) {
+                const result = await handler(payload);
+                if (result !== undefined && result !== null) {
+                    lastResult = result;
+                }
+            }
+        }
+        return lastResult;
+    }
+}
+//# sourceMappingURL=runner.js.map

package/dist/extensions/runner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runner.js","sourceRoot":"","sources":["../../src/extensions/runner.ts"],"names":[],"mappings":"AAAA,kEAAkE;AAgClE,MAAM,OAAO,eAAe;IAClB,WAAW,GAA0B,EAAE,CAAC;IACxC,gBAAgB,GAAmB,EAAE,CAAC;IAE9C,KAAK,CAAC,cAAc,CAAC,SAA6B;QAChD,KAAK,MAAM,OAAO,IAAI,SAAS,EAAE,CAAC;YAChC,MAAM,GAAG,GAAwB;gBAC/B,QAAQ,EAAE,IAAI,GAAG,EAAE;gBACnB,KAAK,EAAE,EAAE;gBACT,QAAQ,EAAE,IAAI,GAAG,EAAE;aACpB,CAAC;YAEF,MAAM,GAAG,GAAiB;gBACxB,EAAE,EAAE,CAA+B,KAAQ,EAAE,OAA4B,EAAE,EAAE;oBAC3E,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC;wBAAE,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;oBAC1D,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzC,CAAC;gBACD,YAAY,EAAE,CAA0B,IAAwB,EAAE,EAAE;oBAClE,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAiB,CAAC,CAAC;gBACpC,CAAC;gBACD,eAAe,EAAE,CAAC,IAAY,EAAE,OAAuB,EAAE,EAAE;oBACzD,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;gBAClC,CAAC;gBACD,WAAW,EAAE,CAAC,OAAqB,EAAE,EAAE;oBACrC,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACtC,CAAC;aACF,CAAC;YAEF,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC;YACnB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC;IAED,UAAU,CAAC,IAAY;QACrB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,OAAO;gBAAE,OAAO,OAAO,CAAC;QAC9B,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,oBAAoB;QAClB,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACxC,IAAI,CAAC,gBAAgB,GAAG,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,mEAAmE;IAEnE,KAAK,CAAC,oBAAoB,CAAC,KAA4B;QACrD,OAAO,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,yBAAyB,CAAC,KAAiC;QAC/D,OAAO,IAAI,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAoB;QACrC,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,KAAsB;QACzC,OAAO,IAAI,CAAC,KAAK,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,KAAmB;QACnC,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,KAAyB;QAC/C,OAAO,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;IAC7C,CAAC;IAED,iDAAiD;IAEjD,KAAK,CAAC,UAAU,CAAC,SAA6B,EAAE,OAAa;QAC3D,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED,mBAAmB;IAEX,KAAK,CAAC,KAAK,CACjB,SAAY,EACZ,OAAwC;QAExC,IAAI,UAAe,CAAC;QAEpB,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC7C,IAAI,CAAC,QAAQ;gBAAE,SAAS;YAExB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC;gBACtC,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;oBAC5C,UAAU,GAAG,MAAM,CAAC;gBACtB,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;CACF"}

package/dist/extensions/types.d.ts

@@ -0,0 +1,179 @@
+import type { TSchema } from "@sinclair/typebox";
+import type { AgentTool, AgentToolResult } from "../tools/types.js";
+import type { AgentMessage, ThinkingLevel } from "../types.js";
+import type { Message, ToolDefinition } from "../llm/types.js";
+export type ExtensionFactory = (api: ExtensionAPI) => void | Promise<void>;
+export interface ExtensionAPI {
+    on<E extends ExtensionEventType>(event: E, handler: ExtensionHandler<E>): void;
+    registerTool<TParams extends TSchema>(tool: AgentTool<TParams>): void;
+    registerCommand(name: string, handler: CommandHandler): void;
+    sendMessage(message: AgentMessage): void;
+}
+export type ExtensionEventType = "before_agent_start" | "agent_start" | "agent_end" | "turn_start" | "turn_end" | "message_start" | "message_end" | "tool_call" | "tool_result" | "before_provider_request" | "context" | "before_compact" | "after_compact" | "session_start" | "session_switch" | "session_fork" | "task_started" | "task_completed" | "task_failed";
+export interface BeforeAgentStartEvent {
+    systemPrompt: string;
+    tools: AgentTool[];
+    modelId: string;
+    thinkingLevel?: ThinkingLevel;
+}
+export interface BeforeAgentStartResult {
+    systemPrompt?: string;
+    tools?: AgentTool[];
+    modelId?: string;
+    thinkingLevel?: ThinkingLevel;
+}
+export interface BeforeProviderRequestEvent {
+    model: string;
+    systemPrompt: string;
+    messages: Message[];
+    tools?: ToolDefinition[];
+    thinkingLevel?: ThinkingLevel;
+}
+export interface BeforeProviderRequestResult {
+    systemPrompt?: string;
+    messages?: Message[];
+    tools?: ToolDefinition[];
+    thinkingLevel?: ThinkingLevel;
+}
+export interface ToolCallEvent {
+    toolName: string;
+    toolCallId: string;
+    args: unknown;
+}
+export interface ToolCallEventResult {
+    block?: boolean;
+    reason?: string;
+}
+export interface ToolResultEvent {
+    toolName: string;
+    toolCallId: string;
+    args: unknown;
+    result: AgentToolResult;
+    isError: boolean;
+}
+export interface ToolResultEventResult {
+    content?: AgentToolResult["content"];
+    details?: unknown;
+    isError?: boolean;
+}
+export interface ContextEvent {
+    messages: AgentMessage[];
+}
+export interface ContextEventResult {
+    messages?: AgentMessage[];
+}
+export interface BeforeCompactEvent {
+    messages: AgentMessage[];
+}
+export interface BeforeCompactResult {
+    summary?: string;
+    skip?: boolean;
+}
+export interface AfterCompactEvent {
+    summary: string;
+    tokensBefore: number;
+    tokensAfter: number;
+}
+export interface SessionEvent {
+    sessionId: string;
+    previousSessionId?: string;
+}
+export interface ExtensionEventMap {
+    before_agent_start: {
+        payload: BeforeAgentStartEvent;
+        result: BeforeAgentStartResult | void;
+    };
+    agent_start: {
+        payload: void;
+        result: void;
+    };
+    agent_end: {
+        payload: {
+            messages: AgentMessage[];
+        };
+        result: void;
+    };
+    turn_start: {
+        payload: void;
+        result: void;
+    };
+    turn_end: {
+        payload: {
+            message: AgentMessage;
+        };
+        result: void;
+    };
+    message_start: {
+        payload: {
+            message: AgentMessage;
+        };
+        result: void;
+    };
+    message_end: {
+        payload: {
+            message: AgentMessage;
+        };
+        result: void;
+    };
+    tool_call: {
+        payload: ToolCallEvent;
+        result: ToolCallEventResult | void;
+    };
+    tool_result: {
+        payload: ToolResultEvent;
+        result: ToolResultEventResult | void;
+    };
+    before_provider_request: {
+        payload: BeforeProviderRequestEvent;
+        result: BeforeProviderRequestResult | void;
+    };
+    context: {
+        payload: ContextEvent;
+        result: ContextEventResult | void;
+    };
+    before_compact: {
+        payload: BeforeCompactEvent;
+        result: BeforeCompactResult | void;
+    };
+    after_compact: {
+        payload: AfterCompactEvent;
+        result: void;
+    };
+    session_start: {
+        payload: SessionEvent;
+        result: void;
+    };
+    session_switch: {
+        payload: SessionEvent;
+        result: void;
+    };
+    session_fork: {
+        payload: SessionEvent;
+        result: void;
+    };
+    task_started: {
+        payload: {
+            taskId: string;
+            taskName: string;
+        };
+        result: void;
+    };
+    task_completed: {
+        payload: {
+            taskId: string;
+            taskName: string;
+            result: unknown;
+        };
+        result: void;
+    };
+    task_failed: {
+        payload: {
+            taskId: string;
+            taskName: string;
+            error: string;
+        };
+        result: void;
+    };
+}
+export type ExtensionHandler<E extends ExtensionEventType> = (event: ExtensionEventMap[E]["payload"]) => ExtensionEventMap[E]["result"] | Promise<ExtensionEventMap[E]["result"]>;
+export type CommandHandler = (args: string) => Promise<void>;

package/dist/extensions/types.js

@@ -0,0 +1,3 @@
+// Extension system types
+export {};
+//# sourceMappingURL=types.js.map

package/dist/extensions/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/extensions/types.ts"],"names":[],"mappings":"AAAA,yBAAyB"}