@mcp-ts/sdk 1.0.1 → 1.2.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/README.md +42 -19
- package/dist/adapters/agui-adapter.d.mts +6 -6
- package/dist/adapters/agui-adapter.d.ts +6 -6
- package/dist/adapters/agui-adapter.js +50 -25
- package/dist/adapters/agui-adapter.js.map +1 -1
- package/dist/adapters/agui-adapter.mjs +50 -25
- package/dist/adapters/agui-adapter.mjs.map +1 -1
- package/dist/adapters/agui-middleware.d.mts +23 -8
- package/dist/adapters/agui-middleware.d.ts +23 -8
- package/dist/adapters/agui-middleware.js +71 -43
- package/dist/adapters/agui-middleware.js.map +1 -1
- package/dist/adapters/agui-middleware.mjs +71 -44
- package/dist/adapters/agui-middleware.mjs.map +1 -1
- package/dist/adapters/ai-adapter.d.mts +2 -2
- package/dist/adapters/ai-adapter.d.ts +2 -2
- package/dist/adapters/langchain-adapter.d.mts +2 -2
- package/dist/adapters/langchain-adapter.d.ts +2 -2
- package/dist/adapters/mastra-adapter.d.mts +2 -2
- package/dist/adapters/mastra-adapter.d.ts +2 -2
- package/dist/client/index.d.mts +182 -55
- package/dist/client/index.d.ts +182 -55
- package/dist/client/index.js +535 -130
- package/dist/client/index.js.map +1 -1
- package/dist/client/index.mjs +535 -131
- package/dist/client/index.mjs.map +1 -1
- package/dist/client/react.d.mts +386 -4
- package/dist/client/react.d.ts +386 -4
- package/dist/client/react.js +890 -143
- package/dist/client/react.js.map +1 -1
- package/dist/client/react.mjs +883 -145
- package/dist/client/react.mjs.map +1 -1
- package/dist/client/vue.d.mts +3 -3
- package/dist/client/vue.d.ts +3 -3
- package/dist/client/vue.js +546 -141
- package/dist/client/vue.js.map +1 -1
- package/dist/client/vue.mjs +546 -142
- package/dist/client/vue.mjs.map +1 -1
- package/dist/{events-BP6WyRNh.d.mts → events-BgeztGYZ.d.mts} +12 -1
- package/dist/{events-BP6WyRNh.d.ts → events-BgeztGYZ.d.ts} +12 -1
- package/dist/index.d.mts +4 -4
- package/dist/index.d.ts +4 -4
- package/dist/index.js +797 -248
- package/dist/index.js.map +1 -1
- package/dist/index.mjs +791 -245
- package/dist/index.mjs.map +1 -1
- package/dist/{multi-session-client-DMF3ED2O.d.mts → multi-session-client-CxogNckF.d.mts} +1 -1
- package/dist/{multi-session-client-BOFgPypS.d.ts → multi-session-client-cox_WXUj.d.ts} +1 -1
- package/dist/server/index.d.mts +41 -37
- package/dist/server/index.d.ts +41 -37
- package/dist/server/index.js +241 -116
- package/dist/server/index.js.map +1 -1
- package/dist/server/index.mjs +237 -112
- package/dist/server/index.mjs.map +1 -1
- package/dist/shared/index.d.mts +39 -3
- package/dist/shared/index.d.ts +39 -3
- package/dist/shared/index.js +19 -0
- package/dist/shared/index.js.map +1 -1
- package/dist/shared/index.mjs +18 -1
- package/dist/shared/index.mjs.map +1 -1
- package/package.json +9 -2
- package/src/adapters/agui-adapter.ts +58 -35
- package/src/adapters/agui-middleware.ts +83 -45
- package/src/client/core/app-host.ts +417 -0
- package/src/client/core/sse-client.ts +365 -212
- package/src/client/core/types.ts +31 -0
- package/src/client/index.ts +1 -0
- package/src/client/react/agui-subscriber.ts +275 -0
- package/src/client/react/index.ts +23 -3
- package/src/client/react/use-agui-subscriber.ts +270 -0
- package/src/client/react/use-app-host.ts +73 -0
- package/src/client/react/use-mcp-app-iframe.ts +164 -0
- package/src/client/react/{useMcp.ts → use-mcp.ts} +18 -0
- package/src/client/vue/index.ts +1 -1
- package/src/server/handlers/nextjs-handler.ts +8 -7
- package/src/server/handlers/sse-handler.ts +129 -165
- package/src/server/mcp/oauth-client.ts +32 -2
- package/src/server/storage/index.ts +17 -1
- package/src/server/storage/sqlite-backend.ts +185 -0
- package/src/shared/events.ts +12 -0
- package/src/shared/index.ts +6 -1
- package/src/shared/tool-utils.ts +61 -0
- package/src/shared/types.ts +3 -1
- /package/src/client/vue/{useMcp.ts → use-mcp.ts} +0 -0
|
@@ -7,16 +7,33 @@ var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { en
|
|
|
7
7
|
var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
|
|
8
8
|
|
|
9
9
|
// src/adapters/agui-adapter.ts
|
|
10
|
+
var PYDANTIC_FORBIDDEN_PROPS = [
|
|
11
|
+
// JSON Schema meta-properties
|
|
12
|
+
"$schema",
|
|
13
|
+
"$id",
|
|
14
|
+
"$comment",
|
|
15
|
+
"$defs",
|
|
16
|
+
"definitions",
|
|
17
|
+
// Extended properties used by some MCP servers (e.g., Apify)
|
|
18
|
+
"prefill",
|
|
19
|
+
"examples",
|
|
20
|
+
"enumTitles",
|
|
21
|
+
"enumDescriptions",
|
|
22
|
+
// Other common extensions
|
|
23
|
+
"deprecated",
|
|
24
|
+
"readOnly",
|
|
25
|
+
"writeOnly",
|
|
26
|
+
"contentMediaType",
|
|
27
|
+
"contentEncoding"
|
|
28
|
+
];
|
|
10
29
|
function cleanSchema(schema) {
|
|
11
30
|
if (!schema) {
|
|
12
31
|
return { type: "object", properties: {} };
|
|
13
32
|
}
|
|
14
33
|
const cleaned = { ...schema };
|
|
15
|
-
|
|
16
|
-
|
|
17
|
-
|
|
18
|
-
delete cleaned.$defs;
|
|
19
|
-
delete cleaned.definitions;
|
|
34
|
+
for (const prop of PYDANTIC_FORBIDDEN_PROPS) {
|
|
35
|
+
delete cleaned[prop];
|
|
36
|
+
}
|
|
20
37
|
if (cleaned.properties && typeof cleaned.properties === "object") {
|
|
21
38
|
const cleanedProps = {};
|
|
22
39
|
for (const [key, value] of Object.entries(cleaned.properties)) {
|
|
@@ -38,20 +55,29 @@ function cleanSchema(schema) {
|
|
|
38
55
|
}
|
|
39
56
|
|
|
40
57
|
// src/adapters/agui-middleware.ts
|
|
58
|
+
var MCP_APP_UI_EVENT = "mcp-apps-ui";
|
|
41
59
|
var McpMiddleware = class extends Middleware {
|
|
42
60
|
constructor(config) {
|
|
43
61
|
super();
|
|
44
62
|
__publicField(this, "tools");
|
|
45
63
|
__publicField(this, "toolSchemas");
|
|
46
|
-
__publicField(this, "maxResultLength");
|
|
47
64
|
this.tools = config.tools;
|
|
48
|
-
this.maxResultLength = config.maxResultLength ?? 5e4;
|
|
49
65
|
this.toolSchemas = this.tools.map((t) => ({
|
|
50
66
|
name: t.name,
|
|
51
67
|
description: t.description,
|
|
52
|
-
parameters: cleanSchema(t.parameters)
|
|
68
|
+
parameters: cleanSchema(t.parameters),
|
|
69
|
+
_meta: t._meta
|
|
70
|
+
// Include _meta in the tool definition passed to the agent
|
|
53
71
|
}));
|
|
54
72
|
}
|
|
73
|
+
/**
|
|
74
|
+
* Extract base tool name from prefixed format for event emission
|
|
75
|
+
* e.g., "tool_abc123_get-time" -> "get-time"
|
|
76
|
+
*/
|
|
77
|
+
getBaseToolName(toolName) {
|
|
78
|
+
const match = toolName.match(/^tool_[^_]+_(.+)$/);
|
|
79
|
+
return match ? match[1] : toolName;
|
|
80
|
+
}
|
|
55
81
|
isMcpTool(toolName) {
|
|
56
82
|
return this.tools.some((t) => t.name === toolName);
|
|
57
83
|
}
|
|
@@ -76,24 +102,22 @@ var McpMiddleware = class extends Middleware {
|
|
|
76
102
|
async executeTool(toolName, args) {
|
|
77
103
|
const tool = this.tools.find((t) => t.name === toolName);
|
|
78
104
|
if (!tool?.handler) {
|
|
79
|
-
return `Error: Tool ${tool ? "has no handler" : "not found"}: ${toolName}
|
|
105
|
+
return { resultStr: `Error: Tool ${tool ? "has no handler" : "not found"}: ${toolName}` };
|
|
80
106
|
}
|
|
81
107
|
try {
|
|
82
|
-
console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);
|
|
83
108
|
const result = await tool.handler(args);
|
|
84
|
-
let resultStr
|
|
85
|
-
if (
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
109
|
+
let resultStr;
|
|
110
|
+
if (typeof result === "string") {
|
|
111
|
+
resultStr = result;
|
|
112
|
+
} else if (result && typeof result === "object") {
|
|
113
|
+
resultStr = JSON.stringify(result);
|
|
114
|
+
} else {
|
|
115
|
+
resultStr = String(result);
|
|
91
116
|
}
|
|
92
|
-
|
|
93
|
-
return resultStr;
|
|
117
|
+
return { resultStr, rawResult: result };
|
|
94
118
|
} catch (error) {
|
|
95
119
|
console.error(`[McpMiddleware] Error executing tool:`, error);
|
|
96
|
-
return `Error: ${error.message || String(error)}
|
|
120
|
+
return { resultStr: `Error: ${error.message || String(error)}` };
|
|
97
121
|
}
|
|
98
122
|
}
|
|
99
123
|
generateId(prefix) {
|
|
@@ -120,7 +144,6 @@ var McpMiddleware = class extends Middleware {
|
|
|
120
144
|
if (this.isMcpTool(e.toolCallName)) {
|
|
121
145
|
pendingMcpCalls.add(e.toolCallId);
|
|
122
146
|
}
|
|
123
|
-
console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);
|
|
124
147
|
}
|
|
125
148
|
}
|
|
126
149
|
if (event.type === EventType.TOOL_CALL_ARGS) {
|
|
@@ -130,10 +153,7 @@ var McpMiddleware = class extends Middleware {
|
|
|
130
153
|
toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);
|
|
131
154
|
}
|
|
132
155
|
}
|
|
133
|
-
if (event.type === EventType.TOOL_CALL_END)
|
|
134
|
-
const e = event;
|
|
135
|
-
console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? "unknown"} (id: ${e.toolCallId})`);
|
|
136
|
-
}
|
|
156
|
+
if (event.type === EventType.TOOL_CALL_END) ;
|
|
137
157
|
if (event.type === EventType.MESSAGES_SNAPSHOT) {
|
|
138
158
|
const messages = event.messages || [];
|
|
139
159
|
if (messages.length > 0) {
|
|
@@ -151,7 +171,6 @@ var McpMiddleware = class extends Middleware {
|
|
|
151
171
|
toolCallArgsBuffer.set(tc.id, tc.function.arguments || "{}");
|
|
152
172
|
if (this.isMcpTool(tc.function.name)) {
|
|
153
173
|
pendingMcpCalls.add(tc.id);
|
|
154
|
-
console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);
|
|
155
174
|
}
|
|
156
175
|
}
|
|
157
176
|
}
|
|
@@ -169,12 +188,12 @@ var McpMiddleware = class extends Middleware {
|
|
|
169
188
|
const toolName = toolCallNames.get(toolCallId);
|
|
170
189
|
if (!toolName) return;
|
|
171
190
|
const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || "{}");
|
|
172
|
-
|
|
173
|
-
const result = await this.executeTool(toolName, args);
|
|
191
|
+
const { resultStr, rawResult } = await this.executeTool(toolName, args);
|
|
174
192
|
results.push({
|
|
175
193
|
toolCallId,
|
|
176
194
|
toolName,
|
|
177
|
-
result,
|
|
195
|
+
result: resultStr,
|
|
196
|
+
rawResult,
|
|
178
197
|
messageId: this.generateId("mcp_result")
|
|
179
198
|
});
|
|
180
199
|
pendingMcpCalls.delete(toolCallId);
|
|
@@ -182,9 +201,29 @@ var McpMiddleware = class extends Middleware {
|
|
|
182
201
|
await Promise.all(promises);
|
|
183
202
|
return results;
|
|
184
203
|
}
|
|
185
|
-
/** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */
|
|
186
204
|
emitToolResults(observer, results) {
|
|
187
|
-
for (const { toolCallId, toolName, result, messageId } of results) {
|
|
205
|
+
for (const { toolCallId, toolName, result, rawResult, messageId } of results) {
|
|
206
|
+
const toolDef = this.tools.find((t) => t.name === toolName);
|
|
207
|
+
const sessionId = toolDef?._meta?.sessionId;
|
|
208
|
+
const resourceUri = rawResult?._meta?.ui?.resourceUri ?? rawResult?._meta?.["ui/resourceUri"] ?? toolDef?._meta?.ui?.resourceUri ?? toolDef?._meta?.["ui/resourceUri"];
|
|
209
|
+
if (resourceUri) {
|
|
210
|
+
const baseToolName = this.getBaseToolName(toolName);
|
|
211
|
+
const payload = {
|
|
212
|
+
toolCallId,
|
|
213
|
+
resourceUri,
|
|
214
|
+
sessionId,
|
|
215
|
+
toolName: baseToolName,
|
|
216
|
+
// Use base name to match metadata
|
|
217
|
+
result: rawResult ?? result
|
|
218
|
+
};
|
|
219
|
+
observer.next({
|
|
220
|
+
type: EventType.CUSTOM,
|
|
221
|
+
name: MCP_APP_UI_EVENT,
|
|
222
|
+
value: payload,
|
|
223
|
+
timestamp: Date.now(),
|
|
224
|
+
role: "tool"
|
|
225
|
+
});
|
|
226
|
+
}
|
|
188
227
|
observer.next({
|
|
189
228
|
type: EventType.TOOL_CALL_RESULT,
|
|
190
229
|
toolCallId,
|
|
@@ -193,7 +232,6 @@ var McpMiddleware = class extends Middleware {
|
|
|
193
232
|
role: "tool",
|
|
194
233
|
timestamp: Date.now()
|
|
195
234
|
});
|
|
196
|
-
console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);
|
|
197
235
|
}
|
|
198
236
|
}
|
|
199
237
|
run(input, next) {
|
|
@@ -207,12 +245,8 @@ var McpMiddleware = class extends Middleware {
|
|
|
207
245
|
};
|
|
208
246
|
this.ensureIds(input);
|
|
209
247
|
const anyInput = input;
|
|
210
|
-
console.log(`[McpMiddleware] === NEW RUN ===`);
|
|
211
|
-
console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);
|
|
212
|
-
console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);
|
|
213
248
|
if (this.toolSchemas?.length) {
|
|
214
249
|
input.tools = [...input.tools || [], ...this.toolSchemas];
|
|
215
|
-
console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t) => t.name));
|
|
216
250
|
}
|
|
217
251
|
const handleRunFinished = async () => {
|
|
218
252
|
if (state.error) return;
|
|
@@ -226,7 +260,6 @@ var McpMiddleware = class extends Middleware {
|
|
|
226
260
|
observer.complete();
|
|
227
261
|
return;
|
|
228
262
|
}
|
|
229
|
-
console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);
|
|
230
263
|
const toolCalls = [];
|
|
231
264
|
for (const toolCallId of state.pendingMcpCalls) {
|
|
232
265
|
const name = state.toolCallNames.get(toolCallId);
|
|
@@ -248,11 +281,9 @@ var McpMiddleware = class extends Middleware {
|
|
|
248
281
|
tool_calls: toolCalls.length > 0 ? toolCalls : void 0
|
|
249
282
|
};
|
|
250
283
|
input.messages.push(assistantMsg);
|
|
251
|
-
console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);
|
|
252
284
|
}
|
|
253
285
|
const results = await this.executeTools(state);
|
|
254
286
|
this.emitToolResults(observer, results);
|
|
255
|
-
console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);
|
|
256
287
|
for (const { toolCallId, result, messageId } of results) {
|
|
257
288
|
input.messages.push({
|
|
258
289
|
id: messageId,
|
|
@@ -265,20 +296,17 @@ var McpMiddleware = class extends Middleware {
|
|
|
265
296
|
state.toolCallNames.clear();
|
|
266
297
|
state.textContent = "";
|
|
267
298
|
anyInput.runId = this.generateId("mcp_run");
|
|
268
|
-
console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);
|
|
269
299
|
next.run(input).subscribe({
|
|
270
300
|
next: (event) => {
|
|
271
301
|
if (state.error) return;
|
|
272
302
|
this.handleToolCallEvent(event, state);
|
|
273
303
|
if (event.type === EventType.RUN_ERROR) {
|
|
274
|
-
console.log(`[McpMiddleware] RUN_ERROR received in continuation`);
|
|
275
304
|
state.error = true;
|
|
276
305
|
observer.next(event);
|
|
277
306
|
observer.complete();
|
|
278
307
|
return;
|
|
279
308
|
}
|
|
280
309
|
if (event.type === EventType.RUN_STARTED) {
|
|
281
|
-
console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);
|
|
282
310
|
return;
|
|
283
311
|
}
|
|
284
312
|
if (event.type === EventType.RUN_FINISHED) {
|
|
@@ -306,7 +334,6 @@ var McpMiddleware = class extends Middleware {
|
|
|
306
334
|
if (state.error) return;
|
|
307
335
|
this.handleToolCallEvent(event, state);
|
|
308
336
|
if (event.type === EventType.RUN_ERROR) {
|
|
309
|
-
console.log(`[McpMiddleware] RUN_ERROR received`);
|
|
310
337
|
state.error = true;
|
|
311
338
|
observer.next(event);
|
|
312
339
|
observer.complete();
|
|
@@ -337,6 +364,6 @@ function createMcpMiddleware(options) {
|
|
|
337
364
|
};
|
|
338
365
|
}
|
|
339
366
|
|
|
340
|
-
export { McpMiddleware, McpMiddleware as McpToolExecutorMiddleware, createMcpMiddleware, createMcpMiddleware as createMcpToolMiddleware };
|
|
367
|
+
export { MCP_APP_UI_EVENT, McpMiddleware, McpMiddleware as McpToolExecutorMiddleware, createMcpMiddleware, createMcpMiddleware as createMcpToolMiddleware };
|
|
341
368
|
//# sourceMappingURL=agui-middleware.mjs.map
|
|
342
369
|
//# sourceMappingURL=agui-middleware.mjs.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":[],"mappings":";;;;;;;;;AAwCO,SAAS,YAAY,MAAA,EAA8D;AACtF,EAAA,IAAI,CAAC,MAAA,EAAQ;AACT,IAAA,OAAO,EAAE,IAAA,EAAM,QAAA,EAAU,UAAA,EAAY,EAAC,EAAE;AAAA,EAC5C;AAEA,EAAA,MAAM,OAAA,GAAU,EAAE,GAAG,MAAA,EAAO;AAG5B,EAAA,OAAO,OAAA,CAAQ,OAAA;AACf,EAAA,OAAO,OAAA,CAAQ,GAAA;AACf,EAAA,OAAO,OAAA,CAAQ,QAAA;AACf,EAAA,OAAO,OAAA,CAAQ,KAAA;AACf,EAAA,OAAO,OAAA,CAAQ,WAAA;AAGf,EAAA,IAAI,OAAA,CAAQ,UAAA,IAAc,OAAO,OAAA,CAAQ,eAAe,QAAA,EAAU;AAC9D,IAAA,MAAM,eAAoC,EAAC;AAC3C,IAAA,KAAA,MAAW,CAAC,KAAK,KAAK,CAAA,IAAK,OAAO,OAAA,CAAQ,OAAA,CAAQ,UAAU,CAAA,EAAG;AAC3D,MAAA,IAAI,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,EAAM;AAC7C,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,WAAA,CAAY,KAA4B,CAAA;AAAA,MAChE,CAAA,MAAO;AACH,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,KAAA;AAAA,MACxB;AAAA,IACJ;AACA,IAAA,OAAA,CAAQ,UAAA,GAAa,YAAA;AAAA,EACzB;AAGA,EAAA,IAAI,OAAA,CAAQ,KAAA,IAAS,OAAO,OAAA,CAAQ,UAAU,QAAA,EAAU;AACpD,IAAA,OAAA,CAAQ,KAAA,GAAQ,WAAA,CAAY,OAAA,CAAQ,KAAK,CAAA;AAAA,EAC7C;AAGA,EAAA,IAAI,OAAA,CAAQ,oBAAA,IAAwB,OAAO,OAAA,CAAQ,yBAAyB,QAAA,EAAU;AAClF,IAAA,OAAA,CAAQ,oBAAA,GAAuB,WAAA,CAAY,OAAA,CAAQ,oBAAoB,CAAA;AAAA,EAC3E;AAEA,EAAA,OAAO,OAAA;AACX;;;AC1BO,IAAM,aAAA,GAAN,cAA4B,UAAA,CAAW;AAAA,EAK1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AALV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,iBAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,eAAA,GAAkB,OAAO,eAAA,IAAmB,GAAA;AACjD,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU;AAAA,KACxC,CAAE,CAAA;AAAA,EACN;AAAA,EAEQ,UAAU,QAAA,EAA2B;AACzC,IAAA,OAAO,KAAK,KAAA,CAAM,IAAA,CAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AAAA,EACnD;AAAA,EAEQ,UAAU,UAAA,EAAyC;AACvD,IAAA,IAAI,CAAC,UAAA,EAAY,IAAA,EAAK,SAAU,EAAC;AAEjC,IAAA,IAAI;AACA,MAAA,OAAO,IAAA,CAAK,MAAM,UAAU,CAAA;AAAA,IAChC,CAAA,CAAA,MAAQ;AAEJ,MAAA,MAAM,OAAA,GAAU,WAAW,IAAA,EAAK;AAChC,MAAA,IAAI,OAAA,CAAQ,QAAA,CAAS,IAAI,CAAA,EAAG;AACxB,QAAA,MAAM,WAAA,GAAc,QAAQ,KAAA,CAAM,CAAA,EAAG,QAAQ,OAAA,CAAQ,IAAI,IAAI,CAAC,CAAA;AAC9D,QAAA,IAAI;AACA,UAAA,OAAO,IAAA,CAAK,MAAM,WAAW,CAAA;AAAA,QACjC,CAAA,CAAA,MAAQ;AACJ,UAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,WAAW,CAAA;AAAA,QACtE;AAAA,MACJ;AACA,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,UAAU,CAAA;AACjE,MAAA,OAAO,EAAC;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,MAAc,WAAA,CAAY,QAAA,EAAkB,IAAA,EAA4C;AACpF,IAAA,MAAM,OAAO,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA,CAAA,KAAK,CAAA,CAAE,SAAS,QAAQ,CAAA;AACrD,IAAA,IAAI,CAAC,MAAM,OAAA,EAAS;AAChB,MAAA,OAAO,CAAA,YAAA,EAAe,IAAA,GAAO,gBAAA,GAAmB,WAAW,KAAK,QAAQ,CAAA,CAAA;AAAA,IAC5E;AAEA,IAAA,IAAI;AACA,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,gCAAA,EAAmC,QAAQ,CAAA,CAAA,EAAI,IAAI,CAAA;AAC/D,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA;AACtC,MAAA,IAAI,YAAY,OAAO,MAAA,KAAW,WAAW,MAAA,GAAS,IAAA,CAAK,UAAU,MAAM,CAAA;AAE3E,MAAA,IAAI,SAAA,CAAU,MAAA,GAAS,IAAA,CAAK,eAAA,EAAiB;AACzC,QAAA,MAAM,WAAW,SAAA,CAAU,MAAA;AAC3B,QAAA,SAAA,GAAY,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,eAAe,CAAA,GAC/C;;AAAA,oBAAA,EAA2B,QAAQ,CAAA,IAAA,EAAO,IAAA,CAAK,eAAe,CAAA,OAAA,CAAA;AAClE,QAAA,OAAA,CAAQ,IAAI,CAAA,2CAAA,EAA8C,QAAQ,CAAA,IAAA,EAAO,IAAA,CAAK,eAAe,CAAA,MAAA,CAAQ,CAAA;AAAA,MACzG;AAEA,MAAA,OAAA,CAAQ,IAAI,CAAA,4BAAA,CAAA,EAAgC,SAAA,CAAU,KAAA,CAAM,CAAA,EAAG,GAAG,CAAC,CAAA;AACnE,MAAA,OAAO,SAAA;AAAA,IACX,SAAS,KAAA,EAAY;AACjB,MAAA,OAAA,CAAQ,KAAA,CAAM,yCAAyC,KAAK,CAAA;AAC5D,MAAA,OAAO,CAAA,OAAA,EAAU,KAAA,CAAM,OAAA,IAAW,MAAA,CAAO,KAAK,CAAC,CAAA,CAAA;AAAA,IACnD;AAAA,EACJ;AAAA,EAEQ,WAAW,MAAA,EAAwB;AACvC,IAAA,OAAO,GAAG,MAAM,CAAA,CAAA,EAAI,IAAA,CAAK,GAAA,EAAK,CAAA,CAAA,EAAI,IAAA,CAAK,MAAA,EAAO,CAAE,SAAS,EAAE,CAAA,CAAE,KAAA,CAAM,CAAA,EAAG,CAAC,CAAC,CAAA,CAAA;AAAA,EAC5E;AAAA,EAEQ,UAAU,KAAA,EAA4B;AAC1C,IAAA,MAAM,QAAA,GAAW,KAAA;AACjB,IAAA,IAAI,CAAC,QAAA,CAAS,QAAA,WAAmB,QAAA,GAAW,IAAA,CAAK,WAAW,YAAY,CAAA;AACxE,IAAA,IAAI,CAAC,QAAA,CAAS,KAAA,WAAgB,KAAA,GAAQ,IAAA,CAAK,WAAW,SAAS,CAAA;AAAA,EACnE;AAAA;AAAA,EAGQ,mBAAA,CAAoB,OAAkB,KAAA,EAAuB;AACjE,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAG/D,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AACA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iCAAA,EAAoC,CAAA,CAAE,YAAY,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,SAAA,EAAY,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAC,CAAA,CAAA,CAAG,CAAA;AAAA,MACpI;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AACxC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+BAAA,EAAkC,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,SAAS,CAAA,MAAA,EAAS,CAAA,CAAE,UAAU,CAAA,CAAA,CAAG,CAAA;AAAA,IACtH;AAGA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;AACzB,kBAAA,OAAA,CAAQ,GAAA,CAAI,iDAAiD,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,MAAA,EAAS,EAAA,CAAG,EAAE,CAAA,CAAA,CAAG,CAAA;AAAA,gBAClG;AAAA,cACJ;AAAA,YACJ;AACA,YAAA;AAAA,UACJ;AAAA,QACJ;AAAA,MACJ;AAAA,IACJ;AAAA,EACJ;AAAA;AAAA,EAGA,MAAc,aAAa,KAAA,EAAwC;AAC/D,IAAA,MAAM,EAAE,kBAAA,EAAoB,aAAA,EAAe,eAAA,EAAgB,GAAI,KAAA;AAC/D,IAAA,MAAM,UAAwB,EAAC;AAE/B,IAAA,MAAM,WAAW,CAAC,GAAG,eAAe,CAAA,CAAE,GAAA,CAAI,OAAO,UAAA,KAAe;AAC5D,MAAA,MAAM,QAAA,GAAW,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC7C,MAAA,IAAI,CAAC,QAAA,EAAU;AAEf,MAAA,MAAM,OAAO,IAAA,CAAK,SAAA,CAAU,mBAAmB,GAAA,CAAI,UAAU,KAAK,IAAI,CAAA;AACtE,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,wCAAA,EAA2C,QAAQ,CAAA,CAAE,CAAA;AAEjE,MAAA,MAAM,MAAA,GAAS,MAAM,IAAA,CAAK,WAAA,CAAY,UAAU,IAAI,CAAA;AACpD,MAAA,OAAA,CAAQ,IAAA,CAAK;AAAA,QACT,UAAA;AAAA,QACA,QAAA;AAAA,QACA,MAAA;AAAA,QACA,SAAA,EAAW,IAAA,CAAK,UAAA,CAAW,YAAY;AAAA,OAC1C,CAAA;AACD,MAAA,eAAA,CAAgB,OAAO,UAAU,CAAA;AAAA,IACrC,CAAC,CAAA;AAED,IAAA,MAAM,OAAA,CAAQ,IAAI,QAAQ,CAAA;AAC1B,IAAA,OAAO,OAAA;AAAA,EACX;AAAA;AAAA,EAGQ,eAAA,CAAgB,UAAiC,OAAA,EAA6B;AAClF,IAAA,KAAA,MAAW,EAAE,UAAA,EAAY,QAAA,EAAU,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AAC/D,MAAA,QAAA,CAAS,IAAA,CAAK;AAAA,QACV,MAAM,SAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AACR,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,+CAAA,EAAkD,QAAQ,CAAA,CAAE,CAAA;AAAA,IAC5E;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAI,UAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;AAEjB,MAAA,OAAA,CAAQ,IAAI,CAAA,+BAAA,CAAiC,CAAA;AAC7C,MAAA,OAAA,CAAQ,IAAI,CAAA,0BAAA,EAA6B,QAAA,CAAS,QAAQ,CAAA,SAAA,EAAY,QAAA,CAAS,KAAK,CAAA,CAAE,CAAA;AACtF,MAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,0BAAA,EAA6B,KAAA,CAAM,QAAA,EAAU,MAAA,IAAU,CAAC,CAAA,SAAA,EAAY,IAAA,CAAK,KAAA,EAAO,MAAA,IAAU,CAAC,CAAA,CAAE,CAAA;AAGzG,MAAA,IAAI,IAAA,CAAK,aAAa,MAAA,EAAQ;AAC1B,QAAA,KAAA,CAAM,KAAA,GAAQ,CAAC,GAAI,KAAA,CAAM,SAAS,EAAC,EAAI,GAAG,IAAA,CAAK,WAAW,CAAA;AAC1D,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,yBAAA,EAA4B,IAAA,CAAK,WAAA,CAAY,MAAM,CAAA,OAAA,CAAA,EAAW,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,CAAC,CAAA,KAAY,CAAA,CAAE,IAAI,CAAC,CAAA;AAAA,MACvH;AAEA,MAAA,MAAM,oBAAoB,YAAY;AAClC,QAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,QAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,KAAS,CAAA,EAAG;AAClC,UAAA,QAAA,CAAS,IAAA,CAAK;AAAA,YACV,MAAM,SAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;AAEA,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,kCAAA,EAAqC,KAAA,CAAM,eAAA,CAAgB,IAAI,CAAA,cAAA,CAAgB,CAAA;AAG3F,QAAA,MAAM,YAAY,EAAC;AACnB,QAAA,KAAA,MAAW,UAAA,IAAc,MAAM,eAAA,EAAiB;AAC5C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,aAAA,CAAc,GAAA,CAAI,UAAU,CAAA;AAC/C,UAAA,MAAM,IAAA,GAAO,KAAA,CAAM,kBAAA,CAAmB,GAAA,CAAI,UAAU,CAAA,IAAK,IAAA;AACzD,UAAA,IAAI,IAAA,EAAM;AACN,YAAA,SAAA,CAAU,IAAA,CAAK;AAAA,cACX,EAAA,EAAI,UAAA;AAAA,cACJ,IAAA,EAAM,UAAA;AAAA,cACN,QAAA,EAAU,EAAE,IAAA,EAAM,SAAA,EAAW,IAAA;AAAK,aACrC,CAAA;AAAA,UACL;AAAA,QACJ;AAGA,QAAA,IAAI,SAAA,CAAU,MAAA,GAAS,CAAA,IAAK,KAAA,CAAM,WAAA,EAAa;AAC3C,UAAA,MAAM,YAAA,GAAe;AAAA,YACjB,EAAA,EAAI,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAAA,YAC7B,IAAA,EAAM,WAAA;AAAA,YACN,OAAA,EAAS,MAAM,WAAA,IAAe,IAAA;AAAA;AAAA,YAC9B,UAAA,EAAY,SAAA,CAAU,MAAA,GAAS,CAAA,GAAI,SAAA,GAAY;AAAA,WACnD;AACA,UAAA,KAAA,CAAM,QAAA,CAAS,KAAK,YAAmB,CAAA;AACvC,UAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,iEAAA,EAAoE,KAAA,CAAM,WAAA,EAAa,KAAA,CAAM,CAAA,EAAG,EAAE,CAAC,CAAA,KAAA,EAAQ,SAAA,CAAU,MAAM,CAAA,OAAA,CAAS,CAAA;AAAA,QACpJ;AAGA,QAAA,MAAM,OAAA,GAAU,MAAM,IAAA,CAAK,YAAA,CAAa,KAAK,CAAA;AAC7C,QAAA,IAAA,CAAK,eAAA,CAAgB,UAAU,OAAO,CAAA;AAGtC,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,6CAAA,EAAgD,OAAA,CAAQ,MAAM,CAAA,QAAA,CAAU,CAAA;AAGpF,QAAA,KAAA,MAAW,EAAE,UAAA,EAAY,MAAA,EAAQ,SAAA,MAAe,OAAA,EAAS;AACrD,UAAA,KAAA,CAAM,SAAS,IAAA,CAAK;AAAA,YAChB,EAAA,EAAI,SAAA;AAAA,YACJ,IAAA,EAAM,MAAA;AAAA,YACN,YAAA,EAAc,UAAA;AAAA,YACd,OAAA,EAAS;AAAA,WACL,CAAA;AAAA,QACZ;AAGA,QAAA,KAAA,CAAM,mBAAmB,KAAA,EAAM;AAC/B,QAAA,KAAA,CAAM,cAAc,KAAA,EAAM;AAC1B,QAAA,KAAA,CAAM,WAAA,GAAc,EAAA;AAEpB,QAAA,QAAA,CAAS,KAAA,GAAQ,IAAA,CAAK,UAAA,CAAW,SAAS,CAAA;AAC1C,QAAA,OAAA,CAAQ,GAAA,CAAI,CAAA,mDAAA,EAAsD,KAAA,CAAM,QAAA,CAAS,MAAM,CAAA,CAAE,CAAA;AAGzF,QAAA,IAAA,CAAK,GAAA,CAAI,KAAK,CAAA,CAAE,SAAA,CAAU;AAAA,UACtB,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,YAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,YAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,cAAA,OAAA,CAAQ,IAAI,CAAA,kDAAA,CAAoD,CAAA;AAChE,cAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,cAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,cAAA,QAAA,CAAS,QAAA,EAAS;AAClB,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,WAAA,EAAa;AACtC,cAAA,OAAA,CAAQ,IAAI,CAAA,uDAAA,CAAyD,CAAA;AACrE,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,YAAA,OAAA,CAAQ,IAAI,CAAA,kCAAA,CAAoC,CAAA;AAChD,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,oBACZ,OAAA,EACF;AACE,EAAA,MAAM,UAAA,GAAa,IAAI,aAAA,CAAc,OAAO,CAAA;AAC5C,EAAA,OAAO,CAAC,OAAsB,IAAA,KAA+C;AACzE,IAAA,OAAO,UAAA,CAAW,GAAA,CAAI,KAAA,EAAO,IAAI,CAAA;AAAA,EACrC,CAAA;AACJ","file":"agui-middleware.mjs","sourcesContent":["/**\r\n * MCP Adapter for AG-UI Integration\r\n *\r\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\r\n * It provides tools with handlers for server-side execution and tool definitions\r\n * in JSON Schema format for passing to remote agents.\r\n *\r\n * @example\r\n * ```typescript\r\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\r\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\r\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\r\n * import { HttpAgent } from '@ag-ui/client';\r\n *\r\n * // Create MCP client\r\n * const mcpClient = new MultiSessionClient('user_123');\r\n * await mcpClient.connect();\r\n *\r\n * // Create adapter and get tools\r\n * const adapter = new AguiAdapter(mcpClient);\r\n * const tools = await adapter.getTools();\r\n *\r\n * // Use with AG-UI middleware\r\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\r\n * agent.use(createMcpMiddleware({ tools }));\r\n * ```\r\n */\r\n\r\nimport { MCPClient } from '../server/mcp/oauth-client.js';\r\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\r\n\r\n/**\r\n * Cleans a JSON Schema by removing meta-properties that cause issues with\r\n * strict Pydantic validation (e.g., Google ADK).\r\n *\r\n * Removes: $schema, $id, $comment, $defs, definitions\r\n *\r\n * @param schema - The JSON Schema to clean\r\n * @returns Cleaned schema without meta-properties\r\n */\r\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\r\n if (!schema) {\r\n return { type: 'object', properties: {} };\r\n }\r\n\r\n const cleaned = { ...schema };\r\n\r\n // Remove JSON Schema meta-properties that cause Pydantic validation errors\r\n delete cleaned.$schema;\r\n delete cleaned.$id;\r\n delete cleaned.$comment;\r\n delete cleaned.$defs;\r\n delete cleaned.definitions;\r\n\r\n // Recursively clean nested properties\r\n if (cleaned.properties && typeof cleaned.properties === 'object') {\r\n const cleanedProps: Record<string, any> = {};\r\n for (const [key, value] of Object.entries(cleaned.properties)) {\r\n if (typeof value === 'object' && value !== null) {\r\n cleanedProps[key] = cleanSchema(value as Record<string, any>);\r\n } else {\r\n cleanedProps[key] = value;\r\n }\r\n }\r\n cleaned.properties = cleanedProps;\r\n }\r\n\r\n // Clean items if it's an array schema\r\n if (cleaned.items && typeof cleaned.items === 'object') {\r\n cleaned.items = cleanSchema(cleaned.items);\r\n }\r\n\r\n // Clean additionalProperties if it's an object schema\r\n if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\r\n cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\r\n }\r\n\r\n return cleaned;\r\n}\r\n\r\n/**\r\n * Configuration options for AguiAdapter\r\n */\r\nexport interface AguiAdapterOptions {\r\n /**\r\n * Prefix for tool names to avoid collision with other tools.\r\n * @default serverId or 'mcp'\r\n */\r\n prefix?: string;\r\n}\r\n\r\n/**\r\n * AG-UI Tool with handler for server-side execution.\r\n */\r\nexport interface AguiTool {\r\n name: string;\r\n description: string;\r\n parameters?: Record<string, any>;\r\n handler?: (args: any) => any | Promise<any>;\r\n}\r\n\r\n/**\r\n * Tool definition format for passing to remote agents (without handler).\r\n */\r\nexport interface AguiToolDefinition {\r\n name: string;\r\n description: string;\r\n parameters: Record<string, any>;\r\n}\r\n\r\n/**\r\n * Adapter that transforms MCP tools into AG-UI compatible formats.\r\n */\r\nexport class AguiAdapter {\r\n constructor(\r\n private client: MCPClient | MultiSessionClient,\r\n private options: AguiAdapterOptions = {}\r\n ) { }\r\n\r\n /**\r\n * Get tools with handlers for MCP tool execution.\r\n */\r\n async getTools(): Promise<AguiTool[]> {\r\n if (this.isMultiSession()) {\r\n const clients = (this.client as MultiSessionClient).getClients();\r\n const allTools: AguiTool[] = [];\r\n for (const client of clients) {\r\n allTools.push(...await this.transformTools(client));\r\n }\r\n return allTools;\r\n }\r\n return this.transformTools(this.client as MCPClient);\r\n }\r\n\r\n /**\r\n * Get tool definitions in JSON Schema format for passing to remote agents.\r\n */\r\n async getToolDefinitions(): Promise<AguiToolDefinition[]> {\r\n if (this.isMultiSession()) {\r\n const clients = (this.client as MultiSessionClient).getClients();\r\n const allTools: AguiToolDefinition[] = [];\r\n for (const client of clients) {\r\n allTools.push(...await this.transformToolDefinitions(client));\r\n }\r\n return allTools;\r\n }\r\n return this.transformToolDefinitions(this.client as MCPClient);\r\n }\r\n\r\n /**\r\n * Get tools as a function (for dynamic loading).\r\n */\r\n getToolsFunction(): () => Promise<AguiTool[]> {\r\n return () => this.getTools();\r\n }\r\n\r\n private isMultiSession(): boolean {\r\n return typeof (this.client as any).getClients === 'function';\r\n }\r\n\r\n private async transformTools(client: MCPClient): Promise<AguiTool[]> {\r\n if (!client.isConnected()) return [];\r\n\r\n const result = await client.listTools();\r\n const prefix = this.options.prefix ?? `tool_${client.getServerId() ?? 'mcp'}`;\r\n\r\n return result.tools.map(tool => ({\r\n name: `${prefix}_${tool.name}`,\r\n description: tool.description || `Execute ${tool.name}`,\r\n parameters: cleanSchema(tool.inputSchema),\r\n handler: async (args: any) => {\r\n console.log(`[AguiAdapter] Executing MCP tool: ${tool.name}`, args);\r\n const result = await client.callTool(tool.name, args);\r\n\r\n if (result.content && Array.isArray(result.content)) {\r\n const textContent = result.content\r\n .filter((c: any) => c.type === 'text')\r\n .map((c: any) => c.text)\r\n .join('\\n');\r\n return textContent || result;\r\n }\r\n return result;\r\n }\r\n }));\r\n }\r\n\r\n private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\r\n if (!client.isConnected()) return [];\r\n\r\n const result = await client.listTools();\r\n const prefix = this.options.prefix ?? `tool_${client.getServerId() ?? 'mcp'}`;\r\n\r\n return result.tools.map(tool => ({\r\n name: `${prefix}_${tool.name}`,\r\n description: tool.description || `Execute ${tool.name}`,\r\n parameters: cleanSchema(tool.inputSchema),\r\n }));\r\n }\r\n}\r\n","/**\r\n * AG-UI Middleware for MCP Tool Execution\r\n *\r\n * This middleware intercepts tool calls from remote agents and executes\r\n * MCP tools server-side, returning results back to the agent.\r\n *\r\n * @requires @ag-ui/client - Peer dependency for AG-UI types\r\n * @requires rxjs - Uses RxJS Observables for event streaming\r\n */\r\n\r\nimport { Observable, Subscriber } from 'rxjs';\r\nimport {\r\n Middleware,\r\n EventType,\r\n type AbstractAgent,\r\n type RunAgentInput,\r\n type BaseEvent,\r\n type ToolCallEndEvent,\r\n type Tool,\r\n} from '@ag-ui/client';\r\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\r\n\r\n/** Tool execution result for continuation */\r\ninterface ToolResult {\r\n toolCallId: string;\r\n toolName: string;\r\n result: string;\r\n messageId: string;\r\n}\r\n\r\n/** State for tracking tool calls during a run */\r\ninterface RunState {\r\n toolCallArgsBuffer: Map<string, string>;\r\n toolCallNames: Map<string, string>;\r\n pendingMcpCalls: Set<string>;\r\n textContent?: string;\r\n error: boolean;\r\n}\r\n\r\n/**\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 /** Max result length in chars (default: 50000) */\r\n maxResultLength?: number;\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 private maxResultLength: number;\r\n\r\n constructor(config: McpMiddlewareConfig) {\r\n super();\r\n this.tools = config.tools;\r\n this.maxResultLength = config.maxResultLength ?? 50000;\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 }));\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<string> {\r\n const tool = this.tools.find(t => t.name === toolName);\r\n if (!tool?.handler) {\r\n return `Error: Tool ${tool ? 'has no handler' : 'not found'}: ${toolName}`;\r\n }\r\n\r\n try {\r\n console.log(`[McpMiddleware] Executing tool: ${toolName}`, args);\r\n const result = await tool.handler(args);\r\n let resultStr = typeof result === 'string' ? result : JSON.stringify(result);\r\n\r\n if (resultStr.length > this.maxResultLength) {\r\n const original = resultStr.length;\r\n resultStr = resultStr.slice(0, this.maxResultLength) +\r\n `\\n\\n[... Truncated from ${original} to ${this.maxResultLength} chars]`;\r\n console.log(`[McpMiddleware] Tool result truncated from ${original} to ${this.maxResultLength} chars`);\r\n }\r\n\r\n console.log(`[McpMiddleware] Tool result:`, resultStr.slice(0, 200));\r\n return resultStr;\r\n } catch (error: any) {\r\n console.error(`[McpMiddleware] Error executing tool:`, error);\r\n return `Error: ${error.message || String(error)}`;\r\n }\r\n }\r\n\r\n private generateId(prefix: string): string {\r\n return `${prefix}_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;\r\n }\r\n\r\n private ensureIds(input: RunAgentInput): void {\r\n const anyInput = input as any;\r\n if (!anyInput.threadId) anyInput.threadId = this.generateId('mcp_thread');\r\n if (!anyInput.runId) anyInput.runId = this.generateId('mcp_run');\r\n }\r\n\r\n /** Process tool call events and update state */\r\n private handleToolCallEvent(event: BaseEvent, state: RunState): void {\r\n const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n\r\n // Accumulate text content for reconstruction\r\n if (event.type === EventType.TEXT_MESSAGE_CHUNK) {\r\n const e = event as any;\r\n if (e.delta) {\r\n state.textContent = (state.textContent || '') + e.delta;\r\n }\r\n }\r\n\r\n if (event.type === EventType.TOOL_CALL_START) {\r\n const e = event as any;\r\n if (e.toolCallId && e.toolCallName) {\r\n toolCallNames.set(e.toolCallId, e.toolCallName);\r\n if (this.isMcpTool(e.toolCallName)) {\r\n pendingMcpCalls.add(e.toolCallId);\r\n }\r\n console.log(`[McpMiddleware] TOOL_CALL_START: ${e.toolCallName} (id: ${e.toolCallId}, isMCP: ${this.isMcpTool(e.toolCallName)})`);\r\n }\r\n }\r\n\r\n if (event.type === EventType.TOOL_CALL_ARGS) {\r\n const e = event as any;\r\n if (e.toolCallId && e.delta) {\r\n const existing = toolCallArgsBuffer.get(e.toolCallId) || '';\r\n toolCallArgsBuffer.set(e.toolCallId, existing + e.delta);\r\n }\r\n }\r\n\r\n if (event.type === EventType.TOOL_CALL_END) {\r\n const e = event as ToolCallEndEvent;\r\n console.log(`[McpMiddleware] TOOL_CALL_END: ${toolCallNames.get(e.toolCallId) ?? 'unknown'} (id: ${e.toolCallId})`);\r\n }\r\n\r\n // Workaround: Extract parallel tool calls from MESSAGES_SNAPSHOT\r\n if (event.type === EventType.MESSAGES_SNAPSHOT) {\r\n const messages = (event as any).messages || [];\r\n if (messages.length > 0) {\r\n const lastMsg = messages[messages.length - 1];\r\n // Update text content from snapshot if available (often more reliable)\r\n if (lastMsg.role === 'assistant' && lastMsg.content) {\r\n state.textContent = lastMsg.content;\r\n }\r\n\r\n // Discover tools\r\n for (let i = messages.length - 1; i >= 0; i--) {\r\n const msg = messages[i];\r\n const tools = Array.isArray(msg.toolCalls) ? msg.toolCalls :\r\n (Array.isArray(msg.tool_calls) ? msg.tool_calls : []);\r\n\r\n if (msg.role === 'assistant' && tools.length > 0) {\r\n for (const tc of tools) {\r\n if (tc.id && tc.function?.name && !toolCallNames.has(tc.id)) {\r\n toolCallNames.set(tc.id, tc.function.name);\r\n toolCallArgsBuffer.set(tc.id, tc.function.arguments || '{}');\r\n if (this.isMcpTool(tc.function.name)) {\r\n pendingMcpCalls.add(tc.id);\r\n console.log(`[McpMiddleware] MESSAGES_SNAPSHOT: Discovered ${tc.function.name} (id: ${tc.id})`);\r\n }\r\n }\r\n }\r\n break;\r\n }\r\n }\r\n }\r\n }\r\n }\r\n\r\n /** Execute pending MCP tools and return results */\r\n private async executeTools(state: RunState): Promise<ToolResult[]> {\r\n const { toolCallArgsBuffer, toolCallNames, pendingMcpCalls } = state;\r\n const results: ToolResult[] = [];\r\n\r\n const promises = [...pendingMcpCalls].map(async (toolCallId) => {\r\n const toolName = toolCallNames.get(toolCallId);\r\n if (!toolName) return;\r\n\r\n const args = this.parseArgs(toolCallArgsBuffer.get(toolCallId) || '{}');\r\n console.log(`[McpMiddleware] Executing pending tool: ${toolName}`);\r\n\r\n const result = await this.executeTool(toolName, args);\r\n results.push({\r\n toolCallId,\r\n toolName,\r\n result,\r\n messageId: this.generateId('mcp_result'),\r\n });\r\n pendingMcpCalls.delete(toolCallId);\r\n });\r\n\r\n await Promise.all(promises);\r\n return results;\r\n }\r\n\r\n /** Emit tool results (without RUN_FINISHED - that's emitted when truly done) */\r\n private emitToolResults(observer: Subscriber<BaseEvent>, results: ToolResult[]): void {\r\n for (const { toolCallId, toolName, result, messageId } of results) {\r\n observer.next({\r\n type: EventType.TOOL_CALL_RESULT,\r\n toolCallId,\r\n messageId,\r\n content: result,\r\n role: 'tool',\r\n timestamp: Date.now(),\r\n } as any);\r\n console.log(`[McpMiddleware] Emitting TOOL_CALL_RESULT for: ${toolName}`);\r\n }\r\n }\r\n\r\n run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\r\n return new Observable<BaseEvent>((observer: Subscriber<BaseEvent>) => {\r\n const state: RunState = {\r\n toolCallArgsBuffer: new Map(),\r\n toolCallNames: new Map(),\r\n pendingMcpCalls: new Set(),\r\n textContent: '',\r\n error: false,\r\n };\r\n\r\n this.ensureIds(input);\r\n const anyInput = input as any;\r\n\r\n console.log(`[McpMiddleware] === NEW RUN ===`);\r\n console.log(`[McpMiddleware] threadId: ${anyInput.threadId}, runId: ${anyInput.runId}`);\r\n console.log(`[McpMiddleware] messages: ${input.messages?.length ?? 0}, tools: ${this.tools?.length ?? 0}`);\r\n\r\n // Inject MCP tools\r\n if (this.toolSchemas?.length) {\r\n input.tools = [...(input.tools || []), ...this.toolSchemas];\r\n console.log(`[McpMiddleware] Injected ${this.toolSchemas.length} tools:`, this.toolSchemas.map((t: Tool) => t.name));\r\n }\r\n\r\n const handleRunFinished = async () => {\r\n if (state.error) return; // Don't continue after error\r\n\r\n if (state.pendingMcpCalls.size === 0) {\r\n observer.next({\r\n type: EventType.RUN_FINISHED,\r\n threadId: anyInput.threadId,\r\n runId: anyInput.runId,\r\n timestamp: Date.now(),\r\n } as any);\r\n observer.complete();\r\n return;\r\n }\r\n\r\n console.log(`[McpMiddleware] RUN_FINISHED with ${state.pendingMcpCalls.size} pending calls`);\r\n\r\n // Reconstruct the Assistant Message that triggered these tools\r\n const toolCalls = [];\r\n for (const toolCallId of state.pendingMcpCalls) {\r\n const name = state.toolCallNames.get(toolCallId);\r\n const args = state.toolCallArgsBuffer.get(toolCallId) || '{}';\r\n if (name) {\r\n toolCalls.push({\r\n id: toolCallId,\r\n type: 'function',\r\n function: { name, arguments: args }\r\n });\r\n }\r\n }\r\n\r\n // Add the Assistant Message to history FIRST\r\n if (toolCalls.length > 0 || state.textContent) {\r\n const assistantMsg = {\r\n id: this.generateId('msg_ast'),\r\n role: 'assistant',\r\n content: state.textContent || null, // Ensure null if empty string for strict LLMs\r\n tool_calls: toolCalls.length > 0 ? toolCalls : undefined\r\n };\r\n input.messages.push(assistantMsg as any);\r\n console.log(`[McpMiddleware] Added assistant message to history before tools: ${state.textContent?.slice(0, 50)}... [${toolCalls.length} tools]`);\r\n }\r\n\r\n // Execute tools and emit results (no RUN_FINISHED yet - continuation follows)\r\n const results = await this.executeTools(state);\r\n this.emitToolResults(observer, results);\r\n\r\n // Prepare continuation\r\n console.log(`[McpMiddleware] Triggering continuation with ${results.length} results`);\r\n\r\n // Add tool result messages to history\r\n for (const { toolCallId, result, messageId } of results) {\r\n input.messages.push({\r\n id: messageId,\r\n role: 'tool',\r\n tool_call_id: toolCallId,\r\n content: result,\r\n } as any);\r\n }\r\n\r\n // Reset state for next turn\r\n state.toolCallArgsBuffer.clear();\r\n state.toolCallNames.clear();\r\n state.textContent = ''; // Clear text content for next turn\r\n\r\n anyInput.runId = this.generateId('mcp_run');\r\n console.log(`[McpMiddleware] === CONTINUATION RUN === messages: ${input.messages.length}`);\r\n\r\n // Subscribe to continuation\r\n next.run(input).subscribe({\r\n next: (event) => {\r\n if (state.error) return;\r\n\r\n this.handleToolCallEvent(event, state);\r\n\r\n if (event.type === EventType.RUN_ERROR) {\r\n console.log(`[McpMiddleware] RUN_ERROR received in continuation`);\r\n state.error = true;\r\n observer.next(event);\r\n observer.complete();\r\n return;\r\n }\r\n\r\n if (event.type === EventType.RUN_STARTED) {\r\n console.log(`[McpMiddleware] Filtering RUN_STARTED from continuation`);\r\n return;\r\n }\r\n\r\n if (event.type === EventType.RUN_FINISHED) {\r\n if (state.pendingMcpCalls.size > 0) {\r\n handleRunFinished();\r\n } else {\r\n observer.next(event);\r\n observer.complete();\r\n }\r\n return;\r\n }\r\n observer.next(event);\r\n },\r\n error: (err) => {\r\n state.error = true;\r\n observer.error(err);\r\n },\r\n complete: () => {\r\n if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n },\r\n });\r\n };\r\n\r\n const subscription = next.run(input).subscribe({\r\n next: (event) => {\r\n if (state.error) return;\r\n\r\n this.handleToolCallEvent(event, state);\r\n\r\n if (event.type === EventType.RUN_ERROR) {\r\n console.log(`[McpMiddleware] RUN_ERROR received`);\r\n state.error = true;\r\n observer.next(event);\r\n observer.complete();\r\n return;\r\n }\r\n\r\n if (event.type === EventType.RUN_FINISHED) {\r\n handleRunFinished();\r\n return;\r\n }\r\n observer.next(event);\r\n },\r\n error: (err) => {\r\n state.error = true;\r\n observer.error(err);\r\n },\r\n complete: () => {\r\n if (!state.error && state.pendingMcpCalls.size === 0) observer.complete();\r\n },\r\n });\r\n\r\n return () => subscription.unsubscribe();\r\n });\r\n }\r\n}\r\n\r\n/**\r\n * Factory function to create MCP middleware.\r\n */\r\nexport function createMcpMiddleware(\r\n options: { tools: AguiTool[]; maxResultLength?: number }\r\n) {\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"]}
|
|
1
|
+
{"version":3,"sources":["../../src/adapters/agui-adapter.ts","../../src/adapters/agui-middleware.ts"],"names":[],"mappings":";;;;;;;;;AAmCA,IAAM,wBAAA,GAA2B;AAAA;AAAA,EAE7B,SAAA;AAAA,EAAW,KAAA;AAAA,EAAO,UAAA;AAAA,EAAY,OAAA;AAAA,EAAS,aAAA;AAAA;AAAA,EAEvC,SAAA;AAAA,EAAW,UAAA;AAAA,EAAY,YAAA;AAAA,EAAc,kBAAA;AAAA;AAAA,EAErC,YAAA;AAAA,EAAc,UAAA;AAAA,EAAY,WAAA;AAAA,EAAa,kBAAA;AAAA,EAAoB;AAC/D,CAAA;AASO,SAAS,YAAY,MAAA,EAA8D;AACtF,EAAA,IAAI,CAAC,MAAA,EAAQ;AACT,IAAA,OAAO,EAAE,IAAA,EAAM,QAAA,EAAU,UAAA,EAAY,EAAC,EAAE;AAAA,EAC5C;AAEA,EAAA,MAAM,OAAA,GAAU,EAAE,GAAG,MAAA,EAAO;AAG5B,EAAA,KAAA,MAAW,QAAQ,wBAAA,EAA0B;AACzC,IAAA,OAAO,QAAQ,IAAI,CAAA;AAAA,EACvB;AAGA,EAAA,IAAI,OAAA,CAAQ,UAAA,IAAc,OAAO,OAAA,CAAQ,eAAe,QAAA,EAAU;AAC9D,IAAA,MAAM,eAAoC,EAAC;AAC3C,IAAA,KAAA,MAAW,CAAC,KAAK,KAAK,CAAA,IAAK,OAAO,OAAA,CAAQ,OAAA,CAAQ,UAAU,CAAA,EAAG;AAC3D,MAAA,IAAI,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,EAAM;AAC7C,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,WAAA,CAAY,KAA4B,CAAA;AAAA,MAChE,CAAA,MAAO;AACH,QAAA,YAAA,CAAa,GAAG,CAAA,GAAI,KAAA;AAAA,MACxB;AAAA,IACJ;AACA,IAAA,OAAA,CAAQ,UAAA,GAAa,YAAA;AAAA,EACzB;AAGA,EAAA,IAAI,OAAA,CAAQ,KAAA,IAAS,OAAO,OAAA,CAAQ,UAAU,QAAA,EAAU;AACpD,IAAA,OAAA,CAAQ,KAAA,GAAQ,WAAA,CAAY,OAAA,CAAQ,KAAK,CAAA;AAAA,EAC7C;AAGA,EAAA,IAAI,OAAA,CAAQ,oBAAA,IAAwB,OAAO,OAAA,CAAQ,yBAAyB,QAAA,EAAU;AAClF,IAAA,OAAA,CAAQ,oBAAA,GAAuB,WAAA,CAAY,OAAA,CAAQ,oBAAoB,CAAA;AAAA,EAC3E;AAEA,EAAA,OAAO,OAAA;AACX;;;AChEO,IAAM,gBAAA,GAAmB;AAgDzB,IAAM,aAAA,GAAN,cAA4B,UAAA,CAAW;AAAA,EAI1C,YAAY,MAAA,EAA6B;AACrC,IAAA,KAAA,EAAM;AAJV,IAAA,aAAA,CAAA,IAAA,EAAQ,OAAA,CAAA;AACR,IAAA,aAAA,CAAA,IAAA,EAAQ,aAAA,CAAA;AAIJ,IAAA,IAAA,CAAK,QAAQ,MAAA,CAAO,KAAA;AACpB,IAAA,IAAA,CAAK,WAAA,GAAc,IAAA,CAAK,KAAA,CAAM,GAAA,CAAI,CAAC,CAAA,MAAiB;AAAA,MAChD,MAAM,CAAA,CAAE,IAAA;AAAA,MACR,aAAa,CAAA,CAAE,WAAA;AAAA,MACf,UAAA,EAAY,WAAA,CAAY,CAAA,CAAE,UAAU,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,KAAS,SAAA,CAAU,kBAAA,EAAoB;AAC7C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,EAAE,KAAA,EAAO;AACT,QAAA,KAAA,CAAM,WAAA,GAAA,CAAe,KAAA,CAAM,WAAA,IAAe,EAAA,IAAM,CAAA,CAAE,KAAA;AAAA,MACtD;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,eAAA,EAAiB;AAC1C,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,YAAA,EAAc;AAChC,QAAA,aAAA,CAAc,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,CAAA,CAAE,YAAY,CAAA;AAC9C,QAAA,IAAI,IAAA,CAAK,SAAA,CAAU,CAAA,CAAE,YAAY,CAAA,EAAG;AAChC,UAAA,eAAA,CAAgB,GAAA,CAAI,EAAE,UAAU,CAAA;AAAA,QACpC;AAAA,MACJ;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,cAAA,EAAgB;AACzC,MAAA,MAAM,CAAA,GAAI,KAAA;AACV,MAAA,IAAI,CAAA,CAAE,UAAA,IAAc,CAAA,CAAE,KAAA,EAAO;AACzB,QAAA,MAAM,QAAA,GAAW,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAU,CAAA,IAAK,EAAA;AACzD,QAAA,kBAAA,CAAmB,GAAA,CAAI,CAAA,CAAE,UAAA,EAAY,QAAA,GAAW,EAAE,KAAK,CAAA;AAAA,MAC3D;AAAA,IACJ;AAEA,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,aAAA,EAAe;AAK5C,IAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,iBAAA,EAAmB;AAC5C,MAAA,MAAM,QAAA,GAAY,KAAA,CAAc,QAAA,IAAY,EAAC;AAC7C,MAAA,IAAI,QAAA,CAAS,SAAS,CAAA,EAAG;AACrB,QAAA,MAAM,OAAA,GAAU,QAAA,CAAS,QAAA,CAAS,MAAA,GAAS,CAAC,CAAA;AAE5C,QAAA,IAAI,OAAA,CAAQ,IAAA,KAAS,WAAA,IAAe,OAAA,CAAQ,OAAA,EAAS;AACjD,UAAA,KAAA,CAAM,cAAc,OAAA,CAAQ,OAAA;AAAA,QAChC;AAGA,QAAA,KAAA,IAAS,IAAI,QAAA,CAAS,MAAA,GAAS,CAAA,EAAG,CAAA,IAAK,GAAG,CAAA,EAAA,EAAK;AAC3C,UAAA,MAAM,GAAA,GAAM,SAAS,CAAC,CAAA;AACtB,UAAA,MAAM,KAAA,GAAQ,KAAA,CAAM,OAAA,CAAQ,GAAA,CAAI,SAAS,CAAA,GAAI,GAAA,CAAI,SAAA,GAC5C,KAAA,CAAM,QAAQ,GAAA,CAAI,UAAU,CAAA,GAAI,GAAA,CAAI,aAAa,EAAC;AAEvD,UAAA,IAAI,GAAA,CAAI,IAAA,KAAS,WAAA,IAAe,KAAA,CAAM,SAAS,CAAA,EAAG;AAC9C,YAAA,KAAA,MAAW,MAAM,KAAA,EAAO;AACpB,cAAA,IAAI,EAAA,CAAG,EAAA,IAAM,EAAA,CAAG,QAAA,EAAU,IAAA,IAAQ,CAAC,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAE,CAAA,EAAG;AACzD,gBAAA,aAAA,CAAc,GAAA,CAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,SAAS,IAAI,CAAA;AACzC,gBAAA,kBAAA,CAAmB,IAAI,EAAA,CAAG,EAAA,EAAI,EAAA,CAAG,QAAA,CAAS,aAAa,IAAI,CAAA;AAC3D,gBAAA,IAAI,IAAA,CAAK,SAAA,CAAU,EAAA,CAAG,QAAA,CAAS,IAAI,CAAA,EAAG;AAClC,kBAAA,eAAA,CAAgB,GAAA,CAAI,GAAG,EAAE,CAAA;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,MAAM,SAAA,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,MAAM,SAAA,CAAU,gBAAA;AAAA,QAChB,UAAA;AAAA,QACA,SAAA;AAAA,QACA,OAAA,EAAS,MAAA;AAAA,QACT,IAAA,EAAM,MAAA;AAAA,QACN,SAAA,EAAW,KAAK,GAAA;AAAI,OAChB,CAAA;AAAA,IACZ;AAAA,EACJ;AAAA,EAEA,GAAA,CAAI,OAAsB,IAAA,EAA4C;AAClE,IAAA,OAAO,IAAI,UAAA,CAAsB,CAAC,QAAA,KAAoC;AAClE,MAAA,MAAM,KAAA,GAAkB;AAAA,QACpB,kBAAA,sBAAwB,GAAA,EAAI;AAAA,QAC5B,aAAA,sBAAmB,GAAA,EAAI;AAAA,QACvB,eAAA,sBAAqB,GAAA,EAAI;AAAA,QACzB,WAAA,EAAa,EAAA;AAAA,QACb,KAAA,EAAO;AAAA,OACX;AAEA,MAAA,IAAA,CAAK,UAAU,KAAK,CAAA;AACpB,MAAA,MAAM,QAAA,GAAW,KAAA;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,MAAM,SAAA,CAAU,YAAA;AAAA,YAChB,UAAU,QAAA,CAAS,QAAA;AAAA,YACnB,OAAO,QAAA,CAAS,KAAA;AAAA,YAChB,SAAA,EAAW,KAAK,GAAA;AAAI,WAChB,CAAA;AACR,UAAA,QAAA,CAAS,QAAA,EAAS;AAClB,UAAA;AAAA,QACJ;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,KAAS,SAAA,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,KAAS,SAAA,CAAU,WAAA,EAAa;AACtC,cAAA;AAAA,YACJ;AAEA,YAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,cAAA,IAAI,KAAA,CAAM,eAAA,CAAgB,IAAA,GAAO,CAAA,EAAG;AAChC,gBAAA,iBAAA,EAAkB;AAAA,cACtB,CAAA,MAAO;AACH,gBAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,gBAAA,QAAA,CAAS,QAAA,EAAS;AAAA,cACtB;AACA,cAAA;AAAA,YACJ;AACA,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,UACvB,CAAA;AAAA,UACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,UACtB,CAAA;AAAA,UACA,UAAU,MAAM;AACZ,YAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,UAC5E;AAAA,SACH,CAAA;AAAA,MACL,CAAA;AAEA,MAAA,MAAM,YAAA,GAAe,IAAA,CAAK,GAAA,CAAI,KAAK,EAAE,SAAA,CAAU;AAAA,QAC3C,IAAA,EAAM,CAAC,KAAA,KAAU;AACb,UAAA,IAAI,MAAM,KAAA,EAAO;AAEjB,UAAA,IAAA,CAAK,mBAAA,CAAoB,OAAO,KAAK,CAAA;AAErC,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,SAAA,EAAW;AACpC,YAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,YAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AACnB,YAAA,QAAA,CAAS,QAAA,EAAS;AAClB,YAAA;AAAA,UACJ;AAEA,UAAA,IAAI,KAAA,CAAM,IAAA,KAAS,SAAA,CAAU,YAAA,EAAc;AACvC,YAAA,iBAAA,EAAkB;AAClB,YAAA;AAAA,UACJ;AACA,UAAA,QAAA,CAAS,KAAK,KAAK,CAAA;AAAA,QACvB,CAAA;AAAA,QACA,KAAA,EAAO,CAAC,GAAA,KAAQ;AACZ,UAAA,KAAA,CAAM,KAAA,GAAQ,IAAA;AACd,UAAA,QAAA,CAAS,MAAM,GAAG,CAAA;AAAA,QACtB,CAAA;AAAA,QACA,UAAU,MAAM;AACZ,UAAA,IAAI,CAAC,MAAM,KAAA,IAAS,KAAA,CAAM,gBAAgB,IAAA,KAAS,CAAA,WAAY,QAAA,EAAS;AAAA,QAC5E;AAAA,OACH,CAAA;AAED,MAAA,OAAO,MAAM,aAAa,WAAA,EAAY;AAAA,IAC1C,CAAC,CAAA;AAAA,EACL;AACJ;AAKO,SAAS,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.mjs","sourcesContent":["/**\r\n * MCP Adapter for AG-UI Integration\r\n *\r\n * This adapter transforms MCP tools into formats compatible with AG-UI agents.\r\n * It provides tools with handlers for server-side execution and tool definitions\r\n * in JSON Schema format for passing to remote agents.\r\n *\r\n * @example\r\n * ```typescript\r\n * import { MultiSessionClient } from '@mcp-ts/sdk/server';\r\n * import { AguiAdapter } from '@mcp-ts/sdk/adapters/agui-adapter';\r\n * import { createMcpMiddleware } from '@mcp-ts/sdk/adapters/agui-middleware';\r\n * import { HttpAgent } from '@ag-ui/client';\r\n *\r\n * // Create MCP client\r\n * const mcpClient = new MultiSessionClient('user_123');\r\n * await mcpClient.connect();\r\n *\r\n * // Create adapter and get tools\r\n * const adapter = new AguiAdapter(mcpClient);\r\n * const tools = await adapter.getTools();\r\n *\r\n * // Use with AG-UI middleware\r\n * const agent = new HttpAgent({ url: 'http://localhost:8000/agent' });\r\n * agent.use(createMcpMiddleware({ tools }));\r\n * ```\r\n */\r\n\r\nimport { MCPClient } from '../server/mcp/oauth-client.js';\r\nimport { MultiSessionClient } from '../server/mcp/multi-session-client.js';\r\n\r\n/**\r\n * Extended JSON Schema properties that Pydantic's strict validation rejects.\r\n * These are valid JSON Schema extensions but not part of the core spec.\r\n */\r\nconst PYDANTIC_FORBIDDEN_PROPS = [\r\n // JSON Schema meta-properties\r\n '$schema', '$id', '$comment', '$defs', 'definitions',\r\n // Extended properties used by some MCP servers (e.g., Apify)\r\n 'prefill', 'examples', 'enumTitles', 'enumDescriptions',\r\n // Other common extensions\r\n 'deprecated', 'readOnly', 'writeOnly', 'contentMediaType', 'contentEncoding',\r\n];\r\n\r\n/**\r\n * Cleans a JSON Schema by removing meta-properties that cause issues with\r\n * strict Pydantic validation (e.g., Google ADK, LangGraph).\r\n *\r\n * @param schema - The JSON Schema to clean\r\n * @returns Cleaned schema without forbidden properties\r\n */\r\nexport function cleanSchema(schema: Record<string, any> | undefined): Record<string, any> {\r\n if (!schema) {\r\n return { type: 'object', properties: {} };\r\n }\r\n\r\n const cleaned = { ...schema };\r\n\r\n // Remove all forbidden properties\r\n for (const prop of PYDANTIC_FORBIDDEN_PROPS) {\r\n delete cleaned[prop];\r\n }\r\n\r\n // Recursively clean nested properties\r\n if (cleaned.properties && typeof cleaned.properties === 'object') {\r\n const cleanedProps: Record<string, any> = {};\r\n for (const [key, value] of Object.entries(cleaned.properties)) {\r\n if (typeof value === 'object' && value !== null) {\r\n cleanedProps[key] = cleanSchema(value as Record<string, any>);\r\n } else {\r\n cleanedProps[key] = value;\r\n }\r\n }\r\n cleaned.properties = cleanedProps;\r\n }\r\n\r\n // Clean items if it's an array schema\r\n if (cleaned.items && typeof cleaned.items === 'object') {\r\n cleaned.items = cleanSchema(cleaned.items);\r\n }\r\n\r\n // Clean additionalProperties if it's an object schema\r\n if (cleaned.additionalProperties && typeof cleaned.additionalProperties === 'object') {\r\n cleaned.additionalProperties = cleanSchema(cleaned.additionalProperties);\r\n }\r\n\r\n return cleaned;\r\n}\r\n\r\n/**\r\n * Configuration options for AguiAdapter\r\n */\r\nexport interface AguiAdapterOptions {\r\n /**\r\n * Prefix for tool names to avoid collision with other tools.\r\n * @default serverId or 'mcp'\r\n */\r\n prefix?: string;\r\n}\r\n\r\n/**\r\n * AG-UI Tool with handler for server-side execution.\r\n */\r\nexport interface AguiTool {\r\n name: string;\r\n description: string;\r\n parameters?: Record<string, any>;\r\n _meta?: Record<string, any>; // Add _meta to AguiTool\r\n handler?: (args: any) => any | Promise<any>;\r\n}\r\n\r\n/**\r\n * Tool definition format for passing to remote agents (without handler).\r\n */\r\nexport interface AguiToolDefinition {\r\n name: string;\r\n description: string;\r\n parameters: Record<string, any>;\r\n _meta?: Record<string, any>; // Add _meta to AguiToolDefinition\r\n}\r\n\r\n/**\r\n * Adapter that transforms MCP tools into AG-UI compatible formats.\r\n */\r\nexport class AguiAdapter {\r\n constructor(\r\n private client: MCPClient | MultiSessionClient,\r\n private options: AguiAdapterOptions = {}\r\n ) { }\r\n\r\n /**\r\n * Get tools with handlers for MCP tool execution.\r\n */\r\n async getTools(): Promise<AguiTool[]> {\r\n if (this.isMultiSession()) {\r\n const clients = (this.client as MultiSessionClient).getClients();\r\n const allTools: AguiTool[] = [];\r\n for (const client of clients) {\r\n allTools.push(...await this.transformTools(client));\r\n }\r\n return allTools;\r\n }\r\n return this.transformTools(this.client as MCPClient);\r\n }\r\n\r\n /**\r\n * Get tool definitions in JSON Schema format for passing to remote agents.\r\n */\r\n async getToolDefinitions(): Promise<AguiToolDefinition[]> {\r\n if (this.isMultiSession()) {\r\n const clients = (this.client as MultiSessionClient).getClients();\r\n const allTools: AguiToolDefinition[] = [];\r\n for (const client of clients) {\r\n allTools.push(...await this.transformToolDefinitions(client));\r\n }\r\n return allTools;\r\n }\r\n return this.transformToolDefinitions(this.client as MCPClient);\r\n }\r\n\r\n /**\r\n * Get tools as a function (for dynamic loading).\r\n */\r\n getToolsFunction(): () => Promise<AguiTool[]> {\r\n return () => this.getTools();\r\n }\r\n\r\n private isMultiSession(): boolean {\r\n return typeof (this.client as any).getClients === 'function';\r\n }\r\n\r\n private async transformTools(client: MCPClient): Promise<AguiTool[]> {\r\n if (!client.isConnected()) return [];\r\n\r\n const result = await client.listTools();\r\n const serverId = (typeof (client as any).getServerId === 'function'\r\n ? (client as any).getServerId()\r\n : undefined) as string | undefined;\r\n const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n const prefix = `tool_${normalizedPrefix}`;\r\n\r\n return result.tools.map(tool => {\r\n // Type assertion to access _meta if it exists on the tool object (it comes from MCP SDK)\r\n const mcpTool = tool as any;\r\n const mcpToolName = tool.name;\r\n return {\r\n name: `${prefix}_${tool.name}`,\r\n description: tool.description || `Execute ${tool.name}`,\r\n parameters: cleanSchema(tool.inputSchema),\r\n _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n handler: async (args: any) => {\r\n // Call the actual MCP tool\r\n const callResult = await (client as any).callTool(mcpToolName, args ?? {});\r\n\r\n // Return the raw result object so middleware can inspect `_meta` (e.g. for UI triggers)\r\n return callResult;\r\n }\r\n }\r\n });\r\n }\r\n\r\n private async transformToolDefinitions(client: MCPClient): Promise<AguiToolDefinition[]> {\r\n if (!client.isConnected()) return [];\r\n\r\n const result = await client.listTools();\r\n const serverId = (typeof (client as any).getServerId === 'function'\r\n ? (client as any).getServerId()\r\n : undefined) as string | undefined;\r\n const normalizedPrefix = (this.options.prefix ?? serverId ?? 'mcp').replace(/-/g, '');\r\n const prefix = `tool_${normalizedPrefix}`;\r\n\r\n return result.tools.map(tool => {\r\n const mcpTool = tool as any;\r\n return {\r\n name: `${prefix}_${tool.name}`,\r\n description: tool.description || `Execute ${tool.name}`,\r\n parameters: cleanSchema(tool.inputSchema),\r\n _meta: { ...mcpTool._meta, sessionId: (client as any).getSessionId?.() },\r\n };\r\n });\r\n }\r\n}\r\n","/**\r\n * AG-UI Middleware for MCP Tool Execution\r\n *\r\n * This middleware intercepts tool calls from remote agents and executes\r\n * MCP tools server-side, returning results back to the agent.\r\n *\r\n * @requires @ag-ui/client - Peer dependency for AG-UI types\r\n * @requires rxjs - Uses RxJS Observables for event streaming\r\n */\r\n\r\nimport { Observable, Subscriber } from 'rxjs';\r\nimport {\r\n Middleware,\r\n EventType,\r\n type AbstractAgent,\r\n type RunAgentInput,\r\n type BaseEvent,\r\n type ToolCallEndEvent,\r\n type Tool,\r\n} from '@ag-ui/client';\r\nimport { type AguiTool, cleanSchema } from './agui-adapter.js';\r\n\r\n/** 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"]}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CxogNckF.mjs';
|
|
2
2
|
import { ToolSet } from 'ai';
|
|
3
|
-
import '../events-
|
|
3
|
+
import '../events-BgeztGYZ.mjs';
|
|
4
4
|
import '@modelcontextprotocol/sdk/types.js';
|
|
5
5
|
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
6
|
import '@modelcontextprotocol/sdk/client/auth.js';
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-cox_WXUj.js';
|
|
2
2
|
import { ToolSet } from 'ai';
|
|
3
|
-
import '../events-
|
|
3
|
+
import '../events-BgeztGYZ.js';
|
|
4
4
|
import '@modelcontextprotocol/sdk/types.js';
|
|
5
5
|
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
6
|
import '@modelcontextprotocol/sdk/client/auth.js';
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CxogNckF.mjs';
|
|
2
2
|
import { StructuredTool } from '@langchain/core/tools';
|
|
3
|
-
import '../events-
|
|
3
|
+
import '../events-BgeztGYZ.mjs';
|
|
4
4
|
import '@modelcontextprotocol/sdk/types.js';
|
|
5
5
|
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
6
|
import '@modelcontextprotocol/sdk/client/auth.js';
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-cox_WXUj.js';
|
|
2
2
|
import { StructuredTool } from '@langchain/core/tools';
|
|
3
|
-
import '../events-
|
|
3
|
+
import '../events-BgeztGYZ.js';
|
|
4
4
|
import '@modelcontextprotocol/sdk/types.js';
|
|
5
5
|
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
6
|
import '@modelcontextprotocol/sdk/client/auth.js';
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-CxogNckF.mjs';
|
|
2
2
|
import { z } from 'zod';
|
|
3
|
-
import '../events-
|
|
3
|
+
import '../events-BgeztGYZ.mjs';
|
|
4
4
|
import '@modelcontextprotocol/sdk/types.js';
|
|
5
5
|
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
6
|
import '@modelcontextprotocol/sdk/client/auth.js';
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-
|
|
1
|
+
import { M as MCPClient, a as MultiSessionClient } from '../multi-session-client-cox_WXUj.js';
|
|
2
2
|
import { z } from 'zod';
|
|
3
|
-
import '../events-
|
|
3
|
+
import '../events-BgeztGYZ.js';
|
|
4
4
|
import '@modelcontextprotocol/sdk/types.js';
|
|
5
5
|
import '@modelcontextprotocol/sdk/shared/auth.js';
|
|
6
6
|
import '@modelcontextprotocol/sdk/client/auth.js';
|