@mcp-ts/sdk 1.1.0 → 1.2.0

package/README.md

@@ -1,14 +1,10 @@
 <div align="center">
-  <a href="https://github.com/zonlabs/mcp-ts">
-    <img src="docs/static/img/logo.svg" alt="mcp-ts Logo" width="80" height="80" />
-  </a>
-  <h1 align="center">@mcp-ts</h1>
-  <p>TypeScript SDK providing MCP capabilities to agents across JavaScript/cross-runtime environments.</p>
+  <img src="docs/static/img/mcp-ts-banner.svg" alt="MCP-TS Banner" width="100%" style="max-width: 1200px;" />
 </div>
 
 <div align="center">
-  <a href="https://github.com/zonlabs/mcp-ts/raw/main/docs/static/vid/langchain-agui.mp4">
-    <em>Click to watch demo video</em>
+  <a href="https://zonlabs.github.io/mcp-ts/#ag-ui-demo">
+    <em>Watch AG-UI + LangChain demo</em>
   </a>
 </div>
 <br />
@@ -24,7 +20,7 @@
 
 <p align="center">
   <a href="https://www.npmjs.com/package/@mcp-ts/sdk">
-    <img src="https://badge.fury.io/js/@mcp-ts%2Fcore.svg" alt="npm version" />
+    <img src="https://img.shields.io/npm/v/@mcp-ts/sdk.svg" alt="npm version" />
   </a>
   <a href="https://opensource.org/licenses/MIT">
     <img src="https://img.shields.io/badge/License-MIT-yellow.svg" alt="License: MIT" />
@@ -44,8 +40,23 @@
 - **Agent Adapters** - Built-in adapters for AI SDK, LangChain, Mastra, and AG-UI
 - **MCP Apps Extension (SEP-1865)** - Interactive UI-driven tool interfaces
 
+## Examples
+
+Check out working examples demonstrating the MCP Apps extension and agent integrations in the [examples/agents](examples/agents) directory.
+
+> Examples MCP Apps referred from [modelcontextprotocol/ext-apps](https://github.com/modelcontextprotocol/ext-apps/tree/main/examples)
+
 <div align="center">
-  <img src="docs/static/img/mcp-apps.png" alt="MCP Apps" width="100%" />
+  <table cellspacing="5" cellpadding="0">
+    <tr>
+      <td width="50%">
+        <img src="docs/static/img/mcp-apps-img-1.png" alt="MCP Apps 1" width="100%" />
+      </td>
+      <td width="50%">
+        <img src="docs/static/img/mcp-apps-img-2.png" alt="MCP Apps 2" width="100%" />
+      </td>
+    </tr>
+  </table>
   <p><em>Interactive UIs for MCP tools</em></p>
 </div>
 
@@ -299,7 +310,7 @@ For more details, refer to the documentation and follow the **installation guide
 
 ## Contributing
 
-Contributions are welcome! Please read [CLAUDE.md](./CLAUDE.md) for development guidelines.
+Contributions are welcome! Please read [CONTRIBUTING.md](./CONTRIBUTING.md) for guidelines on how to contribute.
 
 ## License
 

package/dist/adapters/agui-adapter.js

@@ -104,7 +104,6 @@ var AguiAdapter = class {
         parameters: cleanSchema(tool.inputSchema),
         _meta: { ...mcpTool._meta, sessionId: client.getSessionId?.() },
         handler: async (args) => {
-          console.log(`[AguiAdapter] Executing MCP tool: ${mcpToolName}`, args);
           const callResult = await client.callTool(mcpToolName, args ?? {});
           return callResult;
         }

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;AAC1B,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,WAAW,CAAA,CAAA,EAAI,IAAI,CAAA;AAGpE,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                    console.log(`[AguiAdapter] Executing MCP tool: ${mcpToolName}`, args);\r\n\r\n                    // IMPORTANT: call the actual MCP tool. (Previously this mistakenly returned the listTools() result.)\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,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"]}

package/dist/adapters/agui-adapter.mjs

@@ -102,7 +102,6 @@ var AguiAdapter = class {
         parameters: cleanSchema(tool.inputSchema),
         _meta: { ...mcpTool._meta, sessionId: client.getSessionId?.() },
         handler: async (args) => {
-          console.log(`[AguiAdapter] Executing MCP tool: ${mcpToolName}`, args);
           const callResult = await client.callTool(mcpToolName, args ?? {});
           return callResult;
         }

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;AAC1B,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,WAAW,CAAA,CAAA,EAAI,IAAI,CAAA;AAGpE,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                    console.log(`[AguiAdapter] Executing MCP tool: ${mcpToolName}`, args);\r\n\r\n                    // IMPORTANT: call the actual MCP tool. (Previously this mistakenly returned the listTools() result.)\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,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"]}

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

@@ -47,6 +47,11 @@ declare class McpMiddleware extends Middleware {
     private tools;
     private toolSchemas;
     constructor(config: McpMiddlewareConfig);
+    /**
+     * Extract base tool name from prefixed format for event emission
+     * e.g., "tool_abc123_get-time" -> "get-time"
+     */
+    private getBaseToolName;
     private isMcpTool;
     private parseArgs;
     private executeTool;

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

@@ -47,6 +47,11 @@ declare class McpMiddleware extends Middleware {
     private tools;
     private toolSchemas;
     constructor(config: McpMiddlewareConfig);
+    /**
+     * Extract base tool name from prefixed format for event emission
+     * e.g., "tool_abc123_get-time" -> "get-time"
+     */
+    private getBaseToolName;
     private isMcpTool;
     private parseArgs;
     private executeTool;

package/dist/adapters/agui-middleware.js

@@ -71,6 +71,14 @@ var McpMiddleware = class extends client.Middleware {
       // Include _meta in the tool definition passed to the agent
     }));
   }
+  /**
+   * Extract base tool name from prefixed format for event emission
+   * e.g., "tool_abc123_get-time" -> "get-time"
+   */
+  getBaseToolName(toolName) {
+    const match = toolName.match(/^tool_[^_]+_(.+)$/);
+    return match ? match[1] : toolName;
+  }
   isMcpTool(toolName) {
     return this.tools.some((t) => t.name === toolName);
   }
@@ -98,7 +106,6 @@ var McpMiddleware = class extends client.Middleware {
       return { resultStr: `Error: Tool ${tool ? "has no handler" : "not found"}: ${toolName}` };
     }
     try {
-      console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);
       const result = await tool.handler(args);
       let resultStr;
       if (typeof result === "string") {
@@ -108,7 +115,6 @@ var McpMiddleware = class extends client.Middleware {
       } else {
         resultStr = String(result);
       }
-      console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));
       return { resultStr, rawResult: result };
     } catch (error) {
       console.error(`[McpMiddleware] Error executing tool:`, error);
@@ -139,7 +145,6 @@ var McpMiddleware = class extends client.Middleware {
         if (this.isMcpTool(e.toolCallName)) {
           pendingMcpCalls.add(e.toolCallId);
         }
-        console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);
       }
     }
     if (event.type === client.EventType.TOOL_CALL_ARGS) {
@@ -149,10 +154,7 @@ var McpMiddleware = class extends client.Middleware {
         toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);
       }
     }
-    if (event.type === client.EventType.TOOL_CALL_END) {
-      const e = event;
-      console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? "unknown"} (id: ${e.toolCallId})`);
-    }
+    if (event.type === client.EventType.TOOL_CALL_END) ;
     if (event.type === client.EventType.MESSAGES_SNAPSHOT) {
       const messages = event.messages || [];
       if (messages.length > 0) {
@@ -170,7 +172,6 @@ var McpMiddleware = class extends client.Middleware {
                 toolCallArgsBuffer.set(tc.id, tc.function.arguments || "{}");
                 if (this.isMcpTool(tc.function.name)) {
                   pendingMcpCalls.add(tc.id);
-                  console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);
                 }
               }
             }
@@ -188,7 +189,6 @@ var McpMiddleware = class extends client.Middleware {
       const toolName = toolCallNames.get(toolCallId);
       if (!toolName) return;
       const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || "{}");
-      console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);
       const { resultStr, rawResult } = await this.executeTool(toolName, args);
       results.push({
         toolCallId,
@@ -208,11 +208,13 @@ var McpMiddleware = class extends client.Middleware {
       const sessionId = toolDef?._meta?.sessionId;
       const resourceUri = rawResult?._meta?.ui?.resourceUri ?? rawResult?._meta?.["ui/resourceUri"] ?? toolDef?._meta?.ui?.resourceUri ?? toolDef?._meta?.["ui/resourceUri"];
       if (resourceUri) {
+        const baseToolName = this.getBaseToolName(toolName);
         const payload = {
           toolCallId,
           resourceUri,
           sessionId,
-          toolName,
+          toolName: baseToolName,
+          // Use base name to match metadata
           result: rawResult ?? result
         };
         observer.next({
@@ -222,7 +224,6 @@ var McpMiddleware = class extends client.Middleware {
           timestamp: Date.now(),
           role: "tool"
         });
-        console.log(`[McpMiddleware] Emitting CustomEvent(${MCP_APP_UI_EVENT}) for: ${toolName} (session: ${sessionId})`);
       }
       observer.next({
         type: client.EventType.TOOL_CALL_RESULT,
@@ -232,7 +233,6 @@ var McpMiddleware = class extends client.Middleware {
         role: "tool",
         timestamp: Date.now()
       });
-      console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);
     }
   }
   run(input, next) {
@@ -246,12 +246,8 @@ var McpMiddleware = class extends client.Middleware {
       };
       this.ensureIds(input);
       const anyInput = input;
-      console.log(`[McpMiddleware] === NEW RUN ===`);
-      console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);
-      console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);
       if (this.toolSchemas?.length) {
         input.tools = [...input.tools || [], ...this.toolSchemas];
-        console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t) => t.name));
       }
       const handleRunFinished = async () => {
         if (state.error) return;
@@ -265,7 +261,6 @@ var McpMiddleware = class extends client.Middleware {
           observer.complete();
           return;
         }
-        console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);
         const toolCalls = [];
         for (const toolCallId of state.pendingMcpCalls) {
           const name = state.toolCallNames.get(toolCallId);
@@ -287,11 +282,9 @@ var McpMiddleware = class extends client.Middleware {
             tool_calls: toolCalls.length > 0 ? toolCalls : void 0
           };
           input.messages.push(assistantMsg);
-          console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);
         }
         const results = await this.executeTools(state);
         this.emitToolResults(observer, results);
-        console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);
         for (const { toolCallId, result, messageId } of results) {
           input.messages.push({
             id: messageId,
@@ -304,20 +297,17 @@ var McpMiddleware = class extends client.Middleware {
         state.toolCallNames.clear();
         state.textContent = "";
         anyInput.runId = this.generateId("mcp_run");
-        console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);
         next.run(input).subscribe({
           next: (event) => {
             if (state.error) return;
             this.handleToolCallEvent(event, state);
             if (event.type === client.EventType.RUN_ERROR) {
-              console.log(`[McpMiddleware] RUN_ERROR received in continuation`);
               state.error = true;
               observer.next(event);
               observer.complete();
               return;
             }
             if (event.type === client.EventType.RUN_STARTED) {
-              console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);
               return;
             }
             if (event.type === client.EventType.RUN_FINISHED) {
@@ -345,7 +335,6 @@ var McpMiddleware = class extends client.Middleware {
           if (state.error) return;
           this.handleToolCallEvent(event, state);
           if (event.type === client.EventType.RUN_ERROR) {
-            console.log(`[McpMiddleware] RUN_ERROR received`);
             state.error = true;
             observer.next(event);
             observer.complete();

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;;;AChEO,IAAM,gBAAA,GAAmB;AAgDzB,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,CAAA;AAAA,MACpC,OAAO,CAAA,CAAE;AAAA;AAAA,KACb,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,EAA4E;AACpH,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,EAAE,WAAW,CAAA,YAAA,EAAe,IAAA,GAAO,mBAAmB,WAAW,CAAA,EAAA,EAAK,QAAQ,CAAA,CAAA,EAAG;AAAA,IAC5F;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAE/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AAEtC,MAAA,IAAI,SAAA;AAEJ,MAAA,IAAI,OAAO,WAAW,QAAA,EAAU;AAC5B,QAAA,SAAA,GAAY,MAAA;AAAA,MAChB,CAAA,MAAA,IAAW,MAAA,IAAU,OAAO,MAAA,KAAW,QAAA,EAAU;AAE7C,QAAA,SAAA,GAAY,IAAA,CAAK,UAAU,MAAM,CAAA;AAAA,MACrC,CAAA,MAAO;AACH,QAAA,SAAA,GAAY,OAAO,MAAM,CAAA;AAAA,MAC7B;AAEA,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,EAAE,SAAA,EAAW,SAAA,EAAW,MAAA,EAAO;AAAA,IAC1C,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,EAAE,WAAW,CAAA,OAAA,EAAU,KAAA,CAAM,WAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA,EAAG;AAAA,IACnE;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,EAAE,WAAW,SAAA,EAAU,GAAI,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA,EAAQ,SAAA;AAAA,QACR,SAAA;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,EAEQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,QAAQ,SAAA,EAAW,SAAA,MAAe,OAAA,EAAS;AAG1E,MAAA,MAAM,UAAU,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACxD,MAAA,MAAM,SAAA,GAAY,SAAS,KAAA,EAAO,SAAA;AAClC,MAAA,MAAM,WAAA,GACF,SAAA,EAAW,KAAA,EAAO,EAAA,EAAI,eACtB,SAAA,EAAW,KAAA,GAAQ,gBAAgB,CAAA,IACnC,SAAS,KAAA,EAAO,EAAA,EAAI,WAAA,IACpB,OAAA,EAAS,QAAQ,gBAAgB,CAAA;AAErC,MAAA,IAAI,WAAA,EAAa;AACb,QAAA,MAAM,OAAA,GAAgC;AAAA,UAClC,UAAA;AAAA,UACA,WAAA;AAAA,UACA,SAAA;AAAA,UACA,QAAA;AAAA,UACA,QAAQ,SAAA,IAAa;AAAA,SACzB;AAEA,QAAA,QAAA,CAAS,IAAA,CAAK;AAAA,UACV,MAAMA,gBAAA,CAAU,MAAA;AAAA,UAChB,IAAA,EAAM,gBAAA;AAAA,UACN,KAAA,EAAO,OAAA;AAAA,UACP,SAAA,EAAW,KAAK,GAAA,EAAI;AAAA,UACpB,IAAA,EAAM;AAAA,SACF,CAAA;AAER,QAAA,OAAA,CAAQ,IAAI,CAAA,qCAAA,EAAwC,gBAAgB,UAAU,QAAQ,CAAA,WAAA,EAAc,SAAS,CAAA,CAAA,CAAG,CAAA;AAAA,MACpH;AAEA,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,oBAAoB,OAAA,EAAgC;AAChE,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                    console.log(`[AguiAdapter] Executing MCP tool: ${mcpToolName}`, args);\r\n\r\n                    // IMPORTANT: call the actual MCP tool. (Previously this mistakenly returned the listTools() result.)\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/** New event type for MCP UI triggers */\r\nexport const MCP_APP_UI_EVENT = 'mcp-apps-ui';\r\n/**\r\n * MCP Apps UI trigger event.\r\n *\r\n * IMPORTANT: This must be emitted as an AG-UI CustomEvent so subscribers\r\n * (e.g. CopilotKit `onCustomEvent`) can receive it.\r\n */\r\nexport interface McpAppUiEventPayload {\r\n    toolCallId: string;\r\n    resourceUri: string;\r\n    sessionId?: string;\r\n    toolName: string;\r\n    result?: any;\r\n}\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    /**\r\n     * Raw result object (if available).\r\n     * Used to preserve metadata (e.g. `_meta`) that is lost in the stringified `result`.\r\n     */\r\n    rawResult?: any;\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/**\r\n * Configuration for McpMiddleware\r\n */\r\nexport interface McpMiddlewareConfig {\r\n    /** Pre-loaded tools with handlers (required) */\r\n    tools: AguiTool[];\r\n}\r\n\r\n/**\r\n * AG-UI Middleware that executes MCP tools server-side.\r\n */\r\nexport class McpMiddleware extends Middleware {\r\n    private tools: AguiTool[];\r\n    private toolSchemas: Tool[];\r\n\r\n    constructor(config: McpMiddlewareConfig) {\r\n        super();\r\n        this.tools = config.tools;\r\n        this.toolSchemas = this.tools.map((t: AguiTool) => ({\r\n            name: t.name,\r\n            description: t.description,\r\n            parameters: cleanSchema(t.parameters),\r\n            _meta: t._meta, // Include _meta in the tool definition passed to the agent\r\n        }));\r\n    }\r\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<{ resultStr: string, rawResult?: any }> {\r\n        const tool = this.tools.find(t => t.name === toolName);\r\n        if (!tool?.handler) {\r\n            return { resultStr: `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}` };\r\n        }\r\n\r\n        try {\r\n            console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\r\n            // Result can be a string (legacy) or an object (MCP Result with content array)\r\n            const result = await tool.handler(args);\r\n\r\n            let resultStr: string;\r\n\r\n            if (typeof result === 'string') {\r\n                resultStr = result;\r\n            } else if (result && typeof result === 'object') {\r\n                // Determine if we should preserve the object structure (e.g. for MCP Tool Results)\r\n                resultStr = JSON.stringify(result);\r\n            } else {\r\n                resultStr = String(result);\r\n            }\r\n\r\n            console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\r\n            return { resultStr, rawResult: result };\r\n        } catch (error: any) {\r\n            console.error(`[McpMiddleware] Error executing tool:`, error);\r\n            return { resultStr: `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 { resultStr, rawResult } = await this.executeTool(toolName, args);\r\n            results.push({\r\n                toolCallId,\r\n                toolName,\r\n                result: resultStr,\r\n                rawResult,\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    private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\r\n        for (const { toolCallId, toolName, result, rawResult, messageId } of results) {\r\n            // UI metadata may appear either on the tool CALL result (rawResult._meta)\r\n            // or only on the tool DEFINITION (listTools result). We support both.\r\n            const toolDef = this.tools.find(t => t.name === toolName);\r\n            const sessionId = toolDef?._meta?.sessionId;\r\n            const resourceUri =\r\n                rawResult?._meta?.ui?.resourceUri ??\r\n                rawResult?._meta?.['ui/resourceUri'] ??\r\n                toolDef?._meta?.ui?.resourceUri ??\r\n                toolDef?._meta?.['ui/resourceUri'];\r\n\r\n            if (resourceUri) {\r\n                const payload: McpAppUiEventPayload = {\r\n                    toolCallId,\r\n                    resourceUri,\r\n                    sessionId,\r\n                    toolName,\r\n                    result: rawResult ?? result,\r\n                };\r\n\r\n                observer.next({\r\n                    type: EventType.CUSTOM,\r\n                    name: MCP_APP_UI_EVENT,\r\n                    value: payload,\r\n                    timestamp: Date.now(),\r\n                    role: 'tool',\r\n                } as any);\r\n\r\n                console.log(`[McpMiddleware] Emitting CustomEvent(${MCP_APP_UI_EVENT}) for: ${toolName} (session: ${sessionId})`);\r\n            }\r\n\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(options: { tools: AguiTool[] }) {\r\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 };\r\n"]}
+{"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;;;AChEO,IAAM,gBAAA,GAAmB;AAgDzB,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,CAAA;AAAA,MACpC,OAAO,CAAA,CAAE;AAAA;AAAA,KACb,CAAE,CAAA;AAAA,EACN;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,gBAAgB,QAAA,EAA0B;AAC9C,IAAA,MAAM,KAAA,GAAQ,QAAA,CAAS,KAAA,CAAM,mBAAmB,CAAA;AAChD,IAAA,OAAO,KAAA,GAAQ,KAAA,CAAM,CAAC,CAAA,GAAI,QAAA;AAAA,EAC9B;AAAA,EAEQ,UAAU,QAAA,EAA2B;AAEzC,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,EAA4E;AACpH,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,EAAE,WAAW,CAAA,YAAA,EAAe,IAAA,GAAO,mBAAmB,WAAW,CAAA,EAAA,EAAK,QAAQ,CAAA,CAAA,EAAG;AAAA,IAC5F;AAEA,IAAA,IAAI;AAEA,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AAEtC,MAAA,IAAI,SAAA;AAEJ,MAAA,IAAI,OAAO,WAAW,QAAA,EAAU;AAC5B,QAAA,SAAA,GAAY,MAAA;AAAA,MAChB,CAAA,MAAA,IAAW,MAAA,IAAU,OAAO,MAAA,KAAW,QAAA,EAAU;AAE7C,QAAA,SAAA,GAAY,IAAA,CAAK,UAAU,MAAM,CAAA;AAAA,MACrC,CAAA,MAAO;AACH,QAAA,SAAA,GAAY,OAAO,MAAM,CAAA;AAAA,MAC7B;AAEA,MAAA,OAAO,EAAE,SAAA,EAAW,SAAA,EAAW,MAAA,EAAO;AAAA,IAC1C,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,EAAE,WAAW,CAAA,OAAA,EAAU,KAAA,CAAM,WAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA,EAAG;AAAA,IACnE;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;AAAA,MACJ;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;AAK5C,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;AAAA,gBAC7B;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,MAAM,EAAE,WAAW,SAAA,EAAU,GAAI,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA,EAAQ,SAAA;AAAA,QACR,SAAA;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,EAEQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,QAAQ,SAAA,EAAW,SAAA,MAAe,OAAA,EAAS;AAG1E,MAAA,MAAM,UAAU,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACxD,MAAA,MAAM,SAAA,GAAY,SAAS,KAAA,EAAO,SAAA;AAClC,MAAA,MAAM,WAAA,GACF,SAAA,EAAW,KAAA,EAAO,EAAA,EAAI,eACtB,SAAA,EAAW,KAAA,GAAQ,gBAAgB,CAAA,IACnC,SAAS,KAAA,EAAO,EAAA,EAAI,WAAA,IACpB,OAAA,EAAS,QAAQ,gBAAgB,CAAA;AAErC,MAAA,IAAI,WAAA,EAAa;AAEb,QAAA,MAAM,YAAA,GAAe,IAAA,CAAK,eAAA,CAAgB,QAAQ,CAAA;AAElD,QAAA,MAAM,OAAA,GAAgC;AAAA,UAClC,UAAA;AAAA,UACA,WAAA;AAAA,UACA,SAAA;AAAA,UACA,QAAA,EAAU,YAAA;AAAA;AAAA,UACV,QAAQ,SAAA,IAAa;AAAA,SACzB;AAEA,QAAA,QAAA,CAAS,IAAA,CAAK;AAAA,UACV,MAAMA,gBAAA,CAAU,MAAA;AAAA,UAChB,IAAA,EAAM,gBAAA;AAAA,UACN,KAAA,EAAO,OAAA;AAAA,UACP,SAAA,EAAW,KAAK,GAAA,EAAI;AAAA,UACpB,IAAA,EAAM;AAAA,SACF,CAAA;AAAA,MACZ;AAEA,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;AAAA,IACZ;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;AAGjB,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;AAAA,MAC9D;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;AAGA,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;AAAA,QAC3C;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,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;AAG1C,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,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;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,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,oBAAoB,OAAA,EAAgC;AAChE,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/** New event type for MCP UI triggers */\r\nexport const MCP_APP_UI_EVENT = 'mcp-apps-ui';\r\n/**\r\n * MCP Apps UI trigger event.\r\n *\r\n * IMPORTANT: This must be emitted as an AG-UI CustomEvent so subscribers\r\n * (e.g. CopilotKit `onCustomEvent`) can receive it.\r\n */\r\nexport interface McpAppUiEventPayload {\r\n    toolCallId: string;\r\n    resourceUri: string;\r\n    sessionId?: string;\r\n    toolName: string;\r\n    result?: any;\r\n}\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    /**\r\n     * Raw result object (if available).\r\n     * Used to preserve metadata (e.g. `_meta`) that is lost in the stringified `result`.\r\n     */\r\n    rawResult?: any;\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/**\r\n * Configuration for McpMiddleware\r\n */\r\nexport interface McpMiddlewareConfig {\r\n    /** Pre-loaded tools with handlers (required) */\r\n    tools: AguiTool[];\r\n}\r\n\r\n/**\r\n * AG-UI Middleware that executes MCP tools server-side.\r\n */\r\nexport class McpMiddleware extends Middleware {\r\n    private tools: AguiTool[];\r\n    private toolSchemas: Tool[];\r\n\r\n    constructor(config: McpMiddlewareConfig) {\r\n        super();\r\n        this.tools = config.tools;\r\n        this.toolSchemas = this.tools.map((t: AguiTool) => ({\r\n            name: t.name,\r\n            description: t.description,\r\n            parameters: cleanSchema(t.parameters),\r\n            _meta: t._meta, // Include _meta in the tool definition passed to the agent\r\n        }));\r\n    }\r\n\r\n    /**\r\n     * Extract base tool name from prefixed format for event emission\r\n     * e.g., \"tool_abc123_get-time\" -> \"get-time\"\r\n     */\r\n    private getBaseToolName(toolName: string): string {\r\n        const match = toolName.match(/^tool_[^_]+_(.+)$/);\r\n        return match ? match[1] : toolName;\r\n    }\r\n\r\n    private isMcpTool(toolName: string): boolean {\r\n        // Direct comparison - tool names should match as-is\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<{ resultStr: string, rawResult?: any }> {\r\n        const tool = this.tools.find(t => t.name === toolName);\r\n        if (!tool?.handler) {\r\n            return { resultStr: `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}` };\r\n        }\r\n\r\n        try {\r\n            // Result can be a string (legacy) or an object (MCP Result with content array)\r\n            const result = await tool.handler(args);\r\n\r\n            let resultStr: string;\r\n\r\n            if (typeof result === 'string') {\r\n                resultStr = result;\r\n            } else if (result && typeof result === 'object') {\r\n                // Determine if we should preserve the object structure (e.g. for MCP Tool Results)\r\n                resultStr = JSON.stringify(result);\r\n            } else {\r\n                resultStr = String(result);\r\n            }\r\n\r\n            return { resultStr, rawResult: result };\r\n        } catch (error: any) {\r\n            console.error(`[McpMiddleware] Error executing tool:`, error);\r\n            return { resultStr: `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            }\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            // Track tool call end event\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                                }\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            const { resultStr, rawResult } = await this.executeTool(toolName, args);\r\n            results.push({\r\n                toolCallId,\r\n                toolName,\r\n                result: resultStr,\r\n                rawResult,\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    private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\r\n        for (const { toolCallId, toolName, result, rawResult, messageId } of results) {\r\n            // UI metadata may appear either on the tool CALL result (rawResult._meta)\r\n            // or only on the tool DEFINITION (listTools result). We support both.\r\n            const toolDef = this.tools.find(t => t.name === toolName);\r\n            const sessionId = toolDef?._meta?.sessionId;\r\n            const resourceUri =\r\n                rawResult?._meta?.ui?.resourceUri ??\r\n                rawResult?._meta?.['ui/resourceUri'] ??\r\n                toolDef?._meta?.ui?.resourceUri ??\r\n                toolDef?._meta?.['ui/resourceUri'];\r\n\r\n            if (resourceUri) {\r\n                // Extract base name for event emission to match metadata\r\n                const baseToolName = this.getBaseToolName(toolName);\r\n\r\n                const payload: McpAppUiEventPayload = {\r\n                    toolCallId,\r\n                    resourceUri,\r\n                    sessionId,\r\n                    toolName: baseToolName, // Use base name to match metadata\r\n                    result: rawResult ?? result,\r\n                };\r\n\r\n                observer.next({\r\n                    type: EventType.CUSTOM,\r\n                    name: MCP_APP_UI_EVENT,\r\n                    value: payload,\r\n                    timestamp: Date.now(),\r\n                    role: 'tool',\r\n                } as any);\r\n            }\r\n\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        }\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            // Inject MCP tools\r\n            if (this.toolSchemas?.length) {\r\n                input.tools = [...(input.tools || []), ...this.toolSchemas];\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                // 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                }\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                // 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\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                            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                            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                        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(options: { tools: AguiTool[] }) {\r\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 };\r\n"]}