@mcp-ts/sdk 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +297 -0
- package/dist/adapters/agui-adapter.d.mts +119 -0
- package/dist/adapters/agui-adapter.d.ts +119 -0
- package/dist/adapters/agui-adapter.js +109 -0
- package/dist/adapters/agui-adapter.js.map +1 -0
- package/dist/adapters/agui-adapter.mjs +107 -0
- package/dist/adapters/agui-adapter.mjs.map +1 -0
- package/dist/adapters/agui-middleware.d.mts +171 -0
- package/dist/adapters/agui-middleware.d.ts +171 -0
- package/dist/adapters/agui-middleware.js +429 -0
- package/dist/adapters/agui-middleware.js.map +1 -0
- package/dist/adapters/agui-middleware.mjs +417 -0
- package/dist/adapters/agui-middleware.mjs.map +1 -0
- package/dist/adapters/ai-adapter.d.mts +38 -0
- package/dist/adapters/ai-adapter.d.ts +38 -0
- package/dist/adapters/ai-adapter.js +82 -0
- package/dist/adapters/ai-adapter.js.map +1 -0
- package/dist/adapters/ai-adapter.mjs +80 -0
- package/dist/adapters/ai-adapter.mjs.map +1 -0
- package/dist/adapters/langchain-adapter.d.mts +46 -0
- package/dist/adapters/langchain-adapter.d.ts +46 -0
- package/dist/adapters/langchain-adapter.js +102 -0
- package/dist/adapters/langchain-adapter.js.map +1 -0
- package/dist/adapters/langchain-adapter.mjs +100 -0
- package/dist/adapters/langchain-adapter.mjs.map +1 -0
- package/dist/adapters/mastra-adapter.d.mts +49 -0
- package/dist/adapters/mastra-adapter.d.ts +49 -0
- package/dist/adapters/mastra-adapter.js +95 -0
- package/dist/adapters/mastra-adapter.js.map +1 -0
- package/dist/adapters/mastra-adapter.mjs +93 -0
- package/dist/adapters/mastra-adapter.mjs.map +1 -0
- package/dist/client/index.d.mts +119 -0
- package/dist/client/index.d.ts +119 -0
- package/dist/client/index.js +225 -0
- package/dist/client/index.js.map +1 -0
- package/dist/client/index.mjs +223 -0
- package/dist/client/index.mjs.map +1 -0
- package/dist/client/react.d.mts +151 -0
- package/dist/client/react.d.ts +151 -0
- package/dist/client/react.js +492 -0
- package/dist/client/react.js.map +1 -0
- package/dist/client/react.mjs +489 -0
- package/dist/client/react.mjs.map +1 -0
- package/dist/client/vue.d.mts +157 -0
- package/dist/client/vue.d.ts +157 -0
- package/dist/client/vue.js +474 -0
- package/dist/client/vue.js.map +1 -0
- package/dist/client/vue.mjs +471 -0
- package/dist/client/vue.mjs.map +1 -0
- package/dist/events-BP6WyRNh.d.mts +110 -0
- package/dist/events-BP6WyRNh.d.ts +110 -0
- package/dist/index.d.mts +10 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.js +2784 -0
- package/dist/index.js.map +1 -0
- package/dist/index.mjs +2723 -0
- package/dist/index.mjs.map +1 -0
- package/dist/multi-session-client-BOFgPypS.d.ts +389 -0
- package/dist/multi-session-client-DMF3ED2O.d.mts +389 -0
- package/dist/server/index.d.mts +269 -0
- package/dist/server/index.d.ts +269 -0
- package/dist/server/index.js +2444 -0
- package/dist/server/index.js.map +1 -0
- package/dist/server/index.mjs +2414 -0
- package/dist/server/index.mjs.map +1 -0
- package/dist/shared/index.d.mts +24 -0
- package/dist/shared/index.d.ts +24 -0
- package/dist/shared/index.js +223 -0
- package/dist/shared/index.js.map +1 -0
- package/dist/shared/index.mjs +190 -0
- package/dist/shared/index.mjs.map +1 -0
- package/dist/types-SbDlA2VX.d.mts +153 -0
- package/dist/types-SbDlA2VX.d.ts +153 -0
- package/dist/utils-0qmYrqoa.d.mts +92 -0
- package/dist/utils-0qmYrqoa.d.ts +92 -0
- package/package.json +165 -0
- package/src/adapters/agui-adapter.ts +210 -0
- package/src/adapters/agui-middleware.ts +512 -0
- package/src/adapters/ai-adapter.ts +115 -0
- package/src/adapters/langchain-adapter.ts +127 -0
- package/src/adapters/mastra-adapter.ts +126 -0
- package/src/client/core/sse-client.ts +340 -0
- package/src/client/index.ts +26 -0
- package/src/client/react/index.ts +10 -0
- package/src/client/react/useMcp.ts +558 -0
- package/src/client/vue/index.ts +10 -0
- package/src/client/vue/useMcp.ts +542 -0
- package/src/index.ts +11 -0
- package/src/server/handlers/nextjs-handler.ts +216 -0
- package/src/server/handlers/sse-handler.ts +699 -0
- package/src/server/index.ts +57 -0
- package/src/server/mcp/multi-session-client.ts +132 -0
- package/src/server/mcp/oauth-client.ts +1168 -0
- package/src/server/mcp/storage-oauth-provider.ts +239 -0
- package/src/server/storage/file-backend.ts +169 -0
- package/src/server/storage/index.ts +115 -0
- package/src/server/storage/memory-backend.ts +132 -0
- package/src/server/storage/redis-backend.ts +210 -0
- package/src/server/storage/redis.ts +160 -0
- package/src/server/storage/types.ts +109 -0
- package/src/shared/constants.ts +29 -0
- package/src/shared/errors.ts +133 -0
- package/src/shared/events.ts +166 -0
- package/src/shared/index.ts +70 -0
- package/src/shared/types.ts +274 -0
- package/src/shared/utils.ts +16 -0
|
@@ -0,0 +1,417 @@
|
|
|
1
|
+
import { Observable } from 'rxjs';
|
|
2
|
+
import { Middleware, EventType } from '@ag-ui/client';
|
|
3
|
+
export { EventType, Middleware } from '@ag-ui/client';
|
|
4
|
+
|
|
5
|
+
var __defProp = Object.defineProperty;
|
|
6
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
7
|
+
var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
|
|
8
|
+
var __esm = (fn, res) => function __init() {
|
|
9
|
+
return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
|
|
10
|
+
};
|
|
11
|
+
var __export = (target, all) => {
|
|
12
|
+
for (var name in all)
|
|
13
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
14
|
+
};
|
|
15
|
+
var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
|
|
16
|
+
|
|
17
|
+
// src/adapters/agui-adapter.ts
|
|
18
|
+
var agui_adapter_exports = {};
|
|
19
|
+
__export(agui_adapter_exports, {
|
|
20
|
+
AguiAdapter: () => AguiAdapter
|
|
21
|
+
});
|
|
22
|
+
var AguiAdapter;
|
|
23
|
+
var init_agui_adapter = __esm({
|
|
24
|
+
"src/adapters/agui-adapter.ts"() {
|
|
25
|
+
AguiAdapter = class {
|
|
26
|
+
constructor(client, options = {}) {
|
|
27
|
+
this.client = client;
|
|
28
|
+
this.options = options;
|
|
29
|
+
}
|
|
30
|
+
/**
|
|
31
|
+
* Get tools with handlers for MCP tool execution.
|
|
32
|
+
*
|
|
33
|
+
* Each tool includes a handler function that:
|
|
34
|
+
* 1. Calls the MCP tool via the client
|
|
35
|
+
* 2. Extracts text content from the result
|
|
36
|
+
* 3. Returns the result as a string or JSON
|
|
37
|
+
*
|
|
38
|
+
* @returns Array of AguiTool objects
|
|
39
|
+
*/
|
|
40
|
+
async getTools() {
|
|
41
|
+
const isMultiSession = typeof this.client.getClients === "function";
|
|
42
|
+
if (isMultiSession) {
|
|
43
|
+
const clients = this.client.getClients();
|
|
44
|
+
const allTools = [];
|
|
45
|
+
for (const client of clients) {
|
|
46
|
+
const tools = await this.transformTools(client);
|
|
47
|
+
allTools.push(...tools);
|
|
48
|
+
}
|
|
49
|
+
return allTools;
|
|
50
|
+
}
|
|
51
|
+
return this.transformTools(this.client);
|
|
52
|
+
}
|
|
53
|
+
async transformTools(client) {
|
|
54
|
+
if (!client.isConnected()) {
|
|
55
|
+
return [];
|
|
56
|
+
}
|
|
57
|
+
const result = await client.listTools();
|
|
58
|
+
const prefix = this.options.prefix ?? client.getServerId() ?? "mcp";
|
|
59
|
+
const tools = [];
|
|
60
|
+
for (const tool of result.tools) {
|
|
61
|
+
const toolName = `${prefix}_${tool.name}`;
|
|
62
|
+
tools.push({
|
|
63
|
+
name: toolName,
|
|
64
|
+
description: tool.description || `Execute ${tool.name}`,
|
|
65
|
+
parameters: tool.inputSchema || { type: "object", properties: {} },
|
|
66
|
+
handler: async (args) => {
|
|
67
|
+
console.log(`[AguiAdapter] Executing MCP tool: ${tool.name}`, args);
|
|
68
|
+
const result2 = await client.callTool(tool.name, args);
|
|
69
|
+
if (result2.content && Array.isArray(result2.content)) {
|
|
70
|
+
const textContent = result2.content.filter((c) => c.type === "text").map((c) => c.text).join("\n");
|
|
71
|
+
return textContent || result2;
|
|
72
|
+
}
|
|
73
|
+
return result2;
|
|
74
|
+
}
|
|
75
|
+
});
|
|
76
|
+
}
|
|
77
|
+
return tools;
|
|
78
|
+
}
|
|
79
|
+
/**
|
|
80
|
+
* Get tools as a function (for dynamic loading).
|
|
81
|
+
*
|
|
82
|
+
* @returns Function that returns a Promise of tools
|
|
83
|
+
*/
|
|
84
|
+
getToolsFunction() {
|
|
85
|
+
return async () => this.getTools();
|
|
86
|
+
}
|
|
87
|
+
/**
|
|
88
|
+
* Get tool definitions in JSON Schema format for passing to remote agents.
|
|
89
|
+
*
|
|
90
|
+
* This format is compatible with:
|
|
91
|
+
* - OpenAI's function calling API
|
|
92
|
+
* - AG-UI input.tools format
|
|
93
|
+
* - Most LLM tool/function calling implementations
|
|
94
|
+
*
|
|
95
|
+
* @returns Array of AguiToolDefinition objects
|
|
96
|
+
*/
|
|
97
|
+
async getToolDefinitions() {
|
|
98
|
+
const isMultiSession = typeof this.client.getClients === "function";
|
|
99
|
+
if (isMultiSession) {
|
|
100
|
+
const clients = this.client.getClients();
|
|
101
|
+
const allTools = [];
|
|
102
|
+
for (const client of clients) {
|
|
103
|
+
const tools = await this.transformToolDefinitions(client);
|
|
104
|
+
allTools.push(...tools);
|
|
105
|
+
}
|
|
106
|
+
return allTools;
|
|
107
|
+
}
|
|
108
|
+
return this.transformToolDefinitions(this.client);
|
|
109
|
+
}
|
|
110
|
+
async transformToolDefinitions(client) {
|
|
111
|
+
if (!client.isConnected()) {
|
|
112
|
+
return [];
|
|
113
|
+
}
|
|
114
|
+
const result = await client.listTools();
|
|
115
|
+
const prefix = this.options.prefix ?? client.getServerId() ?? "mcp";
|
|
116
|
+
const tools = [];
|
|
117
|
+
for (const tool of result.tools) {
|
|
118
|
+
tools.push({
|
|
119
|
+
name: `${prefix}_${tool.name}`,
|
|
120
|
+
description: tool.description || `Execute ${tool.name}`,
|
|
121
|
+
parameters: tool.inputSchema || { type: "object", properties: {} }
|
|
122
|
+
});
|
|
123
|
+
}
|
|
124
|
+
return tools;
|
|
125
|
+
}
|
|
126
|
+
};
|
|
127
|
+
}
|
|
128
|
+
});
|
|
129
|
+
var McpMiddleware = class extends Middleware {
|
|
130
|
+
constructor(config) {
|
|
131
|
+
super();
|
|
132
|
+
__publicField(this, "client");
|
|
133
|
+
__publicField(this, "toolPrefix");
|
|
134
|
+
__publicField(this, "actions");
|
|
135
|
+
__publicField(this, "tools");
|
|
136
|
+
__publicField(this, "actionsLoaded", false);
|
|
137
|
+
this.client = config.client;
|
|
138
|
+
this.toolPrefix = config.toolPrefix ?? "server-";
|
|
139
|
+
this.actions = config.tools ?? null;
|
|
140
|
+
this.tools = null;
|
|
141
|
+
if (this.actions) {
|
|
142
|
+
this.actionsLoaded = true;
|
|
143
|
+
this.tools = this.actionsToTools(this.actions);
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
/**
|
|
147
|
+
* Convert actions to AG-UI tool format
|
|
148
|
+
*/
|
|
149
|
+
actionsToTools(actions) {
|
|
150
|
+
return actions.map((action) => ({
|
|
151
|
+
name: action.name,
|
|
152
|
+
description: action.description,
|
|
153
|
+
parameters: action.parameters || { type: "object", properties: {} }
|
|
154
|
+
}));
|
|
155
|
+
}
|
|
156
|
+
/**
|
|
157
|
+
* Check if a tool name is an MCP tool (matches the configured prefix)
|
|
158
|
+
*/
|
|
159
|
+
isMcpTool(toolName) {
|
|
160
|
+
return toolName.startsWith(this.toolPrefix);
|
|
161
|
+
}
|
|
162
|
+
/**
|
|
163
|
+
* Load actions from the MCP client if not already loaded
|
|
164
|
+
*/
|
|
165
|
+
async ensureActionsLoaded() {
|
|
166
|
+
if (this.actionsLoaded) return;
|
|
167
|
+
const { AguiAdapter: AguiAdapter2 } = await Promise.resolve().then(() => (init_agui_adapter(), agui_adapter_exports));
|
|
168
|
+
const adapter = new AguiAdapter2(this.client);
|
|
169
|
+
this.actions = await adapter.getTools();
|
|
170
|
+
this.actionsLoaded = true;
|
|
171
|
+
}
|
|
172
|
+
/**
|
|
173
|
+
* Execute an MCP tool and return the result as a string
|
|
174
|
+
*/
|
|
175
|
+
async executeTool(toolName, args) {
|
|
176
|
+
await this.ensureActionsLoaded();
|
|
177
|
+
const action = this.actions?.find((a) => a.name === toolName);
|
|
178
|
+
if (!action) {
|
|
179
|
+
return `Error: Tool not found: ${toolName}`;
|
|
180
|
+
}
|
|
181
|
+
if (!action.handler) {
|
|
182
|
+
return `Error: Tool has no handler: ${toolName}`;
|
|
183
|
+
}
|
|
184
|
+
try {
|
|
185
|
+
console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);
|
|
186
|
+
const result = await action.handler(args);
|
|
187
|
+
console.log(`[McpMiddleware] Tool result:`, typeof result === "string" ? result.slice(0, 200) : result);
|
|
188
|
+
return typeof result === "string" ? result : JSON.stringify(result);
|
|
189
|
+
} catch (error) {
|
|
190
|
+
console.error(`[McpMiddleware] Error executing tool:`, error);
|
|
191
|
+
return `Error executing tool: ${error.message || String(error)}`;
|
|
192
|
+
}
|
|
193
|
+
}
|
|
194
|
+
/**
|
|
195
|
+
* Generate a unique message ID for tool results
|
|
196
|
+
*/
|
|
197
|
+
generateMessageId() {
|
|
198
|
+
return `mcp_result_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;
|
|
199
|
+
}
|
|
200
|
+
/**
|
|
201
|
+
* Run the middleware, intercepting and executing MCP tool calls
|
|
202
|
+
*/
|
|
203
|
+
run(input, next) {
|
|
204
|
+
return new Observable((observer) => {
|
|
205
|
+
const toolCallArgsBuffer = /* @__PURE__ */ new Map();
|
|
206
|
+
const toolCallNames = /* @__PURE__ */ new Map();
|
|
207
|
+
const pendingMcpCalls = /* @__PURE__ */ new Set();
|
|
208
|
+
console.log(`[McpMiddleware] Starting run with ${this.actions?.length ?? 0} registered actions`);
|
|
209
|
+
console.log(`[McpMiddleware] Tool prefix: "${this.toolPrefix}"`);
|
|
210
|
+
if (this.tools && this.tools.length > 0) {
|
|
211
|
+
const existingTools = input.tools || [];
|
|
212
|
+
input.tools = [...existingTools, ...this.tools];
|
|
213
|
+
console.log(`[McpMiddleware] Injected ${this.tools.length} MCP tools into input.tools`);
|
|
214
|
+
console.log(`[McpMiddleware] Total tools: ${input.tools.length}`);
|
|
215
|
+
console.log(`[McpMiddleware] Tool names:`, this.tools.map((t) => t.name));
|
|
216
|
+
}
|
|
217
|
+
const handleRunFinished = async (event) => {
|
|
218
|
+
if (pendingMcpCalls.size === 0) {
|
|
219
|
+
observer.next(event);
|
|
220
|
+
observer.complete();
|
|
221
|
+
return;
|
|
222
|
+
}
|
|
223
|
+
console.log(`[McpMiddleware] RUN_FINISHED received with ${pendingMcpCalls.size} pending MCP calls`);
|
|
224
|
+
const callPromises = [...pendingMcpCalls].map(async (toolCallId) => {
|
|
225
|
+
const toolName = toolCallNames.get(toolCallId);
|
|
226
|
+
if (!toolName) return;
|
|
227
|
+
const argsString = toolCallArgsBuffer.get(toolCallId) || "{}";
|
|
228
|
+
let args = {};
|
|
229
|
+
try {
|
|
230
|
+
args = JSON.parse(argsString);
|
|
231
|
+
} catch (e) {
|
|
232
|
+
console.error(`[McpMiddleware] Failed to parse args:`, argsString);
|
|
233
|
+
}
|
|
234
|
+
console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);
|
|
235
|
+
const result = await this.executeTool(toolName, args);
|
|
236
|
+
const messageId = this.generateMessageId();
|
|
237
|
+
const resultEvent = {
|
|
238
|
+
type: EventType.TOOL_CALL_RESULT,
|
|
239
|
+
toolCallId,
|
|
240
|
+
messageId,
|
|
241
|
+
content: result,
|
|
242
|
+
role: "tool",
|
|
243
|
+
timestamp: Date.now()
|
|
244
|
+
};
|
|
245
|
+
console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);
|
|
246
|
+
observer.next(resultEvent);
|
|
247
|
+
input.messages.push({
|
|
248
|
+
id: messageId,
|
|
249
|
+
role: "tool",
|
|
250
|
+
toolCallId,
|
|
251
|
+
content: result
|
|
252
|
+
});
|
|
253
|
+
pendingMcpCalls.delete(toolCallId);
|
|
254
|
+
});
|
|
255
|
+
await Promise.all(callPromises);
|
|
256
|
+
console.log(`[McpMiddleware] All MCP tools executed, emitting RUN_FINISHED`);
|
|
257
|
+
observer.next({
|
|
258
|
+
type: EventType.RUN_FINISHED,
|
|
259
|
+
threadId: input.threadId,
|
|
260
|
+
runId: input.runId,
|
|
261
|
+
timestamp: Date.now()
|
|
262
|
+
});
|
|
263
|
+
console.log(`[McpMiddleware] Triggering new run`);
|
|
264
|
+
this.triggerNewRun(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls);
|
|
265
|
+
};
|
|
266
|
+
const subscription = next.run(input).subscribe({
|
|
267
|
+
next: (event) => {
|
|
268
|
+
if (event.type === EventType.TOOL_CALL_START) {
|
|
269
|
+
const startEvent = event;
|
|
270
|
+
if (startEvent.toolCallId && startEvent.toolCallName) {
|
|
271
|
+
toolCallNames.set(startEvent.toolCallId, startEvent.toolCallName);
|
|
272
|
+
const isMcp = this.isMcpTool(startEvent.toolCallName);
|
|
273
|
+
console.log(`[McpMiddleware] TOOL_CALL_START: ${startEvent.toolCallName} (id: ${startEvent.toolCallId}, isMCP: ${isMcp})`);
|
|
274
|
+
if (isMcp) {
|
|
275
|
+
pendingMcpCalls.add(startEvent.toolCallId);
|
|
276
|
+
}
|
|
277
|
+
}
|
|
278
|
+
}
|
|
279
|
+
if (event.type === EventType.TOOL_CALL_ARGS) {
|
|
280
|
+
const argsEvent = event;
|
|
281
|
+
if (argsEvent.toolCallId && argsEvent.delta) {
|
|
282
|
+
const existing = toolCallArgsBuffer.get(argsEvent.toolCallId) || "";
|
|
283
|
+
toolCallArgsBuffer.set(argsEvent.toolCallId, existing + argsEvent.delta);
|
|
284
|
+
}
|
|
285
|
+
}
|
|
286
|
+
if (event.type === EventType.TOOL_CALL_END) {
|
|
287
|
+
const endEvent = event;
|
|
288
|
+
const toolName = toolCallNames.get(endEvent.toolCallId);
|
|
289
|
+
console.log(`[McpMiddleware] TOOL_CALL_END: ${toolName ?? "unknown"} (id: ${endEvent.toolCallId})`);
|
|
290
|
+
}
|
|
291
|
+
if (event.type === EventType.RUN_FINISHED) {
|
|
292
|
+
handleRunFinished(event);
|
|
293
|
+
return;
|
|
294
|
+
}
|
|
295
|
+
observer.next(event);
|
|
296
|
+
},
|
|
297
|
+
error: (error) => {
|
|
298
|
+
observer.error(error);
|
|
299
|
+
},
|
|
300
|
+
complete: () => {
|
|
301
|
+
if (pendingMcpCalls.size === 0) {
|
|
302
|
+
observer.complete();
|
|
303
|
+
}
|
|
304
|
+
}
|
|
305
|
+
});
|
|
306
|
+
return () => {
|
|
307
|
+
subscription.unsubscribe();
|
|
308
|
+
};
|
|
309
|
+
});
|
|
310
|
+
}
|
|
311
|
+
triggerNewRun(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls) {
|
|
312
|
+
toolCallArgsBuffer.clear();
|
|
313
|
+
toolCallNames.clear();
|
|
314
|
+
pendingMcpCalls.clear();
|
|
315
|
+
console.log(`[McpMiddleware] Starting new run with updated messages`);
|
|
316
|
+
next.run(input).subscribe({
|
|
317
|
+
next: (event) => {
|
|
318
|
+
if (event.type === EventType.TOOL_CALL_START) {
|
|
319
|
+
const startEvent = event;
|
|
320
|
+
if (startEvent.toolCallId && startEvent.toolCallName) {
|
|
321
|
+
toolCallNames.set(startEvent.toolCallId, startEvent.toolCallName);
|
|
322
|
+
const isMcp = this.isMcpTool(startEvent.toolCallName);
|
|
323
|
+
console.log(`[McpMiddleware] TOOL_CALL_START: ${startEvent.toolCallName} (id: ${startEvent.toolCallId}, isMCP: ${isMcp})`);
|
|
324
|
+
if (isMcp) {
|
|
325
|
+
pendingMcpCalls.add(startEvent.toolCallId);
|
|
326
|
+
}
|
|
327
|
+
}
|
|
328
|
+
}
|
|
329
|
+
if (event.type === EventType.TOOL_CALL_ARGS) {
|
|
330
|
+
const argsEvent = event;
|
|
331
|
+
if (argsEvent.toolCallId && argsEvent.delta) {
|
|
332
|
+
const existing = toolCallArgsBuffer.get(argsEvent.toolCallId) || "";
|
|
333
|
+
toolCallArgsBuffer.set(argsEvent.toolCallId, existing + argsEvent.delta);
|
|
334
|
+
}
|
|
335
|
+
}
|
|
336
|
+
if (event.type === EventType.TOOL_CALL_END) {
|
|
337
|
+
const endEvent = event;
|
|
338
|
+
const toolName = toolCallNames.get(endEvent.toolCallId);
|
|
339
|
+
console.log(`[McpMiddleware] TOOL_CALL_END: ${toolName ?? "unknown"} (id: ${endEvent.toolCallId})`);
|
|
340
|
+
}
|
|
341
|
+
if (event.type === EventType.RUN_FINISHED) {
|
|
342
|
+
if (pendingMcpCalls.size > 0) {
|
|
343
|
+
console.log(`[McpMiddleware] RUN_FINISHED with ${pendingMcpCalls.size} pending calls, executing...`);
|
|
344
|
+
this.handlePendingCalls(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls);
|
|
345
|
+
} else {
|
|
346
|
+
observer.next(event);
|
|
347
|
+
observer.complete();
|
|
348
|
+
}
|
|
349
|
+
return;
|
|
350
|
+
}
|
|
351
|
+
observer.next(event);
|
|
352
|
+
},
|
|
353
|
+
error: (error) => observer.error(error),
|
|
354
|
+
complete: () => {
|
|
355
|
+
if (pendingMcpCalls.size === 0) {
|
|
356
|
+
observer.complete();
|
|
357
|
+
}
|
|
358
|
+
}
|
|
359
|
+
});
|
|
360
|
+
}
|
|
361
|
+
async handlePendingCalls(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls) {
|
|
362
|
+
const callPromises = [...pendingMcpCalls].map(async (toolCallId) => {
|
|
363
|
+
const toolName = toolCallNames.get(toolCallId);
|
|
364
|
+
if (!toolName) return;
|
|
365
|
+
const argsString = toolCallArgsBuffer.get(toolCallId) || "{}";
|
|
366
|
+
let args = {};
|
|
367
|
+
try {
|
|
368
|
+
args = JSON.parse(argsString);
|
|
369
|
+
} catch (e) {
|
|
370
|
+
console.error(`[McpMiddleware] Failed to parse args:`, argsString);
|
|
371
|
+
}
|
|
372
|
+
console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);
|
|
373
|
+
const result = await this.executeTool(toolName, args);
|
|
374
|
+
const messageId = this.generateMessageId();
|
|
375
|
+
const resultEvent = {
|
|
376
|
+
type: EventType.TOOL_CALL_RESULT,
|
|
377
|
+
toolCallId,
|
|
378
|
+
messageId,
|
|
379
|
+
content: result,
|
|
380
|
+
role: "tool",
|
|
381
|
+
timestamp: Date.now()
|
|
382
|
+
};
|
|
383
|
+
console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);
|
|
384
|
+
observer.next(resultEvent);
|
|
385
|
+
input.messages.push({
|
|
386
|
+
id: messageId,
|
|
387
|
+
role: "tool",
|
|
388
|
+
toolCallId,
|
|
389
|
+
content: result
|
|
390
|
+
});
|
|
391
|
+
pendingMcpCalls.delete(toolCallId);
|
|
392
|
+
});
|
|
393
|
+
await Promise.all(callPromises);
|
|
394
|
+
console.log(`[McpMiddleware] Pending tools executed, emitting RUN_FINISHED`);
|
|
395
|
+
observer.next({
|
|
396
|
+
type: EventType.RUN_FINISHED,
|
|
397
|
+
threadId: input.threadId,
|
|
398
|
+
runId: input.runId,
|
|
399
|
+
timestamp: Date.now()
|
|
400
|
+
});
|
|
401
|
+
console.log(`[McpMiddleware] Triggering new run`);
|
|
402
|
+
this.triggerNewRun(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls);
|
|
403
|
+
}
|
|
404
|
+
};
|
|
405
|
+
function createMcpMiddleware(client, options = {}) {
|
|
406
|
+
const middleware = new McpMiddleware({
|
|
407
|
+
client,
|
|
408
|
+
...options
|
|
409
|
+
});
|
|
410
|
+
return (input, next) => {
|
|
411
|
+
return middleware.run(input, next);
|
|
412
|
+
};
|
|
413
|
+
}
|
|
414
|
+
|
|
415
|
+
export { McpMiddleware, McpMiddleware as McpToolExecutorMiddleware, createMcpMiddleware, createMcpMiddleware as createMcpToolMiddleware };
|
|
416
|
+
//# sourceMappingURL=agui-middleware.mjs.map
|
|
417
|
+
//# sourceMappingURL=agui-middleware.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":["result","AguiAdapter"],"mappings":";;;;;;;;;;;;;;;;;AAAA,IAAA,oBAAA,GAAA,EAAA;AAAA,QAAA,CAAA,oBAAA,EAAA;AAAA,EAAA,WAAA,EAAA,MAAA;AAAA,CAAA,CAAA;AAAA,IAkFa,WAAA;AAlFb,IAAA,iBAAA,GAAA,KAAA,CAAA;AAAA,EAAA,8BAAA,GAAA;AAkFO,IAAM,cAAN,MAAkB;AAAA,MACrB,WAAA,CACY,MAAA,EACA,OAAA,GAA8B,EAAC,EACzC;AAFU,QAAA,IAAA,CAAA,MAAA,GAAA,MAAA;AACA,QAAA,IAAA,CAAA,OAAA,GAAA,OAAA;AAAA,MACR;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAYJ,MAAM,QAAA,GAAgC;AAClC,QAAA,MAAM,cAAA,GAAiB,OAAQ,IAAA,CAAK,MAAA,CAAe,UAAA,KAAe,UAAA;AAElE,QAAA,IAAI,cAAA,EAAgB;AAChB,UAAA,MAAM,OAAA,GAAW,IAAA,CAAK,MAAA,CAA8B,UAAA,EAAW;AAC/D,UAAA,MAAM,WAAuB,EAAC;AAE9B,UAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC1B,YAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,cAAA,CAAe,MAAM,CAAA;AAC9C,YAAA,QAAA,CAAS,IAAA,CAAK,GAAG,KAAK,CAAA;AAAA,UAC1B;AAEA,UAAA,OAAO,QAAA;AAAA,QACX;AAEA,QAAA,OAAO,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,MAAmB,CAAA;AAAA,MACvD;AAAA,MAEA,MAAc,eAAe,MAAA,EAAwC;AACjE,QAAA,IAAI,CAAC,MAAA,CAAO,WAAA,EAAY,EAAG;AACvB,UAAA,OAAO,EAAC;AAAA,QACZ;AAEA,QAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AACtC,QAAA,MAAM,SAAS,IAAA,CAAK,OAAA,CAAQ,MAAA,IAAU,MAAA,CAAO,aAAY,IAAK,KAAA;AAC9D,QAAA,MAAM,QAAoB,EAAC;AAE3B,QAAA,KAAA,MAAW,IAAA,IAAQ,OAAO,KAAA,EAAO;AAC7B,UAAA,MAAM,QAAA,GAAW,CAAA,EAAG,MAAM,CAAA,CAAA,EAAI,KAAK,IAAI,CAAA,CAAA;AAEvC,UAAA,KAAA,CAAM,IAAA,CAAK;AAAA,YACP,IAAA,EAAM,QAAA;AAAA,YACN,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,QAAA,EAAW,KAAK,IAAI,CAAA,CAAA;AAAA,YACrD,UAAA,EAAY,KAAK,WAAA,IAAe,EAAE,MAAM,QAAA,EAAU,UAAA,EAAY,EAAC,EAAE;AAAA,YACjE,OAAA,EAAS,OAAO,IAAA,KAAc;AAC1B,cAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,IAAA,CAAK,IAAI,IAAI,IAAI,CAAA;AAClE,cAAA,MAAMA,UAAS,MAAM,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,IAAI,CAAA;AAGpD,cAAA,IAAIA,QAAO,OAAA,IAAW,KAAA,CAAM,OAAA,CAAQA,OAAAA,CAAO,OAAO,CAAA,EAAG;AACjD,gBAAA,MAAM,cAAcA,OAAAA,CAAO,OAAA,CACtB,MAAA,CAAO,CAAC,MAAW,CAAA,CAAE,IAAA,KAAS,MAAM,CAAA,CACpC,IAAI,CAAC,CAAA,KAAW,EAAE,IAAI,CAAA,CACtB,KAAK,IAAI,CAAA;AACd,gBAAA,OAAO,WAAA,IAAeA,OAAAA;AAAA,cAC1B;AAEA,cAAA,OAAOA,OAAAA;AAAA,YACX;AAAA,WACH,CAAA;AAAA,QACL;AAEA,QAAA,OAAO,KAAA;AAAA,MACX;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAOA,gBAAA,GAA8C;AAC1C,QAAA,OAAO,YAAY,KAAK,QAAA,EAAS;AAAA,MACrC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAYA,MAAM,kBAAA,GAAoD;AACtD,QAAA,MAAM,cAAA,GAAiB,OAAQ,IAAA,CAAK,MAAA,CAAe,UAAA,KAAe,UAAA;AAElE,QAAA,IAAI,cAAA,EAAgB;AAChB,UAAA,MAAM,OAAA,GAAW,IAAA,CAAK,MAAA,CAA8B,UAAA,EAAW;AAC/D,UAAA,MAAM,WAAiC,EAAC;AAExC,UAAA,KAAA,MAAW,UAAU,OAAA,EAAS;AAC1B,YAAA,MAAM,KAAA,GAAQ,MAAM,IAAA,CAAK,wBAAA,CAAyB,MAAM,CAAA;AACxD,YAAA,QAAA,CAAS,IAAA,CAAK,GAAG,KAAK,CAAA;AAAA,UAC1B;AAEA,UAAA,OAAO,QAAA;AAAA,QACX;AAEA,QAAA,OAAO,IAAA,CAAK,wBAAA,CAAyB,IAAA,CAAK,MAAmB,CAAA;AAAA,MACjE;AAAA,MAEA,MAAc,yBAAyB,MAAA,EAAkD;AACrF,QAAA,IAAI,CAAC,MAAA,CAAO,WAAA,EAAY,EAAG;AACvB,UAAA,OAAO,EAAC;AAAA,QACZ;AAEA,QAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,SAAA,EAAU;AACtC,QAAA,MAAM,SAAS,IAAA,CAAK,OAAA,CAAQ,MAAA,IAAU,MAAA,CAAO,aAAY,IAAK,KAAA;AAC9D,QAAA,MAAM,QAA8B,EAAC;AAErC,QAAA,KAAA,MAAW,IAAA,IAAQ,OAAO,KAAA,EAAO;AAC7B,UAAA,KAAA,CAAM,IAAA,CAAK;AAAA,YACP,IAAA,EAAM,CAAA,EAAG,MAAM,CAAA,CAAA,EAAI,KAAK,IAAI,CAAA,CAAA;AAAA,YAC5B,WAAA,EAAa,IAAA,CAAK,WAAA,IAAe,CAAA,QAAA,EAAW,KAAK,IAAI,CAAA,CAAA;AAAA,YACrD,UAAA,EAAY,KAAK,WAAA,IAAe,EAAE,MAAM,QAAA,EAAU,UAAA,EAAY,EAAC;AAAE,WACpE,CAAA;AAAA,QACL;AAEA,QAAA,OAAO,KAAA;AAAA,MACX;AAAA,KACJ;AAAA,EAAA;AAAA,CAAA,CAAA;AC7FO,IAAM,aAAA,GAAN,cAA4B,UAAA,CAAW;AAAA,EAO1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAPV,IAAA,aAAA,CAAA,IAAA,EAAQ,QAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,YAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,SAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,eAAA,EAAyB,KAAA,CAAA;AAI7B,IAAA,IAAA,CAAK,SAAS,MAAA,CAAO,MAAA;AACrB,IAAA,IAAA,CAAK,UAAA,GAAa,OAAO,UAAA,IAAc,SAAA;AACvC,IAAA,IAAA,CAAK,OAAA,GAAU,OAAO,KAAA,IAAS,IAAA;AAC/B,IAAA,IAAA,CAAK,KAAA,GAAQ,IAAA;AACb,IAAA,IAAI,KAAK,OAAA,EAAS;AACd,MAAA,IAAA,CAAK,aAAA,GAAgB,IAAA;AACrB,MAAA,IAAA,CAAK,KAAA,GAAQ,IAAA,CAAK,cAAA,CAAe,IAAA,CAAK,OAAO,CAAA;AAAA,IACjD;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKQ,eAAe,OAAA,EAAiC;AACpD,IAAA,OAAO,OAAA,CAAQ,IAAI,CAAA,MAAA,MAAW;AAAA,MAC1B,MAAM,MAAA,CAAO,IAAA;AAAA,MACb,aAAa,MAAA,CAAO,WAAA;AAAA,MACpB,UAAA,EAAY,OAAO,UAAA,IAAc,EAAE,MAAM,QAAA,EAAU,UAAA,EAAY,EAAC;AAAE,KACtE,CAAE,CAAA;AAAA,EACN;AAAA;AAAA;AAAA;AAAA,EAKQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,QAAA,CAAS,UAAA,CAAW,IAAA,CAAK,UAAU,CAAA;AAAA,EAC9C;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,mBAAA,GAAqC;AAC/C,IAAA,IAAI,KAAK,aAAA,EAAe;AAExB,IAAA,MAAM,EAAE,WAAA,EAAAC,YAAAA,EAAY,GAAI,MAAM,OAAA,CAAA,OAAA,EAAA,CAAA,IAAA,CAAA,OAAA,iBAAA,EAAA,EAAA,oBAAA,CAAA,CAAA;AAC9B,IAAA,MAAM,OAAA,GAAU,IAAIA,YAAAA,CAAY,IAAA,CAAK,MAAM,CAAA;AAC3C,IAAA,IAAA,CAAK,OAAA,GAAU,MAAM,OAAA,CAAQ,QAAA,EAAS;AACtC,IAAA,IAAA,CAAK,aAAA,GAAgB,IAAA;AAAA,EACzB;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,KAAK,mBAAA,EAAoB;AAE/B,IAAA,MAAM,SAAS,IAAA,CAAK,OAAA,EAAS,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAC1D,IAAA,IAAI,CAAC,MAAA,EAAQ;AACT,MAAA,OAAO,0BAA0B,QAAQ,CAAA,CAAA;AAAA,IAC7C;AAEA,IAAA,IAAI,CAAC,OAAO,OAAA,EAAS;AACjB,MAAA,OAAO,+BAA+B,QAAQ,CAAA,CAAA;AAAA,IAClD;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,MAAA,CAAO,OAAA,CAAQ,IAAI,CAAA;AACxC,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,4BAAA,CAAA,EAAgC,OAAO,MAAA,KAAW,QAAA,GAAW,OAAO,KAAA,CAAM,CAAA,EAAG,GAAG,CAAA,GAAI,MAAM,CAAA;AACtG,MAAA,OAAO,OAAO,MAAA,KAAW,QAAA,GAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAAA,IACtE,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,sBAAA,EAAyB,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IAClE;AAAA,EACJ;AAAA;AAAA;AAAA;AAAA,EAKQ,iBAAA,GAA4B;AAChC,IAAA,OAAO,CAAA,WAAA,EAAc,IAAA,CAAK,GAAA,EAAK,IAAI,IAAA,CAAK,MAAA,EAAO,CAAE,QAAA,CAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC7E;AAAA;AAAA;AAAA;AAAA,EAKA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAI,UAAA,CAAsB,CAAC,QAAA,KAAoC;AAElE,MAAA,MAAM,kBAAA,uBAAyB,GAAA,EAAoB;AACnD,MAAA,MAAM,aAAA,uBAAoB,GAAA,EAAoB;AAC9C,MAAA,MAAM,eAAA,uBAAsB,GAAA,EAAY;AAExC,MAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,EAAqC,IAAA,CAAK,OAAA,EAAS,MAAA,IAAU,CAAC,CAAA,mBAAA,CAAqB,CAAA;AAC/F,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,8BAAA,EAAiC,IAAA,CAAK,UAAU,CAAA,CAAA,CAAG,CAAA;AAG/D,MAAA,IAAI,IAAA,CAAK,KAAA,IAAS,IAAA,CAAK,KAAA,CAAM,SAAS,CAAA,EAAG;AACrC,QAAA,MAAM,aAAA,GAAgB,KAAA,CAAM,KAAA,IAAS,EAAC;AACtC,QAAA,KAAA,CAAM,QAAQ,CAAC,GAAG,aAAA,EAAe,GAAG,KAAK,KAAK,CAAA;AAC9C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,KAAA,CAAM,MAAM,CAAA,2BAAA,CAA6B,CAAA;AACtF,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6BAAA,EAAgC,KAAA,CAAM,KAAA,CAAM,MAAM,CAAA,CAAE,CAAA;AAChE,QAAA,OAAA,CAAQ,GAAA,CAAI,+BAA+B,IAAA,CAAK,KAAA,CAAM,IAAI,CAAA,CAAA,KAAK,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MAC1E;AAEA,MAAA,MAAM,iBAAA,GAAoB,OAAO,KAAA,KAAqB;AAClD,QAAA,IAAI,eAAA,CAAgB,SAAS,CAAA,EAAG;AAC5B,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,2CAAA,EAA8C,eAAA,CAAgB,IAAI,CAAA,kBAAA,CAAoB,CAAA;AAGlG,QAAA,MAAM,eAAe,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAChE,UAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,UAAA,IAAI,CAAC,QAAA,EAAU;AAEf,UAAA,MAAM,UAAA,GAAa,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,OAA4B,EAAC;AACjC,UAAA,IAAI;AACA,YAAA,IAAA,GAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,UAChC,SAAS,CAAA,EAAG;AACR,YAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AAAA,UACrE;AAEA,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AACjE,UAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,UAAA,MAAM,SAAA,GAAY,KAAK,iBAAA,EAAkB;AAGzC,UAAA,MAAM,WAAA,GAAyB;AAAA,YAC3B,MAAM,SAAA,CAAU,gBAAA;AAAA,YAChB,UAAA;AAAA,YACA,SAAA;AAAA,YACA,OAAA,EAAS,MAAA;AAAA,YACT,IAAA,EAAM,MAAA;AAAA,YACN,SAAA,EAAW,KAAK,GAAA;AAAI,WACxB;AAEA,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AACxE,UAAA,QAAA,CAAS,KAAK,WAAW,CAAA;AAGzB,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,UAAA;AAAA,YACA,OAAA,EAAS;AAAA,WACL,CAAA;AAER,UAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,QACrC,CAAC,CAAA;AAED,QAAA,MAAM,OAAA,CAAQ,IAAI,YAAY,CAAA;AAG9B,QAAA,OAAA,CAAQ,IAAI,CAAA,6DAAA,CAA+D,CAAA;AAC3E,QAAA,QAAA,CAAS,IAAA,CAAK;AAAA,UACV,MAAM,SAAA,CAAU,YAAA;AAAA,UAChB,UAAW,KAAA,CAAc,QAAA;AAAA,UACzB,OAAQ,KAAA,CAAc,KAAA;AAAA,UACtB,SAAA,EAAW,KAAK,GAAA;AAAI,SAChB,CAAA;AAGR,QAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,QAAA,IAAA,CAAK,cAAc,QAAA,EAAU,KAAA,EAAO,IAAA,EAAM,kBAAA,EAAoB,eAAe,eAAe,CAAA;AAAA,MAChG,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAqB;AAExB,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,YAAA,MAAM,UAAA,GAAa,KAAA;AACnB,YAAA,IAAI,UAAA,CAAW,UAAA,IAAc,UAAA,CAAW,YAAA,EAAc;AAClD,cAAA,aAAA,CAAc,GAAA,CAAI,UAAA,CAAW,UAAA,EAAY,UAAA,CAAW,YAAY,CAAA;AAChE,cAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,SAAA,CAAU,UAAA,CAAW,YAAY,CAAA;AACpD,cAAA,OAAA,CAAQ,GAAA,CAAI,oCAAoC,UAAA,CAAW,YAAY,SAAS,UAAA,CAAW,UAAU,CAAA,SAAA,EAAY,KAAK,CAAA,CAAA,CAAG,CAAA;AAEzH,cAAA,IAAI,KAAA,EAAO;AACP,gBAAA,eAAA,CAAgB,GAAA,CAAI,WAAW,UAAU,CAAA;AAAA,cAC7C;AAAA,YACJ;AAAA,UACJ;AAGA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,YAAA,MAAM,SAAA,GAAY,KAAA;AAClB,YAAA,IAAI,SAAA,CAAU,UAAA,IAAc,SAAA,CAAU,KAAA,EAAO;AACzC,cAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,SAAA,CAAU,UAAU,CAAA,IAAK,EAAA;AACjE,cAAA,kBAAA,CAAmB,GAAA,CAAI,SAAA,CAAU,UAAA,EAAY,QAAA,GAAW,UAAU,KAAK,CAAA;AAAA,YAC3E;AAAA,UACJ;AAGA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AACxC,YAAA,MAAM,QAAA,GAAW,KAAA;AACjB,YAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,QAAA,CAAS,UAAU,CAAA;AACtD,YAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,EAAkC,QAAA,IAAY,SAAS,CAAA,MAAA,EAAS,QAAA,CAAS,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,UACtG;AAGA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,CAAkB,KAAK,CAAA;AACvB,YAAA;AAAA,UACJ;AAGA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,KAAA,KAAU;AACd,UAAA,QAAA,CAAS,MAAM,KAAK,CAAA;AAAA,QACxB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,eAAA,CAAgB,SAAS,CAAA,EAAG;AAC5B,YAAA,QAAA,CAAS,QAAA,EAAS;AAAA,UACtB;AAAA,QACJ;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM;AACT,QAAA,YAAA,CAAa,WAAA,EAAY;AAAA,MAC7B,CAAA;AAAA,IACJ,CAAC,CAAA;AAAA,EACL;AAAA,EAEQ,cACJ,QAAA,EACA,KAAA,EACA,IAAA,EACA,kBAAA,EACA,eACA,eAAA,EACI;AACJ,IAAA,kBAAA,CAAmB,KAAA,EAAM;AACzB,IAAA,aAAA,CAAc,KAAA,EAAM;AACpB,IAAA,eAAA,CAAgB,KAAA,EAAM;AAEtB,IAAA,OAAA,CAAQ,IAAI,CAAA,sDAAA,CAAwD,CAAA;AAEpE,IAAqB,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,MAC3C,IAAA,EAAM,CAAC,KAAA,KAAqB;AACxB,QAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,UAAA,MAAM,UAAA,GAAa,KAAA;AACnB,UAAA,IAAI,UAAA,CAAW,UAAA,IAAc,UAAA,CAAW,YAAA,EAAc;AAClD,YAAA,aAAA,CAAc,GAAA,CAAI,UAAA,CAAW,UAAA,EAAY,UAAA,CAAW,YAAY,CAAA;AAChE,YAAA,MAAM,KAAA,GAAQ,IAAA,CAAK,SAAA,CAAU,UAAA,CAAW,YAAY,CAAA;AACpD,YAAA,OAAA,CAAQ,GAAA,CAAI,oCAAoC,UAAA,CAAW,YAAY,SAAS,UAAA,CAAW,UAAU,CAAA,SAAA,EAAY,KAAK,CAAA,CAAA,CAAG,CAAA;AAEzH,YAAA,IAAI,KAAA,EAAO;AACP,cAAA,eAAA,CAAgB,GAAA,CAAI,WAAW,UAAU,CAAA;AAAA,YAC7C;AAAA,UACJ;AAAA,QACJ;AAEA,QAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,UAAA,MAAM,SAAA,GAAY,KAAA;AAClB,UAAA,IAAI,SAAA,CAAU,UAAA,IAAc,SAAA,CAAU,KAAA,EAAO;AACzC,YAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,SAAA,CAAU,UAAU,CAAA,IAAK,EAAA;AACjE,YAAA,kBAAA,CAAmB,GAAA,CAAI,SAAA,CAAU,UAAA,EAAY,QAAA,GAAW,UAAU,KAAK,CAAA;AAAA,UAC3E;AAAA,QACJ;AAEA,QAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AACxC,UAAA,MAAM,QAAA,GAAW,KAAA;AACjB,UAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,QAAA,CAAS,UAAU,CAAA;AACtD,UAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,EAAkC,QAAA,IAAY,SAAS,CAAA,MAAA,EAAS,QAAA,CAAS,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,QACtG;AAEA,QAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,UAAA,IAAI,eAAA,CAAgB,OAAO,CAAA,EAAG;AAC1B,YAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,eAAA,CAAgB,IAAI,CAAA,4BAAA,CAA8B,CAAA;AACnG,YAAA,IAAA,CAAK,mBAAmB,QAAA,EAAU,KAAA,EAAO,IAAA,EAAM,kBAAA,EAAoB,eAAe,eAAe,CAAA;AAAA,UACrG,CAAA,MAAO;AACH,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAAA,UACtB;AACA,UAAA;AAAA,QACJ;AAEA,QAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,MACvB,CAAA;AAAA,MACA,KAAA,EAAO,CAAC,KAAA,KAAU,QAAA,CAAS,MAAM,KAAK,CAAA;AAAA,MACtC,UAAU,MAAM;AACZ,QAAA,IAAI,eAAA,CAAgB,SAAS,CAAA,EAAG;AAC5B,UAAA,QAAA,CAAS,QAAA,EAAS;AAAA,QACtB;AAAA,MACJ;AAAA,KACH;AAAA,EACL;AAAA,EAEA,MAAc,kBAAA,CACV,QAAA,EACA,OACA,IAAA,EACA,kBAAA,EACA,eACA,eAAA,EACa;AACb,IAAA,MAAM,eAAe,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAChE,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,UAAA,GAAa,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,MAAA,IAAI,OAA4B,EAAC;AACjC,MAAA,IAAI;AACA,QAAA,IAAA,GAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,MAChC,SAAS,CAAA,EAAG;AACR,QAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AAAA,MACrE;AAEA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AACjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,MAAM,SAAA,GAAY,KAAK,iBAAA,EAAkB;AAEzC,MAAA,MAAM,WAAA,GAAyB;AAAA,QAC3B,MAAM,SAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OACxB;AAEA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AACxE,MAAA,QAAA,CAAS,KAAK,WAAW,CAAA;AAEzB,MAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,QAChB,EAAA,EAAI,SAAA;AAAA,QACJ,IAAA,EAAM,MAAA;AAAA,QACN,UAAA;AAAA,QACA,OAAA,EAAS;AAAA,OACL,CAAA;AAER,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,YAAY,CAAA;AAE9B,IAAA,OAAA,CAAQ,IAAI,CAAA,6DAAA,CAA+D,CAAA;AAC3E,IAAA,QAAA,CAAS,IAAA,CAAK;AAAA,MACV,MAAM,SAAA,CAAU,YAAA;AAAA,MAChB,UAAW,KAAA,CAAc,QAAA;AAAA,MACzB,OAAQ,KAAA,CAAc,KAAA;AAAA,MACtB,SAAA,EAAW,KAAK,GAAA;AAAI,KAChB,CAAA;AAER,IAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,IAAA,IAAA,CAAK,cAAc,QAAA,EAAU,KAAA,EAAO,IAAA,EAAM,kBAAA,EAAoB,eAAe,eAAe,CAAA;AAAA,EAChG;AACJ;AAwBO,SAAS,mBAAA,CACZ,MAAA,EACA,OAAA,GAAuD,EAAC,EAC1D;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc;AAAA,IACjC,MAAA;AAAA,IACA,GAAG;AAAA,GACN,CAAA;AAED,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.mjs","sourcesContent":["/**\r\n * MCP Adapter for AG-UI Integration\r\n *\r\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\r\n * It provides tools with handlers for server-side execution and tool definitions\r\n * in JSON Schema format for passing to remote agents.\r\n *\r\n * @example\r\n * ```typescript\r\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\r\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/mcp-adapter';\r\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\r\n * import { HttpAgent } from '@ag-ui/client';\r\n *\r\n * // Create MCP client\r\n * const mcpClient = new MultiSessionClient('user_123');\r\n * await mcpClient.connect();\r\n *\r\n * // Create adapter and get tools\r\n * const adapter = new AguiAdapter(mcpClient);\r\n * const tools = await adapter.getTools();\r\n *\r\n * // Use with AG-UI middleware\r\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\r\n * agent.use(createMcpMiddleware(mcpClient, { tools }));\r\n * ```\r\n */\r\n\r\nimport { MCPClient } from '../server/mcp/oauth-client.js';\r\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\r\n\r\n/**\r\n * Configuration options for AguiAdapter\r\n */\r\nexport interface AguiAdapterOptions {\r\n /**\r\n * Prefix for tool names to avoid collision with other tools.\r\n * @default serverId or 'mcp'\r\n */\r\n prefix?: string;\r\n}\r\n\r\n/**\r\n * AG-UI Tool with handler for server-side execution.\r\n *\r\n * Tools contain:\r\n * - `name`: Unique identifier (prefixed with server ID)\r\n * - `description`: Human-readable description for the LLM\r\n * - `parameters`: JSON Schema defining the input format\r\n * - `handler`: Function that executes the tool via MCP client\r\n */\r\nexport interface AguiTool {\r\n /** Unique tool name (e.g., \"server-abc_get_weather\") */\r\n name: string;\r\n /** Human-readable description for the LLM */\r\n description: string;\r\n /** JSON Schema format parameters */\r\n parameters?: Record<string, any>;\r\n /** Handler function that executes the MCP tool */\r\n handler?: (args: any) => any | Promise<any>;\r\n}\r\n\r\n/**\r\n * Tool definition format for passing to remote agents (without handler).\r\n * Compatible with OpenAI's function calling API.\r\n */\r\nexport interface AguiToolDefinition {\r\n /** Tool name (e.g., \"server-abc_get_weather\") */\r\n name: string;\r\n /** Human-readable description */\r\n description: string;\r\n /** JSON Schema format parameters */\r\n parameters: Record<string, any>;\r\n}\r\n\r\n/**\r\n * Adapter that transforms MCP tools into AG-UI compatible formats.\r\n *\r\n * This adapter provides two main outputs:\r\n * - `getTools()`: Returns tools with handlers for server-side execution\r\n * - `getToolDefinitions()`: Returns tool definitions in JSON Schema format for remote agents\r\n */\r\nexport class AguiAdapter {\r\n constructor(\r\n private client: MCPClient | MultiSessionClient,\r\n private options: AguiAdapterOptions = {}\r\n ) { }\r\n\r\n /**\r\n * Get tools with handlers for MCP tool execution.\r\n *\r\n * Each tool includes a handler function that:\r\n * 1. Calls the MCP tool via the client\r\n * 2. Extracts text content from the result\r\n * 3. Returns the result as a string or JSON\r\n *\r\n * @returns Array of AguiTool objects\r\n */\r\n async getTools(): Promise<AguiTool[]> {\r\n const isMultiSession = typeof (this.client as any).getClients === 'function';\r\n\r\n if (isMultiSession) {\r\n const clients = (this.client as MultiSessionClient).getClients();\r\n const allTools: AguiTool[] = [];\r\n\r\n for (const client of clients) {\r\n const tools = await this.transformTools(client);\r\n allTools.push(...tools);\r\n }\r\n\r\n return allTools;\r\n }\r\n\r\n return this.transformTools(this.client as MCPClient);\r\n }\r\n\r\n private async transformTools(client: MCPClient): Promise<AguiTool[]> {\r\n if (!client.isConnected()) {\r\n return [];\r\n }\r\n\r\n const result = await client.listTools();\r\n const prefix = this.options.prefix ?? client.getServerId() ?? 'mcp';\r\n const tools: AguiTool[] = [];\r\n\r\n for (const tool of result.tools) {\r\n const toolName = `${prefix}_${tool.name}`;\r\n\r\n tools.push({\r\n name: toolName,\r\n description: tool.description || `Execute ${tool.name}`,\r\n parameters: tool.inputSchema || { type: 'object', properties: {} },\r\n handler: async (args: any) => {\r\n console.log(`[AguiAdapter] Executing MCP tool: ${tool.name}`, args);\r\n const result = await client.callTool(tool.name, args);\r\n\r\n // Extract text content from result\r\n if (result.content && Array.isArray(result.content)) {\r\n const textContent = result.content\r\n .filter((c: any) => c.type === 'text')\r\n .map((c: any) => c.text)\r\n .join('\\n');\r\n return textContent || result;\r\n }\r\n\r\n return result;\r\n }\r\n });\r\n }\r\n\r\n return tools;\r\n }\r\n\r\n /**\r\n * Get tools as a function (for dynamic loading).\r\n *\r\n * @returns Function that returns a Promise of tools\r\n */\r\n getToolsFunction(): () => Promise<AguiTool[]> {\r\n return async () => this.getTools();\r\n }\r\n\r\n /**\r\n * Get tool definitions in JSON Schema format for passing to remote agents.\r\n *\r\n * This format is compatible with:\r\n * - OpenAI's function calling API\r\n * - AG-UI input.tools format\r\n * - Most LLM tool/function calling implementations\r\n *\r\n * @returns Array of AguiToolDefinition objects\r\n */\r\n async getToolDefinitions(): Promise<AguiToolDefinition[]> {\r\n const isMultiSession = typeof (this.client as any).getClients === 'function';\r\n\r\n if (isMultiSession) {\r\n const clients = (this.client as MultiSessionClient).getClients();\r\n const allTools: AguiToolDefinition[] = [];\r\n\r\n for (const client of clients) {\r\n const tools = await this.transformToolDefinitions(client);\r\n allTools.push(...tools);\r\n }\r\n\r\n return allTools;\r\n }\r\n\r\n return this.transformToolDefinitions(this.client as MCPClient);\r\n }\r\n\r\n private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\r\n if (!client.isConnected()) {\r\n return [];\r\n }\r\n\r\n const result = await client.listTools();\r\n const prefix = this.options.prefix ?? client.getServerId() ?? 'mcp';\r\n const tools: AguiToolDefinition[] = [];\r\n\r\n for (const tool of result.tools) {\r\n tools.push({\r\n name: `${prefix}_${tool.name}`,\r\n description: tool.description || `Execute ${tool.name}`,\r\n parameters: tool.inputSchema || { type: 'object', properties: {} },\r\n });\r\n }\r\n\r\n return tools;\r\n }\r\n}\r\n","/**\n * AG-UI Middleware for MCP Tool Execution\n *\n * This middleware intercepts tool calls from remote agents (e.g., LangGraph, AutoGen)\n * and executes MCP tools server-side, returning results back to the agent.\n *\n * ## How It Works\n *\n * 1. **Tool Injection**: When a run starts, the middleware injects MCP tool definitions\n * into `input.tools` so the remote agent knows about available MCP tools.\n *\n * 2. **Event Interception**: The middleware subscribes to the agent's event stream and\n * tracks tool calls using AG-UI events:\n * - `TOOL_CALL_START`: Records tool name and ID\n * - `TOOL_CALL_ARGS`: Accumulates streamed arguments\n * - `TOOL_CALL_END`: Marks tool call as complete\n * - `RUN_FINISHED`: Triggers execution of pending MCP tools\n *\n * 3. **Server-Side Execution**: When `RUN_FINISHED` arrives with pending MCP tool calls,\n * the middleware:\n * - Executes each MCP tool via the MCP client\n * - Emits `TOOL_CALL_RESULT` events with the results\n * - Adds results to `input.messages` for context\n * - Emits `RUN_FINISHED` to close the current run\n * - Triggers a new run so the agent can process tool results\n *\n * 4. **Recursive Processing**: If the new run makes more MCP tool calls, the cycle\n * repeats until the agent completes without pending MCP calls.\n *\n * ## Tool Identification\n *\n * MCP tools are identified by a configurable prefix (default: `server-`).\n * Tools not matching this prefix are passed through without interception.\n *\n * @requires @ag-ui/client - This middleware requires @ag-ui/client as a peer dependency\n * @requires rxjs - Uses RxJS Observables for event streaming\n *\n * @example\n * ```typescript\n * import { HttpAgent } from '@ag-ui/client';\n * import { McpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\n *\n * // Create MCP client and adapter\n * const mcpClient = new MultiSessionClient('user_123');\n * await mcpClient.connect();\n *\n * const adapter = new AguiAdapter(mcpClient);\n * const actions = await adapter.getActions();\n *\n * // Create middleware with pre-loaded actions\n * const middleware = new McpMiddleware({\n * client: mcpClient,\n * actions,\n * toolPrefix: 'server-',\n * });\n *\n * // Use with HttpAgent\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(middleware);\n * ```\n */\n\nimport { Observable, Subscriber } from 'rxjs';\nimport {\n Middleware,\n EventType,\n type AbstractAgent,\n type RunAgentInput,\n type BaseEvent,\n type ToolCallEndEvent,\n} from '@ag-ui/client';\nimport { MCPClient } from '../server/mcp/oauth-client.js';\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\nimport type { AguiTool } from './agui-adapter.js';\n\n/**\n * Tool definition format for AG-UI input.tools\n */\nexport interface AgUiTool {\n name: string;\n description: string;\n parameters?: Record<string, any>;\n}\n\n/**\n * Configuration for McpMiddleware\n */\nexport interface McpMiddlewareConfig {\n /**\n * MCP client or MultiSessionClient for executing tools\n */\n client: MCPClient | MultiSessionClient;\n\n /**\n * Prefix used to identify MCP tool names.\n * Tools starting with this prefix will be executed server-side.\n * @default 'server-'\n */\n toolPrefix?: string;\n\n /**\n * Pre-loaded tools with handlers for execution.\n * If not provided, tools will be loaded from the MCP client on first use.\n */\n tools?: AguiTool[];\n}\n\n/**\n * AG-UI Middleware that executes MCP tools server-side.\n *\n * This middleware intercepts tool calls for MCP tools (identified by prefix),\n * executes them via the MCP client, and returns results to the agent.\n *\n * @see {@link createMcpMiddleware} for a simpler factory function\n */\nexport class McpMiddleware extends Middleware {\n private client: MCPClient | MultiSessionClient;\n private toolPrefix: string;\n private actions: AguiTool[] | null;\n private tools: AgUiTool[] | null;\n private actionsLoaded: boolean = false;\n\n constructor(config: McpMiddlewareConfig) {\n super();\n this.client = config.client;\n this.toolPrefix = config.toolPrefix ?? 'server-';\n this.actions = config.tools ?? null;\n this.tools = null;\n if (this.actions) {\n this.actionsLoaded = true;\n this.tools = this.actionsToTools(this.actions);\n }\n }\n\n /**\n * Convert actions to AG-UI tool format\n */\n private actionsToTools(actions: AguiTool[]): AgUiTool[] {\n return actions.map(action => ({\n name: action.name,\n description: action.description,\n parameters: action.parameters || { type: 'object', properties: {} },\n }));\n }\n\n /**\n * Check if a tool name is an MCP tool (matches the configured prefix)\n */\n private isMcpTool(toolName: string): boolean {\n return toolName.startsWith(this.toolPrefix);\n }\n\n /**\n * Load actions from the MCP client if not already loaded\n */\n private async ensureActionsLoaded(): Promise<void> {\n if (this.actionsLoaded) return;\n\n const { AguiAdapter } = await import('./agui-adapter.js');\n const adapter = new AguiAdapter(this.client);\n this.actions = await adapter.getTools();\n this.actionsLoaded = true;\n }\n\n /**\n * Execute an MCP tool and return the result as a string\n */\n private async executeTool(toolName: string, args: Record<string, any>): Promise<string> {\n await this.ensureActionsLoaded();\n\n const action = this.actions?.find(a => a.name === toolName);\n if (!action) {\n return `Error: Tool not found: ${toolName}`;\n }\n\n if (!action.handler) {\n return `Error: Tool has no handler: ${toolName}`;\n }\n\n try {\n console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\n const result = await action.handler(args);\n console.log(`[McpMiddleware] Tool result:`, typeof result === 'string' ? result.slice(0, 200) : result);\n return typeof result === 'string' ? result : JSON.stringify(result);\n } catch (error: any) {\n console.error(`[McpMiddleware] Error executing tool:`, error);\n return `Error executing tool: ${error.message || String(error)}`;\n }\n }\n\n /**\n * Generate a unique message ID for tool results\n */\n private generateMessageId(): string {\n return `mcp_result_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\n }\n\n /**\n * Run the middleware, intercepting and executing MCP tool calls\n */\n run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\n return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\n // State for this run\n const toolCallArgsBuffer = new Map<string, string>();\n const toolCallNames = new Map<string, string>();\n const pendingMcpCalls = new Set<string>();\n\n console.log(`[McpMiddleware] Starting run with ${this.actions?.length ?? 0} registered actions`);\n console.log(`[McpMiddleware] Tool prefix: \"${this.toolPrefix}\"`);\n\n // Inject MCP tools into input.tools\n if (this.tools && this.tools.length > 0) {\n const existingTools = input.tools || [];\n input.tools = [...existingTools, ...this.tools];\n console.log(`[McpMiddleware] Injected ${this.tools.length} MCP tools into input.tools`);\n console.log(`[McpMiddleware] Total tools: ${input.tools.length}`);\n console.log(`[McpMiddleware] Tool names:`, this.tools.map(t => t.name));\n }\n\n const handleRunFinished = async (event: BaseEvent) => {\n if (pendingMcpCalls.size === 0) {\n observer.next(event);\n observer.complete();\n return;\n }\n\n console.log(`[McpMiddleware] RUN_FINISHED received with ${pendingMcpCalls.size} pending MCP calls`);\n\n // Execute all pending MCP tool calls\n const callPromises = [...pendingMcpCalls].map(async (toolCallId) => {\n const toolName = toolCallNames.get(toolCallId);\n if (!toolName) return;\n\n const argsString = toolCallArgsBuffer.get(toolCallId) || '{}';\n let args: Record<string, any> = {};\n try {\n args = JSON.parse(argsString);\n } catch (e) {\n console.error(`[McpMiddleware] Failed to parse args:`, argsString);\n }\n\n console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\n const result = await this.executeTool(toolName, args);\n const messageId = this.generateMessageId();\n\n // Emit tool result event\n const resultEvent: BaseEvent = {\n type: EventType.TOOL_CALL_RESULT,\n toolCallId,\n messageId,\n content: result,\n role: 'tool',\n timestamp: Date.now(),\n } as any;\n\n console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\n observer.next(resultEvent);\n\n // Add tool result to messages for the next run\n input.messages.push({\n id: messageId,\n role: 'tool',\n toolCallId,\n content: result,\n } as any);\n\n pendingMcpCalls.delete(toolCallId);\n });\n\n await Promise.all(callPromises);\n\n // Emit RUN_FINISHED before starting new run\n console.log(`[McpMiddleware] All MCP tools executed, emitting RUN_FINISHED`);\n observer.next({\n type: EventType.RUN_FINISHED,\n threadId: (input as any).threadId,\n runId: (input as any).runId,\n timestamp: Date.now(),\n } as any);\n\n // Trigger a new run to continue the conversation\n console.log(`[McpMiddleware] Triggering new run`);\n this.triggerNewRun(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls);\n };\n\n const subscription = next.run(input).subscribe({\n next: (event: BaseEvent) => {\n // Track tool call names from TOOL_CALL_START events\n if (event.type === EventType.TOOL_CALL_START) {\n const startEvent = event as any;\n if (startEvent.toolCallId && startEvent.toolCallName) {\n toolCallNames.set(startEvent.toolCallId, startEvent.toolCallName);\n const isMcp = this.isMcpTool(startEvent.toolCallName);\n console.log(`[McpMiddleware] TOOL_CALL_START: ${startEvent.toolCallName} (id: ${startEvent.toolCallId}, isMCP: ${isMcp})`);\n\n if (isMcp) {\n pendingMcpCalls.add(startEvent.toolCallId);\n }\n }\n }\n\n // Accumulate tool call arguments from TOOL_CALL_ARGS events\n if (event.type === EventType.TOOL_CALL_ARGS) {\n const argsEvent = event as any;\n if (argsEvent.toolCallId && argsEvent.delta) {\n const existing = toolCallArgsBuffer.get(argsEvent.toolCallId) || '';\n toolCallArgsBuffer.set(argsEvent.toolCallId, existing + argsEvent.delta);\n }\n }\n\n // Track TOOL_CALL_END\n if (event.type === EventType.TOOL_CALL_END) {\n const endEvent = event as ToolCallEndEvent;\n const toolName = toolCallNames.get(endEvent.toolCallId);\n console.log(`[McpMiddleware] TOOL_CALL_END: ${toolName ?? 'unknown'} (id: ${endEvent.toolCallId})`);\n }\n\n // Handle RUN_FINISHED - execute pending MCP tools\n if (event.type === EventType.RUN_FINISHED) {\n handleRunFinished(event);\n return;\n }\n\n // Pass through all other events\n observer.next(event);\n },\n error: (error) => {\n observer.error(error);\n },\n complete: () => {\n if (pendingMcpCalls.size === 0) {\n observer.complete();\n }\n },\n });\n\n return () => {\n subscription.unsubscribe();\n };\n });\n }\n\n private triggerNewRun(\n observer: Subscriber<BaseEvent>,\n input: RunAgentInput,\n next: AbstractAgent,\n toolCallArgsBuffer: Map<string, string>,\n toolCallNames: Map<string, string>,\n pendingMcpCalls: Set<string>,\n ): void {\n toolCallArgsBuffer.clear();\n toolCallNames.clear();\n pendingMcpCalls.clear();\n\n console.log(`[McpMiddleware] Starting new run with updated messages`);\n\n const subscription = next.run(input).subscribe({\n next: (event: BaseEvent) => {\n if (event.type === EventType.TOOL_CALL_START) {\n const startEvent = event as any;\n if (startEvent.toolCallId && startEvent.toolCallName) {\n toolCallNames.set(startEvent.toolCallId, startEvent.toolCallName);\n const isMcp = this.isMcpTool(startEvent.toolCallName);\n console.log(`[McpMiddleware] TOOL_CALL_START: ${startEvent.toolCallName} (id: ${startEvent.toolCallId}, isMCP: ${isMcp})`);\n\n if (isMcp) {\n pendingMcpCalls.add(startEvent.toolCallId);\n }\n }\n }\n\n if (event.type === EventType.TOOL_CALL_ARGS) {\n const argsEvent = event as any;\n if (argsEvent.toolCallId && argsEvent.delta) {\n const existing = toolCallArgsBuffer.get(argsEvent.toolCallId) || '';\n toolCallArgsBuffer.set(argsEvent.toolCallId, existing + argsEvent.delta);\n }\n }\n\n if (event.type === EventType.TOOL_CALL_END) {\n const endEvent = event as ToolCallEndEvent;\n const toolName = toolCallNames.get(endEvent.toolCallId);\n console.log(`[McpMiddleware] TOOL_CALL_END: ${toolName ?? 'unknown'} (id: ${endEvent.toolCallId})`);\n }\n\n if (event.type === EventType.RUN_FINISHED) {\n if (pendingMcpCalls.size > 0) {\n console.log(`[McpMiddleware] RUN_FINISHED with ${pendingMcpCalls.size} pending calls, executing...`);\n this.handlePendingCalls(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls);\n } else {\n observer.next(event);\n observer.complete();\n }\n return;\n }\n\n observer.next(event);\n },\n error: (error) => observer.error(error),\n complete: () => {\n if (pendingMcpCalls.size === 0) {\n observer.complete();\n }\n },\n });\n }\n\n private async handlePendingCalls(\n observer: Subscriber<BaseEvent>,\n input: RunAgentInput,\n next: AbstractAgent,\n toolCallArgsBuffer: Map<string, string>,\n toolCallNames: Map<string, string>,\n pendingMcpCalls: Set<string>,\n ): Promise<void> {\n const callPromises = [...pendingMcpCalls].map(async (toolCallId) => {\n const toolName = toolCallNames.get(toolCallId);\n if (!toolName) return;\n\n const argsString = toolCallArgsBuffer.get(toolCallId) || '{}';\n let args: Record<string, any> = {};\n try {\n args = JSON.parse(argsString);\n } catch (e) {\n console.error(`[McpMiddleware] Failed to parse args:`, argsString);\n }\n\n console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\n const result = await this.executeTool(toolName, args);\n const messageId = this.generateMessageId();\n\n const resultEvent: BaseEvent = {\n type: EventType.TOOL_CALL_RESULT,\n toolCallId,\n messageId,\n content: result,\n role: 'tool',\n timestamp: Date.now(),\n } as any;\n\n console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\n observer.next(resultEvent);\n\n input.messages.push({\n id: messageId,\n role: 'tool',\n toolCallId,\n content: result,\n } as any);\n\n pendingMcpCalls.delete(toolCallId);\n });\n\n await Promise.all(callPromises);\n\n console.log(`[McpMiddleware] Pending tools executed, emitting RUN_FINISHED`);\n observer.next({\n type: EventType.RUN_FINISHED,\n threadId: (input as any).threadId,\n runId: (input as any).runId,\n timestamp: Date.now(),\n } as any);\n\n console.log(`[McpMiddleware] Triggering new run`);\n this.triggerNewRun(observer, input, next, toolCallArgsBuffer, toolCallNames, pendingMcpCalls);\n }\n}\n\n/**\n * Factory function to create MCP middleware.\n *\n * This is a convenience wrapper around McpMiddleware that returns a function\n * compatible with the AG-UI middleware pattern.\n *\n * @param client - MCP client or MultiSessionClient\n * @param options - Configuration options\n * @returns Middleware function\n *\n * @example\n * ```typescript\n * import { HttpAgent } from '@ag-ui/client';\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\n *\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\n * agent.use(createMcpMiddleware(multiSessionClient, {\n * toolPrefix: 'server-',\n * actions: mcpActions,\n * }));\n * ```\n */\nexport function createMcpMiddleware(\n client: MCPClient | MultiSessionClient,\n options: { toolPrefix?: string; tools?: AguiTool[] } = {}\n) {\n const middleware = new McpMiddleware({\n client,\n ...options,\n });\n\n return (input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> => {\n return middleware.run(input, next);\n };\n}\n\n// Legacy exports for backward compatibility\nexport { McpMiddleware as McpToolExecutorMiddleware };\nexport { createMcpMiddleware as createMcpToolMiddleware };\n\n// Re-export types for convenience\nexport { Middleware, EventType };\nexport type { RunAgentInput, BaseEvent, AbstractAgent, ToolCallEndEvent };\n"]}
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-DMF3ED2O.mjs';
|
|
2
|
+
import { ToolSet } from 'ai';
|
|
3
|
+
import '../events-BP6WyRNh.mjs';
|
|
4
|
+
import '@modelcontextprotocol/sdk/types.js';
|
|
5
|
+
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
|
+
import '@modelcontextprotocol/sdk/client/auth.js';
|
|
7
|
+
|
|
8
|
+
interface AIAdapterOptions {
|
|
9
|
+
/**
|
|
10
|
+
* Prefix for tool names to avoid collision with other tools.
|
|
11
|
+
* Defaults to the client's serverId.
|
|
12
|
+
*/
|
|
13
|
+
prefix?: string;
|
|
14
|
+
}
|
|
15
|
+
/**
|
|
16
|
+
* Adapter to use MCP tools with the Vercel AI SDK.
|
|
17
|
+
*/
|
|
18
|
+
declare class AIAdapter {
|
|
19
|
+
private client;
|
|
20
|
+
private options;
|
|
21
|
+
private jsonSchema;
|
|
22
|
+
constructor(client: MCPClient | MultiSessionClient, options?: AIAdapterOptions);
|
|
23
|
+
/**
|
|
24
|
+
* Lazy-loads the jsonSchema function from the AI SDK.
|
|
25
|
+
*/
|
|
26
|
+
private ensureJsonSchema;
|
|
27
|
+
private transformTools;
|
|
28
|
+
/**
|
|
29
|
+
* Fetches tools from the client(s) and converts them to AI SDK tools.
|
|
30
|
+
*/
|
|
31
|
+
getTools(): Promise<ToolSet>;
|
|
32
|
+
/**
|
|
33
|
+
* Convenience static method to fetch tools in a single line.
|
|
34
|
+
*/
|
|
35
|
+
static getTools(client: MCPClient | MultiSessionClient, options?: AIAdapterOptions): Promise<ToolSet>;
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
export { AIAdapter, type AIAdapterOptions };
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-BOFgPypS.js';
|
|
2
|
+
import { ToolSet } from 'ai';
|
|
3
|
+
import '../events-BP6WyRNh.js';
|
|
4
|
+
import '@modelcontextprotocol/sdk/types.js';
|
|
5
|
+
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
|
+
import '@modelcontextprotocol/sdk/client/auth.js';
|
|
7
|
+
|
|
8
|
+
interface AIAdapterOptions {
|
|
9
|
+
/**
|
|
10
|
+
* Prefix for tool names to avoid collision with other tools.
|
|
11
|
+
* Defaults to the client's serverId.
|
|
12
|
+
*/
|
|
13
|
+
prefix?: string;
|
|
14
|
+
}
|
|
15
|
+
/**
|
|
16
|
+
* Adapter to use MCP tools with the Vercel AI SDK.
|
|
17
|
+
*/
|
|
18
|
+
declare class AIAdapter {
|
|
19
|
+
private client;
|
|
20
|
+
private options;
|
|
21
|
+
private jsonSchema;
|
|
22
|
+
constructor(client: MCPClient | MultiSessionClient, options?: AIAdapterOptions);
|
|
23
|
+
/**
|
|
24
|
+
* Lazy-loads the jsonSchema function from the AI SDK.
|
|
25
|
+
*/
|
|
26
|
+
private ensureJsonSchema;
|
|
27
|
+
private transformTools;
|
|
28
|
+
/**
|
|
29
|
+
* Fetches tools from the client(s) and converts them to AI SDK tools.
|
|
30
|
+
*/
|
|
31
|
+
getTools(): Promise<ToolSet>;
|
|
32
|
+
/**
|
|
33
|
+
* Convenience static method to fetch tools in a single line.
|
|
34
|
+
*/
|
|
35
|
+
static getTools(client: MCPClient | MultiSessionClient, options?: AIAdapterOptions): Promise<ToolSet>;
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
export { AIAdapter, type AIAdapterOptions };
|
|
@@ -0,0 +1,82 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
|
|
3
|
+
var __defProp = Object.defineProperty;
|
|
4
|
+
var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
|
|
5
|
+
var __publicField = (obj, key, value) => __defNormalProp(obj, key + "" , value);
|
|
6
|
+
|
|
7
|
+
// src/adapters/ai-adapter.ts
|
|
8
|
+
var AIAdapter = class _AIAdapter {
|
|
9
|
+
constructor(client, options = {}) {
|
|
10
|
+
this.client = client;
|
|
11
|
+
this.options = options;
|
|
12
|
+
__publicField(this, "jsonSchema");
|
|
13
|
+
}
|
|
14
|
+
/**
|
|
15
|
+
* Lazy-loads the jsonSchema function from the AI SDK.
|
|
16
|
+
*/
|
|
17
|
+
async ensureJsonSchema() {
|
|
18
|
+
if (!this.jsonSchema) {
|
|
19
|
+
const { jsonSchema } = await import('ai');
|
|
20
|
+
this.jsonSchema = jsonSchema;
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
async transformTools(client) {
|
|
24
|
+
const isConnected = typeof client.isConnected === "function" ? client.isConnected() : false;
|
|
25
|
+
if (!isConnected) {
|
|
26
|
+
return {};
|
|
27
|
+
}
|
|
28
|
+
const result = await client.listTools();
|
|
29
|
+
return Object.fromEntries(
|
|
30
|
+
result.tools.map((tool) => {
|
|
31
|
+
const serverId = typeof client.getServerId === "function" ? client.getServerId() : void 0;
|
|
32
|
+
const prefix = this.options.prefix ?? serverId?.replace(/-/g, "") ?? "mcp";
|
|
33
|
+
return [
|
|
34
|
+
`tool_${prefix}_${tool.name}`,
|
|
35
|
+
{
|
|
36
|
+
description: tool.description,
|
|
37
|
+
inputSchema: this.jsonSchema(tool.inputSchema),
|
|
38
|
+
execute: async (args) => {
|
|
39
|
+
try {
|
|
40
|
+
const response = await client.callTool(tool.name, args);
|
|
41
|
+
return response;
|
|
42
|
+
} catch (error) {
|
|
43
|
+
const errorMessage = error instanceof Error ? error.message : String(error);
|
|
44
|
+
throw new Error(`Tool execution failed: ${errorMessage}`);
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
];
|
|
49
|
+
})
|
|
50
|
+
);
|
|
51
|
+
}
|
|
52
|
+
/**
|
|
53
|
+
* Fetches tools from the client(s) and converts them to AI SDK tools.
|
|
54
|
+
*/
|
|
55
|
+
async getTools() {
|
|
56
|
+
await this.ensureJsonSchema();
|
|
57
|
+
const isMultiSession = typeof this.client.getClients === "function";
|
|
58
|
+
const clients = isMultiSession ? this.client.getClients() : [this.client];
|
|
59
|
+
const results = await Promise.all(
|
|
60
|
+
clients.map(async (client) => {
|
|
61
|
+
try {
|
|
62
|
+
return await this.transformTools(client);
|
|
63
|
+
} catch (error) {
|
|
64
|
+
const serverId = typeof client.getServerId === "function" ? client.getServerId() ?? "unknown" : "unknown";
|
|
65
|
+
console.error(`[AIAdapter] Failed to fetch tools from ${serverId}:`, error);
|
|
66
|
+
return {};
|
|
67
|
+
}
|
|
68
|
+
})
|
|
69
|
+
);
|
|
70
|
+
return results.reduce((acc, tools) => ({ ...acc, ...tools }), {});
|
|
71
|
+
}
|
|
72
|
+
/**
|
|
73
|
+
* Convenience static method to fetch tools in a single line.
|
|
74
|
+
*/
|
|
75
|
+
static async getTools(client, options = {}) {
|
|
76
|
+
return new _AIAdapter(client, options).getTools();
|
|
77
|
+
}
|
|
78
|
+
};
|
|
79
|
+
|
|
80
|
+
exports.AIAdapter = AIAdapter;
|
|
81
|
+
//# sourceMappingURL=ai-adapter.js.map
|
|
82
|
+
//# sourceMappingURL=ai-adapter.js.map
|