@mcp-ts/sdk 1.0.0 → 1.0.1

package/dist/adapters/agui-adapter.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/agui-adapter.ts"],"names":["result"],"mappings":";AAkFO,IAAM,cAAN,MAAkB;AAAA,EACrB,WAAA,CACY,MAAA,EACA,OAAA,GAA8B,EAAC,EACzC;AAFU,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYJ,MAAM,QAAA,GAAgC;AAClC,IAAA,MAAM,cAAA,GAAiB,OAAQ,IAAA,CAAK,MAAA,CAAe,UAAA,KAAe,UAAA;AAElE,IAAA,IAAI,cAAA,EAAgB;AAChB,MAAA,MAAM,OAAA,GAAW,IAAA,CAAK,MAAA,CAA8B,UAAA,EAAW;AAC/D,MAAA,MAAM,WAAuB,EAAC;AAE9B,MAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC1B,QAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,cAAA,CAAe,MAAM,CAAA;AAC9C,QAAA,QAAA,CAAS,IAAA,CAAK,GAAG,KAAK,CAAA;AAAA,MAC1B;AAEA,MAAA,OAAO,QAAA;AAAA,IACX;AAEA,IAAA,OAAO,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,MAAmB,CAAA;AAAA,EACvD;AAAA,EAEA,MAAc,eAAe,MAAA,EAAwC;AACjE,IAAA,IAAI,CAAC,MAAA,CAAO,WAAA,EAAY,EAAG;AACvB,MAAA,OAAO,EAAC;AAAA,IACZ;AAEA,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AACtC,IAAA,MAAM,SAAS,IAAA,CAAK,OAAA,CAAQ,MAAA,IAAU,MAAA,CAAO,aAAY,IAAK,KAAA;AAC9D,IAAA,MAAM,QAAoB,EAAC;AAE3B,IAAA,KAAA,MAAW,IAAA,IAAQ,OAAO,KAAA,EAAO;AAC7B,MAAA,MAAM,QAAA,GAAW,CAAA,EAAG,MAAM,CAAA,CAAA,EAAI,KAAK,IAAI,CAAA,CAAA;AAEvC,MAAA,KAAA,CAAM,IAAA,CAAK;AAAA,QACP,IAAA,EAAM,QAAA;AAAA,QACN,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,QAAA,EAAW,KAAK,IAAI,CAAA,CAAA;AAAA,QACrD,UAAA,EAAY,KAAK,WAAA,IAAe,EAAE,MAAM,QAAA,EAAU,UAAA,EAAY,EAAC,EAAE;AAAA,QACjE,OAAA,EAAS,OAAO,IAAA,KAAc;AAC1B,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,IAAA,CAAK,IAAI,IAAI,IAAI,CAAA;AAClE,UAAA,MAAMA,UAAS,MAAM,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,IAAI,CAAA;AAGpD,UAAA,IAAIA,QAAO,OAAA,IAAW,KAAA,CAAM,OAAA,CAAQA,OAAAA,CAAO,OAAO,CAAA,EAAG;AACjD,YAAA,MAAM,cAAcA,OAAAA,CAAO,OAAA,CACtB,MAAA,CAAO,CAAC,MAAW,CAAA,CAAE,IAAA,KAAS,MAAM,CAAA,CACpC,IAAI,CAAC,CAAA,KAAW,EAAE,IAAI,CAAA,CACtB,KAAK,IAAI,CAAA;AACd,YAAA,OAAO,WAAA,IAAeA,OAAAA;AAAA,UAC1B;AAEA,UAAA,OAAOA,OAAAA;AAAA,QACX;AAAA,OACH,CAAA;AAAA,IACL;AAEA,IAAA,OAAO,KAAA;AAAA,EACX;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,gBAAA,GAA8C;AAC1C,IAAA,OAAO,YAAY,KAAK,QAAA,EAAS;AAAA,EACrC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA,MAAM,kBAAA,GAAoD;AACtD,IAAA,MAAM,cAAA,GAAiB,OAAQ,IAAA,CAAK,MAAA,CAAe,UAAA,KAAe,UAAA;AAElE,IAAA,IAAI,cAAA,EAAgB;AAChB,MAAA,MAAM,OAAA,GAAW,IAAA,CAAK,MAAA,CAA8B,UAAA,EAAW;AAC/D,MAAA,MAAM,WAAiC,EAAC;AAExC,MAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC1B,QAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,wBAAA,CAAyB,MAAM,CAAA;AACxD,QAAA,QAAA,CAAS,IAAA,CAAK,GAAG,KAAK,CAAA;AAAA,MAC1B;AAEA,MAAA,OAAO,QAAA;AAAA,IACX;AAEA,IAAA,OAAO,IAAA,CAAK,wBAAA,CAAyB,IAAA,CAAK,MAAmB,CAAA;AAAA,EACjE;AAAA,EAEA,MAAc,yBAAyB,MAAA,EAAkD;AACrF,IAAA,IAAI,CAAC,MAAA,CAAO,WAAA,EAAY,EAAG;AACvB,MAAA,OAAO,EAAC;AAAA,IACZ;AAEA,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AACtC,IAAA,MAAM,SAAS,IAAA,CAAK,OAAA,CAAQ,MAAA,IAAU,MAAA,CAAO,aAAY,IAAK,KAAA;AAC9D,IAAA,MAAM,QAA8B,EAAC;AAErC,IAAA,KAAA,MAAW,IAAA,IAAQ,OAAO,KAAA,EAAO;AAC7B,MAAA,KAAA,CAAM,IAAA,CAAK;AAAA,QACP,IAAA,EAAM,CAAA,EAAG,MAAM,CAAA,CAAA,EAAI,KAAK,IAAI,CAAA,CAAA;AAAA,QAC5B,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,QAAA,EAAW,KAAK,IAAI,CAAA,CAAA;AAAA,QACrD,UAAA,EAAY,KAAK,WAAA,IAAe,EAAE,MAAM,QAAA,EAAU,UAAA,EAAY,EAAC;AAAE,OACpE,CAAA;AAAA,IACL;AAEA,IAAA,OAAO,KAAA;AAAA,EACX;AACJ","file":"agui-adapter.mjs","sourcesContent":["/**\r\n * MCP Adapter for AG-UI Integration\r\n *\r\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\r\n * It provides tools with handlers for server-side execution and tool definitions\r\n * in JSON Schema format for passing to remote agents.\r\n *\r\n * @example\r\n * ```typescript\r\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\r\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/mcp-adapter';\r\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\r\n * import { HttpAgent } from '@ag-ui/client';\r\n *\r\n * // Create MCP client\r\n * const mcpClient = new MultiSessionClient('user_123');\r\n * await mcpClient.connect();\r\n *\r\n * // Create adapter and get tools\r\n * const adapter = new AguiAdapter(mcpClient);\r\n * const tools = await adapter.getTools();\r\n *\r\n * // Use with AG-UI middleware\r\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\r\n * agent.use(createMcpMiddleware(mcpClient, { tools }));\r\n * ```\r\n */\r\n\r\nimport { MCPClient } from '../server/mcp/oauth-client.js';\r\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\r\n\r\n/**\r\n * Configuration options for AguiAdapter\r\n */\r\nexport interface AguiAdapterOptions {\r\n    /**\r\n     * Prefix for tool names to avoid collision with other tools.\r\n     * @default serverId or 'mcp'\r\n     */\r\n    prefix?: string;\r\n}\r\n\r\n/**\r\n * AG-UI Tool with handler for server-side execution.\r\n *\r\n * Tools contain:\r\n * - `name`: Unique identifier (prefixed with server ID)\r\n * - `description`: Human-readable description for the LLM\r\n * - `parameters`: JSON Schema defining the input format\r\n * - `handler`: Function that executes the tool via MCP client\r\n */\r\nexport interface AguiTool {\r\n    /** Unique tool name (e.g., \"server-abc_get_weather\") */\r\n    name: string;\r\n    /** Human-readable description for the LLM */\r\n    description: string;\r\n    /** JSON Schema format parameters */\r\n    parameters?: Record<string, any>;\r\n    /** Handler function that executes the MCP tool */\r\n    handler?: (args: any) => any | Promise<any>;\r\n}\r\n\r\n/**\r\n * Tool definition format for passing to remote agents (without handler).\r\n * Compatible with OpenAI's function calling API.\r\n */\r\nexport interface AguiToolDefinition {\r\n    /** Tool name (e.g., \"server-abc_get_weather\") */\r\n    name: string;\r\n    /** Human-readable description */\r\n    description: string;\r\n    /** JSON Schema format parameters */\r\n    parameters: Record<string, any>;\r\n}\r\n\r\n/**\r\n * Adapter that transforms MCP tools into AG-UI compatible formats.\r\n *\r\n * This adapter provides two main outputs:\r\n * - `getTools()`: Returns tools with handlers for server-side execution\r\n * - `getToolDefinitions()`: Returns tool definitions in JSON Schema format for remote agents\r\n */\r\nexport class AguiAdapter {\r\n    constructor(\r\n        private client: MCPClient | MultiSessionClient,\r\n        private options: AguiAdapterOptions = {}\r\n    ) { }\r\n\r\n    /**\r\n     * Get tools with handlers for MCP tool execution.\r\n     *\r\n     * Each tool includes a handler function that:\r\n     * 1. Calls the MCP tool via the client\r\n     * 2. Extracts text content from the result\r\n     * 3. Returns the result as a string or JSON\r\n     *\r\n     * @returns Array of AguiTool objects\r\n     */\r\n    async getTools(): Promise<AguiTool[]> {\r\n        const isMultiSession = typeof (this.client as any).getClients === 'function';\r\n\r\n        if (isMultiSession) {\r\n            const clients = (this.client as MultiSessionClient).getClients();\r\n            const allTools: AguiTool[] = [];\r\n\r\n            for (const client of clients) {\r\n                const tools = await this.transformTools(client);\r\n                allTools.push(...tools);\r\n            }\r\n\r\n            return allTools;\r\n        }\r\n\r\n        return this.transformTools(this.client as MCPClient);\r\n    }\r\n\r\n    private async transformTools(client: MCPClient): Promise<AguiTool[]> {\r\n        if (!client.isConnected()) {\r\n            return [];\r\n        }\r\n\r\n        const result = await client.listTools();\r\n        const prefix = this.options.prefix ?? client.getServerId() ?? 'mcp';\r\n        const tools: AguiTool[] = [];\r\n\r\n        for (const tool of result.tools) {\r\n            const toolName = `${prefix}_${tool.name}`;\r\n\r\n            tools.push({\r\n                name: toolName,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: tool.inputSchema || { type: 'object', properties: {} },\r\n                handler: async (args: any) => {\r\n                    console.log(`[AguiAdapter] Executing MCP tool: ${tool.name}`, args);\r\n                    const result = await client.callTool(tool.name, args);\r\n\r\n                    // Extract text content from result\r\n                    if (result.content && Array.isArray(result.content)) {\r\n                        const textContent = result.content\r\n                            .filter((c: any) => c.type === 'text')\r\n                            .map((c: any) => c.text)\r\n                            .join('\\n');\r\n                        return textContent || result;\r\n                    }\r\n\r\n                    return result;\r\n                }\r\n            });\r\n        }\r\n\r\n        return tools;\r\n    }\r\n\r\n    /**\r\n     * Get tools as a function (for dynamic loading).\r\n     *\r\n     * @returns Function that returns a Promise of tools\r\n     */\r\n    getToolsFunction(): () => Promise<AguiTool[]> {\r\n        return async () => this.getTools();\r\n    }\r\n\r\n    /**\r\n     * Get tool definitions in JSON Schema format for passing to remote agents.\r\n     *\r\n     * This format is compatible with:\r\n     * - OpenAI's function calling API\r\n     * - AG-UI input.tools format\r\n     * - Most LLM tool/function calling implementations\r\n     *\r\n     * @returns Array of AguiToolDefinition objects\r\n     */\r\n    async getToolDefinitions(): Promise<AguiToolDefinition[]> {\r\n        const isMultiSession = typeof (this.client as any).getClients === 'function';\r\n\r\n        if (isMultiSession) {\r\n            const clients = (this.client as MultiSessionClient).getClients();\r\n            const allTools: AguiToolDefinition[] = [];\r\n\r\n            for (const client of clients) {\r\n                const tools = await this.transformToolDefinitions(client);\r\n                allTools.push(...tools);\r\n            }\r\n\r\n            return allTools;\r\n        }\r\n\r\n        return this.transformToolDefinitions(this.client as MCPClient);\r\n    }\r\n\r\n    private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\r\n        if (!client.isConnected()) {\r\n            return [];\r\n        }\r\n\r\n        const result = await client.listTools();\r\n        const prefix = this.options.prefix ?? client.getServerId() ?? 'mcp';\r\n        const tools: AguiToolDefinition[] = [];\r\n\r\n        for (const tool of result.tools) {\r\n            tools.push({\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: tool.inputSchema || { type: 'object', properties: {} },\r\n            });\r\n        }\r\n\r\n        return tools;\r\n    }\r\n}\r\n"]}
+{"version":3,"sources":["../../src/adapters/agui-adapter.ts"],"names":["result"],"mappings":";AAwCO,SAAS,YAAY,MAAA,EAA8D;AACtF,EAAA,IAAI,CAAC,MAAA,EAAQ;AACT,IAAA,OAAO,EAAE,IAAA,EAAM,QAAA,EAAU,UAAA,EAAY,EAAC,EAAE;AAAA,EAC5C;AAEA,EAAA,MAAM,OAAA,GAAU,EAAE,GAAG,MAAA,EAAO;AAG5B,EAAA,OAAO,OAAA,CAAQ,OAAA;AACf,EAAA,OAAO,OAAA,CAAQ,GAAA;AACf,EAAA,OAAO,OAAA,CAAQ,QAAA;AACf,EAAA,OAAO,OAAA,CAAQ,KAAA;AACf,EAAA,OAAO,OAAA,CAAQ,WAAA;AAGf,EAAA,IAAI,OAAA,CAAQ,UAAA,IAAc,OAAO,OAAA,CAAQ,eAAe,QAAA,EAAU;AAC9D,IAAA,MAAM,eAAoC,EAAC;AAC3C,IAAA,KAAA,MAAW,CAAC,KAAK,KAAK,CAAA,IAAK,OAAO,OAAA,CAAQ,OAAA,CAAQ,UAAU,CAAA,EAAG;AAC3D,MAAA,IAAI,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,EAAM;AAC7C,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,WAAA,CAAY,KAA4B,CAAA;AAAA,MAChE,CAAA,MAAO;AACH,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,KAAA;AAAA,MACxB;AAAA,IACJ;AACA,IAAA,OAAA,CAAQ,UAAA,GAAa,YAAA;AAAA,EACzB;AAGA,EAAA,IAAI,OAAA,CAAQ,KAAA,IAAS,OAAO,OAAA,CAAQ,UAAU,QAAA,EAAU;AACpD,IAAA,OAAA,CAAQ,KAAA,GAAQ,WAAA,CAAY,OAAA,CAAQ,KAAK,CAAA;AAAA,EAC7C;AAGA,EAAA,IAAI,OAAA,CAAQ,oBAAA,IAAwB,OAAO,OAAA,CAAQ,yBAAyB,QAAA,EAAU;AAClF,IAAA,OAAA,CAAQ,oBAAA,GAAuB,WAAA,CAAY,OAAA,CAAQ,oBAAoB,CAAA;AAAA,EAC3E;AAEA,EAAA,OAAO,OAAA;AACX;AAmCO,IAAM,cAAN,MAAkB;AAAA,EACrB,WAAA,CACY,MAAA,EACA,OAAA,GAA8B,EAAC,EACzC;AAFU,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKJ,MAAM,QAAA,GAAgC;AAClC,IAAA,IAAI,IAAA,CAAK,gBAAe,EAAG;AACvB,MAAA,MAAM,OAAA,GAAW,IAAA,CAAK,MAAA,CAA8B,UAAA,EAAW;AAC/D,MAAA,MAAM,WAAuB,EAAC;AAC9B,MAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC1B,QAAA,QAAA,CAAS,KAAK,GAAG,MAAM,IAAA,CAAK,cAAA,CAAe,MAAM,CAAC,CAAA;AAAA,MACtD;AACA,MAAA,OAAO,QAAA;AAAA,IACX;AACA,IAAA,OAAO,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,MAAmB,CAAA;AAAA,EACvD;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,kBAAA,GAAoD;AACtD,IAAA,IAAI,IAAA,CAAK,gBAAe,EAAG;AACvB,MAAA,MAAM,OAAA,GAAW,IAAA,CAAK,MAAA,CAA8B,UAAA,EAAW;AAC/D,MAAA,MAAM,WAAiC,EAAC;AACxC,MAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC1B,QAAA,QAAA,CAAS,KAAK,GAAG,MAAM,IAAA,CAAK,wBAAA,CAAyB,MAAM,CAAC,CAAA;AAAA,MAChE;AACA,MAAA,OAAO,QAAA;AAAA,IACX;AACA,IAAA,OAAO,IAAA,CAAK,wBAAA,CAAyB,IAAA,CAAK,MAAmB,CAAA;AAAA,EACjE;AAAA;AAAA;AAAA;AAAA,EAKA,gBAAA,GAA8C;AAC1C,IAAA,OAAO,MAAM,KAAK,QAAA,EAAS;AAAA,EAC/B;AAAA,EAEQ,cAAA,GAA0B;AAC9B,IAAA,OAAO,OAAQ,IAAA,CAAK,MAAA,CAAe,UAAA,KAAe,UAAA;AAAA,EACtD;AAAA,EAEA,MAAc,eAAe,MAAA,EAAwC;AACjE,IAAA,IAAI,CAAC,MAAA,CAAO,WAAA,EAAY,SAAU,EAAC;AAEnC,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AACtC,IAAA,MAAM,MAAA,GAAS,KAAK,OAAA,CAAQ,MAAA,IAAU,QAAQ,MAAA,CAAO,WAAA,MAAiB,KAAK,CAAA,CAAA;AAE3E,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,MAAS;AAAA,MAC7B,IAAA,EAAM,CAAA,EAAG,MAAM,CAAA,CAAA,EAAI,KAAK,IAAI,CAAA,CAAA;AAAA,MAC5B,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,QAAA,EAAW,KAAK,IAAI,CAAA,CAAA;AAAA,MACrD,UAAA,EAAY,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,MACxC,OAAA,EAAS,OAAO,IAAA,KAAc;AAC1B,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,IAAA,CAAK,IAAI,IAAI,IAAI,CAAA;AAClE,QAAA,MAAMA,UAAS,MAAM,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,IAAI,CAAA;AAEpD,QAAA,IAAIA,QAAO,OAAA,IAAW,KAAA,CAAM,OAAA,CAAQA,OAAAA,CAAO,OAAO,CAAA,EAAG;AACjD,UAAA,MAAM,cAAcA,OAAAA,CAAO,OAAA,CACtB,MAAA,CAAO,CAAC,MAAW,CAAA,CAAE,IAAA,KAAS,MAAM,CAAA,CACpC,IAAI,CAAC,CAAA,KAAW,EAAE,IAAI,CAAA,CACtB,KAAK,IAAI,CAAA;AACd,UAAA,OAAO,WAAA,IAAeA,OAAAA;AAAA,QAC1B;AACA,QAAA,OAAOA,OAAAA;AAAA,MACX;AAAA,KACJ,CAAE,CAAA;AAAA,EACN;AAAA,EAEA,MAAc,yBAAyB,MAAA,EAAkD;AACrF,IAAA,IAAI,CAAC,MAAA,CAAO,WAAA,EAAY,SAAU,EAAC;AAEnC,IAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AACtC,IAAA,MAAM,MAAA,GAAS,KAAK,OAAA,CAAQ,MAAA,IAAU,QAAQ,MAAA,CAAO,WAAA,MAAiB,KAAK,CAAA,CAAA;AAE3E,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,MAAS;AAAA,MAC7B,IAAA,EAAM,CAAA,EAAG,MAAM,CAAA,CAAA,EAAI,KAAK,IAAI,CAAA,CAAA;AAAA,MAC5B,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,QAAA,EAAW,KAAK,IAAI,CAAA,CAAA;AAAA,MACrD,UAAA,EAAY,WAAA,CAAY,IAAA,CAAK,WAAW;AAAA,KAC5C,CAAE,CAAA;AAAA,EACN;AACJ","file":"agui-adapter.mjs","sourcesContent":["/**\r\n * MCP Adapter for AG-UI Integration\r\n *\r\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\r\n * It provides tools with handlers for server-side execution and tool definitions\r\n * in JSON Schema format for passing to remote agents.\r\n *\r\n * @example\r\n * ```typescript\r\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\r\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\r\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\r\n * import { HttpAgent } from '@ag-ui/client';\r\n *\r\n * // Create MCP client\r\n * const mcpClient = new MultiSessionClient('user_123');\r\n * await mcpClient.connect();\r\n *\r\n * // Create adapter and get tools\r\n * const adapter = new AguiAdapter(mcpClient);\r\n * const tools = await adapter.getTools();\r\n *\r\n * // Use with AG-UI middleware\r\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\r\n * agent.use(createMcpMiddleware({ tools }));\r\n * ```\r\n */\r\n\r\nimport { MCPClient } from '../server/mcp/oauth-client.js';\r\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\r\n\r\n/**\r\n * Cleans a JSON Schema by removing meta-properties that cause issues with\r\n * strict Pydantic validation (e.g., Google ADK).\r\n *\r\n * Removes: $schema, $id, $comment, $defs, definitions\r\n *\r\n * @param schema - The JSON Schema to clean\r\n * @returns Cleaned schema without meta-properties\r\n */\r\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\r\n    if (!schema) {\r\n        return { type: 'object', properties: {} };\r\n    }\r\n\r\n    const cleaned = { ...schema };\r\n\r\n    // Remove JSON Schema meta-properties that cause Pydantic validation errors\r\n    delete cleaned.$schema;\r\n    delete cleaned.$id;\r\n    delete cleaned.$comment;\r\n    delete cleaned.$defs;\r\n    delete cleaned.definitions;\r\n\r\n    // Recursively clean nested properties\r\n    if (cleaned.properties && typeof cleaned.properties === 'object') {\r\n        const cleanedProps: Record<string, any> = {};\r\n        for (const [key, value] of Object.entries(cleaned.properties)) {\r\n            if (typeof value === 'object' && value !== null) {\r\n                cleanedProps[key] = cleanSchema(value as Record<string, any>);\r\n            } else {\r\n                cleanedProps[key] = value;\r\n            }\r\n        }\r\n        cleaned.properties = cleanedProps;\r\n    }\r\n\r\n    // Clean items if it's an array schema\r\n    if (cleaned.items && typeof cleaned.items === 'object') {\r\n        cleaned.items = cleanSchema(cleaned.items);\r\n    }\r\n\r\n    // Clean additionalProperties if it's an object schema\r\n    if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\r\n        cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\r\n    }\r\n\r\n    return cleaned;\r\n}\r\n\r\n/**\r\n * Configuration options for AguiAdapter\r\n */\r\nexport interface AguiAdapterOptions {\r\n    /**\r\n     * Prefix for tool names to avoid collision with other tools.\r\n     * @default serverId or 'mcp'\r\n     */\r\n    prefix?: string;\r\n}\r\n\r\n/**\r\n * AG-UI Tool with handler for server-side execution.\r\n */\r\nexport interface AguiTool {\r\n    name: string;\r\n    description: string;\r\n    parameters?: Record<string, any>;\r\n    handler?: (args: any) => any | Promise<any>;\r\n}\r\n\r\n/**\r\n * Tool definition format for passing to remote agents (without handler).\r\n */\r\nexport interface AguiToolDefinition {\r\n    name: string;\r\n    description: string;\r\n    parameters: Record<string, any>;\r\n}\r\n\r\n/**\r\n * Adapter that transforms MCP tools into AG-UI compatible formats.\r\n */\r\nexport class AguiAdapter {\r\n    constructor(\r\n        private client: MCPClient | MultiSessionClient,\r\n        private options: AguiAdapterOptions = {}\r\n    ) { }\r\n\r\n    /**\r\n     * Get tools with handlers for MCP tool execution.\r\n     */\r\n    async getTools(): Promise<AguiTool[]> {\r\n        if (this.isMultiSession()) {\r\n            const clients = (this.client as MultiSessionClient).getClients();\r\n            const allTools: AguiTool[] = [];\r\n            for (const client of clients) {\r\n                allTools.push(...await this.transformTools(client));\r\n            }\r\n            return allTools;\r\n        }\r\n        return this.transformTools(this.client as MCPClient);\r\n    }\r\n\r\n    /**\r\n     * Get tool definitions in JSON Schema format for passing to remote agents.\r\n     */\r\n    async getToolDefinitions(): Promise<AguiToolDefinition[]> {\r\n        if (this.isMultiSession()) {\r\n            const clients = (this.client as MultiSessionClient).getClients();\r\n            const allTools: AguiToolDefinition[] = [];\r\n            for (const client of clients) {\r\n                allTools.push(...await this.transformToolDefinitions(client));\r\n            }\r\n            return allTools;\r\n        }\r\n        return this.transformToolDefinitions(this.client as MCPClient);\r\n    }\r\n\r\n    /**\r\n     * Get tools as a function (for dynamic loading).\r\n     */\r\n    getToolsFunction(): () => Promise<AguiTool[]> {\r\n        return () => this.getTools();\r\n    }\r\n\r\n    private isMultiSession(): boolean {\r\n        return typeof (this.client as any).getClients === 'function';\r\n    }\r\n\r\n    private async transformTools(client: MCPClient): Promise<AguiTool[]> {\r\n        if (!client.isConnected()) return [];\r\n\r\n        const result = await client.listTools();\r\n        const prefix = this.options.prefix ?? `tool_${client.getServerId() ?? 'mcp'}`;\r\n\r\n        return result.tools.map(tool => ({\r\n            name: `${prefix}_${tool.name}`,\r\n            description: tool.description || `Execute ${tool.name}`,\r\n            parameters: cleanSchema(tool.inputSchema),\r\n            handler: async (args: any) => {\r\n                console.log(`[AguiAdapter] Executing MCP tool: ${tool.name}`, args);\r\n                const result = await client.callTool(tool.name, args);\r\n\r\n                if (result.content && Array.isArray(result.content)) {\r\n                    const textContent = result.content\r\n                        .filter((c: any) => c.type === 'text')\r\n                        .map((c: any) => c.text)\r\n                        .join('\\n');\r\n                    return textContent || result;\r\n                }\r\n                return result;\r\n            }\r\n        }));\r\n    }\r\n\r\n    private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\r\n        if (!client.isConnected()) return [];\r\n\r\n        const result = await client.listTools();\r\n        const prefix = this.options.prefix ?? `tool_${client.getServerId() ?? 'mcp'}`;\r\n\r\n        return result.tools.map(tool => ({\r\n            name: `${prefix}_${tool.name}`,\r\n            description: tool.description || `Execute ${tool.name}`,\r\n            parameters: cleanSchema(tool.inputSchema),\r\n        }));\r\n    }\r\n}\r\n"]}

package/dist/adapters/agui-middleware.d.mts

@@ -1,8 +1,8 @@
 import { Observable } from 'rxjs';
 import { Middleware, RunAgentInput, AbstractAgent, BaseEvent } from '@ag-ui/client';
-export { AbstractAgent, BaseEvent, EventType, Middleware, RunAgentInput, ToolCallEndEvent } from '@ag-ui/client';
-import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-DMF3ED2O.mjs';
+export { AbstractAgent, BaseEvent, EventType, Middleware, RunAgentInput, Tool, ToolCallEndEvent } from '@ag-ui/client';
 import { AguiTool } from './agui-adapter.mjs';
+import '../multi-session-client-DMF3ED2O.mjs';
 import '../events-BP6WyRNh.mjs';
 import '@modelcontextprotocol/sdk/types.js';
 import '@modelcontextprotocol/sdk/shared/auth.js';
@@ -11,161 +11,49 @@ import '@modelcontextprotocol/sdk/client/auth.js';
 /**
  * AG-UI Middleware for MCP Tool Execution
  *
- * This middleware intercepts tool calls from remote agents (e.g., LangGraph, AutoGen)
- * and executes MCP tools server-side, returning results back to the agent.
+ * This middleware intercepts tool calls from remote agents and executes
+ * MCP tools server-side, returning results back to the agent.
  *
- * ## How It Works
- *
- * 1. **Tool Injection**: When a run starts, the middleware injects MCP tool definitions
- *    into `input.tools` so the remote agent knows about available MCP tools.
- *
- * 2. **Event Interception**: The middleware subscribes to the agent's event stream and
- *    tracks tool calls using AG-UI events:
- *    - `TOOL_CALL_START`: Records tool name and ID
- *    - `TOOL_CALL_ARGS`: Accumulates streamed arguments
- *    - `TOOL_CALL_END`: Marks tool call as complete
- *    - `RUN_FINISHED`: Triggers execution of pending MCP tools
- *
- * 3. **Server-Side Execution**: When `RUN_FINISHED` arrives with pending MCP tool calls,
- *    the middleware:
- *    - Executes each MCP tool via the MCP client
- *    - Emits `TOOL_CALL_RESULT` events with the results
- *    - Adds results to `input.messages` for context
- *    - Emits `RUN_FINISHED` to close the current run
- *    - Triggers a new run so the agent can process tool results
- *
- * 4. **Recursive Processing**: If the new run makes more MCP tool calls, the cycle
- *    repeats until the agent completes without pending MCP calls.
- *
- * ## Tool Identification
- *
- * MCP tools are identified by a configurable prefix (default: `server-`).
- * Tools not matching this prefix are passed through without interception.
- *
- * @requires @ag-ui/client - This middleware requires @ag-ui/client as a peer dependency
+ * @requires @ag-ui/client - Peer dependency for AG-UI types
  * @requires rxjs - Uses RxJS Observables for event streaming
- *
- * @example
- * ```typescript
- * import { HttpAgent } from '@ag-ui/client';
- * import { McpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';
- * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';
- *
- * // Create MCP client and adapter
- * const mcpClient = new MultiSessionClient('user_123');
- * await mcpClient.connect();
- *
- * const adapter = new AguiAdapter(mcpClient);
- * const actions = await adapter.getActions();
- *
- * // Create middleware with pre-loaded actions
- * const middleware = new McpMiddleware({
- *   client: mcpClient,
- *   actions,
- *   toolPrefix: 'server-',
- * });
- *
- * // Use with HttpAgent
- * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });
- * agent.use(middleware);
- * ```
  */
 
-/**
- * Tool definition format for AG-UI input.tools
- */
-interface AgUiTool {
-    name: string;
-    description: string;
-    parameters?: Record<string, any>;
-}
 /**
  * Configuration for McpMiddleware
  */
 interface McpMiddlewareConfig {
-    /**
-     * MCP client or MultiSessionClient for executing tools
-     */
-    client: MCPClient | MultiSessionClient;
-    /**
-     * Prefix used to identify MCP tool names.
-     * Tools starting with this prefix will be executed server-side.
-     * @default 'server-'
-     */
-    toolPrefix?: string;
-    /**
-     * Pre-loaded tools with handlers for execution.
-     * If not provided, tools will be loaded from the MCP client on first use.
-     */
-    tools?: AguiTool[];
+    /** Pre-loaded tools with handlers (required) */
+    tools: AguiTool[];
+    /** Max result length in chars (default: 50000) */
+    maxResultLength?: number;
 }
 /**
  * AG-UI Middleware that executes MCP tools server-side.
- *
- * This middleware intercepts tool calls for MCP tools (identified by prefix),
- * executes them via the MCP client, and returns results to the agent.
- *
- * @see {@link createMcpMiddleware} for a simpler factory function
  */
 declare class McpMiddleware extends Middleware {
-    private client;
-    private toolPrefix;
-    private actions;
     private tools;
-    private actionsLoaded;
+    private toolSchemas;
+    private maxResultLength;
     constructor(config: McpMiddlewareConfig);
-    /**
-     * Convert actions to AG-UI tool format
-     */
-    private actionsToTools;
-    /**
-     * Check if a tool name is an MCP tool (matches the configured prefix)
-     */
     private isMcpTool;
-    /**
-     * Load actions from the MCP client if not already loaded
-     */
-    private ensureActionsLoaded;
-    /**
-     * Execute an MCP tool and return the result as a string
-     */
+    private parseArgs;
     private executeTool;
-    /**
-     * Generate a unique message ID for tool results
-     */
-    private generateMessageId;
-    /**
-     * Run the middleware, intercepting and executing MCP tool calls
-     */
+    private generateId;
+    private ensureIds;
+    /** Process tool call events and update state */
+    private handleToolCallEvent;
+    /** Execute pending MCP tools and return results */
+    private executeTools;
+    /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */
+    private emitToolResults;
     run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent>;
-    private triggerNewRun;
-    private handlePendingCalls;
 }
 /**
  * Factory function to create MCP middleware.
- *
- * This is a convenience wrapper around McpMiddleware that returns a function
- * compatible with the AG-UI middleware pattern.
- *
- * @param client - MCP client or MultiSessionClient
- * @param options - Configuration options
- * @returns Middleware function
- *
- * @example
- * ```typescript
- * import { HttpAgent } from '@ag-ui/client';
- * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';
- *
- * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });
- * agent.use(createMcpMiddleware(multiSessionClient, {
- *   toolPrefix: 'server-',
- *   actions: mcpActions,
- * }));
- * ```
  */
-declare function createMcpMiddleware(client: MCPClient | MultiSessionClient, options?: {
-    toolPrefix?: string;
-    tools?: AguiTool[];
+declare function createMcpMiddleware(options: {
+    tools: AguiTool[];
+    maxResultLength?: number;
 }): (input: RunAgentInput, next: AbstractAgent) => Observable<BaseEvent>;
 
-export { type AgUiTool, McpMiddleware, type McpMiddlewareConfig, McpMiddleware as McpToolExecutorMiddleware, createMcpMiddleware, createMcpMiddleware as createMcpToolMiddleware };
+export { McpMiddleware, type McpMiddlewareConfig, McpMiddleware as McpToolExecutorMiddleware, createMcpMiddleware, createMcpMiddleware as createMcpToolMiddleware };

package/dist/adapters/agui-middleware.d.ts

@@ -1,8 +1,8 @@
 import { Observable } from 'rxjs';
 import { Middleware, RunAgentInput, AbstractAgent, BaseEvent } from '@ag-ui/client';
-export { AbstractAgent, BaseEvent, EventType, Middleware, RunAgentInput, ToolCallEndEvent } from '@ag-ui/client';
-import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-BOFgPypS.js';
+export { AbstractAgent, BaseEvent, EventType, Middleware, RunAgentInput, Tool, ToolCallEndEvent } from '@ag-ui/client';
 import { AguiTool } from './agui-adapter.js';
+import '../multi-session-client-BOFgPypS.js';
 import '../events-BP6WyRNh.js';
 import '@modelcontextprotocol/sdk/types.js';
 import '@modelcontextprotocol/sdk/shared/auth.js';
@@ -11,161 +11,49 @@ import '@modelcontextprotocol/sdk/client/auth.js';
 /**
  * AG-UI Middleware for MCP Tool Execution
  *
- * This middleware intercepts tool calls from remote agents (e.g., LangGraph, AutoGen)
- * and executes MCP tools server-side, returning results back to the agent.
+ * This middleware intercepts tool calls from remote agents and executes
+ * MCP tools server-side, returning results back to the agent.
  *
- * ## How It Works
- *
- * 1. **Tool Injection**: When a run starts, the middleware injects MCP tool definitions
- *    into `input.tools` so the remote agent knows about available MCP tools.
- *
- * 2. **Event Interception**: The middleware subscribes to the agent's event stream and
- *    tracks tool calls using AG-UI events:
- *    - `TOOL_CALL_START`: Records tool name and ID
- *    - `TOOL_CALL_ARGS`: Accumulates streamed arguments
- *    - `TOOL_CALL_END`: Marks tool call as complete
- *    - `RUN_FINISHED`: Triggers execution of pending MCP tools
- *
- * 3. **Server-Side Execution**: When `RUN_FINISHED` arrives with pending MCP tool calls,
- *    the middleware:
- *    - Executes each MCP tool via the MCP client
- *    - Emits `TOOL_CALL_RESULT` events with the results
- *    - Adds results to `input.messages` for context
- *    - Emits `RUN_FINISHED` to close the current run
- *    - Triggers a new run so the agent can process tool results
- *
- * 4. **Recursive Processing**: If the new run makes more MCP tool calls, the cycle
- *    repeats until the agent completes without pending MCP calls.
- *
- * ## Tool Identification
- *
- * MCP tools are identified by a configurable prefix (default: `server-`).
- * Tools not matching this prefix are passed through without interception.
- *
- * @requires @ag-ui/client - This middleware requires @ag-ui/client as a peer dependency
+ * @requires @ag-ui/client - Peer dependency for AG-UI types
  * @requires rxjs - Uses RxJS Observables for event streaming
- *
- * @example
- * ```typescript
- * import { HttpAgent } from '@ag-ui/client';
- * import { McpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';
- * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';
- *
- * // Create MCP client and adapter
- * const mcpClient = new MultiSessionClient('user_123');
- * await mcpClient.connect();
- *
- * const adapter = new AguiAdapter(mcpClient);
- * const actions = await adapter.getActions();
- *
- * // Create middleware with pre-loaded actions
- * const middleware = new McpMiddleware({
- *   client: mcpClient,
- *   actions,
- *   toolPrefix: 'server-',
- * });
- *
- * // Use with HttpAgent
- * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });
- * agent.use(middleware);
- * ```
  */
 
-/**
- * Tool definition format for AG-UI input.tools
- */
-interface AgUiTool {
-    name: string;
-    description: string;
-    parameters?: Record<string, any>;
-}
 /**
  * Configuration for McpMiddleware
  */
 interface McpMiddlewareConfig {
-    /**
-     * MCP client or MultiSessionClient for executing tools
-     */
-    client: MCPClient | MultiSessionClient;
-    /**
-     * Prefix used to identify MCP tool names.
-     * Tools starting with this prefix will be executed server-side.
-     * @default 'server-'
-     */
-    toolPrefix?: string;
-    /**
-     * Pre-loaded tools with handlers for execution.
-     * If not provided, tools will be loaded from the MCP client on first use.
-     */
-    tools?: AguiTool[];
+    /** Pre-loaded tools with handlers (required) */
+    tools: AguiTool[];
+    /** Max result length in chars (default: 50000) */
+    maxResultLength?: number;
 }
 /**
  * AG-UI Middleware that executes MCP tools server-side.
- *
- * This middleware intercepts tool calls for MCP tools (identified by prefix),
- * executes them via the MCP client, and returns results to the agent.
- *
- * @see {@link createMcpMiddleware} for a simpler factory function
  */
 declare class McpMiddleware extends Middleware {
-    private client;
-    private toolPrefix;
-    private actions;
     private tools;
-    private actionsLoaded;
+    private toolSchemas;
+    private maxResultLength;
     constructor(config: McpMiddlewareConfig);
-    /**
-     * Convert actions to AG-UI tool format
-     */
-    private actionsToTools;
-    /**
-     * Check if a tool name is an MCP tool (matches the configured prefix)
-     */
     private isMcpTool;
-    /**
-     * Load actions from the MCP client if not already loaded
-     */
-    private ensureActionsLoaded;
-    /**
-     * Execute an MCP tool and return the result as a string
-     */
+    private parseArgs;
     private executeTool;
-    /**
-     * Generate a unique message ID for tool results
-     */
-    private generateMessageId;
-    /**
-     * Run the middleware, intercepting and executing MCP tool calls
-     */
+    private generateId;
+    private ensureIds;
+    /** Process tool call events and update state */
+    private handleToolCallEvent;
+    /** Execute pending MCP tools and return results */
+    private executeTools;
+    /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */
+    private emitToolResults;
     run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent>;
-    private triggerNewRun;
-    private handlePendingCalls;
 }
 /**
  * Factory function to create MCP middleware.
- *
- * This is a convenience wrapper around McpMiddleware that returns a function
- * compatible with the AG-UI middleware pattern.
- *
- * @param client - MCP client or MultiSessionClient
- * @param options - Configuration options
- * @returns Middleware function
- *
- * @example
- * ```typescript
- * import { HttpAgent } from '@ag-ui/client';
- * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';
- *
- * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });
- * agent.use(createMcpMiddleware(multiSessionClient, {
- *   toolPrefix: 'server-',
- *   actions: mcpActions,
- * }));
- * ```
  */
-declare function createMcpMiddleware(client: MCPClient | MultiSessionClient, options?: {
-    toolPrefix?: string;
-    tools?: AguiTool[];
+declare function createMcpMiddleware(options: {
+    tools: AguiTool[];
+    maxResultLength?: number;
 }): (input: RunAgentInput, next: AbstractAgent) => Observable<BaseEvent>;
 
-export { type AgUiTool, McpMiddleware, type McpMiddlewareConfig, McpMiddleware as McpToolExecutorMiddleware, createMcpMiddleware, createMcpMiddleware as createMcpToolMiddleware };
+export { McpMiddleware, type McpMiddlewareConfig, McpMiddleware as McpToolExecutorMiddleware, createMcpMiddleware, createMcpMiddleware as createMcpToolMiddleware };