@mcp-ts/sdk 1.3.6 → 1.3.7

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

@@ -1,4 +1,4 @@
-import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CzhMkE0k.mjs';
+import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CQsRbxYI.mjs';
 import '../events-CK3N--3g.mjs';
 import '@modelcontextprotocol/sdk/types.js';
 import '@modelcontextprotocol/sdk/client/auth.js';

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

@@ -1,4 +1,4 @@
-import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-BYLarghq.js';
+import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CHE8QpVE.js';
 import '../events-CK3N--3g.js';
 import '@modelcontextprotocol/sdk/types.js';
 import '@modelcontextprotocol/sdk/client/auth.js';

package/dist/adapters/agui-adapter.js

@@ -93,7 +93,7 @@ var AguiAdapter = class {
     if (!client.isConnected()) return [];
     const result = await client.listTools();
     const serverId = typeof client.getServerId === "function" ? client.getServerId() : void 0;
-    const normalizedPrefix = (this.options.prefix ?? serverId ?? "mcp").replace(/-/g, "");
+    const normalizedPrefix = this.options.prefix?.replace(/-/g, "") ?? serverId?.replace(/-/g, "").substring(0, 8) ?? "mcp";
     const prefix = `tool_${normalizedPrefix}`;
     return result.tools.map((tool) => {
       const mcpTool = tool;
@@ -114,7 +114,7 @@ var AguiAdapter = class {
     if (!client.isConnected()) return [];
     const result = await client.listTools();
     const serverId = typeof client.getServerId === "function" ? client.getServerId() : void 0;
-    const normalizedPrefix = (this.options.prefix ?? serverId ?? "mcp").replace(/-/g, "");
+    const normalizedPrefix = this.options.prefix?.replace(/-/g, "") ?? serverId?.replace(/-/g, "").substring(0, 8) ?? "mcp";
     const prefix = `tool_${normalizedPrefix}`;
     return result.tools.map((tool) => {
       const mcpTool = tool;

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

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/agui-adapter.ts"],"names":[],"mappings":";;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;AAqCO,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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,gBAAA,GAAA,CAAoB,KAAK,OAAA,CAAQ,MAAA,IAAU,YAAY,KAAA,EAAO,OAAA,CAAQ,MAAM,EAAE,CAAA;AACpF,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAE5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,MAAM,cAAc,IAAA,CAAK,IAAA;AACzB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe,EAAE;AAAA,QACvE,OAAA,EAAS,OAAO,IAAA,KAAc;AAE1B,UAAA,MAAM,aAAa,MAAO,MAAA,CAAe,SAAS,WAAA,EAAa,IAAA,IAAQ,EAAE,CAAA;AAGzE,UAAA,OAAO,UAAA;AAAA,QACX;AAAA,OACJ;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,gBAAA,GAAA,CAAoB,KAAK,OAAA,CAAQ,MAAA,IAAU,YAAY,KAAA,EAAO,OAAA,CAAQ,MAAM,EAAE,CAAA;AACpF,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAC5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe;AAAE,OAC3E;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;AACJ","file":"agui-adapter.js","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 * Extended JSON Schema properties that Pydantic's strict validation rejects.\r\n * These are valid JSON Schema extensions but not part of the core spec.\r\n */\r\nconst PYDANTIC_FORBIDDEN_PROPS = [\r\n    // JSON Schema meta-properties\r\n    '$schema', '$id', '$comment', '$defs', 'definitions',\r\n    // Extended properties used by some MCP servers (e.g., Apify)\r\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\r\n    // Other common extensions\r\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\r\n];\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, LangGraph).\r\n *\r\n * @param schema - The JSON Schema to clean\r\n * @returns Cleaned schema without forbidden 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 all forbidden properties\r\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\r\n        delete cleaned[prop];\r\n    }\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    _meta?: Record<string, any>; // Add _meta to AguiTool\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    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\r\n            const mcpTool = tool as any;\r\n            const mcpToolName = tool.name;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n                handler: async (args: any) => {\r\n                    // Call the actual MCP tool\r\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\r\n\r\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\r\n                    return callResult;\r\n                }\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            const mcpTool = tool as any;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n            };\r\n        });\r\n    }\r\n}\r\n"]}
+{"version":3,"sources":["../../src/adapters/agui-adapter.ts"],"names":[],"mappings":";;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;AAqCO,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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,mBAAmB,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,OAAA,CAAQ,MAAM,EAAE,CAAA,IAAK,QAAA,EAAU,OAAA,CAAQ,MAAM,EAAE,CAAA,CAAE,SAAA,CAAU,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA;AAClH,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAE5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,MAAM,cAAc,IAAA,CAAK,IAAA;AACzB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe,EAAE;AAAA,QACvE,OAAA,EAAS,OAAO,IAAA,KAAc;AAE1B,UAAA,MAAM,aAAa,MAAO,MAAA,CAAe,SAAS,WAAA,EAAa,IAAA,IAAQ,EAAE,CAAA;AAGzE,UAAA,OAAO,UAAA;AAAA,QACX;AAAA,OACJ;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,mBAAmB,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,OAAA,CAAQ,MAAM,EAAE,CAAA,IAAK,QAAA,EAAU,OAAA,CAAQ,MAAM,EAAE,CAAA,CAAE,SAAA,CAAU,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA;AAClH,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAC5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe;AAAE,OAC3E;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;AACJ","file":"agui-adapter.js","sourcesContent":["/**\n * MCP Adapter for AG-UI Integration\n *\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\n * It provides tools with handlers for server-side execution and tool definitions\n * in JSON Schema format for passing to remote agents.\n *\n * @example\n * ```typescript\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n * import { HttpAgent } from '@ag-ui/client';\n *\n * // Create MCP client\n * const mcpClient = new MultiSessionClient('user_123');\n * await mcpClient.connect();\n *\n * // Create adapter and get tools\n * const adapter = new AguiAdapter(mcpClient);\n * const tools = await adapter.getTools();\n *\n * // Use with AG-UI middleware\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(createMcpMiddleware({ tools }));\n * ```\n */\n\nimport { MCPClient } from '../server/mcp/oauth-client.js';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\n\n/**\n * Extended JSON Schema properties that Pydantic's strict validation rejects.\n * These are valid JSON Schema extensions but not part of the core spec.\n */\nconst PYDANTIC_FORBIDDEN_PROPS = [\n    // JSON Schema meta-properties\n    '$schema', '$id', '$comment', '$defs', 'definitions',\n    // Extended properties used by some MCP servers (e.g., Apify)\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\n    // Other common extensions\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\n];\n\n/**\n * Cleans a JSON Schema by removing meta-properties that cause issues with\n * strict Pydantic validation (e.g., Google ADK, LangGraph).\n *\n * @param schema - The JSON Schema to clean\n * @returns Cleaned schema without forbidden properties\n */\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\n    if (!schema) {\n        return { type: 'object', properties: {} };\n    }\n\n    const cleaned = { ...schema };\n\n    // Remove all forbidden properties\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\n        delete cleaned[prop];\n    }\n\n    // Recursively clean nested properties\n    if (cleaned.properties && typeof cleaned.properties === 'object') {\n        const cleanedProps: Record<string, any> = {};\n        for (const [key, value] of Object.entries(cleaned.properties)) {\n            if (typeof value === 'object' && value !== null) {\n                cleanedProps[key] = cleanSchema(value as Record<string, any>);\n            } else {\n                cleanedProps[key] = value;\n            }\n        }\n        cleaned.properties = cleanedProps;\n    }\n\n    // Clean items if it's an array schema\n    if (cleaned.items && typeof cleaned.items === 'object') {\n        cleaned.items = cleanSchema(cleaned.items);\n    }\n\n    // Clean additionalProperties if it's an object schema\n    if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\n        cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\n    }\n\n    return cleaned;\n}\n\n/**\n * Configuration options for AguiAdapter\n */\nexport interface AguiAdapterOptions {\n    /**\n     * Prefix for tool names to avoid collision with other tools.\n     * @default serverId or 'mcp'\n     */\n    prefix?: string;\n}\n\n/**\n * AG-UI Tool with handler for server-side execution.\n */\nexport interface AguiTool {\n    name: string;\n    description: string;\n    parameters?: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiTool\n    handler?: (args: any) => any | Promise<any>;\n}\n\n/**\n * Tool definition format for passing to remote agents (without handler).\n */\nexport interface AguiToolDefinition {\n    name: string;\n    description: string;\n    parameters: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\n}\n\n/**\n * Adapter that transforms MCP tools into AG-UI compatible formats.\n */\nexport class AguiAdapter {\n    constructor(\n        private client: MCPClient | MultiSessionClient,\n        private options: AguiAdapterOptions = {}\n    ) { }\n\n    /**\n     * Get tools with handlers for MCP tool execution.\n     */\n    async getTools(): Promise<AguiTool[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiTool[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformTools(client));\n            }\n            return allTools;\n        }\n        return this.transformTools(this.client as MCPClient);\n    }\n\n    /**\n     * Get tool definitions in JSON Schema format for passing to remote agents.\n     */\n    async getToolDefinitions(): Promise<AguiToolDefinition[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiToolDefinition[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformToolDefinitions(client));\n            }\n            return allTools;\n        }\n        return this.transformToolDefinitions(this.client as MCPClient);\n    }\n\n    /**\n     * Get tools as a function (for dynamic loading).\n     */\n    getToolsFunction(): () => Promise<AguiTool[]> {\n        return () => this.getTools();\n    }\n\n    private isMultiSession(): boolean {\n        return typeof (this.client as any).getClients === 'function';\n    }\n\n    private async transformTools(client: MCPClient): Promise<AguiTool[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\n            const mcpTool = tool as any;\n            const mcpToolName = tool.name;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n                handler: async (args: any) => {\n                    // Call the actual MCP tool\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\n\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\n                    return callResult;\n                }\n            }\n        });\n    }\n\n    private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            const mcpTool = tool as any;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n            };\n        });\n    }\n}\n"]}

package/dist/adapters/agui-adapter.mjs

@@ -91,7 +91,7 @@ var AguiAdapter = class {
     if (!client.isConnected()) return [];
     const result = await client.listTools();
     const serverId = typeof client.getServerId === "function" ? client.getServerId() : void 0;
-    const normalizedPrefix = (this.options.prefix ?? serverId ?? "mcp").replace(/-/g, "");
+    const normalizedPrefix = this.options.prefix?.replace(/-/g, "") ?? serverId?.replace(/-/g, "").substring(0, 8) ?? "mcp";
     const prefix = `tool_${normalizedPrefix}`;
     return result.tools.map((tool) => {
       const mcpTool = tool;
@@ -112,7 +112,7 @@ var AguiAdapter = class {
     if (!client.isConnected()) return [];
     const result = await client.listTools();
     const serverId = typeof client.getServerId === "function" ? client.getServerId() : void 0;
-    const normalizedPrefix = (this.options.prefix ?? serverId ?? "mcp").replace(/-/g, "");
+    const normalizedPrefix = this.options.prefix?.replace(/-/g, "") ?? serverId?.replace(/-/g, "").substring(0, 8) ?? "mcp";
     const prefix = `tool_${normalizedPrefix}`;
     return result.tools.map((tool) => {
       const mcpTool = tool;

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

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/agui-adapter.ts"],"names":[],"mappings":";AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;AAqCO,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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,gBAAA,GAAA,CAAoB,KAAK,OAAA,CAAQ,MAAA,IAAU,YAAY,KAAA,EAAO,OAAA,CAAQ,MAAM,EAAE,CAAA;AACpF,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAE5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,MAAM,cAAc,IAAA,CAAK,IAAA;AACzB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe,EAAE;AAAA,QACvE,OAAA,EAAS,OAAO,IAAA,KAAc;AAE1B,UAAA,MAAM,aAAa,MAAO,MAAA,CAAe,SAAS,WAAA,EAAa,IAAA,IAAQ,EAAE,CAAA;AAGzE,UAAA,OAAO,UAAA;AAAA,QACX;AAAA,OACJ;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,gBAAA,GAAA,CAAoB,KAAK,OAAA,CAAQ,MAAA,IAAU,YAAY,KAAA,EAAO,OAAA,CAAQ,MAAM,EAAE,CAAA;AACpF,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAC5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe;AAAE,OAC3E;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;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 * Extended JSON Schema properties that Pydantic's strict validation rejects.\r\n * These are valid JSON Schema extensions but not part of the core spec.\r\n */\r\nconst PYDANTIC_FORBIDDEN_PROPS = [\r\n    // JSON Schema meta-properties\r\n    '$schema', '$id', '$comment', '$defs', 'definitions',\r\n    // Extended properties used by some MCP servers (e.g., Apify)\r\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\r\n    // Other common extensions\r\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\r\n];\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, LangGraph).\r\n *\r\n * @param schema - The JSON Schema to clean\r\n * @returns Cleaned schema without forbidden 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 all forbidden properties\r\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\r\n        delete cleaned[prop];\r\n    }\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    _meta?: Record<string, any>; // Add _meta to AguiTool\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    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\r\n            const mcpTool = tool as any;\r\n            const mcpToolName = tool.name;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n                handler: async (args: any) => {\r\n                    // Call the actual MCP tool\r\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\r\n\r\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\r\n                    return callResult;\r\n                }\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            const mcpTool = tool as any;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n            };\r\n        });\r\n    }\r\n}\r\n"]}
+{"version":3,"sources":["../../src/adapters/agui-adapter.ts"],"names":[],"mappings":";AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;AAqCO,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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,mBAAmB,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,OAAA,CAAQ,MAAM,EAAE,CAAA,IAAK,QAAA,EAAU,OAAA,CAAQ,MAAM,EAAE,CAAA,CAAE,SAAA,CAAU,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA;AAClH,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAE5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,MAAM,cAAc,IAAA,CAAK,IAAA;AACzB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe,EAAE;AAAA,QACvE,OAAA,EAAS,OAAO,IAAA,KAAc;AAE1B,UAAA,MAAM,aAAa,MAAO,MAAA,CAAe,SAAS,WAAA,EAAa,IAAA,IAAQ,EAAE,CAAA;AAGzE,UAAA,OAAO,UAAA;AAAA,QACX;AAAA,OACJ;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;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,WAAY,OAAQ,MAAA,CAAe,gBAAgB,UAAA,GAClD,MAAA,CAAe,aAAY,GAC5B,MAAA;AACN,IAAA,MAAM,mBAAmB,IAAA,CAAK,OAAA,CAAQ,MAAA,EAAQ,OAAA,CAAQ,MAAM,EAAE,CAAA,IAAK,QAAA,EAAU,OAAA,CAAQ,MAAM,EAAE,CAAA,CAAE,SAAA,CAAU,CAAA,EAAG,CAAC,CAAA,IAAK,KAAA;AAClH,IAAA,MAAM,MAAA,GAAS,QAAQ,gBAAgB,CAAA,CAAA;AAEvC,IAAA,OAAO,MAAA,CAAO,KAAA,CAAM,GAAA,CAAI,CAAA,IAAA,KAAQ;AAC5B,MAAA,MAAM,OAAA,GAAU,IAAA;AAChB,MAAA,OAAO;AAAA,QACH,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,WAAA,CAAY,IAAA,CAAK,WAAW,CAAA;AAAA,QACxC,KAAA,EAAO,EAAE,GAAG,OAAA,CAAQ,OAAO,SAAA,EAAY,MAAA,CAAe,gBAAe;AAAE,OAC3E;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;AACJ","file":"agui-adapter.mjs","sourcesContent":["/**\n * MCP Adapter for AG-UI Integration\n *\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\n * It provides tools with handlers for server-side execution and tool definitions\n * in JSON Schema format for passing to remote agents.\n *\n * @example\n * ```typescript\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n * import { HttpAgent } from '@ag-ui/client';\n *\n * // Create MCP client\n * const mcpClient = new MultiSessionClient('user_123');\n * await mcpClient.connect();\n *\n * // Create adapter and get tools\n * const adapter = new AguiAdapter(mcpClient);\n * const tools = await adapter.getTools();\n *\n * // Use with AG-UI middleware\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(createMcpMiddleware({ tools }));\n * ```\n */\n\nimport { MCPClient } from '../server/mcp/oauth-client.js';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\n\n/**\n * Extended JSON Schema properties that Pydantic's strict validation rejects.\n * These are valid JSON Schema extensions but not part of the core spec.\n */\nconst PYDANTIC_FORBIDDEN_PROPS = [\n    // JSON Schema meta-properties\n    '$schema', '$id', '$comment', '$defs', 'definitions',\n    // Extended properties used by some MCP servers (e.g., Apify)\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\n    // Other common extensions\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\n];\n\n/**\n * Cleans a JSON Schema by removing meta-properties that cause issues with\n * strict Pydantic validation (e.g., Google ADK, LangGraph).\n *\n * @param schema - The JSON Schema to clean\n * @returns Cleaned schema without forbidden properties\n */\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\n    if (!schema) {\n        return { type: 'object', properties: {} };\n    }\n\n    const cleaned = { ...schema };\n\n    // Remove all forbidden properties\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\n        delete cleaned[prop];\n    }\n\n    // Recursively clean nested properties\n    if (cleaned.properties && typeof cleaned.properties === 'object') {\n        const cleanedProps: Record<string, any> = {};\n        for (const [key, value] of Object.entries(cleaned.properties)) {\n            if (typeof value === 'object' && value !== null) {\n                cleanedProps[key] = cleanSchema(value as Record<string, any>);\n            } else {\n                cleanedProps[key] = value;\n            }\n        }\n        cleaned.properties = cleanedProps;\n    }\n\n    // Clean items if it's an array schema\n    if (cleaned.items && typeof cleaned.items === 'object') {\n        cleaned.items = cleanSchema(cleaned.items);\n    }\n\n    // Clean additionalProperties if it's an object schema\n    if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\n        cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\n    }\n\n    return cleaned;\n}\n\n/**\n * Configuration options for AguiAdapter\n */\nexport interface AguiAdapterOptions {\n    /**\n     * Prefix for tool names to avoid collision with other tools.\n     * @default serverId or 'mcp'\n     */\n    prefix?: string;\n}\n\n/**\n * AG-UI Tool with handler for server-side execution.\n */\nexport interface AguiTool {\n    name: string;\n    description: string;\n    parameters?: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiTool\n    handler?: (args: any) => any | Promise<any>;\n}\n\n/**\n * Tool definition format for passing to remote agents (without handler).\n */\nexport interface AguiToolDefinition {\n    name: string;\n    description: string;\n    parameters: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\n}\n\n/**\n * Adapter that transforms MCP tools into AG-UI compatible formats.\n */\nexport class AguiAdapter {\n    constructor(\n        private client: MCPClient | MultiSessionClient,\n        private options: AguiAdapterOptions = {}\n    ) { }\n\n    /**\n     * Get tools with handlers for MCP tool execution.\n     */\n    async getTools(): Promise<AguiTool[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiTool[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformTools(client));\n            }\n            return allTools;\n        }\n        return this.transformTools(this.client as MCPClient);\n    }\n\n    /**\n     * Get tool definitions in JSON Schema format for passing to remote agents.\n     */\n    async getToolDefinitions(): Promise<AguiToolDefinition[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiToolDefinition[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformToolDefinitions(client));\n            }\n            return allTools;\n        }\n        return this.transformToolDefinitions(this.client as MCPClient);\n    }\n\n    /**\n     * Get tools as a function (for dynamic loading).\n     */\n    getToolsFunction(): () => Promise<AguiTool[]> {\n        return () => this.getTools();\n    }\n\n    private isMultiSession(): boolean {\n        return typeof (this.client as any).getClients === 'function';\n    }\n\n    private async transformTools(client: MCPClient): Promise<AguiTool[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\n            const mcpTool = tool as any;\n            const mcpToolName = tool.name;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n                handler: async (args: any) => {\n                    // Call the actual MCP tool\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\n\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\n                    return callResult;\n                }\n            }\n        });\n    }\n\n    private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            const mcpTool = tool as any;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n            };\n        });\n    }\n}\n"]}

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

@@ -2,7 +2,7 @@ import { Observable } from 'rxjs';
 import { Middleware, RunAgentInput, AbstractAgent, BaseEvent } from '@ag-ui/client';
 export { AbstractAgent, BaseEvent, EventType, Middleware, RunAgentInput, Tool, ToolCallEndEvent } from '@ag-ui/client';
 import { AguiTool } from './agui-adapter.mjs';
-import '../multi-session-client-CzhMkE0k.mjs';
+import '../multi-session-client-CQsRbxYI.mjs';
 import '../events-CK3N--3g.mjs';
 import '@modelcontextprotocol/sdk/types.js';
 import '@modelcontextprotocol/sdk/client/auth.js';

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

@@ -2,7 +2,7 @@ import { Observable } from 'rxjs';
 import { Middleware, RunAgentInput, AbstractAgent, BaseEvent } from '@ag-ui/client';
 export { AbstractAgent, BaseEvent, EventType, Middleware, RunAgentInput, Tool, ToolCallEndEvent } from '@ag-ui/client';
 import { AguiTool } from './agui-adapter.js';
-import '../multi-session-client-BYLarghq.js';
+import '../multi-session-client-CHE8QpVE.js';
 import '../events-CK3N--3g.js';
 import '@modelcontextprotocol/sdk/types.js';
 import '@modelcontextprotocol/sdk/client/auth.js';

package/dist/adapters/agui-middleware.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":["Middleware","EventType","Observable"],"mappings":";;;;;;;;;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;;;ACrCO,IAAM,aAAA,GAAN,cAA4BA,iBAAA,CAAW;AAAA,EAI1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAJV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU;AAAA,KACxC,CAAE,CAAA;AAAA,EACN;AAAA,EAEQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAAA,EACnD;AAAA,EAEQ,UAAU,UAAA,EAAyC;AACvD,IAAA,IAAI,CAAC,UAAA,EAAY,IAAA,EAAK,SAAU,EAAC;AAEjC,IAAA,IAAI;AACA,MAAA,OAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,IAChC,CAAA,CAAA,MAAQ;AAEJ,MAAA,MAAM,OAAA,GAAU,WAAW,IAAA,EAAK;AAChC,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,IAAI,CAAA,EAAG;AACxB,QAAA,MAAM,WAAA,GAAc,QAAQ,KAAA,CAAM,CAAA,EAAG,QAAQ,OAAA,CAAQ,IAAI,IAAI,CAAC,CAAA;AAC9D,QAAA,IAAI;AACA,UAAA,OAAO,IAAA,CAAK,MAAM,WAAW,CAAA;AAAA,QACjC,CAAA,CAAA,MAAQ;AACJ,UAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,WAAW,CAAA;AAAA,QACtE;AAAA,MACJ;AACA,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AACjE,MAAA,OAAO,EAAC;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,OAAO,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACrD,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAChB,MAAA,OAAO,CAAA,YAAA,EAAe,IAAA,GAAO,gBAAA,GAAmB,WAAW,KAAK,QAAQ,CAAA,CAAA;AAAA,IAC5E;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AACtC,MAAA,MAAM,YAAY,OAAO,MAAA,KAAW,WAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAE7E,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,SAAA;AAAA,IACX,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,OAAA,EAAU,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IACnD;AAAA,EACJ;AAAA,EAEQ,WAAW,MAAA,EAAwB;AACvC,IAAA,OAAO,GAAG,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC5E;AAAA,EAEQ,UAAU,KAAA,EAA4B;AAC1C,IAAA,MAAM,QAAA,GAAW,KAAA;AACjB,IAAA,IAAI,CAAC,QAAA,CAAS,QAAA,WAAmB,QAAA,GAAW,IAAA,CAAK,WAAW,YAAY,CAAA;AACxE,IAAA,IAAI,CAAC,QAAA,CAAS,KAAA,WAAgB,KAAA,GAAQ,IAAA,CAAK,WAAW,SAAS,CAAA;AAAA,EACnE;AAAA;AAAA,EAGQ,mBAAA,CAAoB,OAAkB,KAAA,EAAuB;AACjE,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAG/D,IAAA,IAAI,KAAA,CAAM,IAAA,KAASC,gBAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AACA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iCAAA,EAAoC,CAAA,CAAE,YAAY,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,SAAA,EAAY,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAC,CAAA,CAAA,CAAG,CAAA;AAAA,MACpI;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,aAAA,EAAe;AACxC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+BAAA,EAAkC,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,SAAS,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACtH;AAGA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;AACzB,kBAAA,OAAA,CAAQ,GAAA,CAAI,iDAAiD,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,MAAA,EAAS,EAAA,CAAG,EAAE,CAAA,CAAA,CAAG,CAAA;AAAA,gBAClG;AAAA,cACJ;AAAA,YACJ;AACA,YAAA;AAAA,UACJ;AAAA,QACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAc,aAAa,KAAA,EAAwC;AAC/D,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAC/D,IAAA,MAAM,UAAwB,EAAC;AAE/B,IAAA,MAAM,WAAW,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAC5D,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,OAAO,IAAA,CAAK,SAAA,CAAU,mBAAmB,GAAA,CAAI,UAAU,KAAK,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AAEjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,UAAA,CAAW,YAAY;AAAA,OAC1C,CAAA;AACD,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,QAAQ,CAAA;AAC1B,IAAA,OAAO,OAAA;AAAA,EACX;AAAA;AAAA,EAGQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AAC/D,MAAA,QAAA,CAAS,IAAA,CAAK;AAAA,QACV,MAAMA,gBAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AACR,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AAAA,IAC5E;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAIC,eAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;AAEjB,MAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,CAAiC,CAAA;AAC7C,MAAA,OAAA,CAAQ,IAAI,CAAA,0BAAA,EAA6B,QAAA,CAAS,QAAQ,CAAA,SAAA,EAAY,QAAA,CAAS,KAAK,CAAA,CAAE,CAAA;AACtF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,0BAAA,EAA6B,KAAA,CAAM,QAAA,EAAU,MAAA,IAAU,CAAC,CAAA,SAAA,EAAY,IAAA,CAAK,KAAA,EAAO,MAAA,IAAU,CAAC,CAAA,CAAE,CAAA;AAGzG,MAAA,IAAI,IAAA,CAAK,aAAa,MAAA,EAAQ;AAC1B,QAAA,KAAA,CAAM,KAAA,GAAQ,CAAC,GAAI,KAAA,CAAM,SAAS,EAAC,EAAI,GAAG,IAAA,CAAK,WAAW,CAAA;AAC1D,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,WAAA,CAAY,MAAM,CAAA,OAAA,CAAA,EAAW,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MACvH;AAEA,MAAA,MAAM,oBAAoB,YAAY;AAClC,QAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,QAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,KAAS,CAAA,EAAG;AAClC,UAAA,QAAA,CAAS,IAAA,CAAK;AAAA,YACV,MAAMD,gBAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,KAAA,CAAM,eAAA,CAAgB,IAAI,CAAA,cAAA,CAAgB,CAAA;AAG3F,QAAA,MAAM,YAAY,EAAC;AACnB,QAAA,KAAA,MAAW,UAAA,IAAc,MAAM,eAAA,EAAiB;AAC5C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC/C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,IAAA,EAAM;AACN,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACX,EAAA,EAAI,UAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA;AAAK,aACrC,CAAA;AAAA,UACL;AAAA,QACJ;AAGA,QAAA,IAAI,SAAA,CAAU,MAAA,GAAS,CAAA,IAAK,KAAA,CAAM,WAAA,EAAa;AAC3C,UAAA,MAAM,YAAA,GAAe;AAAA,YACjB,EAAA,EAAI,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAAA,YAC7B,IAAA,EAAM,WAAA;AAAA,YACN,OAAA,EAAS,MAAM,WAAA,IAAe,IAAA;AAAA;AAAA,YAC9B,UAAA,EAAY,SAAA,CAAU,MAAA,GAAS,CAAA,GAAI,SAAA,GAAY;AAAA,WACnD;AACA,UAAA,KAAA,CAAM,QAAA,CAAS,KAAK,YAAmB,CAAA;AACvC,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iEAAA,EAAoE,KAAA,CAAM,WAAA,EAAa,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,KAAA,EAAQ,SAAA,CAAU,MAAM,CAAA,OAAA,CAAS,CAAA;AAAA,QACpJ;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6CAAA,EAAgD,OAAA,CAAQ,MAAM,CAAA,QAAA,CAAU,CAAA;AAGpF,QAAA,KAAA,MAAW,EAAE,UAAA,EAAY,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AACrD,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,YAAA,EAAc,UAAA;AAAA,YACd,OAAA,EAAS;AAAA,WACL,CAAA;AAAA,QACZ;AAGA,QAAA,KAAA,CAAM,mBAAmB,KAAA,EAAM;AAC/B,QAAA,KAAA,CAAM,cAAc,KAAA,EAAM;AAC1B,QAAA,KAAA,CAAM,WAAA,GAAc,EAAA;AAEpB,QAAA,QAAA,CAAS,KAAA,GAAQ,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAC1C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,mDAAA,EAAsD,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,CAAE,CAAA;AAGzF,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,CAAA,CAAE,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,YAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,YAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,YAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,SAAA,EAAW;AACpC,cAAA,OAAA,CAAQ,IAAI,CAAA,kDAAA,CAAoD,CAAA;AAChE,cAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,cAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,cAAA,QAAA,CAAS,QAAA,EAAS;AAClB,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,WAAA,EAAa;AACtC,cAAA,OAAA,CAAQ,IAAI,CAAA,uDAAA,CAAyD,CAAA;AACrE,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,SAAA,EAAW;AACpC,YAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,oBACZ,OAAA,EACF;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc,OAAO,CAAA;AAC5C,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.js","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 * Extended JSON Schema properties that Pydantic's strict validation rejects.\r\n * These are valid JSON Schema extensions but not part of the core spec.\r\n */\r\nconst PYDANTIC_FORBIDDEN_PROPS = [\r\n    // JSON Schema meta-properties\r\n    '$schema', '$id', '$comment', '$defs', 'definitions',\r\n    // Extended properties used by some MCP servers (e.g., Apify)\r\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\r\n    // Other common extensions\r\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\r\n];\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, LangGraph).\r\n *\r\n * @param schema - The JSON Schema to clean\r\n * @returns Cleaned schema without forbidden 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 all forbidden properties\r\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\r\n        delete cleaned[prop];\r\n    }\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    _meta?: Record<string, any>; // Add _meta to AguiTool\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    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\r\n            const mcpTool = tool as any;\r\n            const mcpToolName = tool.name;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n                handler: async (args: any) => {\r\n                    // Call the actual MCP tool\r\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\r\n\r\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\r\n                    return callResult;\r\n                }\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            const mcpTool = tool as any;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n            };\r\n        });\r\n    }\r\n}\r\n","/**\r\n * AG-UI Middleware for MCP Tool Execution\r\n *\r\n * This middleware intercepts tool calls from remote agents and executes\r\n * MCP tools server-side, returning results back to the agent.\r\n *\r\n * @requires @ag-ui/client - Peer dependency for AG-UI types\r\n * @requires rxjs - Uses RxJS Observables for event streaming\r\n */\r\n\r\nimport { Observable, Subscriber } from 'rxjs';\r\nimport {\r\n    Middleware,\r\n    EventType,\r\n    type AbstractAgent,\r\n    type RunAgentInput,\r\n    type BaseEvent,\r\n    type ToolCallEndEvent,\r\n    type Tool,\r\n} from '@ag-ui/client';\r\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\r\n\r\n/** Tool execution result for continuation */\r\ninterface ToolResult {\r\n    toolCallId: string;\r\n    toolName: string;\r\n    result: string;\r\n    messageId: string;\r\n}\r\n\r\n/** State for tracking tool calls during a run */\r\ninterface RunState {\r\n    toolCallArgsBuffer: Map<string, string>;\r\n    toolCallNames: Map<string, string>;\r\n    pendingMcpCalls: Set<string>;\r\n    textContent?: string;\r\n    error: boolean;\r\n}\r\n\r\n/**\n * Configuration for McpMiddleware\n */\nexport interface McpMiddlewareConfig {\n    /** Pre-loaded tools with handlers (required) */\n    tools: AguiTool[];\n}\n\r\n/**\r\n * AG-UI Middleware that executes MCP tools server-side.\r\n */\r\nexport class McpMiddleware extends Middleware {\n    private tools: AguiTool[];\n    private toolSchemas: Tool[];\n\n    constructor(config: McpMiddlewareConfig) {\n        super();\n        this.tools = config.tools;\n        this.toolSchemas = this.tools.map((t: AguiTool) => ({\n            name: t.name,\n            description: t.description,\n            parameters: cleanSchema(t.parameters),\n        }));\n    }\n\r\n    private isMcpTool(toolName: string): boolean {\r\n        return this.tools.some(t => t.name === toolName);\r\n    }\r\n\r\n    private parseArgs(argsString: string): Record<string, any> {\r\n        if (!argsString?.trim()) return {};\r\n\r\n        try {\r\n            return JSON.parse(argsString);\r\n        } catch {\r\n            // Handle duplicated JSON from streaming issues: {...}{...}\r\n            const trimmed = argsString.trim();\r\n            if (trimmed.includes('}{')) {\r\n                const firstObject = trimmed.slice(0, trimmed.indexOf('}{') + 1);\r\n                try {\r\n                    return JSON.parse(firstObject);\r\n                } catch {\r\n                    console.error(`[McpMiddleware] Failed to parse JSON:`, firstObject);\r\n                }\r\n            }\r\n            console.error(`[McpMiddleware] Failed to parse args:`, argsString);\r\n            return {};\r\n        }\r\n    }\r\n\r\n    private async executeTool(toolName: string, args: Record<string, any>): Promise<string> {\r\n        const tool = this.tools.find(t => t.name === toolName);\r\n        if (!tool?.handler) {\r\n            return `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}`;\r\n        }\r\n\r\n        try {\n            console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\n            const result = await tool.handler(args);\n            const resultStr = typeof result === 'string' ? result : JSON.stringify(result);\n\n            console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\n            return resultStr;\n        } catch (error: any) {\r\n            console.error(`[McpMiddleware] Error executing tool:`, error);\r\n            return `Error: ${error.message || String(error)}`;\r\n        }\r\n    }\r\n\r\n    private generateId(prefix: string): string {\r\n        return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\r\n    }\r\n\r\n    private ensureIds(input: RunAgentInput): void {\r\n        const anyInput = input as any;\r\n        if (!anyInput.threadId) anyInput.threadId = this.generateId('mcp_thread');\r\n        if (!anyInput.runId) anyInput.runId = this.generateId('mcp_run');\r\n    }\r\n\r\n    /** Process tool call events and update state */\r\n    private handleToolCallEvent(event: BaseEvent, state: RunState): void {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n\r\n        // Accumulate text content for reconstruction\r\n        if (event.type === EventType.TEXT_MESSAGE_CHUNK) {\r\n            const e = event as any;\r\n            if (e.delta) {\r\n                state.textContent = (state.textContent || '') + e.delta;\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_START) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.toolCallName) {\r\n                toolCallNames.set(e.toolCallId, e.toolCallName);\r\n                if (this.isMcpTool(e.toolCallName)) {\r\n                    pendingMcpCalls.add(e.toolCallId);\r\n                }\r\n                console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_ARGS) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.delta) {\r\n                const existing = toolCallArgsBuffer.get(e.toolCallId) || '';\r\n                toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_END) {\r\n            const e = event as ToolCallEndEvent;\r\n            console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? 'unknown'} (id: ${e.toolCallId})`);\r\n        }\r\n\r\n        // Workaround: Extract parallel tool calls from MESSAGES_SNAPSHOT\r\n        if (event.type === EventType.MESSAGES_SNAPSHOT) {\r\n            const messages = (event as any).messages || [];\r\n            if (messages.length > 0) {\r\n                const lastMsg = messages[messages.length - 1];\r\n                // Update text content from snapshot if available (often more reliable)\r\n                if (lastMsg.role === 'assistant' && lastMsg.content) {\r\n                    state.textContent = lastMsg.content;\r\n                }\r\n\r\n                // Discover tools\r\n                for (let i = messages.length - 1; i >= 0; i--) {\r\n                    const msg = messages[i];\r\n                    const tools = Array.isArray(msg.toolCalls) ? msg.toolCalls :\r\n                        (Array.isArray(msg.tool_calls) ? msg.tool_calls : []);\r\n\r\n                    if (msg.role === 'assistant' && tools.length > 0) {\r\n                        for (const tc of tools) {\r\n                            if (tc.id && tc.function?.name && !toolCallNames.has(tc.id)) {\r\n                                toolCallNames.set(tc.id, tc.function.name);\r\n                                toolCallArgsBuffer.set(tc.id, tc.function.arguments || '{}');\r\n                                if (this.isMcpTool(tc.function.name)) {\r\n                                    pendingMcpCalls.add(tc.id);\r\n                                    console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);\r\n                                }\r\n                            }\r\n                        }\r\n                        break;\r\n                    }\r\n                }\r\n            }\r\n        }\r\n    }\r\n\r\n    /** Execute pending MCP tools and return results */\r\n    private async executeTools(state: RunState): Promise<ToolResult[]> {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n        const results: ToolResult[] = [];\r\n\r\n        const promises = [...pendingMcpCalls].map(async (toolCallId) => {\r\n            const toolName = toolCallNames.get(toolCallId);\r\n            if (!toolName) return;\r\n\r\n            const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || '{}');\r\n            console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\r\n\r\n            const result = await this.executeTool(toolName, args);\r\n            results.push({\r\n                toolCallId,\r\n                toolName,\r\n                result,\r\n                messageId: this.generateId('mcp_result'),\r\n            });\r\n            pendingMcpCalls.delete(toolCallId);\r\n        });\r\n\r\n        await Promise.all(promises);\r\n        return results;\r\n    }\r\n\r\n    /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */\r\n    private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\r\n        for (const { toolCallId, toolName, result, messageId } of results) {\r\n            observer.next({\r\n                type: EventType.TOOL_CALL_RESULT,\r\n                toolCallId,\r\n                messageId,\r\n                content: result,\r\n                role: 'tool',\r\n                timestamp: Date.now(),\r\n            } as any);\r\n            console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\r\n        }\r\n    }\r\n\r\n    run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\r\n        return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\r\n            const state: RunState = {\r\n                toolCallArgsBuffer: new Map(),\r\n                toolCallNames: new Map(),\r\n                pendingMcpCalls: new Set(),\r\n                textContent: '',\r\n                error: false,\r\n            };\r\n\r\n            this.ensureIds(input);\r\n            const anyInput = input as any;\r\n\r\n            console.log(`[McpMiddleware] === NEW RUN ===`);\r\n            console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);\r\n            console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);\r\n\r\n            // Inject MCP tools\r\n            if (this.toolSchemas?.length) {\r\n                input.tools = [...(input.tools || []), ...this.toolSchemas];\r\n                console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t: Tool) => t.name));\r\n            }\r\n\r\n            const handleRunFinished = async () => {\r\n                if (state.error) return; // Don't continue after error\r\n\r\n                if (state.pendingMcpCalls.size === 0) {\r\n                    observer.next({\r\n                        type: EventType.RUN_FINISHED,\r\n                        threadId: anyInput.threadId,\r\n                        runId: anyInput.runId,\r\n                        timestamp: Date.now(),\r\n                    } as any);\r\n                    observer.complete();\r\n                    return;\r\n                }\r\n\r\n                console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);\r\n\r\n                // Reconstruct the Assistant Message that triggered these tools\r\n                const toolCalls = [];\r\n                for (const toolCallId of state.pendingMcpCalls) {\r\n                    const name = state.toolCallNames.get(toolCallId);\r\n                    const args = state.toolCallArgsBuffer.get(toolCallId) || '{}';\r\n                    if (name) {\r\n                        toolCalls.push({\r\n                            id: toolCallId,\r\n                            type: 'function',\r\n                            function: { name, arguments: args }\r\n                        });\r\n                    }\r\n                }\r\n\r\n                // Add the Assistant Message to history FIRST\r\n                if (toolCalls.length > 0 || state.textContent) {\r\n                    const assistantMsg = {\r\n                        id: this.generateId('msg_ast'),\r\n                        role: 'assistant',\r\n                        content: state.textContent || null, // Ensure null if empty string for strict LLMs\r\n                        tool_calls: toolCalls.length > 0 ? toolCalls : undefined\r\n                    };\r\n                    input.messages.push(assistantMsg as any);\r\n                    console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);\r\n                }\r\n\r\n                // Execute tools and emit results (no RUN_FINISHED yet - continuation follows)\r\n                const results = await this.executeTools(state);\r\n                this.emitToolResults(observer, results);\r\n\r\n                // Prepare continuation\r\n                console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);\r\n\r\n                // Add tool result messages to history\r\n                for (const { toolCallId, result, messageId } of results) {\r\n                    input.messages.push({\r\n                        id: messageId,\r\n                        role: 'tool',\r\n                        tool_call_id: toolCallId,\r\n                        content: result,\r\n                    } as any);\r\n                }\r\n\r\n                // Reset state for next turn\r\n                state.toolCallArgsBuffer.clear();\r\n                state.toolCallNames.clear();\r\n                state.textContent = ''; // Clear text content for next turn\r\n\r\n                anyInput.runId = this.generateId('mcp_run');\r\n                console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);\r\n\r\n                // Subscribe to continuation\r\n                next.run(input).subscribe({\r\n                    next: (event) => {\r\n                        if (state.error) return;\r\n\r\n                        this.handleToolCallEvent(event, state);\r\n\r\n                        if (event.type === EventType.RUN_ERROR) {\r\n                            console.log(`[McpMiddleware] RUN_ERROR received in continuation`);\r\n                            state.error = true;\r\n                            observer.next(event);\r\n                            observer.complete();\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_STARTED) {\r\n                            console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_FINISHED) {\r\n                            if (state.pendingMcpCalls.size > 0) {\r\n                                handleRunFinished();\r\n                            } else {\r\n                                observer.next(event);\r\n                                observer.complete();\r\n                            }\r\n                            return;\r\n                        }\r\n                        observer.next(event);\r\n                    },\r\n                    error: (err) => {\r\n                        state.error = true;\r\n                        observer.error(err);\r\n                    },\r\n                    complete: () => {\r\n                        if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                    },\r\n                });\r\n            };\r\n\r\n            const subscription = next.run(input).subscribe({\r\n                next: (event) => {\r\n                    if (state.error) return;\r\n\r\n                    this.handleToolCallEvent(event, state);\r\n\r\n                    if (event.type === EventType.RUN_ERROR) {\r\n                        console.log(`[McpMiddleware] RUN_ERROR received`);\r\n                        state.error = true;\r\n                        observer.next(event);\r\n                        observer.complete();\r\n                        return;\r\n                    }\r\n\r\n                    if (event.type === EventType.RUN_FINISHED) {\r\n                        handleRunFinished();\r\n                        return;\r\n                    }\r\n                    observer.next(event);\r\n                },\r\n                error: (err) => {\r\n                    state.error = true;\r\n                    observer.error(err);\r\n                },\r\n                complete: () => {\r\n                    if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                },\r\n            });\r\n\r\n            return () => subscription.unsubscribe();\r\n        });\r\n    }\r\n}\r\n\r\n/**\r\n * Factory function to create MCP middleware.\r\n */\r\nexport function createMcpMiddleware(\n    options: { tools: AguiTool[] }\n) {\n    const middleware = new McpMiddleware(options);\r\n    return (input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> => {\r\n        return middleware.run(input, next);\r\n    };\r\n}\r\n\r\n// Legacy exports\r\nexport { McpMiddleware as McpToolExecutorMiddleware };\r\nexport { createMcpMiddleware as createMcpToolMiddleware };\r\n\r\n// Re-exports\r\nexport { Middleware, EventType };\r\nexport type { RunAgentInput, BaseEvent, AbstractAgent, ToolCallEndEvent, Tool };"]}
+{"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":["Middleware","EventType","Observable"],"mappings":";;;;;;;;;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;;;ACrCO,IAAM,aAAA,GAAN,cAA4BA,iBAAA,CAAW;AAAA,EAI1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAJV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU;AAAA,KACxC,CAAE,CAAA;AAAA,EACN;AAAA,EAEQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAAA,EACnD;AAAA,EAEQ,UAAU,UAAA,EAAyC;AACvD,IAAA,IAAI,CAAC,UAAA,EAAY,IAAA,EAAK,SAAU,EAAC;AAEjC,IAAA,IAAI;AACA,MAAA,OAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,IAChC,CAAA,CAAA,MAAQ;AAEJ,MAAA,MAAM,OAAA,GAAU,WAAW,IAAA,EAAK;AAChC,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,IAAI,CAAA,EAAG;AACxB,QAAA,MAAM,WAAA,GAAc,QAAQ,KAAA,CAAM,CAAA,EAAG,QAAQ,OAAA,CAAQ,IAAI,IAAI,CAAC,CAAA;AAC9D,QAAA,IAAI;AACA,UAAA,OAAO,IAAA,CAAK,MAAM,WAAW,CAAA;AAAA,QACjC,CAAA,CAAA,MAAQ;AACJ,UAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,WAAW,CAAA;AAAA,QACtE;AAAA,MACJ;AACA,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AACjE,MAAA,OAAO,EAAC;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,OAAO,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACrD,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAChB,MAAA,OAAO,CAAA,YAAA,EAAe,IAAA,GAAO,gBAAA,GAAmB,WAAW,KAAK,QAAQ,CAAA,CAAA;AAAA,IAC5E;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AACtC,MAAA,MAAM,YAAY,OAAO,MAAA,KAAW,WAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAE7E,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,SAAA;AAAA,IACX,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,OAAA,EAAU,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IACnD;AAAA,EACJ;AAAA,EAEQ,WAAW,MAAA,EAAwB;AACvC,IAAA,OAAO,GAAG,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC5E;AAAA,EAEQ,UAAU,KAAA,EAA4B;AAC1C,IAAA,MAAM,QAAA,GAAW,KAAA;AACjB,IAAA,IAAI,CAAC,QAAA,CAAS,QAAA,WAAmB,QAAA,GAAW,IAAA,CAAK,WAAW,YAAY,CAAA;AACxE,IAAA,IAAI,CAAC,QAAA,CAAS,KAAA,WAAgB,KAAA,GAAQ,IAAA,CAAK,WAAW,SAAS,CAAA;AAAA,EACnE;AAAA;AAAA,EAGQ,mBAAA,CAAoB,OAAkB,KAAA,EAAuB;AACjE,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAG/D,IAAA,IAAI,KAAA,CAAM,IAAA,KAASC,gBAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AACA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iCAAA,EAAoC,CAAA,CAAE,YAAY,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,SAAA,EAAY,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAC,CAAA,CAAA,CAAG,CAAA;AAAA,MACpI;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,aAAA,EAAe;AACxC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+BAAA,EAAkC,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,SAAS,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACtH;AAGA,IAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;AACzB,kBAAA,OAAA,CAAQ,GAAA,CAAI,iDAAiD,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,MAAA,EAAS,EAAA,CAAG,EAAE,CAAA,CAAA,CAAG,CAAA;AAAA,gBAClG;AAAA,cACJ;AAAA,YACJ;AACA,YAAA;AAAA,UACJ;AAAA,QACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAc,aAAa,KAAA,EAAwC;AAC/D,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAC/D,IAAA,MAAM,UAAwB,EAAC;AAE/B,IAAA,MAAM,WAAW,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAC5D,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,OAAO,IAAA,CAAK,SAAA,CAAU,mBAAmB,GAAA,CAAI,UAAU,KAAK,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AAEjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,UAAA,CAAW,YAAY;AAAA,OAC1C,CAAA;AACD,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,QAAQ,CAAA;AAC1B,IAAA,OAAO,OAAA;AAAA,EACX;AAAA;AAAA,EAGQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AAC/D,MAAA,QAAA,CAAS,IAAA,CAAK;AAAA,QACV,MAAMA,gBAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AACR,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AAAA,IAC5E;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAIC,eAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;AAEjB,MAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,CAAiC,CAAA;AAC7C,MAAA,OAAA,CAAQ,IAAI,CAAA,0BAAA,EAA6B,QAAA,CAAS,QAAQ,CAAA,SAAA,EAAY,QAAA,CAAS,KAAK,CAAA,CAAE,CAAA;AACtF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,0BAAA,EAA6B,KAAA,CAAM,QAAA,EAAU,MAAA,IAAU,CAAC,CAAA,SAAA,EAAY,IAAA,CAAK,KAAA,EAAO,MAAA,IAAU,CAAC,CAAA,CAAE,CAAA;AAGzG,MAAA,IAAI,IAAA,CAAK,aAAa,MAAA,EAAQ;AAC1B,QAAA,KAAA,CAAM,KAAA,GAAQ,CAAC,GAAI,KAAA,CAAM,SAAS,EAAC,EAAI,GAAG,IAAA,CAAK,WAAW,CAAA;AAC1D,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,WAAA,CAAY,MAAM,CAAA,OAAA,CAAA,EAAW,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MACvH;AAEA,MAAA,MAAM,oBAAoB,YAAY;AAClC,QAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,QAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,KAAS,CAAA,EAAG;AAClC,UAAA,QAAA,CAAS,IAAA,CAAK;AAAA,YACV,MAAMD,gBAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,KAAA,CAAM,eAAA,CAAgB,IAAI,CAAA,cAAA,CAAgB,CAAA;AAG3F,QAAA,MAAM,YAAY,EAAC;AACnB,QAAA,KAAA,MAAW,UAAA,IAAc,MAAM,eAAA,EAAiB;AAC5C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC/C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,IAAA,EAAM;AACN,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACX,EAAA,EAAI,UAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA;AAAK,aACrC,CAAA;AAAA,UACL;AAAA,QACJ;AAGA,QAAA,IAAI,SAAA,CAAU,MAAA,GAAS,CAAA,IAAK,KAAA,CAAM,WAAA,EAAa;AAC3C,UAAA,MAAM,YAAA,GAAe;AAAA,YACjB,EAAA,EAAI,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAAA,YAC7B,IAAA,EAAM,WAAA;AAAA,YACN,OAAA,EAAS,MAAM,WAAA,IAAe,IAAA;AAAA;AAAA,YAC9B,UAAA,EAAY,SAAA,CAAU,MAAA,GAAS,CAAA,GAAI,SAAA,GAAY;AAAA,WACnD;AACA,UAAA,KAAA,CAAM,QAAA,CAAS,KAAK,YAAmB,CAAA;AACvC,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iEAAA,EAAoE,KAAA,CAAM,WAAA,EAAa,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,KAAA,EAAQ,SAAA,CAAU,MAAM,CAAA,OAAA,CAAS,CAAA;AAAA,QACpJ;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6CAAA,EAAgD,OAAA,CAAQ,MAAM,CAAA,QAAA,CAAU,CAAA;AAGpF,QAAA,KAAA,MAAW,EAAE,UAAA,EAAY,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AACrD,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,YAAA,EAAc,UAAA;AAAA,YACd,OAAA,EAAS;AAAA,WACL,CAAA;AAAA,QACZ;AAGA,QAAA,KAAA,CAAM,mBAAmB,KAAA,EAAM;AAC/B,QAAA,KAAA,CAAM,cAAc,KAAA,EAAM;AAC1B,QAAA,KAAA,CAAM,WAAA,GAAc,EAAA;AAEpB,QAAA,QAAA,CAAS,KAAA,GAAQ,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAC1C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,mDAAA,EAAsD,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,CAAE,CAAA;AAGzF,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,CAAA,CAAE,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,YAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,YAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,YAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,SAAA,EAAW;AACpC,cAAA,OAAA,CAAQ,IAAI,CAAA,kDAAA,CAAoD,CAAA;AAChE,cAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,cAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,cAAA,QAAA,CAAS,QAAA,EAAS;AAClB,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,WAAA,EAAa;AACtC,cAAA,OAAA,CAAQ,IAAI,CAAA,uDAAA,CAAyD,CAAA;AACrE,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,SAAA,EAAW;AACpC,YAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAASA,gBAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,oBACZ,OAAA,EACF;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc,OAAO,CAAA;AAC5C,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.js","sourcesContent":["/**\n * MCP Adapter for AG-UI Integration\n *\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\n * It provides tools with handlers for server-side execution and tool definitions\n * in JSON Schema format for passing to remote agents.\n *\n * @example\n * ```typescript\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n * import { HttpAgent } from '@ag-ui/client';\n *\n * // Create MCP client\n * const mcpClient = new MultiSessionClient('user_123');\n * await mcpClient.connect();\n *\n * // Create adapter and get tools\n * const adapter = new AguiAdapter(mcpClient);\n * const tools = await adapter.getTools();\n *\n * // Use with AG-UI middleware\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(createMcpMiddleware({ tools }));\n * ```\n */\n\nimport { MCPClient } from '../server/mcp/oauth-client.js';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\n\n/**\n * Extended JSON Schema properties that Pydantic's strict validation rejects.\n * These are valid JSON Schema extensions but not part of the core spec.\n */\nconst PYDANTIC_FORBIDDEN_PROPS = [\n    // JSON Schema meta-properties\n    '$schema', '$id', '$comment', '$defs', 'definitions',\n    // Extended properties used by some MCP servers (e.g., Apify)\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\n    // Other common extensions\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\n];\n\n/**\n * Cleans a JSON Schema by removing meta-properties that cause issues with\n * strict Pydantic validation (e.g., Google ADK, LangGraph).\n *\n * @param schema - The JSON Schema to clean\n * @returns Cleaned schema without forbidden properties\n */\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\n    if (!schema) {\n        return { type: 'object', properties: {} };\n    }\n\n    const cleaned = { ...schema };\n\n    // Remove all forbidden properties\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\n        delete cleaned[prop];\n    }\n\n    // Recursively clean nested properties\n    if (cleaned.properties && typeof cleaned.properties === 'object') {\n        const cleanedProps: Record<string, any> = {};\n        for (const [key, value] of Object.entries(cleaned.properties)) {\n            if (typeof value === 'object' && value !== null) {\n                cleanedProps[key] = cleanSchema(value as Record<string, any>);\n            } else {\n                cleanedProps[key] = value;\n            }\n        }\n        cleaned.properties = cleanedProps;\n    }\n\n    // Clean items if it's an array schema\n    if (cleaned.items && typeof cleaned.items === 'object') {\n        cleaned.items = cleanSchema(cleaned.items);\n    }\n\n    // Clean additionalProperties if it's an object schema\n    if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\n        cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\n    }\n\n    return cleaned;\n}\n\n/**\n * Configuration options for AguiAdapter\n */\nexport interface AguiAdapterOptions {\n    /**\n     * Prefix for tool names to avoid collision with other tools.\n     * @default serverId or 'mcp'\n     */\n    prefix?: string;\n}\n\n/**\n * AG-UI Tool with handler for server-side execution.\n */\nexport interface AguiTool {\n    name: string;\n    description: string;\n    parameters?: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiTool\n    handler?: (args: any) => any | Promise<any>;\n}\n\n/**\n * Tool definition format for passing to remote agents (without handler).\n */\nexport interface AguiToolDefinition {\n    name: string;\n    description: string;\n    parameters: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\n}\n\n/**\n * Adapter that transforms MCP tools into AG-UI compatible formats.\n */\nexport class AguiAdapter {\n    constructor(\n        private client: MCPClient | MultiSessionClient,\n        private options: AguiAdapterOptions = {}\n    ) { }\n\n    /**\n     * Get tools with handlers for MCP tool execution.\n     */\n    async getTools(): Promise<AguiTool[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiTool[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformTools(client));\n            }\n            return allTools;\n        }\n        return this.transformTools(this.client as MCPClient);\n    }\n\n    /**\n     * Get tool definitions in JSON Schema format for passing to remote agents.\n     */\n    async getToolDefinitions(): Promise<AguiToolDefinition[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiToolDefinition[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformToolDefinitions(client));\n            }\n            return allTools;\n        }\n        return this.transformToolDefinitions(this.client as MCPClient);\n    }\n\n    /**\n     * Get tools as a function (for dynamic loading).\n     */\n    getToolsFunction(): () => Promise<AguiTool[]> {\n        return () => this.getTools();\n    }\n\n    private isMultiSession(): boolean {\n        return typeof (this.client as any).getClients === 'function';\n    }\n\n    private async transformTools(client: MCPClient): Promise<AguiTool[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\n            const mcpTool = tool as any;\n            const mcpToolName = tool.name;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n                handler: async (args: any) => {\n                    // Call the actual MCP tool\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\n\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\n                    return callResult;\n                }\n            }\n        });\n    }\n\n    private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            const mcpTool = tool as any;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n            };\n        });\n    }\n}\n","/**\r\n * AG-UI Middleware for MCP Tool Execution\r\n *\r\n * This middleware intercepts tool calls from remote agents and executes\r\n * MCP tools server-side, returning results back to the agent.\r\n *\r\n * @requires @ag-ui/client - Peer dependency for AG-UI types\r\n * @requires rxjs - Uses RxJS Observables for event streaming\r\n */\r\n\r\nimport { Observable, Subscriber } from 'rxjs';\r\nimport {\r\n    Middleware,\r\n    EventType,\r\n    type AbstractAgent,\r\n    type RunAgentInput,\r\n    type BaseEvent,\r\n    type ToolCallEndEvent,\r\n    type Tool,\r\n} from '@ag-ui/client';\r\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\r\n\r\n/** Tool execution result for continuation */\r\ninterface ToolResult {\r\n    toolCallId: string;\r\n    toolName: string;\r\n    result: string;\r\n    messageId: string;\r\n}\r\n\r\n/** State for tracking tool calls during a run */\r\ninterface RunState {\r\n    toolCallArgsBuffer: Map<string, string>;\r\n    toolCallNames: Map<string, string>;\r\n    pendingMcpCalls: Set<string>;\r\n    textContent?: string;\r\n    error: boolean;\r\n}\r\n\r\n/**\n * Configuration for McpMiddleware\n */\nexport interface McpMiddlewareConfig {\n    /** Pre-loaded tools with handlers (required) */\n    tools: AguiTool[];\n}\n\r\n/**\r\n * AG-UI Middleware that executes MCP tools server-side.\r\n */\r\nexport class McpMiddleware extends Middleware {\n    private tools: AguiTool[];\n    private toolSchemas: Tool[];\n\n    constructor(config: McpMiddlewareConfig) {\n        super();\n        this.tools = config.tools;\n        this.toolSchemas = this.tools.map((t: AguiTool) => ({\n            name: t.name,\n            description: t.description,\n            parameters: cleanSchema(t.parameters),\n        }));\n    }\n\r\n    private isMcpTool(toolName: string): boolean {\r\n        return this.tools.some(t => t.name === toolName);\r\n    }\r\n\r\n    private parseArgs(argsString: string): Record<string, any> {\r\n        if (!argsString?.trim()) return {};\r\n\r\n        try {\r\n            return JSON.parse(argsString);\r\n        } catch {\r\n            // Handle duplicated JSON from streaming issues: {...}{...}\r\n            const trimmed = argsString.trim();\r\n            if (trimmed.includes('}{')) {\r\n                const firstObject = trimmed.slice(0, trimmed.indexOf('}{') + 1);\r\n                try {\r\n                    return JSON.parse(firstObject);\r\n                } catch {\r\n                    console.error(`[McpMiddleware] Failed to parse JSON:`, firstObject);\r\n                }\r\n            }\r\n            console.error(`[McpMiddleware] Failed to parse args:`, argsString);\r\n            return {};\r\n        }\r\n    }\r\n\r\n    private async executeTool(toolName: string, args: Record<string, any>): Promise<string> {\r\n        const tool = this.tools.find(t => t.name === toolName);\r\n        if (!tool?.handler) {\r\n            return `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}`;\r\n        }\r\n\r\n        try {\n            console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\n            const result = await tool.handler(args);\n            const resultStr = typeof result === 'string' ? result : JSON.stringify(result);\n\n            console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\n            return resultStr;\n        } catch (error: any) {\r\n            console.error(`[McpMiddleware] Error executing tool:`, error);\r\n            return `Error: ${error.message || String(error)}`;\r\n        }\r\n    }\r\n\r\n    private generateId(prefix: string): string {\r\n        return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\r\n    }\r\n\r\n    private ensureIds(input: RunAgentInput): void {\r\n        const anyInput = input as any;\r\n        if (!anyInput.threadId) anyInput.threadId = this.generateId('mcp_thread');\r\n        if (!anyInput.runId) anyInput.runId = this.generateId('mcp_run');\r\n    }\r\n\r\n    /** Process tool call events and update state */\r\n    private handleToolCallEvent(event: BaseEvent, state: RunState): void {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n\r\n        // Accumulate text content for reconstruction\r\n        if (event.type === EventType.TEXT_MESSAGE_CHUNK) {\r\n            const e = event as any;\r\n            if (e.delta) {\r\n                state.textContent = (state.textContent || '') + e.delta;\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_START) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.toolCallName) {\r\n                toolCallNames.set(e.toolCallId, e.toolCallName);\r\n                if (this.isMcpTool(e.toolCallName)) {\r\n                    pendingMcpCalls.add(e.toolCallId);\r\n                }\r\n                console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_ARGS) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.delta) {\r\n                const existing = toolCallArgsBuffer.get(e.toolCallId) || '';\r\n                toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_END) {\r\n            const e = event as ToolCallEndEvent;\r\n            console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? 'unknown'} (id: ${e.toolCallId})`);\r\n        }\r\n\r\n        // Workaround: Extract parallel tool calls from MESSAGES_SNAPSHOT\r\n        if (event.type === EventType.MESSAGES_SNAPSHOT) {\r\n            const messages = (event as any).messages || [];\r\n            if (messages.length > 0) {\r\n                const lastMsg = messages[messages.length - 1];\r\n                // Update text content from snapshot if available (often more reliable)\r\n                if (lastMsg.role === 'assistant' && lastMsg.content) {\r\n                    state.textContent = lastMsg.content;\r\n                }\r\n\r\n                // Discover tools\r\n                for (let i = messages.length - 1; i >= 0; i--) {\r\n                    const msg = messages[i];\r\n                    const tools = Array.isArray(msg.toolCalls) ? msg.toolCalls :\r\n                        (Array.isArray(msg.tool_calls) ? msg.tool_calls : []);\r\n\r\n                    if (msg.role === 'assistant' && tools.length > 0) {\r\n                        for (const tc of tools) {\r\n                            if (tc.id && tc.function?.name && !toolCallNames.has(tc.id)) {\r\n                                toolCallNames.set(tc.id, tc.function.name);\r\n                                toolCallArgsBuffer.set(tc.id, tc.function.arguments || '{}');\r\n                                if (this.isMcpTool(tc.function.name)) {\r\n                                    pendingMcpCalls.add(tc.id);\r\n                                    console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);\r\n                                }\r\n                            }\r\n                        }\r\n                        break;\r\n                    }\r\n                }\r\n            }\r\n        }\r\n    }\r\n\r\n    /** Execute pending MCP tools and return results */\r\n    private async executeTools(state: RunState): Promise<ToolResult[]> {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n        const results: ToolResult[] = [];\r\n\r\n        const promises = [...pendingMcpCalls].map(async (toolCallId) => {\r\n            const toolName = toolCallNames.get(toolCallId);\r\n            if (!toolName) return;\r\n\r\n            const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || '{}');\r\n            console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\r\n\r\n            const result = await this.executeTool(toolName, args);\r\n            results.push({\r\n                toolCallId,\r\n                toolName,\r\n                result,\r\n                messageId: this.generateId('mcp_result'),\r\n            });\r\n            pendingMcpCalls.delete(toolCallId);\r\n        });\r\n\r\n        await Promise.all(promises);\r\n        return results;\r\n    }\r\n\r\n    /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */\r\n    private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\r\n        for (const { toolCallId, toolName, result, messageId } of results) {\r\n            observer.next({\r\n                type: EventType.TOOL_CALL_RESULT,\r\n                toolCallId,\r\n                messageId,\r\n                content: result,\r\n                role: 'tool',\r\n                timestamp: Date.now(),\r\n            } as any);\r\n            console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\r\n        }\r\n    }\r\n\r\n    run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\r\n        return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\r\n            const state: RunState = {\r\n                toolCallArgsBuffer: new Map(),\r\n                toolCallNames: new Map(),\r\n                pendingMcpCalls: new Set(),\r\n                textContent: '',\r\n                error: false,\r\n            };\r\n\r\n            this.ensureIds(input);\r\n            const anyInput = input as any;\r\n\r\n            console.log(`[McpMiddleware] === NEW RUN ===`);\r\n            console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);\r\n            console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);\r\n\r\n            // Inject MCP tools\r\n            if (this.toolSchemas?.length) {\r\n                input.tools = [...(input.tools || []), ...this.toolSchemas];\r\n                console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t: Tool) => t.name));\r\n            }\r\n\r\n            const handleRunFinished = async () => {\r\n                if (state.error) return; // Don't continue after error\r\n\r\n                if (state.pendingMcpCalls.size === 0) {\r\n                    observer.next({\r\n                        type: EventType.RUN_FINISHED,\r\n                        threadId: anyInput.threadId,\r\n                        runId: anyInput.runId,\r\n                        timestamp: Date.now(),\r\n                    } as any);\r\n                    observer.complete();\r\n                    return;\r\n                }\r\n\r\n                console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);\r\n\r\n                // Reconstruct the Assistant Message that triggered these tools\r\n                const toolCalls = [];\r\n                for (const toolCallId of state.pendingMcpCalls) {\r\n                    const name = state.toolCallNames.get(toolCallId);\r\n                    const args = state.toolCallArgsBuffer.get(toolCallId) || '{}';\r\n                    if (name) {\r\n                        toolCalls.push({\r\n                            id: toolCallId,\r\n                            type: 'function',\r\n                            function: { name, arguments: args }\r\n                        });\r\n                    }\r\n                }\r\n\r\n                // Add the Assistant Message to history FIRST\r\n                if (toolCalls.length > 0 || state.textContent) {\r\n                    const assistantMsg = {\r\n                        id: this.generateId('msg_ast'),\r\n                        role: 'assistant',\r\n                        content: state.textContent || null, // Ensure null if empty string for strict LLMs\r\n                        tool_calls: toolCalls.length > 0 ? toolCalls : undefined\r\n                    };\r\n                    input.messages.push(assistantMsg as any);\r\n                    console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);\r\n                }\r\n\r\n                // Execute tools and emit results (no RUN_FINISHED yet - continuation follows)\r\n                const results = await this.executeTools(state);\r\n                this.emitToolResults(observer, results);\r\n\r\n                // Prepare continuation\r\n                console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);\r\n\r\n                // Add tool result messages to history\r\n                for (const { toolCallId, result, messageId } of results) {\r\n                    input.messages.push({\r\n                        id: messageId,\r\n                        role: 'tool',\r\n                        tool_call_id: toolCallId,\r\n                        content: result,\r\n                    } as any);\r\n                }\r\n\r\n                // Reset state for next turn\r\n                state.toolCallArgsBuffer.clear();\r\n                state.toolCallNames.clear();\r\n                state.textContent = ''; // Clear text content for next turn\r\n\r\n                anyInput.runId = this.generateId('mcp_run');\r\n                console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);\r\n\r\n                // Subscribe to continuation\r\n                next.run(input).subscribe({\r\n                    next: (event) => {\r\n                        if (state.error) return;\r\n\r\n                        this.handleToolCallEvent(event, state);\r\n\r\n                        if (event.type === EventType.RUN_ERROR) {\r\n                            console.log(`[McpMiddleware] RUN_ERROR received in continuation`);\r\n                            state.error = true;\r\n                            observer.next(event);\r\n                            observer.complete();\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_STARTED) {\r\n                            console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_FINISHED) {\r\n                            if (state.pendingMcpCalls.size > 0) {\r\n                                handleRunFinished();\r\n                            } else {\r\n                                observer.next(event);\r\n                                observer.complete();\r\n                            }\r\n                            return;\r\n                        }\r\n                        observer.next(event);\r\n                    },\r\n                    error: (err) => {\r\n                        state.error = true;\r\n                        observer.error(err);\r\n                    },\r\n                    complete: () => {\r\n                        if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                    },\r\n                });\r\n            };\r\n\r\n            const subscription = next.run(input).subscribe({\r\n                next: (event) => {\r\n                    if (state.error) return;\r\n\r\n                    this.handleToolCallEvent(event, state);\r\n\r\n                    if (event.type === EventType.RUN_ERROR) {\r\n                        console.log(`[McpMiddleware] RUN_ERROR received`);\r\n                        state.error = true;\r\n                        observer.next(event);\r\n                        observer.complete();\r\n                        return;\r\n                    }\r\n\r\n                    if (event.type === EventType.RUN_FINISHED) {\r\n                        handleRunFinished();\r\n                        return;\r\n                    }\r\n                    observer.next(event);\r\n                },\r\n                error: (err) => {\r\n                    state.error = true;\r\n                    observer.error(err);\r\n                },\r\n                complete: () => {\r\n                    if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                },\r\n            });\r\n\r\n            return () => subscription.unsubscribe();\r\n        });\r\n    }\r\n}\r\n\r\n/**\r\n * Factory function to create MCP middleware.\r\n */\r\nexport function createMcpMiddleware(\n    options: { tools: AguiTool[] }\n) {\n    const middleware = new McpMiddleware(options);\r\n    return (input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> => {\r\n        return middleware.run(input, next);\r\n    };\r\n}\r\n\r\n// Legacy exports\r\nexport { McpMiddleware as McpToolExecutorMiddleware };\r\nexport { createMcpMiddleware as createMcpToolMiddleware };\r\n\r\n// Re-exports\r\nexport { Middleware, EventType };\r\nexport type { RunAgentInput, BaseEvent, AbstractAgent, ToolCallEndEvent, Tool };"]}

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

@@ -1 +1 @@
-{"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":[],"mappings":";;;;;;;;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;;;ACrCO,IAAM,aAAA,GAAN,cAA4B,UAAA,CAAW;AAAA,EAI1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAJV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU;AAAA,KACxC,CAAE,CAAA;AAAA,EACN;AAAA,EAEQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAAA,EACnD;AAAA,EAEQ,UAAU,UAAA,EAAyC;AACvD,IAAA,IAAI,CAAC,UAAA,EAAY,IAAA,EAAK,SAAU,EAAC;AAEjC,IAAA,IAAI;AACA,MAAA,OAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,IAChC,CAAA,CAAA,MAAQ;AAEJ,MAAA,MAAM,OAAA,GAAU,WAAW,IAAA,EAAK;AAChC,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,IAAI,CAAA,EAAG;AACxB,QAAA,MAAM,WAAA,GAAc,QAAQ,KAAA,CAAM,CAAA,EAAG,QAAQ,OAAA,CAAQ,IAAI,IAAI,CAAC,CAAA;AAC9D,QAAA,IAAI;AACA,UAAA,OAAO,IAAA,CAAK,MAAM,WAAW,CAAA;AAAA,QACjC,CAAA,CAAA,MAAQ;AACJ,UAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,WAAW,CAAA;AAAA,QACtE;AAAA,MACJ;AACA,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AACjE,MAAA,OAAO,EAAC;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,OAAO,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACrD,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAChB,MAAA,OAAO,CAAA,YAAA,EAAe,IAAA,GAAO,gBAAA,GAAmB,WAAW,KAAK,QAAQ,CAAA,CAAA;AAAA,IAC5E;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AACtC,MAAA,MAAM,YAAY,OAAO,MAAA,KAAW,WAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAE7E,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,SAAA;AAAA,IACX,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,OAAA,EAAU,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IACnD;AAAA,EACJ;AAAA,EAEQ,WAAW,MAAA,EAAwB;AACvC,IAAA,OAAO,GAAG,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC5E;AAAA,EAEQ,UAAU,KAAA,EAA4B;AAC1C,IAAA,MAAM,QAAA,GAAW,KAAA;AACjB,IAAA,IAAI,CAAC,QAAA,CAAS,QAAA,WAAmB,QAAA,GAAW,IAAA,CAAK,WAAW,YAAY,CAAA;AACxE,IAAA,IAAI,CAAC,QAAA,CAAS,KAAA,WAAgB,KAAA,GAAQ,IAAA,CAAK,WAAW,SAAS,CAAA;AAAA,EACnE;AAAA;AAAA,EAGQ,mBAAA,CAAoB,OAAkB,KAAA,EAAuB;AACjE,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAG/D,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AACA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iCAAA,EAAoC,CAAA,CAAE,YAAY,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,SAAA,EAAY,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAC,CAAA,CAAA,CAAG,CAAA;AAAA,MACpI;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AACxC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+BAAA,EAAkC,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,SAAS,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACtH;AAGA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;AACzB,kBAAA,OAAA,CAAQ,GAAA,CAAI,iDAAiD,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,MAAA,EAAS,EAAA,CAAG,EAAE,CAAA,CAAA,CAAG,CAAA;AAAA,gBAClG;AAAA,cACJ;AAAA,YACJ;AACA,YAAA;AAAA,UACJ;AAAA,QACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAc,aAAa,KAAA,EAAwC;AAC/D,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAC/D,IAAA,MAAM,UAAwB,EAAC;AAE/B,IAAA,MAAM,WAAW,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAC5D,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,OAAO,IAAA,CAAK,SAAA,CAAU,mBAAmB,GAAA,CAAI,UAAU,KAAK,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AAEjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,UAAA,CAAW,YAAY;AAAA,OAC1C,CAAA;AACD,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,QAAQ,CAAA;AAC1B,IAAA,OAAO,OAAA;AAAA,EACX;AAAA;AAAA,EAGQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AAC/D,MAAA,QAAA,CAAS,IAAA,CAAK;AAAA,QACV,MAAM,SAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AACR,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AAAA,IAC5E;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAI,UAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;AAEjB,MAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,CAAiC,CAAA;AAC7C,MAAA,OAAA,CAAQ,IAAI,CAAA,0BAAA,EAA6B,QAAA,CAAS,QAAQ,CAAA,SAAA,EAAY,QAAA,CAAS,KAAK,CAAA,CAAE,CAAA;AACtF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,0BAAA,EAA6B,KAAA,CAAM,QAAA,EAAU,MAAA,IAAU,CAAC,CAAA,SAAA,EAAY,IAAA,CAAK,KAAA,EAAO,MAAA,IAAU,CAAC,CAAA,CAAE,CAAA;AAGzG,MAAA,IAAI,IAAA,CAAK,aAAa,MAAA,EAAQ;AAC1B,QAAA,KAAA,CAAM,KAAA,GAAQ,CAAC,GAAI,KAAA,CAAM,SAAS,EAAC,EAAI,GAAG,IAAA,CAAK,WAAW,CAAA;AAC1D,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,WAAA,CAAY,MAAM,CAAA,OAAA,CAAA,EAAW,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MACvH;AAEA,MAAA,MAAM,oBAAoB,YAAY;AAClC,QAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,QAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,KAAS,CAAA,EAAG;AAClC,UAAA,QAAA,CAAS,IAAA,CAAK;AAAA,YACV,MAAM,SAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,KAAA,CAAM,eAAA,CAAgB,IAAI,CAAA,cAAA,CAAgB,CAAA;AAG3F,QAAA,MAAM,YAAY,EAAC;AACnB,QAAA,KAAA,MAAW,UAAA,IAAc,MAAM,eAAA,EAAiB;AAC5C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC/C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,IAAA,EAAM;AACN,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACX,EAAA,EAAI,UAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA;AAAK,aACrC,CAAA;AAAA,UACL;AAAA,QACJ;AAGA,QAAA,IAAI,SAAA,CAAU,MAAA,GAAS,CAAA,IAAK,KAAA,CAAM,WAAA,EAAa;AAC3C,UAAA,MAAM,YAAA,GAAe;AAAA,YACjB,EAAA,EAAI,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAAA,YAC7B,IAAA,EAAM,WAAA;AAAA,YACN,OAAA,EAAS,MAAM,WAAA,IAAe,IAAA;AAAA;AAAA,YAC9B,UAAA,EAAY,SAAA,CAAU,MAAA,GAAS,CAAA,GAAI,SAAA,GAAY;AAAA,WACnD;AACA,UAAA,KAAA,CAAM,QAAA,CAAS,KAAK,YAAmB,CAAA;AACvC,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iEAAA,EAAoE,KAAA,CAAM,WAAA,EAAa,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,KAAA,EAAQ,SAAA,CAAU,MAAM,CAAA,OAAA,CAAS,CAAA;AAAA,QACpJ;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6CAAA,EAAgD,OAAA,CAAQ,MAAM,CAAA,QAAA,CAAU,CAAA;AAGpF,QAAA,KAAA,MAAW,EAAE,UAAA,EAAY,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AACrD,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,YAAA,EAAc,UAAA;AAAA,YACd,OAAA,EAAS;AAAA,WACL,CAAA;AAAA,QACZ;AAGA,QAAA,KAAA,CAAM,mBAAmB,KAAA,EAAM;AAC/B,QAAA,KAAA,CAAM,cAAc,KAAA,EAAM;AAC1B,QAAA,KAAA,CAAM,WAAA,GAAc,EAAA;AAEpB,QAAA,QAAA,CAAS,KAAA,GAAQ,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAC1C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,mDAAA,EAAsD,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,CAAE,CAAA;AAGzF,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,CAAA,CAAE,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,YAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,YAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,cAAA,OAAA,CAAQ,IAAI,CAAA,kDAAA,CAAoD,CAAA;AAChE,cAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,cAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,cAAA,QAAA,CAAS,QAAA,EAAS;AAClB,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,WAAA,EAAa;AACtC,cAAA,OAAA,CAAQ,IAAI,CAAA,uDAAA,CAAyD,CAAA;AACrE,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,YAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,oBACZ,OAAA,EACF;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc,OAAO,CAAA;AAC5C,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.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 * Extended JSON Schema properties that Pydantic's strict validation rejects.\r\n * These are valid JSON Schema extensions but not part of the core spec.\r\n */\r\nconst PYDANTIC_FORBIDDEN_PROPS = [\r\n    // JSON Schema meta-properties\r\n    '$schema', '$id', '$comment', '$defs', 'definitions',\r\n    // Extended properties used by some MCP servers (e.g., Apify)\r\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\r\n    // Other common extensions\r\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\r\n];\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, LangGraph).\r\n *\r\n * @param schema - The JSON Schema to clean\r\n * @returns Cleaned schema without forbidden 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 all forbidden properties\r\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\r\n        delete cleaned[prop];\r\n    }\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    _meta?: Record<string, any>; // Add _meta to AguiTool\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    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\r\n            const mcpTool = tool as any;\r\n            const mcpToolName = tool.name;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n                handler: async (args: any) => {\r\n                    // Call the actual MCP tool\r\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\r\n\r\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\r\n                    return callResult;\r\n                }\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 serverId = (typeof (client as any).getServerId === 'function'\r\n            ? (client as any).getServerId()\r\n            : undefined) as string | undefined;\r\n        const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n        const prefix = `tool_${normalizedPrefix}`;\r\n\r\n        return result.tools.map(tool => {\r\n            const mcpTool = tool as any;\r\n            return {\r\n                name: `${prefix}_${tool.name}`,\r\n                description: tool.description || `Execute ${tool.name}`,\r\n                parameters: cleanSchema(tool.inputSchema),\r\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n            };\r\n        });\r\n    }\r\n}\r\n","/**\r\n * AG-UI Middleware for MCP Tool Execution\r\n *\r\n * This middleware intercepts tool calls from remote agents and executes\r\n * MCP tools server-side, returning results back to the agent.\r\n *\r\n * @requires @ag-ui/client - Peer dependency for AG-UI types\r\n * @requires rxjs - Uses RxJS Observables for event streaming\r\n */\r\n\r\nimport { Observable, Subscriber } from 'rxjs';\r\nimport {\r\n    Middleware,\r\n    EventType,\r\n    type AbstractAgent,\r\n    type RunAgentInput,\r\n    type BaseEvent,\r\n    type ToolCallEndEvent,\r\n    type Tool,\r\n} from '@ag-ui/client';\r\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\r\n\r\n/** Tool execution result for continuation */\r\ninterface ToolResult {\r\n    toolCallId: string;\r\n    toolName: string;\r\n    result: string;\r\n    messageId: string;\r\n}\r\n\r\n/** State for tracking tool calls during a run */\r\ninterface RunState {\r\n    toolCallArgsBuffer: Map<string, string>;\r\n    toolCallNames: Map<string, string>;\r\n    pendingMcpCalls: Set<string>;\r\n    textContent?: string;\r\n    error: boolean;\r\n}\r\n\r\n/**\n * Configuration for McpMiddleware\n */\nexport interface McpMiddlewareConfig {\n    /** Pre-loaded tools with handlers (required) */\n    tools: AguiTool[];\n}\n\r\n/**\r\n * AG-UI Middleware that executes MCP tools server-side.\r\n */\r\nexport class McpMiddleware extends Middleware {\n    private tools: AguiTool[];\n    private toolSchemas: Tool[];\n\n    constructor(config: McpMiddlewareConfig) {\n        super();\n        this.tools = config.tools;\n        this.toolSchemas = this.tools.map((t: AguiTool) => ({\n            name: t.name,\n            description: t.description,\n            parameters: cleanSchema(t.parameters),\n        }));\n    }\n\r\n    private isMcpTool(toolName: string): boolean {\r\n        return this.tools.some(t => t.name === toolName);\r\n    }\r\n\r\n    private parseArgs(argsString: string): Record<string, any> {\r\n        if (!argsString?.trim()) return {};\r\n\r\n        try {\r\n            return JSON.parse(argsString);\r\n        } catch {\r\n            // Handle duplicated JSON from streaming issues: {...}{...}\r\n            const trimmed = argsString.trim();\r\n            if (trimmed.includes('}{')) {\r\n                const firstObject = trimmed.slice(0, trimmed.indexOf('}{') + 1);\r\n                try {\r\n                    return JSON.parse(firstObject);\r\n                } catch {\r\n                    console.error(`[McpMiddleware] Failed to parse JSON:`, firstObject);\r\n                }\r\n            }\r\n            console.error(`[McpMiddleware] Failed to parse args:`, argsString);\r\n            return {};\r\n        }\r\n    }\r\n\r\n    private async executeTool(toolName: string, args: Record<string, any>): Promise<string> {\r\n        const tool = this.tools.find(t => t.name === toolName);\r\n        if (!tool?.handler) {\r\n            return `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}`;\r\n        }\r\n\r\n        try {\n            console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\n            const result = await tool.handler(args);\n            const resultStr = typeof result === 'string' ? result : JSON.stringify(result);\n\n            console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\n            return resultStr;\n        } catch (error: any) {\r\n            console.error(`[McpMiddleware] Error executing tool:`, error);\r\n            return `Error: ${error.message || String(error)}`;\r\n        }\r\n    }\r\n\r\n    private generateId(prefix: string): string {\r\n        return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\r\n    }\r\n\r\n    private ensureIds(input: RunAgentInput): void {\r\n        const anyInput = input as any;\r\n        if (!anyInput.threadId) anyInput.threadId = this.generateId('mcp_thread');\r\n        if (!anyInput.runId) anyInput.runId = this.generateId('mcp_run');\r\n    }\r\n\r\n    /** Process tool call events and update state */\r\n    private handleToolCallEvent(event: BaseEvent, state: RunState): void {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n\r\n        // Accumulate text content for reconstruction\r\n        if (event.type === EventType.TEXT_MESSAGE_CHUNK) {\r\n            const e = event as any;\r\n            if (e.delta) {\r\n                state.textContent = (state.textContent || '') + e.delta;\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_START) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.toolCallName) {\r\n                toolCallNames.set(e.toolCallId, e.toolCallName);\r\n                if (this.isMcpTool(e.toolCallName)) {\r\n                    pendingMcpCalls.add(e.toolCallId);\r\n                }\r\n                console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_ARGS) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.delta) {\r\n                const existing = toolCallArgsBuffer.get(e.toolCallId) || '';\r\n                toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_END) {\r\n            const e = event as ToolCallEndEvent;\r\n            console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? 'unknown'} (id: ${e.toolCallId})`);\r\n        }\r\n\r\n        // Workaround: Extract parallel tool calls from MESSAGES_SNAPSHOT\r\n        if (event.type === EventType.MESSAGES_SNAPSHOT) {\r\n            const messages = (event as any).messages || [];\r\n            if (messages.length > 0) {\r\n                const lastMsg = messages[messages.length - 1];\r\n                // Update text content from snapshot if available (often more reliable)\r\n                if (lastMsg.role === 'assistant' && lastMsg.content) {\r\n                    state.textContent = lastMsg.content;\r\n                }\r\n\r\n                // Discover tools\r\n                for (let i = messages.length - 1; i >= 0; i--) {\r\n                    const msg = messages[i];\r\n                    const tools = Array.isArray(msg.toolCalls) ? msg.toolCalls :\r\n                        (Array.isArray(msg.tool_calls) ? msg.tool_calls : []);\r\n\r\n                    if (msg.role === 'assistant' && tools.length > 0) {\r\n                        for (const tc of tools) {\r\n                            if (tc.id && tc.function?.name && !toolCallNames.has(tc.id)) {\r\n                                toolCallNames.set(tc.id, tc.function.name);\r\n                                toolCallArgsBuffer.set(tc.id, tc.function.arguments || '{}');\r\n                                if (this.isMcpTool(tc.function.name)) {\r\n                                    pendingMcpCalls.add(tc.id);\r\n                                    console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);\r\n                                }\r\n                            }\r\n                        }\r\n                        break;\r\n                    }\r\n                }\r\n            }\r\n        }\r\n    }\r\n\r\n    /** Execute pending MCP tools and return results */\r\n    private async executeTools(state: RunState): Promise<ToolResult[]> {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n        const results: ToolResult[] = [];\r\n\r\n        const promises = [...pendingMcpCalls].map(async (toolCallId) => {\r\n            const toolName = toolCallNames.get(toolCallId);\r\n            if (!toolName) return;\r\n\r\n            const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || '{}');\r\n            console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\r\n\r\n            const result = await this.executeTool(toolName, args);\r\n            results.push({\r\n                toolCallId,\r\n                toolName,\r\n                result,\r\n                messageId: this.generateId('mcp_result'),\r\n            });\r\n            pendingMcpCalls.delete(toolCallId);\r\n        });\r\n\r\n        await Promise.all(promises);\r\n        return results;\r\n    }\r\n\r\n    /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */\r\n    private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\r\n        for (const { toolCallId, toolName, result, messageId } of results) {\r\n            observer.next({\r\n                type: EventType.TOOL_CALL_RESULT,\r\n                toolCallId,\r\n                messageId,\r\n                content: result,\r\n                role: 'tool',\r\n                timestamp: Date.now(),\r\n            } as any);\r\n            console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\r\n        }\r\n    }\r\n\r\n    run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\r\n        return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\r\n            const state: RunState = {\r\n                toolCallArgsBuffer: new Map(),\r\n                toolCallNames: new Map(),\r\n                pendingMcpCalls: new Set(),\r\n                textContent: '',\r\n                error: false,\r\n            };\r\n\r\n            this.ensureIds(input);\r\n            const anyInput = input as any;\r\n\r\n            console.log(`[McpMiddleware] === NEW RUN ===`);\r\n            console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);\r\n            console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);\r\n\r\n            // Inject MCP tools\r\n            if (this.toolSchemas?.length) {\r\n                input.tools = [...(input.tools || []), ...this.toolSchemas];\r\n                console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t: Tool) => t.name));\r\n            }\r\n\r\n            const handleRunFinished = async () => {\r\n                if (state.error) return; // Don't continue after error\r\n\r\n                if (state.pendingMcpCalls.size === 0) {\r\n                    observer.next({\r\n                        type: EventType.RUN_FINISHED,\r\n                        threadId: anyInput.threadId,\r\n                        runId: anyInput.runId,\r\n                        timestamp: Date.now(),\r\n                    } as any);\r\n                    observer.complete();\r\n                    return;\r\n                }\r\n\r\n                console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);\r\n\r\n                // Reconstruct the Assistant Message that triggered these tools\r\n                const toolCalls = [];\r\n                for (const toolCallId of state.pendingMcpCalls) {\r\n                    const name = state.toolCallNames.get(toolCallId);\r\n                    const args = state.toolCallArgsBuffer.get(toolCallId) || '{}';\r\n                    if (name) {\r\n                        toolCalls.push({\r\n                            id: toolCallId,\r\n                            type: 'function',\r\n                            function: { name, arguments: args }\r\n                        });\r\n                    }\r\n                }\r\n\r\n                // Add the Assistant Message to history FIRST\r\n                if (toolCalls.length > 0 || state.textContent) {\r\n                    const assistantMsg = {\r\n                        id: this.generateId('msg_ast'),\r\n                        role: 'assistant',\r\n                        content: state.textContent || null, // Ensure null if empty string for strict LLMs\r\n                        tool_calls: toolCalls.length > 0 ? toolCalls : undefined\r\n                    };\r\n                    input.messages.push(assistantMsg as any);\r\n                    console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);\r\n                }\r\n\r\n                // Execute tools and emit results (no RUN_FINISHED yet - continuation follows)\r\n                const results = await this.executeTools(state);\r\n                this.emitToolResults(observer, results);\r\n\r\n                // Prepare continuation\r\n                console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);\r\n\r\n                // Add tool result messages to history\r\n                for (const { toolCallId, result, messageId } of results) {\r\n                    input.messages.push({\r\n                        id: messageId,\r\n                        role: 'tool',\r\n                        tool_call_id: toolCallId,\r\n                        content: result,\r\n                    } as any);\r\n                }\r\n\r\n                // Reset state for next turn\r\n                state.toolCallArgsBuffer.clear();\r\n                state.toolCallNames.clear();\r\n                state.textContent = ''; // Clear text content for next turn\r\n\r\n                anyInput.runId = this.generateId('mcp_run');\r\n                console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);\r\n\r\n                // Subscribe to continuation\r\n                next.run(input).subscribe({\r\n                    next: (event) => {\r\n                        if (state.error) return;\r\n\r\n                        this.handleToolCallEvent(event, state);\r\n\r\n                        if (event.type === EventType.RUN_ERROR) {\r\n                            console.log(`[McpMiddleware] RUN_ERROR received in continuation`);\r\n                            state.error = true;\r\n                            observer.next(event);\r\n                            observer.complete();\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_STARTED) {\r\n                            console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_FINISHED) {\r\n                            if (state.pendingMcpCalls.size > 0) {\r\n                                handleRunFinished();\r\n                            } else {\r\n                                observer.next(event);\r\n                                observer.complete();\r\n                            }\r\n                            return;\r\n                        }\r\n                        observer.next(event);\r\n                    },\r\n                    error: (err) => {\r\n                        state.error = true;\r\n                        observer.error(err);\r\n                    },\r\n                    complete: () => {\r\n                        if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                    },\r\n                });\r\n            };\r\n\r\n            const subscription = next.run(input).subscribe({\r\n                next: (event) => {\r\n                    if (state.error) return;\r\n\r\n                    this.handleToolCallEvent(event, state);\r\n\r\n                    if (event.type === EventType.RUN_ERROR) {\r\n                        console.log(`[McpMiddleware] RUN_ERROR received`);\r\n                        state.error = true;\r\n                        observer.next(event);\r\n                        observer.complete();\r\n                        return;\r\n                    }\r\n\r\n                    if (event.type === EventType.RUN_FINISHED) {\r\n                        handleRunFinished();\r\n                        return;\r\n                    }\r\n                    observer.next(event);\r\n                },\r\n                error: (err) => {\r\n                    state.error = true;\r\n                    observer.error(err);\r\n                },\r\n                complete: () => {\r\n                    if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                },\r\n            });\r\n\r\n            return () => subscription.unsubscribe();\r\n        });\r\n    }\r\n}\r\n\r\n/**\r\n * Factory function to create MCP middleware.\r\n */\r\nexport function createMcpMiddleware(\n    options: { tools: AguiTool[] }\n) {\n    const middleware = new McpMiddleware(options);\r\n    return (input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> => {\r\n        return middleware.run(input, next);\r\n    };\r\n}\r\n\r\n// Legacy exports\r\nexport { McpMiddleware as McpToolExecutorMiddleware };\r\nexport { createMcpMiddleware as createMcpToolMiddleware };\r\n\r\n// Re-exports\r\nexport { Middleware, EventType };\r\nexport type { RunAgentInput, BaseEvent, AbstractAgent, ToolCallEndEvent, Tool };"]}
+{"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":[],"mappings":";;;;;;;;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,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,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,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;;;ACrCO,IAAM,aAAA,GAAN,cAA4B,UAAA,CAAW;AAAA,EAI1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAJV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU;AAAA,KACxC,CAAE,CAAA;AAAA,EACN;AAAA,EAEQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAAA,EACnD;AAAA,EAEQ,UAAU,UAAA,EAAyC;AACvD,IAAA,IAAI,CAAC,UAAA,EAAY,IAAA,EAAK,SAAU,EAAC;AAEjC,IAAA,IAAI;AACA,MAAA,OAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,IAChC,CAAA,CAAA,MAAQ;AAEJ,MAAA,MAAM,OAAA,GAAU,WAAW,IAAA,EAAK;AAChC,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,IAAI,CAAA,EAAG;AACxB,QAAA,MAAM,WAAA,GAAc,QAAQ,KAAA,CAAM,CAAA,EAAG,QAAQ,OAAA,CAAQ,IAAI,IAAI,CAAC,CAAA;AAC9D,QAAA,IAAI;AACA,UAAA,OAAO,IAAA,CAAK,MAAM,WAAW,CAAA;AAAA,QACjC,CAAA,CAAA,MAAQ;AACJ,UAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,WAAW,CAAA;AAAA,QACtE;AAAA,MACJ;AACA,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AACjE,MAAA,OAAO,EAAC;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,OAAO,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACrD,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAChB,MAAA,OAAO,CAAA,YAAA,EAAe,IAAA,GAAO,gBAAA,GAAmB,WAAW,KAAK,QAAQ,CAAA,CAAA;AAAA,IAC5E;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AACtC,MAAA,MAAM,YAAY,OAAO,MAAA,KAAW,WAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAE7E,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,SAAA;AAAA,IACX,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,OAAA,EAAU,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IACnD;AAAA,EACJ;AAAA,EAEQ,WAAW,MAAA,EAAwB;AACvC,IAAA,OAAO,GAAG,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC5E;AAAA,EAEQ,UAAU,KAAA,EAA4B;AAC1C,IAAA,MAAM,QAAA,GAAW,KAAA;AACjB,IAAA,IAAI,CAAC,QAAA,CAAS,QAAA,WAAmB,QAAA,GAAW,IAAA,CAAK,WAAW,YAAY,CAAA;AACxE,IAAA,IAAI,CAAC,QAAA,CAAS,KAAA,WAAgB,KAAA,GAAQ,IAAA,CAAK,WAAW,SAAS,CAAA;AAAA,EACnE;AAAA;AAAA,EAGQ,mBAAA,CAAoB,OAAkB,KAAA,EAAuB;AACjE,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAG/D,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AACA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iCAAA,EAAoC,CAAA,CAAE,YAAY,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,SAAA,EAAY,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAC,CAAA,CAAA,CAAG,CAAA;AAAA,MACpI;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AACxC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+BAAA,EAAkC,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,SAAS,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACtH;AAGA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;AACzB,kBAAA,OAAA,CAAQ,GAAA,CAAI,iDAAiD,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,MAAA,EAAS,EAAA,CAAG,EAAE,CAAA,CAAA,CAAG,CAAA;AAAA,gBAClG;AAAA,cACJ;AAAA,YACJ;AACA,YAAA;AAAA,UACJ;AAAA,QACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAc,aAAa,KAAA,EAAwC;AAC/D,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAC/D,IAAA,MAAM,UAAwB,EAAC;AAE/B,IAAA,MAAM,WAAW,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAC5D,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,OAAO,IAAA,CAAK,SAAA,CAAU,mBAAmB,GAAA,CAAI,UAAU,KAAK,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AAEjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,UAAA,CAAW,YAAY;AAAA,OAC1C,CAAA;AACD,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,QAAQ,CAAA;AAC1B,IAAA,OAAO,OAAA;AAAA,EACX;AAAA;AAAA,EAGQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AAC/D,MAAA,QAAA,CAAS,IAAA,CAAK;AAAA,QACV,MAAM,SAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AACR,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AAAA,IAC5E;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAI,UAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;AAEjB,MAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,CAAiC,CAAA;AAC7C,MAAA,OAAA,CAAQ,IAAI,CAAA,0BAAA,EAA6B,QAAA,CAAS,QAAQ,CAAA,SAAA,EAAY,QAAA,CAAS,KAAK,CAAA,CAAE,CAAA;AACtF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,0BAAA,EAA6B,KAAA,CAAM,QAAA,EAAU,MAAA,IAAU,CAAC,CAAA,SAAA,EAAY,IAAA,CAAK,KAAA,EAAO,MAAA,IAAU,CAAC,CAAA,CAAE,CAAA;AAGzG,MAAA,IAAI,IAAA,CAAK,aAAa,MAAA,EAAQ;AAC1B,QAAA,KAAA,CAAM,KAAA,GAAQ,CAAC,GAAI,KAAA,CAAM,SAAS,EAAC,EAAI,GAAG,IAAA,CAAK,WAAW,CAAA;AAC1D,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,WAAA,CAAY,MAAM,CAAA,OAAA,CAAA,EAAW,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MACvH;AAEA,MAAA,MAAM,oBAAoB,YAAY;AAClC,QAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,QAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,KAAS,CAAA,EAAG;AAClC,UAAA,QAAA,CAAS,IAAA,CAAK;AAAA,YACV,MAAM,SAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,KAAA,CAAM,eAAA,CAAgB,IAAI,CAAA,cAAA,CAAgB,CAAA;AAG3F,QAAA,MAAM,YAAY,EAAC;AACnB,QAAA,KAAA,MAAW,UAAA,IAAc,MAAM,eAAA,EAAiB;AAC5C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC/C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,IAAA,EAAM;AACN,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACX,EAAA,EAAI,UAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA;AAAK,aACrC,CAAA;AAAA,UACL;AAAA,QACJ;AAGA,QAAA,IAAI,SAAA,CAAU,MAAA,GAAS,CAAA,IAAK,KAAA,CAAM,WAAA,EAAa;AAC3C,UAAA,MAAM,YAAA,GAAe;AAAA,YACjB,EAAA,EAAI,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAAA,YAC7B,IAAA,EAAM,WAAA;AAAA,YACN,OAAA,EAAS,MAAM,WAAA,IAAe,IAAA;AAAA;AAAA,YAC9B,UAAA,EAAY,SAAA,CAAU,MAAA,GAAS,CAAA,GAAI,SAAA,GAAY;AAAA,WACnD;AACA,UAAA,KAAA,CAAM,QAAA,CAAS,KAAK,YAAmB,CAAA;AACvC,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iEAAA,EAAoE,KAAA,CAAM,WAAA,EAAa,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,KAAA,EAAQ,SAAA,CAAU,MAAM,CAAA,OAAA,CAAS,CAAA;AAAA,QACpJ;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6CAAA,EAAgD,OAAA,CAAQ,MAAM,CAAA,QAAA,CAAU,CAAA;AAGpF,QAAA,KAAA,MAAW,EAAE,UAAA,EAAY,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AACrD,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,YAAA,EAAc,UAAA;AAAA,YACd,OAAA,EAAS;AAAA,WACL,CAAA;AAAA,QACZ;AAGA,QAAA,KAAA,CAAM,mBAAmB,KAAA,EAAM;AAC/B,QAAA,KAAA,CAAM,cAAc,KAAA,EAAM;AAC1B,QAAA,KAAA,CAAM,WAAA,GAAc,EAAA;AAEpB,QAAA,QAAA,CAAS,KAAA,GAAQ,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAC1C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,mDAAA,EAAsD,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,CAAE,CAAA;AAGzF,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,CAAA,CAAE,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,YAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,YAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,cAAA,OAAA,CAAQ,IAAI,CAAA,kDAAA,CAAoD,CAAA;AAChE,cAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,cAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,cAAA,QAAA,CAAS,QAAA,EAAS;AAClB,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,WAAA,EAAa;AACtC,cAAA,OAAA,CAAQ,IAAI,CAAA,uDAAA,CAAyD,CAAA;AACrE,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,YAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,oBACZ,OAAA,EACF;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc,OAAO,CAAA;AAC5C,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.mjs","sourcesContent":["/**\n * MCP Adapter for AG-UI Integration\n *\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\n * It provides tools with handlers for server-side execution and tool definitions\n * in JSON Schema format for passing to remote agents.\n *\n * @example\n * ```typescript\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n * import { HttpAgent } from '@ag-ui/client';\n *\n * // Create MCP client\n * const mcpClient = new MultiSessionClient('user_123');\n * await mcpClient.connect();\n *\n * // Create adapter and get tools\n * const adapter = new AguiAdapter(mcpClient);\n * const tools = await adapter.getTools();\n *\n * // Use with AG-UI middleware\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(createMcpMiddleware({ tools }));\n * ```\n */\n\nimport { MCPClient } from '../server/mcp/oauth-client.js';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\n\n/**\n * Extended JSON Schema properties that Pydantic's strict validation rejects.\n * These are valid JSON Schema extensions but not part of the core spec.\n */\nconst PYDANTIC_FORBIDDEN_PROPS = [\n    // JSON Schema meta-properties\n    '$schema', '$id', '$comment', '$defs', 'definitions',\n    // Extended properties used by some MCP servers (e.g., Apify)\n    'prefill', 'examples', 'enumTitles', 'enumDescriptions',\n    // Other common extensions\n    'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\n];\n\n/**\n * Cleans a JSON Schema by removing meta-properties that cause issues with\n * strict Pydantic validation (e.g., Google ADK, LangGraph).\n *\n * @param schema - The JSON Schema to clean\n * @returns Cleaned schema without forbidden properties\n */\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\n    if (!schema) {\n        return { type: 'object', properties: {} };\n    }\n\n    const cleaned = { ...schema };\n\n    // Remove all forbidden properties\n    for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\n        delete cleaned[prop];\n    }\n\n    // Recursively clean nested properties\n    if (cleaned.properties && typeof cleaned.properties === 'object') {\n        const cleanedProps: Record<string, any> = {};\n        for (const [key, value] of Object.entries(cleaned.properties)) {\n            if (typeof value === 'object' && value !== null) {\n                cleanedProps[key] = cleanSchema(value as Record<string, any>);\n            } else {\n                cleanedProps[key] = value;\n            }\n        }\n        cleaned.properties = cleanedProps;\n    }\n\n    // Clean items if it's an array schema\n    if (cleaned.items && typeof cleaned.items === 'object') {\n        cleaned.items = cleanSchema(cleaned.items);\n    }\n\n    // Clean additionalProperties if it's an object schema\n    if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\n        cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\n    }\n\n    return cleaned;\n}\n\n/**\n * Configuration options for AguiAdapter\n */\nexport interface AguiAdapterOptions {\n    /**\n     * Prefix for tool names to avoid collision with other tools.\n     * @default serverId or 'mcp'\n     */\n    prefix?: string;\n}\n\n/**\n * AG-UI Tool with handler for server-side execution.\n */\nexport interface AguiTool {\n    name: string;\n    description: string;\n    parameters?: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiTool\n    handler?: (args: any) => any | Promise<any>;\n}\n\n/**\n * Tool definition format for passing to remote agents (without handler).\n */\nexport interface AguiToolDefinition {\n    name: string;\n    description: string;\n    parameters: Record<string, any>;\n    _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\n}\n\n/**\n * Adapter that transforms MCP tools into AG-UI compatible formats.\n */\nexport class AguiAdapter {\n    constructor(\n        private client: MCPClient | MultiSessionClient,\n        private options: AguiAdapterOptions = {}\n    ) { }\n\n    /**\n     * Get tools with handlers for MCP tool execution.\n     */\n    async getTools(): Promise<AguiTool[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiTool[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformTools(client));\n            }\n            return allTools;\n        }\n        return this.transformTools(this.client as MCPClient);\n    }\n\n    /**\n     * Get tool definitions in JSON Schema format for passing to remote agents.\n     */\n    async getToolDefinitions(): Promise<AguiToolDefinition[]> {\n        if (this.isMultiSession()) {\n            const clients = (this.client as MultiSessionClient).getClients();\n            const allTools: AguiToolDefinition[] = [];\n            for (const client of clients) {\n                allTools.push(...await this.transformToolDefinitions(client));\n            }\n            return allTools;\n        }\n        return this.transformToolDefinitions(this.client as MCPClient);\n    }\n\n    /**\n     * Get tools as a function (for dynamic loading).\n     */\n    getToolsFunction(): () => Promise<AguiTool[]> {\n        return () => this.getTools();\n    }\n\n    private isMultiSession(): boolean {\n        return typeof (this.client as any).getClients === 'function';\n    }\n\n    private async transformTools(client: MCPClient): Promise<AguiTool[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\n            const mcpTool = tool as any;\n            const mcpToolName = tool.name;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n                handler: async (args: any) => {\n                    // Call the actual MCP tool\n                    const callResult = await (client as any).callTool(mcpToolName, args ?? {});\n\n                    // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\n                    return callResult;\n                }\n            }\n        });\n    }\n\n    private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\n        if (!client.isConnected()) return [];\n\n        const result = await client.listTools();\n        const serverId = (typeof (client as any).getServerId === 'function'\n            ? (client as any).getServerId()\n            : undefined) as string | undefined;\n        const normalizedPrefix = this.options.prefix?.replace(/-/g, '') ?? serverId?.replace(/-/g, '').substring(0, 8) ?? 'mcp';\n        const prefix = `tool_${normalizedPrefix}`;\n\n        return result.tools.map(tool => {\n            const mcpTool = tool as any;\n            return {\n                name: `${prefix}_${tool.name}`,\n                description: tool.description || `Execute ${tool.name}`,\n                parameters: cleanSchema(tool.inputSchema),\n                _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\n            };\n        });\n    }\n}\n","/**\r\n * AG-UI Middleware for MCP Tool Execution\r\n *\r\n * This middleware intercepts tool calls from remote agents and executes\r\n * MCP tools server-side, returning results back to the agent.\r\n *\r\n * @requires @ag-ui/client - Peer dependency for AG-UI types\r\n * @requires rxjs - Uses RxJS Observables for event streaming\r\n */\r\n\r\nimport { Observable, Subscriber } from 'rxjs';\r\nimport {\r\n    Middleware,\r\n    EventType,\r\n    type AbstractAgent,\r\n    type RunAgentInput,\r\n    type BaseEvent,\r\n    type ToolCallEndEvent,\r\n    type Tool,\r\n} from '@ag-ui/client';\r\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\r\n\r\n/** Tool execution result for continuation */\r\ninterface ToolResult {\r\n    toolCallId: string;\r\n    toolName: string;\r\n    result: string;\r\n    messageId: string;\r\n}\r\n\r\n/** State for tracking tool calls during a run */\r\ninterface RunState {\r\n    toolCallArgsBuffer: Map<string, string>;\r\n    toolCallNames: Map<string, string>;\r\n    pendingMcpCalls: Set<string>;\r\n    textContent?: string;\r\n    error: boolean;\r\n}\r\n\r\n/**\n * Configuration for McpMiddleware\n */\nexport interface McpMiddlewareConfig {\n    /** Pre-loaded tools with handlers (required) */\n    tools: AguiTool[];\n}\n\r\n/**\r\n * AG-UI Middleware that executes MCP tools server-side.\r\n */\r\nexport class McpMiddleware extends Middleware {\n    private tools: AguiTool[];\n    private toolSchemas: Tool[];\n\n    constructor(config: McpMiddlewareConfig) {\n        super();\n        this.tools = config.tools;\n        this.toolSchemas = this.tools.map((t: AguiTool) => ({\n            name: t.name,\n            description: t.description,\n            parameters: cleanSchema(t.parameters),\n        }));\n    }\n\r\n    private isMcpTool(toolName: string): boolean {\r\n        return this.tools.some(t => t.name === toolName);\r\n    }\r\n\r\n    private parseArgs(argsString: string): Record<string, any> {\r\n        if (!argsString?.trim()) return {};\r\n\r\n        try {\r\n            return JSON.parse(argsString);\r\n        } catch {\r\n            // Handle duplicated JSON from streaming issues: {...}{...}\r\n            const trimmed = argsString.trim();\r\n            if (trimmed.includes('}{')) {\r\n                const firstObject = trimmed.slice(0, trimmed.indexOf('}{') + 1);\r\n                try {\r\n                    return JSON.parse(firstObject);\r\n                } catch {\r\n                    console.error(`[McpMiddleware] Failed to parse JSON:`, firstObject);\r\n                }\r\n            }\r\n            console.error(`[McpMiddleware] Failed to parse args:`, argsString);\r\n            return {};\r\n        }\r\n    }\r\n\r\n    private async executeTool(toolName: string, args: Record<string, any>): Promise<string> {\r\n        const tool = this.tools.find(t => t.name === toolName);\r\n        if (!tool?.handler) {\r\n            return `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}`;\r\n        }\r\n\r\n        try {\n            console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\n            const result = await tool.handler(args);\n            const resultStr = typeof result === 'string' ? result : JSON.stringify(result);\n\n            console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\n            return resultStr;\n        } catch (error: any) {\r\n            console.error(`[McpMiddleware] Error executing tool:`, error);\r\n            return `Error: ${error.message || String(error)}`;\r\n        }\r\n    }\r\n\r\n    private generateId(prefix: string): string {\r\n        return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\r\n    }\r\n\r\n    private ensureIds(input: RunAgentInput): void {\r\n        const anyInput = input as any;\r\n        if (!anyInput.threadId) anyInput.threadId = this.generateId('mcp_thread');\r\n        if (!anyInput.runId) anyInput.runId = this.generateId('mcp_run');\r\n    }\r\n\r\n    /** Process tool call events and update state */\r\n    private handleToolCallEvent(event: BaseEvent, state: RunState): void {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n\r\n        // Accumulate text content for reconstruction\r\n        if (event.type === EventType.TEXT_MESSAGE_CHUNK) {\r\n            const e = event as any;\r\n            if (e.delta) {\r\n                state.textContent = (state.textContent || '') + e.delta;\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_START) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.toolCallName) {\r\n                toolCallNames.set(e.toolCallId, e.toolCallName);\r\n                if (this.isMcpTool(e.toolCallName)) {\r\n                    pendingMcpCalls.add(e.toolCallId);\r\n                }\r\n                console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_ARGS) {\r\n            const e = event as any;\r\n            if (e.toolCallId && e.delta) {\r\n                const existing = toolCallArgsBuffer.get(e.toolCallId) || '';\r\n                toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);\r\n            }\r\n        }\r\n\r\n        if (event.type === EventType.TOOL_CALL_END) {\r\n            const e = event as ToolCallEndEvent;\r\n            console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? 'unknown'} (id: ${e.toolCallId})`);\r\n        }\r\n\r\n        // Workaround: Extract parallel tool calls from MESSAGES_SNAPSHOT\r\n        if (event.type === EventType.MESSAGES_SNAPSHOT) {\r\n            const messages = (event as any).messages || [];\r\n            if (messages.length > 0) {\r\n                const lastMsg = messages[messages.length - 1];\r\n                // Update text content from snapshot if available (often more reliable)\r\n                if (lastMsg.role === 'assistant' && lastMsg.content) {\r\n                    state.textContent = lastMsg.content;\r\n                }\r\n\r\n                // Discover tools\r\n                for (let i = messages.length - 1; i >= 0; i--) {\r\n                    const msg = messages[i];\r\n                    const tools = Array.isArray(msg.toolCalls) ? msg.toolCalls :\r\n                        (Array.isArray(msg.tool_calls) ? msg.tool_calls : []);\r\n\r\n                    if (msg.role === 'assistant' && tools.length > 0) {\r\n                        for (const tc of tools) {\r\n                            if (tc.id && tc.function?.name && !toolCallNames.has(tc.id)) {\r\n                                toolCallNames.set(tc.id, tc.function.name);\r\n                                toolCallArgsBuffer.set(tc.id, tc.function.arguments || '{}');\r\n                                if (this.isMcpTool(tc.function.name)) {\r\n                                    pendingMcpCalls.add(tc.id);\r\n                                    console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);\r\n                                }\r\n                            }\r\n                        }\r\n                        break;\r\n                    }\r\n                }\r\n            }\r\n        }\r\n    }\r\n\r\n    /** Execute pending MCP tools and return results */\r\n    private async executeTools(state: RunState): Promise<ToolResult[]> {\r\n        const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n        const results: ToolResult[] = [];\r\n\r\n        const promises = [...pendingMcpCalls].map(async (toolCallId) => {\r\n            const toolName = toolCallNames.get(toolCallId);\r\n            if (!toolName) return;\r\n\r\n            const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || '{}');\r\n            console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\r\n\r\n            const result = await this.executeTool(toolName, args);\r\n            results.push({\r\n                toolCallId,\r\n                toolName,\r\n                result,\r\n                messageId: this.generateId('mcp_result'),\r\n            });\r\n            pendingMcpCalls.delete(toolCallId);\r\n        });\r\n\r\n        await Promise.all(promises);\r\n        return results;\r\n    }\r\n\r\n    /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */\r\n    private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\r\n        for (const { toolCallId, toolName, result, messageId } of results) {\r\n            observer.next({\r\n                type: EventType.TOOL_CALL_RESULT,\r\n                toolCallId,\r\n                messageId,\r\n                content: result,\r\n                role: 'tool',\r\n                timestamp: Date.now(),\r\n            } as any);\r\n            console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\r\n        }\r\n    }\r\n\r\n    run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\r\n        return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\r\n            const state: RunState = {\r\n                toolCallArgsBuffer: new Map(),\r\n                toolCallNames: new Map(),\r\n                pendingMcpCalls: new Set(),\r\n                textContent: '',\r\n                error: false,\r\n            };\r\n\r\n            this.ensureIds(input);\r\n            const anyInput = input as any;\r\n\r\n            console.log(`[McpMiddleware] === NEW RUN ===`);\r\n            console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);\r\n            console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);\r\n\r\n            // Inject MCP tools\r\n            if (this.toolSchemas?.length) {\r\n                input.tools = [...(input.tools || []), ...this.toolSchemas];\r\n                console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t: Tool) => t.name));\r\n            }\r\n\r\n            const handleRunFinished = async () => {\r\n                if (state.error) return; // Don't continue after error\r\n\r\n                if (state.pendingMcpCalls.size === 0) {\r\n                    observer.next({\r\n                        type: EventType.RUN_FINISHED,\r\n                        threadId: anyInput.threadId,\r\n                        runId: anyInput.runId,\r\n                        timestamp: Date.now(),\r\n                    } as any);\r\n                    observer.complete();\r\n                    return;\r\n                }\r\n\r\n                console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);\r\n\r\n                // Reconstruct the Assistant Message that triggered these tools\r\n                const toolCalls = [];\r\n                for (const toolCallId of state.pendingMcpCalls) {\r\n                    const name = state.toolCallNames.get(toolCallId);\r\n                    const args = state.toolCallArgsBuffer.get(toolCallId) || '{}';\r\n                    if (name) {\r\n                        toolCalls.push({\r\n                            id: toolCallId,\r\n                            type: 'function',\r\n                            function: { name, arguments: args }\r\n                        });\r\n                    }\r\n                }\r\n\r\n                // Add the Assistant Message to history FIRST\r\n                if (toolCalls.length > 0 || state.textContent) {\r\n                    const assistantMsg = {\r\n                        id: this.generateId('msg_ast'),\r\n                        role: 'assistant',\r\n                        content: state.textContent || null, // Ensure null if empty string for strict LLMs\r\n                        tool_calls: toolCalls.length > 0 ? toolCalls : undefined\r\n                    };\r\n                    input.messages.push(assistantMsg as any);\r\n                    console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);\r\n                }\r\n\r\n                // Execute tools and emit results (no RUN_FINISHED yet - continuation follows)\r\n                const results = await this.executeTools(state);\r\n                this.emitToolResults(observer, results);\r\n\r\n                // Prepare continuation\r\n                console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);\r\n\r\n                // Add tool result messages to history\r\n                for (const { toolCallId, result, messageId } of results) {\r\n                    input.messages.push({\r\n                        id: messageId,\r\n                        role: 'tool',\r\n                        tool_call_id: toolCallId,\r\n                        content: result,\r\n                    } as any);\r\n                }\r\n\r\n                // Reset state for next turn\r\n                state.toolCallArgsBuffer.clear();\r\n                state.toolCallNames.clear();\r\n                state.textContent = ''; // Clear text content for next turn\r\n\r\n                anyInput.runId = this.generateId('mcp_run');\r\n                console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);\r\n\r\n                // Subscribe to continuation\r\n                next.run(input).subscribe({\r\n                    next: (event) => {\r\n                        if (state.error) return;\r\n\r\n                        this.handleToolCallEvent(event, state);\r\n\r\n                        if (event.type === EventType.RUN_ERROR) {\r\n                            console.log(`[McpMiddleware] RUN_ERROR received in continuation`);\r\n                            state.error = true;\r\n                            observer.next(event);\r\n                            observer.complete();\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_STARTED) {\r\n                            console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);\r\n                            return;\r\n                        }\r\n\r\n                        if (event.type === EventType.RUN_FINISHED) {\r\n                            if (state.pendingMcpCalls.size > 0) {\r\n                                handleRunFinished();\r\n                            } else {\r\n                                observer.next(event);\r\n                                observer.complete();\r\n                            }\r\n                            return;\r\n                        }\r\n                        observer.next(event);\r\n                    },\r\n                    error: (err) => {\r\n                        state.error = true;\r\n                        observer.error(err);\r\n                    },\r\n                    complete: () => {\r\n                        if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                    },\r\n                });\r\n            };\r\n\r\n            const subscription = next.run(input).subscribe({\r\n                next: (event) => {\r\n                    if (state.error) return;\r\n\r\n                    this.handleToolCallEvent(event, state);\r\n\r\n                    if (event.type === EventType.RUN_ERROR) {\r\n                        console.log(`[McpMiddleware] RUN_ERROR received`);\r\n                        state.error = true;\r\n                        observer.next(event);\r\n                        observer.complete();\r\n                        return;\r\n                    }\r\n\r\n                    if (event.type === EventType.RUN_FINISHED) {\r\n                        handleRunFinished();\r\n                        return;\r\n                    }\r\n                    observer.next(event);\r\n                },\r\n                error: (err) => {\r\n                    state.error = true;\r\n                    observer.error(err);\r\n                },\r\n                complete: () => {\r\n                    if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n                },\r\n            });\r\n\r\n            return () => subscription.unsubscribe();\r\n        });\r\n    }\r\n}\r\n\r\n/**\r\n * Factory function to create MCP middleware.\r\n */\r\nexport function createMcpMiddleware(\n    options: { tools: AguiTool[] }\n) {\n    const middleware = new McpMiddleware(options);\r\n    return (input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> => {\r\n        return middleware.run(input, next);\r\n    };\r\n}\r\n\r\n// Legacy exports\r\nexport { McpMiddleware as McpToolExecutorMiddleware };\r\nexport { createMcpMiddleware as createMcpToolMiddleware };\r\n\r\n// Re-exports\r\nexport { Middleware, EventType };\r\nexport type { RunAgentInput, BaseEvent, AbstractAgent, ToolCallEndEvent, Tool };"]}

package/dist/adapters/ai-adapter.d.mts

@@ -1,4 +1,4 @@
-import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CzhMkE0k.mjs';
+import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CQsRbxYI.mjs';
 import { ToolSet } from 'ai';
 import '../events-CK3N--3g.mjs';
 import '@modelcontextprotocol/sdk/types.js';

package/dist/adapters/ai-adapter.d.ts

@@ -1,4 +1,4 @@
-import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-BYLarghq.js';
+import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CHE8QpVE.js';
 import { ToolSet } from 'ai';
 import '../events-CK3N--3g.js';
 import '@modelcontextprotocol/sdk/types.js';