codemate-ai 1.0.0

package/dist/src/mcp/MCPClient.js

@@ -0,0 +1,117 @@
+import { spawn } from 'node:child_process';
+export class MCPClient {
+    process = null;
+    requestId = 0;
+    pendingRequests = new Map();
+    serverName;
+    config;
+    constructor(serverName, config) {
+        this.serverName = serverName;
+        this.config = config;
+    }
+    async connect() {
+        console.log(`🔌 Connecting to MCP server: ${this.serverName}`);
+        this.process = spawn(this.config.command, this.config.args, {
+            env: { ...process.env, ...this.config.env },
+            stdio: ['pipe', 'pipe', 'pipe'],
+        });
+        if (!this.process.stdout || !this.process.stdin) {
+            throw new Error('Failed to create MCP server process');
+        }
+        let buffer = '';
+        this.process.stdout.on('data', (data) => {
+            buffer += data.toString();
+            const lines = buffer.split('\n');
+            buffer = lines.pop() || '';
+            for (const line of lines) {
+                if (line.trim()) {
+                    try {
+                        const response = JSON.parse(line);
+                        this.handleResponse(response);
+                    }
+                    catch (error) {
+                        console.error('Failed to parse MCP response:', error);
+                    }
+                }
+            }
+        });
+        this.process.stderr?.on('data', (data) => {
+            console.error(`[MCP ${this.serverName}]`, data.toString());
+        });
+        await this.initialize();
+        console.log(`✅ Connected to MCP server: ${this.serverName}`);
+    }
+    async initialize() {
+        await this.sendRequest('initialize', {
+            protocolVersion: '2024-11-05',
+            capabilities: {
+                tools: {},
+            },
+            clientInfo: {
+                name: 'aicli',
+                version: '1.0.0',
+            },
+        });
+    }
+    async listTools() {
+        console.log(`🔍 Discovering tools from: ${this.serverName}`);
+        const response = await this.sendRequest('tools/list', {});
+        const tools = response.tools || [];
+        console.log(`✅ Found ${tools.length} tools from ${this.serverName}`);
+        return tools;
+    }
+    async callTool(name, args) {
+        console.log(`🛠️  Calling MCP tool: ${this.serverName}.${name}`);
+        console.log(`   Arguments:`, JSON.stringify(args));
+        const response = await this.sendRequest('tools/call', {
+            name,
+            arguments: args,
+        });
+        console.log(`✅ Tool executed: ${this.serverName}.${name}`);
+        return response;
+    }
+    async sendRequest(method, params) {
+        if (!this.process || !this.process.stdin) {
+            throw new Error('MCP server not connected');
+        }
+        const id = ++this.requestId;
+        const request = {
+            jsonrpc: '2.0',
+            id,
+            method,
+            params,
+        };
+        const promise = new Promise((resolve, reject) => {
+            this.pendingRequests.set(id, { resolve, reject });
+            setTimeout(() => {
+                if (this.pendingRequests.has(id)) {
+                    this.pendingRequests.delete(id);
+                    reject(new Error(`MCP request timeout: ${method}`));
+                }
+            }, 30000);
+        });
+        this.process.stdin.write(JSON.stringify(request) + '\n');
+        return promise;
+    }
+    handleResponse(response) {
+        const pending = this.pendingRequests.get(response.id);
+        if (!pending) {
+            return;
+        }
+        this.pendingRequests.delete(response.id);
+        if (response.error) {
+            pending.reject(new Error(response.error.message));
+        }
+        else {
+            pending.resolve(response.result);
+        }
+    }
+    async disconnect() {
+        if (this.process) {
+            this.process.kill();
+            this.process = null;
+        }
+        console.log(`🔌 Disconnected from MCP server: ${this.serverName}`);
+    }
+}
+//# sourceMappingURL=MCPClient.js.map

package/dist/src/mcp/MCPClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MCPClient.js","sourceRoot":"","sources":["../../../src/mcp/MCPClient.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,oBAAoB,CAAC;AAkB9D,MAAM,OAAO,SAAS;IACZ,OAAO,GAAwB,IAAI,CAAC;IACpC,SAAS,GAAG,CAAC,CAAC;IACd,eAAe,GAAG,IAAI,GAAG,EAM9B,CAAC;IAEI,UAAU,CAAS;IACnB,MAAM,CAAkB;IAEhC,YAAY,UAAkB,EAAE,MAAuB;QACrD,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAKD,KAAK,CAAC,OAAO;QACX,OAAO,CAAC,GAAG,CAAC,gCAAgC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAG/D,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE;YAC1D,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE;YAC3C,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;SAChC,CAAC,CAAC;QAEH,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YAChD,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;QACzD,CAAC;QAGD,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;YAC9C,MAAM,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAG1B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;oBAChB,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAgB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAC/C,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;oBAChC,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;oBACxD,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE;YAC/C,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,CAAC,UAAU,GAAG,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAGH,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;IAC/D,CAAC;IAKO,KAAK,CAAC,UAAU;QACtB,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;YACnC,eAAe,EAAE,YAAY;YAC7B,YAAY,EAAE;gBACZ,KAAK,EAAE,EAAE;aACV;YACD,UAAU,EAAE;gBACV,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,OAAO;aACjB;SACF,CAAC,CAAC;IACL,CAAC;IAKD,KAAK,CAAC,SAAS;QACb,OAAO,CAAC,GAAG,CAAC,8BAA8B,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;QAC1D,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;QAEnC,OAAO,CAAC,GAAG,CAAC,WAAW,KAAK,CAAC,MAAM,eAAe,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAErE,OAAO,KAAK,CAAC;IACf,CAAC;IAKD,KAAK,CAAC,QAAQ,CAAC,IAAY,EAAE,IAAyB;QACpD,OAAO,CAAC,GAAG,CAAC,0BAA0B,IAAI,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC,CAAC;QACjE,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;YACpD,IAAI;YACJ,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC,CAAC;QAE3D,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKO,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,MAAW;QACnD,IAAI,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,EAAE,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC;QAC5B,MAAM,OAAO,GAAe;YAC1B,OAAO,EAAE,KAAK;YACd,EAAE;YACF,MAAM;YACN,MAAM;SACP,CAAC;QAGF,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC9C,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAGlD,UAAU,CAAC,GAAG,EAAE;gBACd,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACjC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAChC,MAAM,CAAC,IAAI,KAAK,CAAC,wBAAwB,MAAM,EAAE,CAAC,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC,EAAE,KAAK,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;QAGH,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC;QAEzD,OAAO,OAAO,CAAC;IACjB,CAAC;IAKO,cAAc,CAAC,QAAqB;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAY,CAAC,CAAC;QAChE,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAY,CAAC,CAAC;QAEnD,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACnB,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QACpD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAKD,KAAK,CAAC,UAAU;QACd,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACpB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACtB,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,oCAAoC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;IACrE,CAAC;CACF"}

package/dist/src/mcp/MCPManager.d.ts

@@ -0,0 +1,12 @@
+import { MCPClient } from './MCPClient';
+import type { Tool } from '../tools/base/Tool';
+export declare class MCPManager {
+    private clients;
+    private tools;
+    initialize(configPath: string): Promise<void>;
+    private connectServer;
+    getTools(): Tool[];
+    getClient(serverName: string): MCPClient | undefined;
+    disconnect(): Promise<void>;
+}
+//# sourceMappingURL=MCPManager.d.ts.map

package/dist/src/mcp/MCPManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MCPManager.d.ts","sourceRoot":"","sources":["../../../src/mcp/MCPManager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAGxC,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAW/C,qBAAa,UAAU;IACrB,OAAO,CAAC,OAAO,CAAgC;IAC/C,OAAO,CAAC,KAAK,CAAc;IAKrB,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YA6BrC,aAAa;IA8B3B,QAAQ,IAAI,IAAI,EAAE;IAOlB,SAAS,CAAC,UAAU,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS;IAO9C,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAclC"}

package/dist/src/mcp/MCPManager.js

@@ -0,0 +1,63 @@
+import * as fs from 'node:fs/promises';
+import { MCPClient } from './MCPClient';
+import { MCPTool } from './MCPTool';
+export class MCPManager {
+    clients = new Map();
+    tools = [];
+    async initialize(configPath) {
+        console.log(`📋 Loading MCP config from: ${configPath}`);
+        try {
+            const configContent = await fs.readFile(configPath, 'utf-8');
+            const config = JSON.parse(configContent);
+            for (const [serverName, serverConfig] of Object.entries(config.servers)) {
+                await this.connectServer(serverName, serverConfig);
+            }
+            console.log(`✅ MCP initialized with ${this.clients.size} servers`);
+            console.log(`✅ Discovered ${this.tools.length} MCP tools`);
+        }
+        catch (error) {
+            if (error.code === 'ENOENT') {
+                console.log(`ℹ️  No MCP config found at ${configPath}`);
+                return;
+            }
+            console.error(`❌ Failed to initialize MCP:`, error);
+            throw error;
+        }
+    }
+    async connectServer(serverName, config) {
+        try {
+            const client = new MCPClient(serverName, config);
+            await client.connect();
+            const toolDefs = await client.listTools();
+            for (const toolDef of toolDefs) {
+                const tool = new MCPTool(client, toolDef, serverName);
+                this.tools.push(tool);
+            }
+            this.clients.set(serverName, client);
+            console.log(`✅ Connected to MCP server: ${serverName} (${toolDefs.length} tools)`);
+        }
+        catch (error) {
+            console.error(`❌ Failed to connect to MCP server ${serverName}:`, error);
+        }
+    }
+    getTools() {
+        return this.tools;
+    }
+    getClient(serverName) {
+        return this.clients.get(serverName);
+    }
+    async disconnect() {
+        for (const [serverName, client] of this.clients.entries()) {
+            try {
+                await client.disconnect();
+            }
+            catch (error) {
+                console.error(`Failed to disconnect from ${serverName}:`, error);
+            }
+        }
+        this.clients.clear();
+        this.tools = [];
+        console.log(`🔌 All MCP servers disconnected`);
+    }
+}
+//# sourceMappingURL=MCPManager.js.map

package/dist/src/mcp/MCPManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MCPManager.js","sourceRoot":"","sources":["../../../src/mcp/MCPManager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAapC,MAAM,OAAO,UAAU;IACb,OAAO,GAAG,IAAI,GAAG,EAAqB,CAAC;IACvC,KAAK,GAAW,EAAE,CAAC;IAK3B,KAAK,CAAC,UAAU,CAAC,UAAkB;QACjC,OAAO,CAAC,GAAG,CAAC,+BAA+B,UAAU,EAAE,CAAC,CAAC;QAEzD,IAAI,CAAC;YAEH,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC7D,MAAM,MAAM,GAAc,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YAGpD,KAAK,MAAM,CAAC,UAAU,EAAE,YAAY,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACxE,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YACrD,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,0BAA0B,IAAI,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC;YACnE,OAAO,CAAC,GAAG,CAAC,gBAAgB,IAAI,CAAC,KAAK,CAAC,MAAM,YAAY,CAAC,CAAC;QAC7D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAK,KAAa,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACrC,OAAO,CAAC,GAAG,CAAC,8BAA8B,UAAU,EAAE,CAAC,CAAC;gBACxD,OAAO;YACT,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YACpD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAKO,KAAK,CAAC,aAAa,CAAC,UAAkB,EAAE,MAAuB;QACrE,IAAI,CAAC;YAEH,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAGjD,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;YAGvB,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;YAG1C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,MAAM,IAAI,GAAG,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC;gBACtD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YAGD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YAErC,OAAO,CAAC,GAAG,CAAC,8BAA8B,UAAU,KAAK,QAAQ,CAAC,MAAM,SAAS,CAAC,CAAC;QACrF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,qCAAqC,UAAU,GAAG,EAAE,KAAK,CAAC,CAAC;QAE3E,CAAC;IACH,CAAC;IAKD,QAAQ;QACN,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAKD,SAAS,CAAC,UAAkB;QAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACtC,CAAC;IAKD,KAAK,CAAC,UAAU;QACd,KAAK,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YAC1D,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YAC5B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,UAAU,GAAG,EAAE,KAAK,CAAC,CAAC;YACnE,CAAC;QACH,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACrB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;QAEhB,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IACjD,CAAC;CACF"}

package/dist/src/mcp/MCPTool.d.ts

@@ -0,0 +1,15 @@
+import { z } from 'zod';
+import { Tool } from '../tools/base/Tool';
+import type { MCPClient } from './MCPClient';
+import type { MCPToolDefinition } from './types';
+export declare class MCPTool extends Tool {
+    name: string;
+    description: string;
+    schema: z.ZodObject<any>;
+    private client;
+    private toolDef;
+    constructor(client: MCPClient, toolDef: MCPToolDefinition, serverName: string);
+    execute(input: any): Promise<any>;
+    private jsonSchemaToZod;
+}
+//# sourceMappingURL=MCPTool.d.ts.map

package/dist/src/mcp/MCPTool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"MCPTool.d.ts","sourceRoot":"","sources":["../../../src/mcp/MCPTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC1C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAOjD,qBAAa,OAAQ,SAAQ,IAAI;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAEzB,OAAO,CAAC,MAAM,CAAY;IAC1B,OAAO,CAAC,OAAO,CAAoB;gBAEvB,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM;IAiBvE,OAAO,CAAC,KAAK,EAAE,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;IA+BvC,OAAO,CAAC,eAAe;CA4CxB"}

package/dist/src/mcp/MCPTool.js

@@ -0,0 +1,73 @@
+import { z } from 'zod';
+import { Tool } from '../tools/base/Tool';
+export class MCPTool extends Tool {
+    name;
+    description;
+    schema;
+    client;
+    toolDef;
+    constructor(client, toolDef, serverName) {
+        super();
+        this.client = client;
+        this.toolDef = toolDef;
+        this.name = `${serverName}_${toolDef.name}`;
+        this.description = `[MCP:${serverName}] ${toolDef.description}`;
+        this.schema = this.jsonSchemaToZod(toolDef.inputSchema);
+    }
+    async execute(input) {
+        try {
+            const result = await this.client.callTool(this.toolDef.name, input);
+            const textContent = result.content
+                .filter((c) => c.type === 'text')
+                .map((c) => c.text)
+                .join('\n');
+            return {
+                success: true,
+                content: textContent,
+                raw: result,
+            };
+        }
+        catch (error) {
+            console.error(`❌ MCP tool execution failed:`, error);
+            if (error instanceof Error) {
+                throw new Error(`MCP tool failed: ${error.message}`);
+            }
+            throw new Error('MCP tool failed: Unknown error');
+        }
+    }
+    jsonSchemaToZod(schema) {
+        const shape = {};
+        for (const [key, prop] of Object.entries(schema.properties || {})) {
+            const p = prop;
+            let zodType;
+            switch (p.type) {
+                case 'string':
+                    zodType = z.string();
+                    break;
+                case 'number':
+                    zodType = z.number();
+                    break;
+                case 'boolean':
+                    zodType = z.boolean();
+                    break;
+                case 'array':
+                    zodType = z.array(z.any());
+                    break;
+                case 'object':
+                    zodType = z.record(z.any());
+                    break;
+                default:
+                    zodType = z.any();
+            }
+            if (p.description) {
+                zodType = zodType.describe(p.description);
+            }
+            if (!schema.required?.includes(key)) {
+                zodType = zodType.optional();
+            }
+            shape[key] = zodType;
+        }
+        return z.object(shape);
+    }
+}
+//# sourceMappingURL=MCPTool.js.map

package/dist/src/mcp/MCPTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MCPTool.js","sourceRoot":"","sources":["../../../src/mcp/MCPTool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAS1C,MAAM,OAAO,OAAQ,SAAQ,IAAI;IAC/B,IAAI,CAAS;IACb,WAAW,CAAS;IACpB,MAAM,CAAmB;IAEjB,MAAM,CAAY;IAClB,OAAO,CAAoB;IAEnC,YAAY,MAAiB,EAAE,OAA0B,EAAE,UAAkB;QAC3E,KAAK,EAAE,CAAC;QAER,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QAGvB,IAAI,CAAC,IAAI,GAAG,GAAG,UAAU,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;QAC5C,IAAI,CAAC,WAAW,GAAG,QAAQ,UAAU,KAAK,OAAO,CAAC,WAAW,EAAE,CAAC;QAGhE,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC1D,CAAC;IAKD,KAAK,CAAC,OAAO,CAAC,KAAU;QACtB,IAAI,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAGpE,MAAM,WAAW,GAAG,MAAM,CAAC,OAAO;iBAC/B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,WAAW;gBACpB,GAAG,EAAE,MAAM;aACZ,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YAErD,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,IAAI,KAAK,CAAC,oBAAoB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YACvD,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAOO,eAAe,CAAC,MAAW;QACjC,MAAM,KAAK,GAAiC,EAAE,CAAC;QAE/C,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,EAAE,CAAC;YAClE,MAAM,CAAC,GAAG,IAAW,CAAC;YAGtB,IAAI,OAAqB,CAAC;YAE1B,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;gBACf,KAAK,QAAQ;oBACX,OAAO,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;oBACrB,MAAM;gBACR,KAAK,QAAQ;oBACX,OAAO,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC;oBACrB,MAAM;gBACR,KAAK,SAAS;oBACZ,OAAO,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC;oBACtB,MAAM;gBACR,KAAK,OAAO;oBACV,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;oBAC3B,MAAM;gBACR,KAAK,QAAQ;oBACX,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;oBAC5B,MAAM;gBACR;oBACE,OAAO,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YACtB,CAAC;YAGD,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;gBAClB,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;YAC5C,CAAC;YAGD,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBACpC,OAAO,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YAC/B,CAAC;YAED,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC;QACvB,CAAC;QAED,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACzB,CAAC;CACF"}

package/dist/src/mcp/types.d.ts

@@ -0,0 +1,42 @@
+export interface MCPServerConfig {
+    command: string;
+    args: string[];
+    env?: Record<string, string>;
+}
+export interface MCPConfig {
+    servers: Record<string, MCPServerConfig>;
+}
+export interface MCPToolDefinition {
+    name: string;
+    description: string;
+    inputSchema: {
+        type: 'object';
+        properties: Record<string, any>;
+        required?: string[];
+    };
+}
+export interface MCPRequest {
+    jsonrpc: '2.0';
+    id: number | string;
+    method: string;
+    params?: any;
+}
+export interface MCPResponse {
+    jsonrpc: '2.0';
+    id: number | string;
+    result?: any;
+    error?: {
+        code: number;
+        message: string;
+        data?: any;
+    };
+}
+export interface MCPToolResult {
+    content: Array<{
+        type: 'text' | 'image' | 'resource';
+        text?: string;
+        data?: string;
+        uri?: string;
+    }>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/mcp/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/mcp/types.ts"],"names":[],"mappings":"AAOA,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAC9B;AAKD,MAAM,WAAW,SAAS;IACxB,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;CAC1C;AAKD,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE;QACX,IAAI,EAAE,QAAQ,CAAC;QACf,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAChC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;KACrB,CAAC;CACH;AAKD,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,KAAK,CAAC;IACf,EAAE,EAAE,MAAM,GAAG,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,GAAG,CAAC;CACd;AAKD,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,KAAK,CAAC;IACf,EAAE,EAAE,MAAM,GAAG,MAAM,CAAC;IACpB,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,KAAK,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;QACb,OAAO,EAAE,MAAM,CAAC;QAChB,IAAI,CAAC,EAAE,GAAG,CAAC;KACZ,CAAC;CACH;AAKD,MAAM,WAAW,aAAa;IAC5B,OAAO,EAAE,KAAK,CAAC;QACb,IAAI,EAAE,MAAM,GAAG,OAAO,GAAG,UAAU,CAAC;QACpC,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,GAAG,CAAC,EAAE,MAAM,CAAC;KACd,CAAC,CAAC;CACJ"}

package/dist/src/mcp/types.js

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

package/dist/src/mcp/types.js.map

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

package/dist/src/optimization/CompactionService.d.ts

@@ -0,0 +1,10 @@
+import type { CompactionResult } from './types';
+import type { ModelService } from '../services/ModelService';
+export declare class CompactionService {
+    private modelService;
+    constructor(modelService: ModelService);
+    compact(messages: any[]): Promise<CompactionResult>;
+    private normalizeMessages;
+    private generateSummary;
+}
+//# sourceMappingURL=CompactionService.d.ts.map

package/dist/src/optimization/CompactionService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CompactionService.d.ts","sourceRoot":"","sources":["../../../src/optimization/CompactionService.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAChD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAW7D,qBAAa,iBAAiB;IAChB,OAAO,CAAC,YAAY;gBAAZ,YAAY,EAAE,YAAY;IAKxC,OAAO,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC,gBAAgB,CAAC;IA0BzD,OAAO,CAAC,iBAAiB;YAWX,eAAe;CAmB9B"}

package/dist/src/optimization/CompactionService.js

@@ -0,0 +1,82 @@
+import { countTotalTokens } from '../utils/tokenCounter';
+export class CompactionService {
+    modelService;
+    constructor(modelService) {
+        this.modelService = modelService;
+    }
+    async compact(messages) {
+        const originalTokens = countTotalTokens(messages);
+        const normalizedMessages = this.normalizeMessages(messages);
+        const summary = await this.generateSummary(normalizedMessages);
+        const compactedTokens = countTotalTokens([{ role: 'user', content: summary }]);
+        console.log(`[Compaction] Compressed ${originalTokens} → ${compactedTokens} tokens (${Math.round((1 - compactedTokens / originalTokens) * 100)}% reduction)`);
+        return {
+            compacted: true,
+            originalTokens,
+            compactedTokens,
+            summary,
+        };
+    }
+    normalizeMessages(messages) {
+        return messages.map((msg) => ({
+            role: msg.role,
+            content: msg.content,
+        }));
+    }
+    async generateSummary(messages) {
+        const response = await this.modelService.chatWithMessages([
+            {
+                role: 'system',
+                content: COMPACT_SYSTEM_PROMPT,
+            },
+            ...messages,
+            {
+                role: 'user',
+                content: COMPACT_USER_PROMPT,
+            },
+        ]);
+        if (!response || !response.content || response.content.trim() === '') {
+            throw new Error('Failed to compact: received empty summary from model');
+        }
+        return response.content;
+    }
+}
+const COMPACT_USER_PROMPT = `
+Provide a detailed but concise summary of our conversation above. Focus on information that would be helpful for continuing the conversation, including what we did, what we're doing, which files we're working on, and what we're going to do next.
+`;
+const COMPACT_SYSTEM_PROMPT = `
+You are a helpful AI assistant tasked with summarizing conversations.
+
+When the conversation history grows too large, you will be invoked to distill the entire history into a concise, structured XML snapshot. This snapshot is CRITICAL, as it will become the agent's *only* memory of the past. The agent will resume its work based solely on this snapshot. All crucial details, plans, errors, and user directives MUST be preserved.
+
+First, you will think through the entire history. Review the user's overall goal, the agent's actions, tool outputs, file modifications, and any unresolved questions. Identify every piece of information that is essential for future actions.
+
+After your reasoning is complete, generate the final <context_summary> XML object. Be incredibly dense with information. Omit any irrelevant conversational filler.
+
+The structure MUST be as follows:
+
+<context_summary>
+  <conversation_overview>
+    <!-- Single paragraph overview of the entire conversation -->
+  </conversation_overview>
+
+  <key_knowledge>
+    <!-- Crucial facts, conventions, and constraints the agent must remember -->
+  </key_knowledge>
+
+  <file_system_state>
+    <!-- List files that have been created, read, modified, or deleted -->
+  </file_system_state>
+
+  <recent_actions>
+    <!-- A summary of the last few significant agent actions and their outcomes -->
+  </recent_actions>
+
+  <current_plan>
+    <!-- The agent's step-by-step plan. Mark completed steps. -->
+  </current_plan>
+</context_summary>
+
+Remember: This summary will serve as the foundation for continuing the conversation and implementation. Ensure all critical information is preserved while maintaining clarity and conciseness.
+`;
+//# sourceMappingURL=CompactionService.js.map

package/dist/src/optimization/CompactionService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CompactionService.js","sourceRoot":"","sources":["../../../src/optimization/CompactionService.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAUzD,MAAM,OAAO,iBAAiB;IACR;IAApB,YAAoB,YAA0B;QAA1B,iBAAY,GAAZ,YAAY,CAAc;IAAG,CAAC;IAKlD,KAAK,CAAC,OAAO,CAAC,QAAe;QAC3B,MAAM,cAAc,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAGlD,MAAM,kBAAkB,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAG5D,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;QAE/D,MAAM,eAAe,GAAG,gBAAgB,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;QAE/E,OAAO,CAAC,GAAG,CACT,2BAA2B,cAAc,MAAM,eAAe,YAAY,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,eAAe,GAAG,cAAc,CAAC,GAAG,GAAG,CAAC,cAAc,CACjJ,CAAC;QAEF,OAAO;YACL,SAAS,EAAE,IAAI;YACf,cAAc;YACd,eAAe;YACf,OAAO;SACR,CAAC;IACJ,CAAC;IAKO,iBAAiB,CAAC,QAAe;QAEvC,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YAC5B,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,OAAO,EAAE,GAAG,CAAC,OAAO;SACrB,CAAC,CAAC,CAAC;IACN,CAAC;IAKO,KAAK,CAAC,eAAe,CAAC,QAAe;QAC3C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC;YACxD;gBACE,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,qBAAqB;aAC/B;YACD,GAAG,QAAQ;YACX;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,mBAAmB;aAC7B;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACrE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,QAAQ,CAAC,OAAO,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,mBAAmB,GAAG;;CAE3B,CAAC;AAEF,MAAM,qBAAqB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkC7B,CAAC"}

package/dist/src/optimization/CompressionService.d.ts

@@ -0,0 +1,18 @@
+import type { CompressionConfig, TokenUsage, ModelLimit, PruneResult, CompactionResult } from './types';
+import { CompactionService } from './CompactionService';
+export declare class CompressionService {
+    private pruningService;
+    private compactionService;
+    private config;
+    constructor(compactionService: CompactionService, config?: Partial<CompressionConfig>);
+    private mergeConfig;
+    isOverflow(tokens: TokenUsage, modelLimit: ModelLimit): boolean;
+    optimize(messages: any[]): Promise<{
+        messages: any[];
+        pruneResult: PruneResult;
+        compactionResult?: CompactionResult;
+    }>;
+    getConfig(): CompressionConfig;
+    updateConfig(config: Partial<CompressionConfig>): void;
+}
+//# sourceMappingURL=CompressionService.d.ts.map

package/dist/src/optimization/CompressionService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CompressionService.d.ts","sourceRoot":"","sources":["../../../src/optimization/CompressionService.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,iBAAiB,EACjB,UAAU,EACV,UAAU,EACV,WAAW,EACX,gBAAgB,EACjB,MAAM,SAAS,CAAC;AAEjB,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAWxD,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,cAAc,CAAiB;IACvC,OAAO,CAAC,iBAAiB,CAAoB;IAC7C,OAAO,CAAC,MAAM,CAAoB;gBAEtB,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,iBAAiB,CAAC;IASrF,OAAO,CAAC,WAAW;IAgCnB,UAAU,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,EAAE,UAAU,GAAG,OAAO;IA8BzD,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE,GAAG,OAAO,CAAC;QACvC,QAAQ,EAAE,GAAG,EAAE,CAAC;QAChB,WAAW,EAAE,WAAW,CAAC;QACzB,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;KACrC,CAAC;IAkDF,SAAS,IAAI,iBAAiB;IAO9B,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,iBAAiB,CAAC,GAAG,IAAI;CAGvD"}

package/dist/src/optimization/CompressionService.js

@@ -0,0 +1,95 @@
+import { PruningService } from './PruningService';
+import { countTotalTokens } from '../utils/tokenCounter';
+export class CompressionService {
+    pruningService;
+    compactionService;
+    config;
+    constructor(compactionService, config) {
+        this.pruningService = new PruningService();
+        this.compactionService = compactionService;
+        this.config = this.mergeConfig(config);
+    }
+    mergeConfig(config) {
+        const DEFAULT_CONFIG = {
+            compaction: {
+                auto: true,
+                outputTokenMax: 8192,
+                autoContinue: true,
+                triggerRatio: 0.7,
+            },
+            pruning: {
+                enabled: true,
+                protectThreshold: 10000,
+                minimumPrune: 1000,
+                protectedTools: ['read_file', 'list_files'],
+                protectTurns: 2,
+            },
+        };
+        return {
+            compaction: {
+                ...DEFAULT_CONFIG.compaction,
+                ...config?.compaction,
+            },
+            pruning: {
+                ...DEFAULT_CONFIG.pruning,
+                ...config?.pruning,
+            },
+        };
+    }
+    isOverflow(tokens, modelLimit) {
+        if (!this.config.compaction.auto) {
+            return false;
+        }
+        const context = modelLimit.context;
+        if (context === 0) {
+            return false;
+        }
+        const currentInputTokens = tokens.input + (tokens.cacheRead || 0);
+        const compressionThreshold = context * this.config.compaction.triggerRatio;
+        const overflow = currentInputTokens > compressionThreshold;
+        console.log(`[Compression] currentInputTokens=${currentInputTokens}, ` +
+            `context=${context}, triggerRatio=${this.config.compaction.triggerRatio}, ` +
+            `threshold=${compressionThreshold}, overflow=${overflow}`);
+        return overflow;
+    }
+    async optimize(messages) {
+        console.log('[Compression] Starting optimization...');
+        const pruneResult = this.pruningService.prune(messages, this.config);
+        if (pruneResult.pruned) {
+            console.log(`[Compression] Pruned ${pruneResult.prunedCount} outputs, ` +
+                `saved ~${pruneResult.prunedTokens} tokens`);
+        }
+        const tokensAfterPrune = countTotalTokens(messages);
+        const needsCompaction = tokensAfterPrune > 50000;
+        if (needsCompaction && this.config.compaction.auto) {
+            console.log('[Compression] Still need compaction, compressing...');
+            const compactionResult = await this.compactionService.compact(messages);
+            const compactedMessages = [
+                {
+                    role: 'user',
+                    content: compactionResult.summary,
+                    type: 'message',
+                    timestamp: new Date().toISOString(),
+                    uuid: 'compacted-summary',
+                    parentUuid: null,
+                },
+            ];
+            return {
+                messages: compactedMessages,
+                pruneResult,
+                compactionResult,
+            };
+        }
+        return {
+            messages,
+            pruneResult,
+        };
+    }
+    getConfig() {
+        return this.config;
+    }
+    updateConfig(config) {
+        this.config = this.mergeConfig(config);
+    }
+}
+//# sourceMappingURL=CompressionService.js.map

package/dist/src/optimization/CompressionService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CompressionService.js","sourceRoot":"","sources":["../../../src/optimization/CompressionService.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAUzD,MAAM,OAAO,kBAAkB;IACrB,cAAc,CAAiB;IAC/B,iBAAiB,CAAoB;IACrC,MAAM,CAAoB;IAElC,YAAY,iBAAoC,EAAE,MAAmC;QACnF,IAAI,CAAC,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC;QAC3C,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAC;QAC3C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IAKO,WAAW,CAAC,MAAmC;QACrD,MAAM,cAAc,GAAsB;YACxC,UAAU,EAAE;gBACV,IAAI,EAAE,IAAI;gBACV,cAAc,EAAE,IAAI;gBACpB,YAAY,EAAE,IAAI;gBAClB,YAAY,EAAE,GAAG;aAClB;YACD,OAAO,EAAE;gBACP,OAAO,EAAE,IAAI;gBACb,gBAAgB,EAAE,KAAK;gBACvB,YAAY,EAAE,IAAI;gBAClB,cAAc,EAAE,CAAC,WAAW,EAAE,YAAY,CAAC;gBAC3C,YAAY,EAAE,CAAC;aAChB;SACF,CAAC;QAEF,OAAO;YACL,UAAU,EAAE;gBACV,GAAG,cAAc,CAAC,UAAU;gBAC5B,GAAG,MAAM,EAAE,UAAU;aACtB;YACD,OAAO,EAAE;gBACP,GAAG,cAAc,CAAC,OAAO;gBACzB,GAAG,MAAM,EAAE,OAAO;aACnB;SACF,CAAC;IACJ,CAAC;IAKD,UAAU,CAAC,MAAkB,EAAE,UAAsB;QACnD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YACjC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;QACnC,IAAI,OAAO,KAAK,CAAC,EAAE,CAAC;YAClB,OAAO,KAAK,CAAC;QACf,CAAC;QAGD,MAAM,kBAAkB,GAAG,MAAM,CAAC,KAAK,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,CAAC,CAAC;QAGlE,MAAM,oBAAoB,GAAG,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,CAAC;QAE3E,MAAM,QAAQ,GAAG,kBAAkB,GAAG,oBAAoB,CAAC;QAE3D,OAAO,CAAC,GAAG,CACT,oCAAoC,kBAAkB,IAAI;YACxD,WAAW,OAAO,kBAAkB,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,YAAY,IAAI;YAC3E,aAAa,oBAAoB,cAAc,QAAQ,EAAE,CAC5D,CAAC;QAEF,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKD,KAAK,CAAC,QAAQ,CAAC,QAAe;QAK5B,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;QAGtD,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAErE,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO,CAAC,GAAG,CACT,wBAAwB,WAAW,CAAC,WAAW,YAAY;gBACzD,UAAU,WAAW,CAAC,YAAY,SAAS,CAC9C,CAAC;QACJ,CAAC;QAGD,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACpD,MAAM,eAAe,GAAG,gBAAgB,GAAG,KAAK,CAAC;QAEjD,IAAI,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,qDAAqD,CAAC,CAAC;YAEnE,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAGxE,MAAM,iBAAiB,GAAG;gBACxB;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,gBAAgB,CAAC,OAAO;oBACjC,IAAI,EAAE,SAAS;oBACf,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,IAAI,EAAE,mBAAmB;oBACzB,UAAU,EAAE,IAAI;iBACjB;aACF,CAAC;YAEF,OAAO;gBACL,QAAQ,EAAE,iBAAiB;gBAC3B,WAAW;gBACX,gBAAgB;aACjB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,QAAQ;YACR,WAAW;SACZ,CAAC;IACJ,CAAC;IAKD,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAKD,YAAY,CAAC,MAAkC;QAC7C,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;CACF"}

package/dist/src/optimization/PerformanceMonitor.d.ts

@@ -0,0 +1,22 @@
+import type { PerformanceMetrics } from './types';
+export declare class PerformanceMonitor {
+    private metrics;
+    private responseTimes;
+    recordRequest(data: {
+        inputTokens: number;
+        outputTokens: number;
+        cacheReadTokens?: number;
+        cacheWriteTokens?: number;
+        responseTime: number;
+    }): void;
+    recordOptimization(data: {
+        pruned: boolean;
+        prunedTokens?: number;
+        compacted: boolean;
+        compactedTokens?: number;
+    }): void;
+    getMetrics(): PerformanceMetrics;
+    generateReport(): string;
+    reset(): void;
+}
+//# sourceMappingURL=PerformanceMonitor.d.ts.map

package/dist/src/optimization/PerformanceMonitor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PerformanceMonitor.d.ts","sourceRoot":"","sources":["../../../src/optimization/PerformanceMonitor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,SAAS,CAAC;AAWlD,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,OAAO,CAUb;IAEF,OAAO,CAAC,aAAa,CAAgB;IAKrC,aAAa,CAAC,IAAI,EAAE;QAClB,WAAW,EAAE,MAAM,CAAC;QACpB,YAAY,EAAE,MAAM,CAAC;QACrB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,GAAG,IAAI;IAwBR,kBAAkB,CAAC,IAAI,EAAE;QACvB,MAAM,EAAE,OAAO,CAAC;QAChB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,SAAS,EAAE,OAAO,CAAC;QACnB,eAAe,CAAC,EAAE,MAAM,CAAC;KAC1B,GAAG,IAAI;IAaR,UAAU,IAAI,kBAAkB;IAOhC,cAAc,IAAI,MAAM;IA0CxB,KAAK,IAAI,IAAI;CAcd"}