agents 0.0.0-8a4558c → 0.0.0-8bf3250
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/dist/ai-chat-agent.d.ts +3 -3
- package/dist/ai-chat-agent.js +34 -26
- package/dist/ai-chat-agent.js.map +1 -1
- package/dist/ai-react.d.ts +7 -7
- package/dist/ai-react.js +27 -27
- package/dist/ai-react.js.map +1 -1
- package/dist/{chunk-6RPGDIE2.js → chunk-4CIGD73X.js} +81 -76
- package/dist/chunk-4CIGD73X.js.map +1 -0
- package/dist/{chunk-BZXOAZUX.js → chunk-767EASBA.js} +5 -5
- package/dist/{chunk-BZXOAZUX.js.map → chunk-767EASBA.js.map} +1 -1
- package/dist/{chunk-OYJXQRRH.js → chunk-E3LCYPCB.js} +22 -18
- package/dist/chunk-E3LCYPCB.js.map +1 -0
- package/dist/{chunk-VCSB47AK.js → chunk-NKZZ66QY.js} +8 -8
- package/dist/chunk-NKZZ66QY.js.map +1 -0
- package/dist/client.js +1 -1
- package/dist/index.d.ts +10 -9
- package/dist/index.js +4 -4
- package/dist/mcp/client.d.ts +281 -9
- package/dist/mcp/client.js +1 -1
- package/dist/mcp/do-oauth-client-provider.js +1 -1
- package/dist/mcp/index.d.ts +6 -6
- package/dist/mcp/index.js +48 -47
- package/dist/mcp/index.js.map +1 -1
- package/dist/react.d.ts +4 -4
- package/dist/react.js +7 -7
- package/dist/react.js.map +1 -1
- package/dist/schedule.d.ts +2 -2
- package/dist/schedule.js +4 -4
- package/dist/schedule.js.map +1 -1
- package/package.json +70 -71
- package/src/index.ts +98 -96
- package/dist/chunk-6RPGDIE2.js.map +0 -1
- package/dist/chunk-OYJXQRRH.js.map +0 -1
- package/dist/chunk-VCSB47AK.js.map +0 -1
|
@@ -1,22 +1,22 @@
|
|
|
1
1
|
import {
|
|
2
2
|
MCPClientManager
|
|
3
|
-
} from "./chunk-
|
|
3
|
+
} from "./chunk-E3LCYPCB.js";
|
|
4
4
|
import {
|
|
5
5
|
DurableObjectOAuthClientProvider
|
|
6
|
-
} from "./chunk-
|
|
6
|
+
} from "./chunk-767EASBA.js";
|
|
7
7
|
import {
|
|
8
8
|
camelCaseToKebabCase
|
|
9
|
-
} from "./chunk-
|
|
9
|
+
} from "./chunk-NKZZ66QY.js";
|
|
10
10
|
|
|
11
11
|
// src/index.ts
|
|
12
|
+
import { AsyncLocalStorage } from "async_hooks";
|
|
13
|
+
import { parseCronExpression } from "cron-schedule";
|
|
14
|
+
import { nanoid } from "nanoid";
|
|
12
15
|
import {
|
|
13
|
-
Server,
|
|
14
16
|
getServerByName,
|
|
15
|
-
routePartykitRequest
|
|
17
|
+
routePartykitRequest,
|
|
18
|
+
Server
|
|
16
19
|
} from "partyserver";
|
|
17
|
-
import { parseCronExpression } from "cron-schedule";
|
|
18
|
-
import { nanoid } from "nanoid";
|
|
19
|
-
import { AsyncLocalStorage } from "async_hooks";
|
|
20
20
|
function isRPCRequest(msg) {
|
|
21
21
|
return typeof msg === "object" && msg !== null && "type" in msg && msg.type === "rpc" && "id" in msg && typeof msg.id === "string" && "method" in msg && typeof msg.method === "string" && "args" in msg && Array.isArray(msg.args);
|
|
22
22
|
}
|
|
@@ -148,13 +148,13 @@ var Agent = class extends Server {
|
|
|
148
148
|
await this.mcp.handleCallbackRequest(request);
|
|
149
149
|
this.broadcast(
|
|
150
150
|
JSON.stringify({
|
|
151
|
-
|
|
152
|
-
|
|
151
|
+
mcp: this.getMcpServers(),
|
|
152
|
+
type: "cf_agent_mcp_servers"
|
|
153
153
|
})
|
|
154
154
|
);
|
|
155
155
|
return new Response("<script>window.close();</script>", {
|
|
156
|
-
|
|
157
|
-
|
|
156
|
+
headers: { "content-type": "text/html" },
|
|
157
|
+
status: 200
|
|
158
158
|
});
|
|
159
159
|
}
|
|
160
160
|
return this._tryCatch(() => _onRequest(request));
|
|
@@ -172,7 +172,7 @@ var Agent = class extends Server {
|
|
|
172
172
|
let parsed;
|
|
173
173
|
try {
|
|
174
174
|
parsed = JSON.parse(message);
|
|
175
|
-
} catch (
|
|
175
|
+
} catch (_e) {
|
|
176
176
|
return this._tryCatch(() => _onMessage(connection, message));
|
|
177
177
|
}
|
|
178
178
|
if (isStateUpdateMessage(parsed)) {
|
|
@@ -197,19 +197,19 @@ var Agent = class extends Server {
|
|
|
197
197
|
}
|
|
198
198
|
const result = await methodFn.apply(this, args);
|
|
199
199
|
const response = {
|
|
200
|
-
|
|
200
|
+
done: true,
|
|
201
201
|
id,
|
|
202
|
-
success: true,
|
|
203
202
|
result,
|
|
204
|
-
|
|
203
|
+
success: true,
|
|
204
|
+
type: "rpc"
|
|
205
205
|
};
|
|
206
206
|
connection.send(JSON.stringify(response));
|
|
207
207
|
} catch (e) {
|
|
208
208
|
const response = {
|
|
209
|
-
|
|
209
|
+
error: e instanceof Error ? e.message : "Unknown error occurred",
|
|
210
210
|
id: parsed.id,
|
|
211
211
|
success: false,
|
|
212
|
-
|
|
212
|
+
type: "rpc"
|
|
213
213
|
};
|
|
214
214
|
connection.send(JSON.stringify(response));
|
|
215
215
|
console.error("RPC error:", e);
|
|
@@ -229,15 +229,15 @@ var Agent = class extends Server {
|
|
|
229
229
|
if (this.state) {
|
|
230
230
|
connection.send(
|
|
231
231
|
JSON.stringify({
|
|
232
|
-
|
|
233
|
-
|
|
232
|
+
state: this.state,
|
|
233
|
+
type: "cf_agent_state"
|
|
234
234
|
})
|
|
235
235
|
);
|
|
236
236
|
}
|
|
237
237
|
connection.send(
|
|
238
238
|
JSON.stringify({
|
|
239
|
-
|
|
240
|
-
|
|
239
|
+
mcp: this.getMcpServers(),
|
|
240
|
+
type: "cf_agent_mcp_servers"
|
|
241
241
|
})
|
|
242
242
|
);
|
|
243
243
|
return this._tryCatch(() => _onConnect(connection, ctx2));
|
|
@@ -253,8 +253,8 @@ var Agent = class extends Server {
|
|
|
253
253
|
const servers = this.sql`
|
|
254
254
|
SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
|
|
255
255
|
`;
|
|
256
|
-
|
|
257
|
-
servers.
|
|
256
|
+
Promise.allSettled(
|
|
257
|
+
servers.map((server) => {
|
|
258
258
|
return this._connectToMcpServerInternal(
|
|
259
259
|
server.name,
|
|
260
260
|
server.server_url,
|
|
@@ -266,13 +266,14 @@ var Agent = class extends Server {
|
|
|
266
266
|
}
|
|
267
267
|
);
|
|
268
268
|
})
|
|
269
|
-
)
|
|
270
|
-
|
|
271
|
-
|
|
272
|
-
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
|
|
269
|
+
).then((_results) => {
|
|
270
|
+
this.broadcast(
|
|
271
|
+
JSON.stringify({
|
|
272
|
+
mcp: this.getMcpServers(),
|
|
273
|
+
type: "cf_agent_mcp_servers"
|
|
274
|
+
})
|
|
275
|
+
);
|
|
276
|
+
});
|
|
276
277
|
await this._tryCatch(() => _onStart());
|
|
277
278
|
}
|
|
278
279
|
);
|
|
@@ -335,8 +336,8 @@ var Agent = class extends Server {
|
|
|
335
336
|
`;
|
|
336
337
|
this.broadcast(
|
|
337
338
|
JSON.stringify({
|
|
338
|
-
|
|
339
|
-
|
|
339
|
+
state,
|
|
340
|
+
type: "cf_agent_state"
|
|
340
341
|
}),
|
|
341
342
|
source !== "server" ? [source.id] : []
|
|
342
343
|
);
|
|
@@ -362,12 +363,14 @@ var Agent = class extends Server {
|
|
|
362
363
|
* @param state Updated state
|
|
363
364
|
* @param source Source of the state update ("server" or a client connection)
|
|
364
365
|
*/
|
|
366
|
+
// biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later
|
|
365
367
|
onStateUpdate(state, source) {
|
|
366
368
|
}
|
|
367
369
|
/**
|
|
368
370
|
* Called when the Agent receives an email
|
|
369
371
|
* @param email Email message to process
|
|
370
372
|
*/
|
|
373
|
+
// biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later
|
|
371
374
|
onEmail(email) {
|
|
372
375
|
return agentContext.run(
|
|
373
376
|
{ agent: this, connection: void 0, request: void 0 },
|
|
@@ -434,8 +437,8 @@ var Agent = class extends Server {
|
|
|
434
437
|
`;
|
|
435
438
|
await this._scheduleNextAlarm();
|
|
436
439
|
return {
|
|
437
|
-
id,
|
|
438
440
|
callback,
|
|
441
|
+
id,
|
|
439
442
|
payload,
|
|
440
443
|
time: timestamp,
|
|
441
444
|
type: "scheduled"
|
|
@@ -452,10 +455,10 @@ var Agent = class extends Server {
|
|
|
452
455
|
`;
|
|
453
456
|
await this._scheduleNextAlarm();
|
|
454
457
|
return {
|
|
455
|
-
id,
|
|
456
458
|
callback,
|
|
457
|
-
payload,
|
|
458
459
|
delayInSeconds: when,
|
|
460
|
+
id,
|
|
461
|
+
payload,
|
|
459
462
|
time: timestamp,
|
|
460
463
|
type: "delayed"
|
|
461
464
|
};
|
|
@@ -471,10 +474,10 @@ var Agent = class extends Server {
|
|
|
471
474
|
`;
|
|
472
475
|
await this._scheduleNextAlarm();
|
|
473
476
|
return {
|
|
474
|
-
id,
|
|
475
477
|
callback,
|
|
476
|
-
payload,
|
|
477
478
|
cron: when,
|
|
479
|
+
id,
|
|
480
|
+
payload,
|
|
478
481
|
time: timestamp,
|
|
479
482
|
type: "cron"
|
|
480
483
|
};
|
|
@@ -587,15 +590,28 @@ var Agent = class extends Server {
|
|
|
587
590
|
callbackUrl,
|
|
588
591
|
options
|
|
589
592
|
);
|
|
593
|
+
this.sql`
|
|
594
|
+
INSERT
|
|
595
|
+
OR REPLACE INTO cf_agents_mcp_servers (id, name, server_url, client_id, auth_url, callback_url, server_options)
|
|
596
|
+
VALUES (
|
|
597
|
+
${result.id},
|
|
598
|
+
${serverName},
|
|
599
|
+
${url},
|
|
600
|
+
${result.clientId ?? null},
|
|
601
|
+
${result.authUrl ?? null},
|
|
602
|
+
${callbackUrl},
|
|
603
|
+
${options ? JSON.stringify(options) : null}
|
|
604
|
+
);
|
|
605
|
+
`;
|
|
590
606
|
this.broadcast(
|
|
591
607
|
JSON.stringify({
|
|
592
|
-
|
|
593
|
-
|
|
608
|
+
mcp: this.getMcpServers(),
|
|
609
|
+
type: "cf_agent_mcp_servers"
|
|
594
610
|
})
|
|
595
611
|
);
|
|
596
612
|
return result;
|
|
597
613
|
}
|
|
598
|
-
async _connectToMcpServerInternal(
|
|
614
|
+
async _connectToMcpServerInternal(_serverName, url, callbackUrl, options, reconnect) {
|
|
599
615
|
const authProvider = new DurableObjectOAuthClientProvider(
|
|
600
616
|
this.ctx.storage,
|
|
601
617
|
this.name,
|
|
@@ -622,28 +638,17 @@ var Agent = class extends Server {
|
|
|
622
638
|
};
|
|
623
639
|
}
|
|
624
640
|
const { id, authUrl, clientId } = await this.mcp.connect(url, {
|
|
641
|
+
client: options?.client,
|
|
625
642
|
reconnect,
|
|
626
643
|
transport: {
|
|
627
644
|
...headerTransportOpts,
|
|
628
645
|
authProvider
|
|
629
|
-
}
|
|
630
|
-
client: options?.client
|
|
646
|
+
}
|
|
631
647
|
});
|
|
632
|
-
this.sql`
|
|
633
|
-
INSERT OR REPLACE INTO cf_agents_mcp_servers (id, name, server_url, client_id, auth_url, callback_url, server_options)
|
|
634
|
-
VALUES (
|
|
635
|
-
${id},
|
|
636
|
-
${serverName},
|
|
637
|
-
${url},
|
|
638
|
-
${clientId ?? null},
|
|
639
|
-
${authUrl ?? null},
|
|
640
|
-
${callbackUrl},
|
|
641
|
-
${options ? JSON.stringify(options) : null}
|
|
642
|
-
);
|
|
643
|
-
`;
|
|
644
648
|
return {
|
|
645
|
-
|
|
646
|
-
|
|
649
|
+
authUrl,
|
|
650
|
+
clientId,
|
|
651
|
+
id
|
|
647
652
|
};
|
|
648
653
|
}
|
|
649
654
|
async removeMcpServer(id) {
|
|
@@ -653,17 +658,17 @@ var Agent = class extends Server {
|
|
|
653
658
|
`;
|
|
654
659
|
this.broadcast(
|
|
655
660
|
JSON.stringify({
|
|
656
|
-
|
|
657
|
-
|
|
661
|
+
mcp: this.getMcpServers(),
|
|
662
|
+
type: "cf_agent_mcp_servers"
|
|
658
663
|
})
|
|
659
664
|
);
|
|
660
665
|
}
|
|
661
666
|
getMcpServers() {
|
|
662
667
|
const mcpState = {
|
|
663
|
-
servers: {},
|
|
664
|
-
tools: this.mcp.listTools(),
|
|
665
668
|
prompts: this.mcp.listPrompts(),
|
|
666
|
-
resources: this.mcp.listResources()
|
|
669
|
+
resources: this.mcp.listResources(),
|
|
670
|
+
servers: {},
|
|
671
|
+
tools: this.mcp.listTools()
|
|
667
672
|
};
|
|
668
673
|
const servers = this.sql`
|
|
669
674
|
SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;
|
|
@@ -671,13 +676,13 @@ var Agent = class extends Server {
|
|
|
671
676
|
for (const server of servers) {
|
|
672
677
|
const serverConn = this.mcp.mcpConnections[server.id];
|
|
673
678
|
mcpState.servers[server.id] = {
|
|
679
|
+
auth_url: server.auth_url,
|
|
680
|
+
capabilities: serverConn?.serverCapabilities ?? null,
|
|
681
|
+
instructions: serverConn?.instructions ?? null,
|
|
674
682
|
name: server.name,
|
|
675
683
|
server_url: server.server_url,
|
|
676
|
-
auth_url: server.auth_url,
|
|
677
684
|
// mark as "authenticating" because the server isn't automatically connected, so it's pending authenticating
|
|
678
|
-
state: serverConn?.connectionState ?? "authenticating"
|
|
679
|
-
instructions: serverConn?.instructions ?? null,
|
|
680
|
-
capabilities: serverConn?.serverCapabilities ?? null
|
|
685
|
+
state: serverConn?.connectionState ?? "authenticating"
|
|
681
686
|
};
|
|
682
687
|
}
|
|
683
688
|
return mcpState;
|
|
@@ -693,9 +698,9 @@ Agent.options = {
|
|
|
693
698
|
};
|
|
694
699
|
async function routeAgentRequest(request, env, options) {
|
|
695
700
|
const corsHeaders = options?.cors === true ? {
|
|
696
|
-
"Access-Control-Allow-Origin": "*",
|
|
697
|
-
"Access-Control-Allow-Methods": "GET, POST, HEAD, OPTIONS",
|
|
698
701
|
"Access-Control-Allow-Credentials": "true",
|
|
702
|
+
"Access-Control-Allow-Methods": "GET, POST, HEAD, OPTIONS",
|
|
703
|
+
"Access-Control-Allow-Origin": "*",
|
|
699
704
|
"Access-Control-Max-Age": "86400"
|
|
700
705
|
} : options?.cors;
|
|
701
706
|
if (request.method === "OPTIONS") {
|
|
@@ -726,7 +731,7 @@ async function routeAgentRequest(request, env, options) {
|
|
|
726
731
|
}
|
|
727
732
|
return response;
|
|
728
733
|
}
|
|
729
|
-
async function routeAgentEmail(
|
|
734
|
+
async function routeAgentEmail(_email, _env, _options) {
|
|
730
735
|
}
|
|
731
736
|
async function getAgentByName(namespace, name, options) {
|
|
732
737
|
return getServerByName(namespace, name, options);
|
|
@@ -746,11 +751,11 @@ var StreamingResponse = class {
|
|
|
746
751
|
throw new Error("StreamingResponse is already closed");
|
|
747
752
|
}
|
|
748
753
|
const response = {
|
|
749
|
-
|
|
754
|
+
done: false,
|
|
750
755
|
id: this._id,
|
|
751
|
-
success: true,
|
|
752
756
|
result: chunk,
|
|
753
|
-
|
|
757
|
+
success: true,
|
|
758
|
+
type: "rpc"
|
|
754
759
|
};
|
|
755
760
|
this._connection.send(JSON.stringify(response));
|
|
756
761
|
}
|
|
@@ -764,11 +769,11 @@ var StreamingResponse = class {
|
|
|
764
769
|
}
|
|
765
770
|
this._closed = true;
|
|
766
771
|
const response = {
|
|
767
|
-
|
|
772
|
+
done: true,
|
|
768
773
|
id: this._id,
|
|
769
|
-
success: true,
|
|
770
774
|
result: finalChunk,
|
|
771
|
-
|
|
775
|
+
success: true,
|
|
776
|
+
type: "rpc"
|
|
772
777
|
};
|
|
773
778
|
this._connection.send(JSON.stringify(response));
|
|
774
779
|
}
|
|
@@ -783,4 +788,4 @@ export {
|
|
|
783
788
|
getAgentByName,
|
|
784
789
|
StreamingResponse
|
|
785
790
|
};
|
|
786
|
-
//# sourceMappingURL=chunk-
|
|
791
|
+
//# sourceMappingURL=chunk-4CIGD73X.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport type { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport type { SSEClientTransportOptions } from \"@modelcontextprotocol/sdk/client/sse.js\";\n\nimport type {\n Prompt,\n Resource,\n ServerCapabilities,\n Tool,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport { parseCronExpression } from \"cron-schedule\";\nimport { nanoid } from \"nanoid\";\nimport {\n type Connection,\n type ConnectionContext,\n getServerByName,\n type PartyServerOptions,\n routePartykitRequest,\n Server,\n type WSMessage,\n} from \"partyserver\";\nimport { camelCaseToKebabCase } from \"./client\";\nimport { MCPClientManager } from \"./mcp/client\";\n// import type { MCPClientConnection } from \"./mcp/client-connection\";\nimport { DurableObjectOAuthClientProvider } from \"./mcp/do-oauth-client-provider\";\n\nexport type { Connection, ConnectionContext, WSMessage } from \"partyserver\";\n\n/**\n * RPC request message from client\n */\nexport type RPCRequest = {\n type: \"rpc\";\n id: string;\n method: string;\n args: unknown[];\n};\n\n/**\n * State update message from client\n */\nexport type StateUpdateMessage = {\n type: \"cf_agent_state\";\n state: unknown;\n};\n\n/**\n * RPC response message to client\n */\nexport type RPCResponse = {\n type: \"rpc\";\n id: string;\n} & (\n | {\n success: true;\n result: unknown;\n done?: false;\n }\n | {\n success: true;\n result: unknown;\n done: true;\n }\n | {\n success: false;\n error: string;\n }\n);\n\n/**\n * Type guard for RPC request messages\n */\nfunction isRPCRequest(msg: unknown): msg is RPCRequest {\n return (\n typeof msg === \"object\" &&\n msg !== null &&\n \"type\" in msg &&\n msg.type === \"rpc\" &&\n \"id\" in msg &&\n typeof msg.id === \"string\" &&\n \"method\" in msg &&\n typeof msg.method === \"string\" &&\n \"args\" in msg &&\n Array.isArray((msg as RPCRequest).args)\n );\n}\n\n/**\n * Type guard for state update messages\n */\nfunction isStateUpdateMessage(msg: unknown): msg is StateUpdateMessage {\n return (\n typeof msg === \"object\" &&\n msg !== null &&\n \"type\" in msg &&\n msg.type === \"cf_agent_state\" &&\n \"state\" in msg\n );\n}\n\n/**\n * Metadata for a callable method\n */\nexport type CallableMetadata = {\n /** Optional description of what the method does */\n description?: string;\n /** Whether the method supports streaming responses */\n streaming?: boolean;\n};\n\nconst callableMetadata = new Map<Function, CallableMetadata>();\n\n/**\n * Decorator that marks a method as callable by clients\n * @param metadata Optional metadata about the callable method\n */\nexport function unstable_callable(metadata: CallableMetadata = {}) {\n return function callableDecorator<This, Args extends unknown[], Return>(\n target: (this: This, ...args: Args) => Return,\n // biome-ignore lint/correctness/noUnusedFunctionParameters: later\n context: ClassMethodDecoratorContext\n ) {\n if (!callableMetadata.has(target)) {\n callableMetadata.set(target, metadata);\n }\n\n return target;\n };\n}\n\n/**\n * Represents a scheduled task within an Agent\n * @template T Type of the payload data\n */\nexport type Schedule<T = string> = {\n /** Unique identifier for the schedule */\n id: string;\n /** Name of the method to be called */\n callback: string;\n /** Data to be passed to the callback */\n payload: T;\n} & (\n | {\n /** Type of schedule for one-time execution at a specific time */\n type: \"scheduled\";\n /** Timestamp when the task should execute */\n time: number;\n }\n | {\n /** Type of schedule for delayed execution */\n type: \"delayed\";\n /** Timestamp when the task should execute */\n time: number;\n /** Number of seconds to delay execution */\n delayInSeconds: number;\n }\n | {\n /** Type of schedule for recurring execution based on cron expression */\n type: \"cron\";\n /** Timestamp for the next execution */\n time: number;\n /** Cron expression defining the schedule */\n cron: string;\n }\n);\n\nfunction getNextCronTime(cron: string) {\n const interval = parseCronExpression(cron);\n return interval.getNextDate();\n}\n\n/**\n * MCP Server state update message from server -> Client\n */\nexport type MCPServerMessage = {\n type: \"cf_agent_mcp_servers\";\n mcp: MCPServersState;\n};\n\nexport type MCPServersState = {\n servers: {\n [id: string]: MCPServer;\n };\n tools: Tool[];\n prompts: Prompt[];\n resources: Resource[];\n};\n\nexport type MCPServer = {\n name: string;\n server_url: string;\n auth_url: string | null;\n // This state is specifically about the temporary process of getting a token (if needed).\n // Scope outside of that can't be relied upon because when the DO sleeps, there's no way\n // to communicate a change to a non-ready state.\n state: \"authenticating\" | \"connecting\" | \"ready\" | \"discovering\" | \"failed\";\n instructions: string | null;\n capabilities: ServerCapabilities | null;\n};\n\n/**\n * MCP Server data stored in DO SQL for resuming MCP Server connections\n */\ntype MCPServerRow = {\n id: string;\n name: string;\n server_url: string;\n client_id: string | null;\n auth_url: string | null;\n callback_url: string;\n server_options: string;\n};\n\nconst STATE_ROW_ID = \"cf_state_row_id\";\nconst STATE_WAS_CHANGED = \"cf_state_was_changed\";\n\nconst DEFAULT_STATE = {} as unknown;\n\nconst agentContext = new AsyncLocalStorage<{\n agent: Agent<unknown>;\n connection: Connection | undefined;\n request: Request | undefined;\n}>();\n\nexport function getCurrentAgent<\n T extends Agent<unknown, unknown> = Agent<unknown, unknown>,\n>(): {\n agent: T | undefined;\n connection: Connection | undefined;\n request: Request<unknown, CfProperties<unknown>> | undefined;\n} {\n const store = agentContext.getStore() as\n | {\n agent: T;\n connection: Connection | undefined;\n request: Request<unknown, CfProperties<unknown>> | undefined;\n }\n | undefined;\n if (!store) {\n return {\n agent: undefined,\n connection: undefined,\n request: undefined,\n };\n }\n return store;\n}\n\n/**\n * Base class for creating Agent implementations\n * @template Env Environment type containing bindings\n * @template State State type to store within the Agent\n */\nexport class Agent<Env, State = unknown> extends Server<Env> {\n private _state = DEFAULT_STATE as State;\n\n private _ParentClass: typeof Agent<Env, State> =\n Object.getPrototypeOf(this).constructor;\n\n mcp: MCPClientManager = new MCPClientManager(this._ParentClass.name, \"0.0.1\");\n\n /**\n * Initial state for the Agent\n * Override to provide default state values\n */\n initialState: State = DEFAULT_STATE as State;\n\n /**\n * Current state of the Agent\n */\n get state(): State {\n if (this._state !== DEFAULT_STATE) {\n // state was previously set, and populated internal state\n return this._state;\n }\n // looks like this is the first time the state is being accessed\n // check if the state was set in a previous life\n const wasChanged = this.sql<{ state: \"true\" | undefined }>`\n SELECT state FROM cf_agents_state WHERE id = ${STATE_WAS_CHANGED}\n `;\n\n // ok, let's pick up the actual state from the db\n const result = this.sql<{ state: State | undefined }>`\n SELECT state FROM cf_agents_state WHERE id = ${STATE_ROW_ID}\n `;\n\n if (\n wasChanged[0]?.state === \"true\" ||\n // we do this check for people who updated their code before we shipped wasChanged\n result[0]?.state\n ) {\n const state = result[0]?.state as string; // could be null?\n\n this._state = JSON.parse(state);\n return this._state;\n }\n\n // ok, this is the first time the state is being accessed\n // and the state was not set in a previous life\n // so we need to set the initial state (if provided)\n if (this.initialState === DEFAULT_STATE) {\n // no initial state provided, so we return undefined\n return undefined as State;\n }\n // initial state provided, so we set the state,\n // update db and return the initial state\n this.setState(this.initialState);\n return this.initialState;\n }\n\n /**\n * Agent configuration options\n */\n static options = {\n /** Whether the Agent should hibernate when inactive */\n hibernate: true, // default to hibernate\n };\n\n /**\n * Execute SQL queries against the Agent's database\n * @template T Type of the returned rows\n * @param strings SQL query template strings\n * @param values Values to be inserted into the query\n * @returns Array of query results\n */\n sql<T = Record<string, string | number | boolean | null>>(\n strings: TemplateStringsArray,\n ...values: (string | number | boolean | null)[]\n ) {\n let query = \"\";\n try {\n // Construct the SQL query with placeholders\n query = strings.reduce(\n (acc, str, i) => acc + str + (i < values.length ? \"?\" : \"\"),\n \"\"\n );\n\n // Execute the SQL query with the provided values\n return [...this.ctx.storage.sql.exec(query, ...values)] as T[];\n } catch (e) {\n console.error(`failed to execute sql query: ${query}`, e);\n throw this.onError(e);\n }\n }\n constructor(ctx: AgentContext, env: Env) {\n super(ctx, env);\n\n this.sql`\n CREATE TABLE IF NOT EXISTS cf_agents_state (\n id TEXT PRIMARY KEY NOT NULL,\n state TEXT\n )\n `;\n\n void this.ctx.blockConcurrencyWhile(async () => {\n return this._tryCatch(async () => {\n // Create alarms table if it doesn't exist\n this.sql`\n CREATE TABLE IF NOT EXISTS cf_agents_schedules (\n id TEXT PRIMARY KEY NOT NULL DEFAULT (randomblob(9)),\n callback TEXT,\n payload TEXT,\n type TEXT NOT NULL CHECK(type IN ('scheduled', 'delayed', 'cron')),\n time INTEGER,\n delayInSeconds INTEGER,\n cron TEXT,\n created_at INTEGER DEFAULT (unixepoch())\n )\n `;\n\n // execute any pending alarms and schedule the next alarm\n await this.alarm();\n });\n });\n\n this.sql`\n CREATE TABLE IF NOT EXISTS cf_agents_mcp_servers (\n id TEXT PRIMARY KEY NOT NULL,\n name TEXT NOT NULL,\n server_url TEXT NOT NULL,\n callback_url TEXT NOT NULL,\n client_id TEXT,\n auth_url TEXT,\n server_options TEXT\n )\n `;\n\n const _onRequest = this.onRequest.bind(this);\n this.onRequest = (request: Request) => {\n return agentContext.run(\n { agent: this, connection: undefined, request },\n async () => {\n if (this.mcp.isCallbackRequest(request)) {\n await this.mcp.handleCallbackRequest(request);\n\n // after the MCP connection handshake, we can send updated mcp state\n this.broadcast(\n JSON.stringify({\n mcp: this.getMcpServers(),\n type: \"cf_agent_mcp_servers\",\n })\n );\n\n // We probably should let the user configure this response/redirect, but this is fine for now.\n return new Response(\"<script>window.close();</script>\", {\n headers: { \"content-type\": \"text/html\" },\n status: 200,\n });\n }\n\n return this._tryCatch(() => _onRequest(request));\n }\n );\n };\n\n const _onMessage = this.onMessage.bind(this);\n this.onMessage = async (connection: Connection, message: WSMessage) => {\n return agentContext.run(\n { agent: this, connection, request: undefined },\n async () => {\n if (typeof message !== \"string\") {\n return this._tryCatch(() => _onMessage(connection, message));\n }\n\n let parsed: unknown;\n try {\n parsed = JSON.parse(message);\n } catch (_e) {\n // silently fail and let the onMessage handler handle it\n return this._tryCatch(() => _onMessage(connection, message));\n }\n\n if (isStateUpdateMessage(parsed)) {\n this._setStateInternal(parsed.state as State, connection);\n return;\n }\n\n if (isRPCRequest(parsed)) {\n try {\n const { id, method, args } = parsed;\n\n // Check if method exists and is callable\n const methodFn = this[method as keyof this];\n if (typeof methodFn !== \"function\") {\n throw new Error(`Method ${method} does not exist`);\n }\n\n if (!this._isCallable(method)) {\n throw new Error(`Method ${method} is not callable`);\n }\n\n const metadata = callableMetadata.get(methodFn as Function);\n\n // For streaming methods, pass a StreamingResponse object\n if (metadata?.streaming) {\n const stream = new StreamingResponse(connection, id);\n await methodFn.apply(this, [stream, ...args]);\n return;\n }\n\n // For regular methods, execute and send response\n const result = await methodFn.apply(this, args);\n const response: RPCResponse = {\n done: true,\n id,\n result,\n success: true,\n type: \"rpc\",\n };\n connection.send(JSON.stringify(response));\n } catch (e) {\n // Send error response\n const response: RPCResponse = {\n error:\n e instanceof Error ? e.message : \"Unknown error occurred\",\n id: parsed.id,\n success: false,\n type: \"rpc\",\n };\n connection.send(JSON.stringify(response));\n console.error(\"RPC error:\", e);\n }\n return;\n }\n\n return this._tryCatch(() => _onMessage(connection, message));\n }\n );\n };\n\n const _onConnect = this.onConnect.bind(this);\n this.onConnect = (connection: Connection, ctx: ConnectionContext) => {\n // TODO: This is a hack to ensure the state is sent after the connection is established\n // must fix this\n return agentContext.run(\n { agent: this, connection, request: ctx.request },\n async () => {\n setTimeout(() => {\n if (this.state) {\n connection.send(\n JSON.stringify({\n state: this.state,\n type: \"cf_agent_state\",\n })\n );\n }\n\n connection.send(\n JSON.stringify({\n mcp: this.getMcpServers(),\n type: \"cf_agent_mcp_servers\",\n })\n );\n\n return this._tryCatch(() => _onConnect(connection, ctx));\n }, 20);\n }\n );\n };\n\n const _onStart = this.onStart.bind(this);\n this.onStart = async () => {\n return agentContext.run(\n { agent: this, connection: undefined, request: undefined },\n async () => {\n const servers = this.sql<MCPServerRow>`\n SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;\n `;\n\n // from DO storage, reconnect to all servers not currently in the oauth flow using our saved auth information\n Promise.allSettled(\n servers.map((server) => {\n return this._connectToMcpServerInternal(\n server.name,\n server.server_url,\n server.callback_url,\n server.server_options\n ? JSON.parse(server.server_options)\n : undefined,\n {\n id: server.id,\n oauthClientId: server.client_id ?? undefined,\n }\n );\n })\n ).then((_results) => {\n this.broadcast(\n JSON.stringify({\n mcp: this.getMcpServers(),\n type: \"cf_agent_mcp_servers\",\n })\n );\n });\n await this._tryCatch(() => _onStart());\n }\n );\n };\n }\n\n private _setStateInternal(\n state: State,\n source: Connection | \"server\" = \"server\"\n ) {\n this._state = state;\n this.sql`\n INSERT OR REPLACE INTO cf_agents_state (id, state)\n VALUES (${STATE_ROW_ID}, ${JSON.stringify(state)})\n `;\n this.sql`\n INSERT OR REPLACE INTO cf_agents_state (id, state)\n VALUES (${STATE_WAS_CHANGED}, ${JSON.stringify(true)})\n `;\n this.broadcast(\n JSON.stringify({\n state: state,\n type: \"cf_agent_state\",\n }),\n source !== \"server\" ? [source.id] : []\n );\n return this._tryCatch(() => {\n const { connection, request } = agentContext.getStore() || {};\n return agentContext.run(\n { agent: this, connection, request },\n async () => {\n return this.onStateUpdate(state, source);\n }\n );\n });\n }\n\n /**\n * Update the Agent's state\n * @param state New state to set\n */\n setState(state: State) {\n this._setStateInternal(state, \"server\");\n }\n\n /**\n * Called when the Agent's state is updated\n * @param state Updated state\n * @param source Source of the state update (\"server\" or a client connection)\n */\n // biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later\n onStateUpdate(state: State | undefined, source: Connection | \"server\") {\n // override this to handle state updates\n }\n\n /**\n * Called when the Agent receives an email\n * @param email Email message to process\n */\n // biome-ignore lint/correctness/noUnusedFunctionParameters: overridden later\n onEmail(email: ForwardableEmailMessage) {\n return agentContext.run(\n { agent: this, connection: undefined, request: undefined },\n async () => {\n console.error(\"onEmail not implemented\");\n }\n );\n }\n\n private async _tryCatch<T>(fn: () => T | Promise<T>) {\n try {\n return await fn();\n } catch (e) {\n throw this.onError(e);\n }\n }\n\n override onError(\n connection: Connection,\n error: unknown\n ): void | Promise<void>;\n override onError(error: unknown): void | Promise<void>;\n override onError(connectionOrError: Connection | unknown, error?: unknown) {\n let theError: unknown;\n if (connectionOrError && error) {\n theError = error;\n // this is a websocket connection error\n console.error(\n \"Error on websocket connection:\",\n (connectionOrError as Connection).id,\n theError\n );\n console.error(\n \"Override onError(connection, error) to handle websocket connection errors\"\n );\n } else {\n theError = connectionOrError;\n // this is a server error\n console.error(\"Error on server:\", theError);\n console.error(\"Override onError(error) to handle server errors\");\n }\n throw theError;\n }\n\n /**\n * Render content (not implemented in base class)\n */\n render() {\n throw new Error(\"Not implemented\");\n }\n\n /**\n * Schedule a task to be executed in the future\n * @template T Type of the payload data\n * @param when When to execute the task (Date, seconds delay, or cron expression)\n * @param callback Name of the method to call\n * @param payload Data to pass to the callback\n * @returns Schedule object representing the scheduled task\n */\n async schedule<T = string>(\n when: Date | string | number,\n callback: keyof this,\n payload?: T\n ): Promise<Schedule<T>> {\n const id = nanoid(9);\n\n if (typeof callback !== \"string\") {\n throw new Error(\"Callback must be a string\");\n }\n\n if (typeof this[callback] !== \"function\") {\n throw new Error(`this.${callback} is not a function`);\n }\n\n if (when instanceof Date) {\n const timestamp = Math.floor(when.getTime() / 1000);\n this.sql`\n INSERT OR REPLACE INTO cf_agents_schedules (id, callback, payload, type, time)\n VALUES (${id}, ${callback}, ${JSON.stringify(\n payload\n )}, 'scheduled', ${timestamp})\n `;\n\n await this._scheduleNextAlarm();\n\n return {\n callback: callback,\n id,\n payload: payload as T,\n time: timestamp,\n type: \"scheduled\",\n };\n }\n if (typeof when === \"number\") {\n const time = new Date(Date.now() + when * 1000);\n const timestamp = Math.floor(time.getTime() / 1000);\n\n this.sql`\n INSERT OR REPLACE INTO cf_agents_schedules (id, callback, payload, type, delayInSeconds, time)\n VALUES (${id}, ${callback}, ${JSON.stringify(\n payload\n )}, 'delayed', ${when}, ${timestamp})\n `;\n\n await this._scheduleNextAlarm();\n\n return {\n callback: callback,\n delayInSeconds: when,\n id,\n payload: payload as T,\n time: timestamp,\n type: \"delayed\",\n };\n }\n if (typeof when === \"string\") {\n const nextExecutionTime = getNextCronTime(when);\n const timestamp = Math.floor(nextExecutionTime.getTime() / 1000);\n\n this.sql`\n INSERT OR REPLACE INTO cf_agents_schedules (id, callback, payload, type, cron, time)\n VALUES (${id}, ${callback}, ${JSON.stringify(\n payload\n )}, 'cron', ${when}, ${timestamp})\n `;\n\n await this._scheduleNextAlarm();\n\n return {\n callback: callback,\n cron: when,\n id,\n payload: payload as T,\n time: timestamp,\n type: \"cron\",\n };\n }\n throw new Error(\"Invalid schedule type\");\n }\n\n /**\n * Get a scheduled task by ID\n * @template T Type of the payload data\n * @param id ID of the scheduled task\n * @returns The Schedule object or undefined if not found\n */\n async getSchedule<T = string>(id: string): Promise<Schedule<T> | undefined> {\n const result = this.sql<Schedule<string>>`\n SELECT * FROM cf_agents_schedules WHERE id = ${id}\n `;\n if (!result) {\n console.error(`schedule ${id} not found`);\n return undefined;\n }\n\n return { ...result[0], payload: JSON.parse(result[0].payload) as T };\n }\n\n /**\n * Get scheduled tasks matching the given criteria\n * @template T Type of the payload data\n * @param criteria Criteria to filter schedules\n * @returns Array of matching Schedule objects\n */\n getSchedules<T = string>(\n criteria: {\n id?: string;\n type?: \"scheduled\" | \"delayed\" | \"cron\";\n timeRange?: { start?: Date; end?: Date };\n } = {}\n ): Schedule<T>[] {\n let query = \"SELECT * FROM cf_agents_schedules WHERE 1=1\";\n const params = [];\n\n if (criteria.id) {\n query += \" AND id = ?\";\n params.push(criteria.id);\n }\n\n if (criteria.type) {\n query += \" AND type = ?\";\n params.push(criteria.type);\n }\n\n if (criteria.timeRange) {\n query += \" AND time >= ? AND time <= ?\";\n const start = criteria.timeRange.start || new Date(0);\n const end = criteria.timeRange.end || new Date(999999999999999);\n params.push(\n Math.floor(start.getTime() / 1000),\n Math.floor(end.getTime() / 1000)\n );\n }\n\n const result = this.ctx.storage.sql\n .exec(query, ...params)\n .toArray()\n .map((row) => ({\n ...row,\n payload: JSON.parse(row.payload as string) as T,\n })) as Schedule<T>[];\n\n return result;\n }\n\n /**\n * Cancel a scheduled task\n * @param id ID of the task to cancel\n * @returns true if the task was cancelled, false otherwise\n */\n async cancelSchedule(id: string): Promise<boolean> {\n this.sql`DELETE FROM cf_agents_schedules WHERE id = ${id}`;\n\n await this._scheduleNextAlarm();\n return true;\n }\n\n private async _scheduleNextAlarm() {\n // Find the next schedule that needs to be executed\n const result = this.sql`\n SELECT time FROM cf_agents_schedules \n WHERE time > ${Math.floor(Date.now() / 1000)}\n ORDER BY time ASC \n LIMIT 1\n `;\n if (!result) return;\n\n if (result.length > 0 && \"time\" in result[0]) {\n const nextTime = (result[0].time as number) * 1000;\n await this.ctx.storage.setAlarm(nextTime);\n }\n }\n\n /**\n * Method called when an alarm fires.\n * Executes any scheduled tasks that are due.\n *\n * @remarks\n * To schedule a task, please use the `this.schedule` method instead.\n * See {@link https://developers.cloudflare.com/agents/api-reference/schedule-tasks/}\n */\n public readonly alarm = async () => {\n const now = Math.floor(Date.now() / 1000);\n\n // Get all schedules that should be executed now\n const result = this.sql<Schedule<string>>`\n SELECT * FROM cf_agents_schedules WHERE time <= ${now}\n `;\n\n for (const row of result || []) {\n const callback = this[row.callback as keyof Agent<Env>];\n if (!callback) {\n console.error(`callback ${row.callback} not found`);\n continue;\n }\n await agentContext.run(\n { agent: this, connection: undefined, request: undefined },\n async () => {\n try {\n await (\n callback as (\n payload: unknown,\n schedule: Schedule<unknown>\n ) => Promise<void>\n ).bind(this)(JSON.parse(row.payload as string), row);\n } catch (e) {\n console.error(`error executing callback \"${row.callback}\"`, e);\n }\n }\n );\n if (row.type === \"cron\") {\n // Update next execution time for cron schedules\n const nextExecutionTime = getNextCronTime(row.cron);\n const nextTimestamp = Math.floor(nextExecutionTime.getTime() / 1000);\n\n this.sql`\n UPDATE cf_agents_schedules SET time = ${nextTimestamp} WHERE id = ${row.id}\n `;\n } else {\n // Delete one-time schedules after execution\n this.sql`\n DELETE FROM cf_agents_schedules WHERE id = ${row.id}\n `;\n }\n }\n\n // Schedule the next alarm\n await this._scheduleNextAlarm();\n };\n\n /**\n * Destroy the Agent, removing all state and scheduled tasks\n */\n async destroy() {\n // drop all tables\n this.sql`DROP TABLE IF EXISTS cf_agents_state`;\n this.sql`DROP TABLE IF EXISTS cf_agents_schedules`;\n this.sql`DROP TABLE IF EXISTS cf_agents_mcp_servers`;\n\n // delete all alarms\n await this.ctx.storage.deleteAlarm();\n await this.ctx.storage.deleteAll();\n this.ctx.abort(\"destroyed\"); // enforce that the agent is evicted\n }\n\n /**\n * Get all methods marked as callable on this Agent\n * @returns A map of method names to their metadata\n */\n private _isCallable(method: string): boolean {\n return callableMetadata.has(this[method as keyof this] as Function);\n }\n\n /**\n * Connect to a new MCP Server\n *\n * @param url MCP Server SSE URL\n * @param callbackHost Base host for the agent, used for the redirect URI.\n * @param agentsPrefix agents routing prefix if not using `agents`\n * @param options MCP client and transport (header) options\n * @returns authUrl\n */\n async addMcpServer(\n serverName: string,\n url: string,\n callbackHost: string,\n agentsPrefix = \"agents\",\n options?: {\n client?: ConstructorParameters<typeof Client>[1];\n transport?: {\n headers: HeadersInit;\n };\n }\n ): Promise<{ id: string; authUrl: string | undefined }> {\n const callbackUrl = `${callbackHost}/${agentsPrefix}/${camelCaseToKebabCase(this._ParentClass.name)}/${this.name}/callback`;\n\n const result = await this._connectToMcpServerInternal(\n serverName,\n url,\n callbackUrl,\n options\n );\n this.sql`\n INSERT\n OR REPLACE INTO cf_agents_mcp_servers (id, name, server_url, client_id, auth_url, callback_url, server_options)\n VALUES (\n ${result.id},\n ${serverName},\n ${url},\n ${result.clientId ?? null},\n ${result.authUrl ?? null},\n ${callbackUrl},\n ${options ? JSON.stringify(options) : null}\n );\n `;\n\n this.broadcast(\n JSON.stringify({\n mcp: this.getMcpServers(),\n type: \"cf_agent_mcp_servers\",\n })\n );\n\n return result;\n }\n\n async _connectToMcpServerInternal(\n _serverName: string,\n url: string,\n callbackUrl: string,\n // it's important that any options here are serializable because we put them into our sqlite DB for reconnection purposes\n options?: {\n client?: ConstructorParameters<typeof Client>[1];\n /**\n * We don't expose the normal set of transport options because:\n * 1) we can't serialize things like the auth provider or a fetch function into the DB for reconnection purposes\n * 2) We probably want these options to be agnostic to the transport type (SSE vs Streamable)\n *\n * This has the limitation that you can't override fetch, but I think headers should handle nearly all cases needed (i.e. non-standard bearer auth).\n */\n transport?: {\n headers?: HeadersInit;\n };\n },\n reconnect?: {\n id: string;\n oauthClientId?: string;\n }\n ): Promise<{\n id: string;\n authUrl: string | undefined;\n clientId: string | undefined;\n }> {\n const authProvider = new DurableObjectOAuthClientProvider(\n this.ctx.storage,\n this.name,\n callbackUrl\n );\n\n if (reconnect) {\n authProvider.serverId = reconnect.id;\n if (reconnect.oauthClientId) {\n authProvider.clientId = reconnect.oauthClientId;\n }\n }\n\n // allows passing through transport headers if necessary\n // this handles some non-standard bearer auth setups (i.e. MCP server behind CF access instead of OAuth)\n let headerTransportOpts: SSEClientTransportOptions = {};\n if (options?.transport?.headers) {\n headerTransportOpts = {\n eventSourceInit: {\n fetch: (url, init) =>\n fetch(url, {\n ...init,\n headers: options?.transport?.headers,\n }),\n },\n requestInit: {\n headers: options?.transport?.headers,\n },\n };\n }\n\n const { id, authUrl, clientId } = await this.mcp.connect(url, {\n client: options?.client,\n reconnect,\n transport: {\n ...headerTransportOpts,\n authProvider,\n },\n });\n\n return {\n authUrl,\n clientId,\n id,\n };\n }\n\n async removeMcpServer(id: string) {\n this.mcp.closeConnection(id);\n this.sql`\n DELETE FROM cf_agents_mcp_servers WHERE id = ${id};\n `;\n this.broadcast(\n JSON.stringify({\n mcp: this.getMcpServers(),\n type: \"cf_agent_mcp_servers\",\n })\n );\n }\n\n getMcpServers(): MCPServersState {\n const mcpState: MCPServersState = {\n prompts: this.mcp.listPrompts(),\n resources: this.mcp.listResources(),\n servers: {},\n tools: this.mcp.listTools(),\n };\n\n const servers = this.sql<MCPServerRow>`\n SELECT id, name, server_url, client_id, auth_url, callback_url, server_options FROM cf_agents_mcp_servers;\n `;\n\n for (const server of servers) {\n const serverConn = this.mcp.mcpConnections[server.id];\n mcpState.servers[server.id] = {\n auth_url: server.auth_url,\n capabilities: serverConn?.serverCapabilities ?? null,\n instructions: serverConn?.instructions ?? null,\n name: server.name,\n server_url: server.server_url,\n // mark as \"authenticating\" because the server isn't automatically connected, so it's pending authenticating\n state: serverConn?.connectionState ?? \"authenticating\",\n };\n }\n\n return mcpState;\n }\n}\n\n/**\n * Namespace for creating Agent instances\n * @template Agentic Type of the Agent class\n */\nexport type AgentNamespace<Agentic extends Agent<unknown>> =\n DurableObjectNamespace<Agentic>;\n\n/**\n * Agent's durable context\n */\nexport type AgentContext = DurableObjectState;\n\n/**\n * Configuration options for Agent routing\n */\nexport type AgentOptions<Env> = PartyServerOptions<Env> & {\n /**\n * Whether to enable CORS for the Agent\n */\n cors?: boolean | HeadersInit | undefined;\n};\n\n/**\n * Route a request to the appropriate Agent\n * @param request Request to route\n * @param env Environment containing Agent bindings\n * @param options Routing options\n * @returns Response from the Agent or undefined if no route matched\n */\nexport async function routeAgentRequest<Env>(\n request: Request,\n env: Env,\n options?: AgentOptions<Env>\n) {\n const corsHeaders =\n options?.cors === true\n ? {\n \"Access-Control-Allow-Credentials\": \"true\",\n \"Access-Control-Allow-Methods\": \"GET, POST, HEAD, OPTIONS\",\n \"Access-Control-Allow-Origin\": \"*\",\n \"Access-Control-Max-Age\": \"86400\",\n }\n : options?.cors;\n\n if (request.method === \"OPTIONS\") {\n if (corsHeaders) {\n return new Response(null, {\n headers: corsHeaders,\n });\n }\n console.warn(\n \"Received an OPTIONS request, but cors was not enabled. Pass `cors: true` or `cors: { ...custom cors headers }` to routeAgentRequest to enable CORS.\"\n );\n }\n\n let response = await routePartykitRequest(\n request,\n env as Record<string, unknown>,\n {\n prefix: \"agents\",\n ...(options as PartyServerOptions<Record<string, unknown>>),\n }\n );\n\n if (\n response &&\n corsHeaders &&\n request.headers.get(\"upgrade\")?.toLowerCase() !== \"websocket\" &&\n request.headers.get(\"Upgrade\")?.toLowerCase() !== \"websocket\"\n ) {\n response = new Response(response.body, {\n headers: {\n ...response.headers,\n ...corsHeaders,\n },\n });\n }\n return response;\n}\n\n/**\n * Route an email to the appropriate Agent\n * @param email Email message to route\n * @param env Environment containing Agent bindings\n * @param options Routing options\n */\nexport async function routeAgentEmail<Env>(\n _email: ForwardableEmailMessage,\n _env: Env,\n _options?: AgentOptions<Env>\n): Promise<void> {}\n\n/**\n * Get or create an Agent by name\n * @template Env Environment type containing bindings\n * @template T Type of the Agent class\n * @param namespace Agent namespace\n * @param name Name of the Agent instance\n * @param options Options for Agent creation\n * @returns Promise resolving to an Agent instance stub\n */\nexport async function getAgentByName<Env, T extends Agent<Env>>(\n namespace: AgentNamespace<T>,\n name: string,\n options?: {\n jurisdiction?: DurableObjectJurisdiction;\n locationHint?: DurableObjectLocationHint;\n }\n) {\n return getServerByName<Env, T>(namespace, name, options);\n}\n\n/**\n * A wrapper for streaming responses in callable methods\n */\nexport class StreamingResponse {\n private _connection: Connection;\n private _id: string;\n private _closed = false;\n\n constructor(connection: Connection, id: string) {\n this._connection = connection;\n this._id = id;\n }\n\n /**\n * Send a chunk of data to the client\n * @param chunk The data to send\n */\n send(chunk: unknown) {\n if (this._closed) {\n throw new Error(\"StreamingResponse is already closed\");\n }\n const response: RPCResponse = {\n done: false,\n id: this._id,\n result: chunk,\n success: true,\n type: \"rpc\",\n };\n this._connection.send(JSON.stringify(response));\n }\n\n /**\n * End the stream and send the final chunk (if any)\n * @param finalChunk Optional final chunk of data to send\n */\n end(finalChunk?: unknown) {\n if (this._closed) {\n throw new Error(\"StreamingResponse is already closed\");\n }\n this._closed = true;\n const response: RPCResponse = {\n done: true,\n id: this._id,\n result: finalChunk,\n success: true,\n type: \"rpc\",\n };\n this._connection.send(JSON.stringify(response));\n }\n}\n"],"mappings":";;;;;;;;;;;AAAA,SAAS,yBAAyB;AAUlC,SAAS,2BAA2B;AACpC,SAAS,cAAc;AACvB;AAAA,EAGE;AAAA,EAEA;AAAA,EACA;AAAA,OAEK;AAoDP,SAAS,aAAa,KAAiC;AACrD,SACE,OAAO,QAAQ,YACf,QAAQ,QACR,UAAU,OACV,IAAI,SAAS,SACb,QAAQ,OACR,OAAO,IAAI,OAAO,YAClB,YAAY,OACZ,OAAO,IAAI,WAAW,YACtB,UAAU,OACV,MAAM,QAAS,IAAmB,IAAI;AAE1C;AAKA,SAAS,qBAAqB,KAAyC;AACrE,SACE,OAAO,QAAQ,YACf,QAAQ,QACR,UAAU,OACV,IAAI,SAAS,oBACb,WAAW;AAEf;AAYA,IAAM,mBAAmB,oBAAI,IAAgC;AAMtD,SAAS,kBAAkB,WAA6B,CAAC,GAAG;AACjE,SAAO,SAAS,kBACd,QAEA,SACA;AACA,QAAI,CAAC,iBAAiB,IAAI,MAAM,GAAG;AACjC,uBAAiB,IAAI,QAAQ,QAAQ;AAAA,IACvC;AAEA,WAAO;AAAA,EACT;AACF;AAsCA,SAAS,gBAAgB,MAAc;AACrC,QAAM,WAAW,oBAAoB,IAAI;AACzC,SAAO,SAAS,YAAY;AAC9B;AA4CA,IAAM,eAAe;AACrB,IAAM,oBAAoB;AAE1B,IAAM,gBAAgB,CAAC;AAEvB,IAAM,eAAe,IAAI,kBAItB;AAEI,SAAS,kBAMd;AACA,QAAM,QAAQ,aAAa,SAAS;AAOpC,MAAI,CAAC,OAAO;AACV,WAAO;AAAA,MACL,OAAO;AAAA,MACP,YAAY;AAAA,MACZ,SAAS;AAAA,IACX;AAAA,EACF;AACA,SAAO;AACT;AAOO,IAAM,QAAN,cAA0C,OAAY;AAAA,EA2F3D,YAAY,KAAmB,KAAU;AACvC,UAAM,KAAK,GAAG;AA3FhB,SAAQ,SAAS;AAEjB,SAAQ,eACN,OAAO,eAAe,IAAI,EAAE;AAE9B,eAAwB,IAAI,iBAAiB,KAAK,aAAa,MAAM,OAAO;AAM5E;AAAA;AAAA;AAAA;AAAA,wBAAsB;AA6kBtB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,SAAgB,QAAQ,YAAY;AAClC,YAAM,MAAM,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI;AAGxC,YAAM,SAAS,KAAK;AAAA,wDACgC,GAAG;AAAA;AAGvD,iBAAW,OAAO,UAAU,CAAC,GAAG;AAC9B,cAAM,WAAW,KAAK,IAAI,QAA4B;AACtD,YAAI,CAAC,UAAU;AACb,kBAAQ,MAAM,YAAY,IAAI,QAAQ,YAAY;AAClD;AAAA,QACF;AACA,cAAM,aAAa;AAAA,UACjB,EAAE,OAAO,MAAM,YAAY,QAAW,SAAS,OAAU;AAAA,UACzD,YAAY;AACV,gBAAI;AACF,oBACE,SAIA,KAAK,IAAI,EAAE,KAAK,MAAM,IAAI,OAAiB,GAAG,GAAG;AAAA,YACrD,SAAS,GAAG;AACV,sBAAQ,MAAM,6BAA6B,IAAI,QAAQ,KAAK,CAAC;AAAA,YAC/D;AAAA,UACF;AAAA,QACF;AACA,YAAI,IAAI,SAAS,QAAQ;AAEvB,gBAAM,oBAAoB,gBAAgB,IAAI,IAAI;AAClD,gBAAM,gBAAgB,KAAK,MAAM,kBAAkB,QAAQ,IAAI,GAAI;AAEnE,eAAK;AAAA,kDACqC,aAAa,eAAe,IAAI,EAAE;AAAA;AAAA,QAE9E,OAAO;AAEL,eAAK;AAAA,uDAC0C,IAAI,EAAE;AAAA;AAAA,QAEvD;AAAA,MACF;AAGA,YAAM,KAAK,mBAAmB;AAAA,IAChC;AA1iBE,SAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAOL,SAAK,KAAK,IAAI,sBAAsB,YAAY;AAC9C,aAAO,KAAK,UAAU,YAAY;AAEhC,aAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAcL,cAAM,KAAK,MAAM;AAAA,MACnB,CAAC;AAAA,IACH,CAAC;AAED,SAAK;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAYL,UAAM,aAAa,KAAK,UAAU,KAAK,IAAI;AAC3C,SAAK,YAAY,CAAC,YAAqB;AACrC,aAAO,aAAa;AAAA,QAClB,EAAE,OAAO,MAAM,YAAY,QAAW,QAAQ;AAAA,QAC9C,YAAY;AACV,cAAI,KAAK,IAAI,kBAAkB,OAAO,GAAG;AACvC,kBAAM,KAAK,IAAI,sBAAsB,OAAO;AAG5C,iBAAK;AAAA,cACH,KAAK,UAAU;AAAA,gBACb,KAAK,KAAK,cAAc;AAAA,gBACxB,MAAM;AAAA,cACR,CAAC;AAAA,YACH;AAGA,mBAAO,IAAI,SAAS,oCAAoC;AAAA,cACtD,SAAS,EAAE,gBAAgB,YAAY;AAAA,cACvC,QAAQ;AAAA,YACV,CAAC;AAAA,UACH;AAEA,iBAAO,KAAK,UAAU,MAAM,WAAW,OAAO,CAAC;AAAA,QACjD;AAAA,MACF;AAAA,IACF;AAEA,UAAM,aAAa,KAAK,UAAU,KAAK,IAAI;AAC3C,SAAK,YAAY,OAAO,YAAwB,YAAuB;AACrE,aAAO,aAAa;AAAA,QAClB,EAAE,OAAO,MAAM,YAAY,SAAS,OAAU;AAAA,QAC9C,YAAY;AACV,cAAI,OAAO,YAAY,UAAU;AAC/B,mBAAO,KAAK,UAAU,MAAM,WAAW,YAAY,OAAO,CAAC;AAAA,UAC7D;AAEA,cAAI;AACJ,cAAI;AACF,qBAAS,KAAK,MAAM,OAAO;AAAA,UAC7B,SAAS,IAAI;AAEX,mBAAO,KAAK,UAAU,MAAM,WAAW,YAAY,OAAO,CAAC;AAAA,UAC7D;AAEA,cAAI,qBAAqB,MAAM,GAAG;AAChC,iBAAK,kBAAkB,OAAO,OAAgB,UAAU;AACxD;AAAA,UACF;AAEA,cAAI,aAAa,MAAM,GAAG;AACxB,gBAAI;AACF,oBAAM,EAAE,IAAI,QAAQ,KAAK,IAAI;AAG7B,oBAAM,WAAW,KAAK,MAAoB;AAC1C,kBAAI,OAAO,aAAa,YAAY;AAClC,sBAAM,IAAI,MAAM,UAAU,MAAM,iBAAiB;AAAA,cACnD;AAEA,kBAAI,CAAC,KAAK,YAAY,MAAM,GAAG;AAC7B,sBAAM,IAAI,MAAM,UAAU,MAAM,kBAAkB;AAAA,cACpD;AAEA,oBAAM,WAAW,iBAAiB,IAAI,QAAoB;AAG1D,kBAAI,UAAU,WAAW;AACvB,sBAAM,SAAS,IAAI,kBAAkB,YAAY,EAAE;AACnD,sBAAM,SAAS,MAAM,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC;AAC5C;AAAA,cACF;AAGA,oBAAM,SAAS,MAAM,SAAS,MAAM,MAAM,IAAI;AAC9C,oBAAM,WAAwB;AAAA,gBAC5B,MAAM;AAAA,gBACN;AAAA,gBACA;AAAA,gBACA,SAAS;AAAA,gBACT,MAAM;AAAA,cACR;AACA,yBAAW,KAAK,KAAK,UAAU,QAAQ,CAAC;AAAA,YAC1C,SAAS,GAAG;AAEV,oBAAM,WAAwB;AAAA,gBAC5B,OACE,aAAa,QAAQ,EAAE,UAAU;AAAA,gBACnC,IAAI,OAAO;AAAA,gBACX,SAAS;AAAA,gBACT,MAAM;AAAA,cACR;AACA,yBAAW,KAAK,KAAK,UAAU,QAAQ,CAAC;AACxC,sBAAQ,MAAM,cAAc,CAAC;AAAA,YAC/B;AACA;AAAA,UACF;AAEA,iBAAO,KAAK,UAAU,MAAM,WAAW,YAAY,OAAO,CAAC;AAAA,QAC7D;AAAA,MACF;AAAA,IACF;AAEA,UAAM,aAAa,KAAK,UAAU,KAAK,IAAI;AAC3C,SAAK,YAAY,CAAC,YAAwBA,SAA2B;AAGnE,aAAO,aAAa;AAAA,QAClB,EAAE,OAAO,MAAM,YAAY,SAASA,KAAI,QAAQ;AAAA,QAChD,YAAY;AACV,qBAAW,MAAM;AACf,gBAAI,KAAK,OAAO;AACd,yBAAW;AAAA,gBACT,KAAK,UAAU;AAAA,kBACb,OAAO,KAAK;AAAA,kBACZ,MAAM;AAAA,gBACR,CAAC;AAAA,cACH;AAAA,YACF;AAEA,uBAAW;AAAA,cACT,KAAK,UAAU;AAAA,gBACb,KAAK,KAAK,cAAc;AAAA,gBACxB,MAAM;AAAA,cACR,CAAC;AAAA,YACH;AAEA,mBAAO,KAAK,UAAU,MAAM,WAAW,YAAYA,IAAG,CAAC;AAAA,UACzD,GAAG,EAAE;AAAA,QACP;AAAA,MACF;AAAA,IACF;AAEA,UAAM,WAAW,KAAK,QAAQ,KAAK,IAAI;AACvC,SAAK,UAAU,YAAY;AACzB,aAAO,aAAa;AAAA,QAClB,EAAE,OAAO,MAAM,YAAY,QAAW,SAAS,OAAU;AAAA,QACzD,YAAY;AACV,gBAAM,UAAU,KAAK;AAAA;AAAA;AAKrB,kBAAQ;AAAA,YACN,QAAQ,IAAI,CAAC,WAAW;AACtB,qBAAO,KAAK;AAAA,gBACV,OAAO;AAAA,gBACP,OAAO;AAAA,gBACP,OAAO;AAAA,gBACP,OAAO,iBACH,KAAK,MAAM,OAAO,cAAc,IAChC;AAAA,gBACJ;AAAA,kBACE,IAAI,OAAO;AAAA,kBACX,eAAe,OAAO,aAAa;AAAA,gBACrC;AAAA,cACF;AAAA,YACF,CAAC;AAAA,UACH,EAAE,KAAK,CAAC,aAAa;AACnB,iBAAK;AAAA,cACH,KAAK,UAAU;AAAA,gBACb,KAAK,KAAK,cAAc;AAAA,gBACxB,MAAM;AAAA,cACR,CAAC;AAAA,YACH;AAAA,UACF,CAAC;AACD,gBAAM,KAAK,UAAU,MAAM,SAAS,CAAC;AAAA,QACvC;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EA/RA,IAAI,QAAe;AACjB,QAAI,KAAK,WAAW,eAAe;AAEjC,aAAO,KAAK;AAAA,IACd;AAGA,UAAM,aAAa,KAAK;AAAA,uDAC2B,iBAAiB;AAAA;AAIpE,UAAM,SAAS,KAAK;AAAA,qDAC6B,YAAY;AAAA;AAG7D,QACE,WAAW,CAAC,GAAG,UAAU;AAAA,IAEzB,OAAO,CAAC,GAAG,OACX;AACA,YAAM,QAAQ,OAAO,CAAC,GAAG;AAEzB,WAAK,SAAS,KAAK,MAAM,KAAK;AAC9B,aAAO,KAAK;AAAA,IACd;AAKA,QAAI,KAAK,iBAAiB,eAAe;AAEvC,aAAO;AAAA,IACT;AAGA,SAAK,SAAS,KAAK,YAAY;AAC/B,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAiBA,IACE,YACG,QACH;AACA,QAAI,QAAQ;AACZ,QAAI;AAEF,cAAQ,QAAQ;AAAA,QACd,CAAC,KAAK,KAAK,MAAM,MAAM,OAAO,IAAI,OAAO,SAAS,MAAM;AAAA,QACxD;AAAA,MACF;AAGA,aAAO,CAAC,GAAG,KAAK,IAAI,QAAQ,IAAI,KAAK,OAAO,GAAG,MAAM,CAAC;AAAA,IACxD,SAAS,GAAG;AACV,cAAQ,MAAM,gCAAgC,KAAK,IAAI,CAAC;AACxD,YAAM,KAAK,QAAQ,CAAC;AAAA,IACtB;AAAA,EACF;AAAA,EAwNQ,kBACN,OACA,SAAgC,UAChC;AACA,SAAK,SAAS;AACd,SAAK;AAAA;AAAA,cAEK,YAAY,KAAK,KAAK,UAAU,KAAK,CAAC;AAAA;AAEhD,SAAK;AAAA;AAAA,cAEK,iBAAiB,KAAK,KAAK,UAAU,IAAI,CAAC;AAAA;AAEpD,SAAK;AAAA,MACH,KAAK,UAAU;AAAA,QACb;AAAA,QACA,MAAM;AAAA,MACR,CAAC;AAAA,MACD,WAAW,WAAW,CAAC,OAAO,EAAE,IAAI,CAAC;AAAA,IACvC;AACA,WAAO,KAAK,UAAU,MAAM;AAC1B,YAAM,EAAE,YAAY,QAAQ,IAAI,aAAa,SAAS,KAAK,CAAC;AAC5D,aAAO,aAAa;AAAA,QAClB,EAAE,OAAO,MAAM,YAAY,QAAQ;AAAA,QACnC,YAAY;AACV,iBAAO,KAAK,cAAc,OAAO,MAAM;AAAA,QACzC;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,SAAS,OAAc;AACrB,SAAK,kBAAkB,OAAO,QAAQ;AAAA,EACxC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,cAAc,OAA0B,QAA+B;AAAA,EAEvE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,QAAQ,OAAgC;AACtC,WAAO,aAAa;AAAA,MAClB,EAAE,OAAO,MAAM,YAAY,QAAW,SAAS,OAAU;AAAA,MACzD,YAAY;AACV,gBAAQ,MAAM,yBAAyB;AAAA,MACzC;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAc,UAAa,IAA0B;AACnD,QAAI;AACF,aAAO,MAAM,GAAG;AAAA,IAClB,SAAS,GAAG;AACV,YAAM,KAAK,QAAQ,CAAC;AAAA,IACtB;AAAA,EACF;AAAA,EAOS,QAAQ,mBAAyC,OAAiB;AACzE,QAAI;AACJ,QAAI,qBAAqB,OAAO;AAC9B,iBAAW;AAEX,cAAQ;AAAA,QACN;AAAA,QACC,kBAAiC;AAAA,QAClC;AAAA,MACF;AACA,cAAQ;AAAA,QACN;AAAA,MACF;AAAA,IACF,OAAO;AACL,iBAAW;AAEX,cAAQ,MAAM,oBAAoB,QAAQ;AAC1C,cAAQ,MAAM,iDAAiD;AAAA,IACjE;AACA,UAAM;AAAA,EACR;AAAA;AAAA;AAAA;AAAA,EAKA,SAAS;AACP,UAAM,IAAI,MAAM,iBAAiB;AAAA,EACnC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,MAAM,SACJ,MACA,UACA,SACsB;AACtB,UAAM,KAAK,OAAO,CAAC;AAEnB,QAAI,OAAO,aAAa,UAAU;AAChC,YAAM,IAAI,MAAM,2BAA2B;AAAA,IAC7C;AAEA,QAAI,OAAO,KAAK,QAAQ,MAAM,YAAY;AACxC,YAAM,IAAI,MAAM,QAAQ,QAAQ,oBAAoB;AAAA,IACtD;AAEA,QAAI,gBAAgB,MAAM;AACxB,YAAM,YAAY,KAAK,MAAM,KAAK,QAAQ,IAAI,GAAI;AAClD,WAAK;AAAA;AAAA,kBAEO,EAAE,KAAK,QAAQ,KAAK,KAAK;AAAA,QACjC;AAAA,MACF,CAAC,kBAAkB,SAAS;AAAA;AAG9B,YAAM,KAAK,mBAAmB;AAE9B,aAAO;AAAA,QACL;AAAA,QACA;AAAA,QACA;AAAA,QACA,MAAM;AAAA,QACN,MAAM;AAAA,MACR;AAAA,IACF;AACA,QAAI,OAAO,SAAS,UAAU;AAC5B,YAAM,OAAO,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,GAAI;AAC9C,YAAM,YAAY,KAAK,MAAM,KAAK,QAAQ,IAAI,GAAI;AAElD,WAAK;AAAA;AAAA,kBAEO,EAAE,KAAK,QAAQ,KAAK,KAAK;AAAA,QACjC;AAAA,MACF,CAAC,gBAAgB,IAAI,KAAK,SAAS;AAAA;AAGrC,YAAM,KAAK,mBAAmB;AAE9B,aAAO;AAAA,QACL;AAAA,QACA,gBAAgB;AAAA,QAChB;AAAA,QACA;AAAA,QACA,MAAM;AAAA,QACN,MAAM;AAAA,MACR;AAAA,IACF;AACA,QAAI,OAAO,SAAS,UAAU;AAC5B,YAAM,oBAAoB,gBAAgB,IAAI;AAC9C,YAAM,YAAY,KAAK,MAAM,kBAAkB,QAAQ,IAAI,GAAI;AAE/D,WAAK;AAAA;AAAA,kBAEO,EAAE,KAAK,QAAQ,KAAK,KAAK;AAAA,QACjC;AAAA,MACF,CAAC,aAAa,IAAI,KAAK,SAAS;AAAA;AAGlC,YAAM,KAAK,mBAAmB;AAE9B,aAAO;AAAA,QACL;AAAA,QACA,MAAM;AAAA,QACN;AAAA,QACA;AAAA,QACA,MAAM;AAAA,QACN,MAAM;AAAA,MACR;AAAA,IACF;AACA,UAAM,IAAI,MAAM,uBAAuB;AAAA,EACzC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,YAAwB,IAA8C;AAC1E,UAAM,SAAS,KAAK;AAAA,qDAC6B,EAAE;AAAA;AAEnD,QAAI,CAAC,QAAQ;AACX,cAAQ,MAAM,YAAY,EAAE,YAAY;AACxC,aAAO;AAAA,IACT;AAEA,WAAO,EAAE,GAAG,OAAO,CAAC,GAAG,SAAS,KAAK,MAAM,OAAO,CAAC,EAAE,OAAO,EAAO;AAAA,EACrE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,aACE,WAII,CAAC,GACU;AACf,QAAI,QAAQ;AACZ,UAAM,SAAS,CAAC;AAEhB,QAAI,SAAS,IAAI;AACf,eAAS;AACT,aAAO,KAAK,SAAS,EAAE;AAAA,IACzB;AAEA,QAAI,SAAS,MAAM;AACjB,eAAS;AACT,aAAO,KAAK,SAAS,IAAI;AAAA,IAC3B;AAEA,QAAI,SAAS,WAAW;AACtB,eAAS;AACT,YAAM,QAAQ,SAAS,UAAU,SAAS,oBAAI,KAAK,CAAC;AACpD,YAAM,MAAM,SAAS,UAAU,OAAO,oBAAI,KAAK,eAAe;AAC9D,aAAO;AAAA,QACL,KAAK,MAAM,MAAM,QAAQ,IAAI,GAAI;AAAA,QACjC,KAAK,MAAM,IAAI,QAAQ,IAAI,GAAI;AAAA,MACjC;AAAA,IACF;AAEA,UAAM,SAAS,KAAK,IAAI,QAAQ,IAC7B,KAAK,OAAO,GAAG,MAAM,EACrB,QAAQ,EACR,IAAI,CAAC,SAAS;AAAA,MACb,GAAG;AAAA,MACH,SAAS,KAAK,MAAM,IAAI,OAAiB;AAAA,IAC3C,EAAE;AAEJ,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,eAAe,IAA8B;AACjD,SAAK,iDAAiD,EAAE;AAExD,UAAM,KAAK,mBAAmB;AAC9B,WAAO;AAAA,EACT;AAAA,EAEA,MAAc,qBAAqB;AAEjC,UAAM,SAAS,KAAK;AAAA;AAAA,qBAEH,KAAK,MAAM,KAAK,IAAI,IAAI,GAAI,CAAC;AAAA;AAAA;AAAA;AAI9C,QAAI,CAAC,OAAQ;AAEb,QAAI,OAAO,SAAS,KAAK,UAAU,OAAO,CAAC,GAAG;AAC5C,YAAM,WAAY,OAAO,CAAC,EAAE,OAAkB;AAC9C,YAAM,KAAK,IAAI,QAAQ,SAAS,QAAQ;AAAA,IAC1C;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EA8DA,MAAM,UAAU;AAEd,SAAK;AACL,SAAK;AACL,SAAK;AAGL,UAAM,KAAK,IAAI,QAAQ,YAAY;AACnC,UAAM,KAAK,IAAI,QAAQ,UAAU;AACjC,SAAK,IAAI,MAAM,WAAW;AAAA,EAC5B;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,YAAY,QAAyB;AAC3C,WAAO,iBAAiB,IAAI,KAAK,MAAoB,CAAa;AAAA,EACpE;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,aACJ,YACA,KACA,cACA,eAAe,UACf,SAMsD;AACtD,UAAM,cAAc,GAAG,YAAY,IAAI,YAAY,IAAI,qBAAqB,KAAK,aAAa,IAAI,CAAC,IAAI,KAAK,IAAI;AAEhH,UAAM,SAAS,MAAM,KAAK;AAAA,MACxB;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AACA,SAAK;AAAA;AAAA;AAAA;AAAA,UAIC,OAAO,EAAE;AAAA,UACT,UAAU;AAAA,UACV,GAAG;AAAA,UACH,OAAO,YAAY,IAAI;AAAA,UACvB,OAAO,WAAW,IAAI;AAAA,UACtB,WAAW;AAAA,UACX,UAAU,KAAK,UAAU,OAAO,IAAI,IAAI;AAAA;AAAA;AAI9C,SAAK;AAAA,MACH,KAAK,UAAU;AAAA,QACb,KAAK,KAAK,cAAc;AAAA,QACxB,MAAM;AAAA,MACR,CAAC;AAAA,IACH;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,4BACJ,aACA,KACA,aAEA,SAaA,WAQC;AACD,UAAM,eAAe,IAAI;AAAA,MACvB,KAAK,IAAI;AAAA,MACT,KAAK;AAAA,MACL;AAAA,IACF;AAEA,QAAI,WAAW;AACb,mBAAa,WAAW,UAAU;AAClC,UAAI,UAAU,eAAe;AAC3B,qBAAa,WAAW,UAAU;AAAA,MACpC;AAAA,IACF;AAIA,QAAI,sBAAiD,CAAC;AACtD,QAAI,SAAS,WAAW,SAAS;AAC/B,4BAAsB;AAAA,QACpB,iBAAiB;AAAA,UACf,OAAO,CAACC,MAAK,SACX,MAAMA,MAAK;AAAA,YACT,GAAG;AAAA,YACH,SAAS,SAAS,WAAW;AAAA,UAC/B,CAAC;AAAA,QACL;AAAA,QACA,aAAa;AAAA,UACX,SAAS,SAAS,WAAW;AAAA,QAC/B;AAAA,MACF;AAAA,IACF;AAEA,UAAM,EAAE,IAAI,SAAS,SAAS,IAAI,MAAM,KAAK,IAAI,QAAQ,KAAK;AAAA,MAC5D,QAAQ,SAAS;AAAA,MACjB;AAAA,MACA,WAAW;AAAA,QACT,GAAG;AAAA,QACH;AAAA,MACF;AAAA,IACF,CAAC;AAED,WAAO;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,gBAAgB,IAAY;AAChC,SAAK,IAAI,gBAAgB,EAAE;AAC3B,SAAK;AAAA,qDAC4C,EAAE;AAAA;AAEnD,SAAK;AAAA,MACH,KAAK,UAAU;AAAA,QACb,KAAK,KAAK,cAAc;AAAA,QACxB,MAAM;AAAA,MACR,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EAEA,gBAAiC;AAC/B,UAAM,WAA4B;AAAA,MAChC,SAAS,KAAK,IAAI,YAAY;AAAA,MAC9B,WAAW,KAAK,IAAI,cAAc;AAAA,MAClC,SAAS,CAAC;AAAA,MACV,OAAO,KAAK,IAAI,UAAU;AAAA,IAC5B;AAEA,UAAM,UAAU,KAAK;AAAA;AAAA;AAIrB,eAAW,UAAU,SAAS;AAC5B,YAAM,aAAa,KAAK,IAAI,eAAe,OAAO,EAAE;AACpD,eAAS,QAAQ,OAAO,EAAE,IAAI;AAAA,QAC5B,UAAU,OAAO;AAAA,QACjB,cAAc,YAAY,sBAAsB;AAAA,QAChD,cAAc,YAAY,gBAAgB;AAAA,QAC1C,MAAM,OAAO;AAAA,QACb,YAAY,OAAO;AAAA;AAAA,QAEnB,OAAO,YAAY,mBAAmB;AAAA,MACxC;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AACF;AAAA;AAAA;AAAA;AAx0Ba,MA4DJ,UAAU;AAAA;AAAA,EAEf,WAAW;AAAA;AACb;AAwyBF,eAAsB,kBACpB,SACA,KACA,SACA;AACA,QAAM,cACJ,SAAS,SAAS,OACd;AAAA,IACE,oCAAoC;AAAA,IACpC,gCAAgC;AAAA,IAChC,+BAA+B;AAAA,IAC/B,0BAA0B;AAAA,EAC5B,IACA,SAAS;AAEf,MAAI,QAAQ,WAAW,WAAW;AAChC,QAAI,aAAa;AACf,aAAO,IAAI,SAAS,MAAM;AAAA,QACxB,SAAS;AAAA,MACX,CAAC;AAAA,IACH;AACA,YAAQ;AAAA,MACN;AAAA,IACF;AAAA,EACF;AAEA,MAAI,WAAW,MAAM;AAAA,IACnB;AAAA,IACA;AAAA,IACA;AAAA,MACE,QAAQ;AAAA,MACR,GAAI;AAAA,IACN;AAAA,EACF;AAEA,MACE,YACA,eACA,QAAQ,QAAQ,IAAI,SAAS,GAAG,YAAY,MAAM,eAClD,QAAQ,QAAQ,IAAI,SAAS,GAAG,YAAY,MAAM,aAClD;AACA,eAAW,IAAI,SAAS,SAAS,MAAM;AAAA,MACrC,SAAS;AAAA,QACP,GAAG,SAAS;AAAA,QACZ,GAAG;AAAA,MACL;AAAA,IACF,CAAC;AAAA,EACH;AACA,SAAO;AACT;AAQA,eAAsB,gBACpB,QACA,MACA,UACe;AAAC;AAWlB,eAAsB,eACpB,WACA,MACA,SAIA;AACA,SAAO,gBAAwB,WAAW,MAAM,OAAO;AACzD;AAKO,IAAM,oBAAN,MAAwB;AAAA,EAK7B,YAAY,YAAwB,IAAY;AAFhD,SAAQ,UAAU;AAGhB,SAAK,cAAc;AACnB,SAAK,MAAM;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,KAAK,OAAgB;AACnB,QAAI,KAAK,SAAS;AAChB,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACvD;AACA,UAAM,WAAwB;AAAA,MAC5B,MAAM;AAAA,MACN,IAAI,KAAK;AAAA,MACT,QAAQ;AAAA,MACR,SAAS;AAAA,MACT,MAAM;AAAA,IACR;AACA,SAAK,YAAY,KAAK,KAAK,UAAU,QAAQ,CAAC;AAAA,EAChD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,YAAsB;AACxB,QAAI,KAAK,SAAS;AAChB,YAAM,IAAI,MAAM,qCAAqC;AAAA,IACvD;AACA,SAAK,UAAU;AACf,UAAM,WAAwB;AAAA,MAC5B,MAAM;AAAA,MACN,IAAI,KAAK;AAAA,MACT,QAAQ;AAAA,MACR,SAAS;AAAA,MACT,MAAM;AAAA,IACR;AACA,SAAK,YAAY,KAAK,KAAK,UAAU,QAAQ,CAAC;AAAA,EAChD;AACF;","names":["ctx","url"]}
|
|
@@ -7,12 +7,12 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
7
7
|
}
|
|
8
8
|
get clientMetadata() {
|
|
9
9
|
return {
|
|
10
|
-
|
|
11
|
-
|
|
10
|
+
client_name: this.clientName,
|
|
11
|
+
client_uri: "example.com",
|
|
12
12
|
grant_types: ["authorization_code", "refresh_token"],
|
|
13
|
+
redirect_uris: [this.redirectUrl],
|
|
13
14
|
response_types: ["code"],
|
|
14
|
-
|
|
15
|
-
client_uri: "example.com"
|
|
15
|
+
token_endpoint_auth_method: "none"
|
|
16
16
|
};
|
|
17
17
|
}
|
|
18
18
|
get redirectUrl() {
|
|
@@ -103,4 +103,4 @@ var DurableObjectOAuthClientProvider = class {
|
|
|
103
103
|
export {
|
|
104
104
|
DurableObjectOAuthClientProvider
|
|
105
105
|
};
|
|
106
|
-
//# sourceMappingURL=chunk-
|
|
106
|
+
//# sourceMappingURL=chunk-767EASBA.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":["../src/mcp/do-oauth-client-provider.ts"],"sourcesContent":["import type { OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type {\n
|
|
1
|
+
{"version":3,"sources":["../src/mcp/do-oauth-client-provider.ts"],"sourcesContent":["import type { OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type {\n OAuthClientInformation,\n OAuthClientInformationFull,\n OAuthClientMetadata,\n OAuthTokens,\n} from \"@modelcontextprotocol/sdk/shared/auth.js\";\n\n// A slight extension to the standard OAuthClientProvider interface because `redirectToAuthorization` doesn't give us the interface we need\n// This allows us to track authentication for a specific server and associated dynamic client registration\nexport interface AgentsOAuthProvider extends OAuthClientProvider {\n authUrl: string | undefined;\n clientId: string | undefined;\n serverId: string | undefined;\n}\n\nexport class DurableObjectOAuthClientProvider implements AgentsOAuthProvider {\n private _authUrl_: string | undefined;\n private _serverId_: string | undefined;\n private _clientId_: string | undefined;\n\n constructor(\n public storage: DurableObjectStorage,\n public clientName: string,\n public baseRedirectUrl: string\n ) {}\n\n get clientMetadata(): OAuthClientMetadata {\n return {\n client_name: this.clientName,\n client_uri: \"example.com\",\n grant_types: [\"authorization_code\", \"refresh_token\"],\n redirect_uris: [this.redirectUrl],\n response_types: [\"code\"],\n token_endpoint_auth_method: \"none\",\n };\n }\n\n get redirectUrl() {\n return `${this.baseRedirectUrl}/${this.serverId}`;\n }\n\n get clientId() {\n if (!this._clientId_) {\n throw new Error(\"Trying to access clientId before it was set\");\n }\n return this._clientId_;\n }\n\n set clientId(clientId_: string) {\n this._clientId_ = clientId_;\n }\n\n get serverId() {\n if (!this._serverId_) {\n throw new Error(\"Trying to access serverId before it was set\");\n }\n return this._serverId_;\n }\n\n set serverId(serverId_: string) {\n this._serverId_ = serverId_;\n }\n\n keyPrefix(clientId: string) {\n return `/${this.clientName}/${this.serverId}/${clientId}`;\n }\n\n clientInfoKey(clientId: string) {\n return `${this.keyPrefix(clientId)}/client_info/`;\n }\n\n async clientInformation(): Promise<OAuthClientInformation | undefined> {\n if (!this._clientId_) {\n return undefined;\n }\n return (\n (await this.storage.get<OAuthClientInformation>(\n this.clientInfoKey(this.clientId)\n )) ?? undefined\n );\n }\n\n async saveClientInformation(\n clientInformation: OAuthClientInformationFull\n ): Promise<void> {\n await this.storage.put(\n this.clientInfoKey(clientInformation.client_id),\n clientInformation\n );\n this.clientId = clientInformation.client_id;\n }\n\n tokenKey(clientId: string) {\n return `${this.keyPrefix(clientId)}/token`;\n }\n\n async tokens(): Promise<OAuthTokens | undefined> {\n if (!this._clientId_) {\n return undefined;\n }\n return (\n (await this.storage.get<OAuthTokens>(this.tokenKey(this.clientId))) ??\n undefined\n );\n }\n\n async saveTokens(tokens: OAuthTokens): Promise<void> {\n await this.storage.put(this.tokenKey(this.clientId), tokens);\n }\n\n get authUrl() {\n return this._authUrl_;\n }\n\n /**\n * Because this operates on the server side (but we need browser auth), we send this url back to the user\n * and require user interact to initiate the redirect flow\n */\n async redirectToAuthorization(authUrl: URL): Promise<void> {\n // We want to track the client ID in state here because the typescript SSE client sometimes does\n // a dynamic client registration AFTER generating this redirect URL.\n const client_id = authUrl.searchParams.get(\"client_id\");\n if (client_id) {\n authUrl.searchParams.append(\"state\", client_id);\n }\n this._authUrl_ = authUrl.toString();\n }\n\n codeVerifierKey(clientId: string) {\n return `${this.keyPrefix(clientId)}/code_verifier`;\n }\n\n async saveCodeVerifier(verifier: string): Promise<void> {\n await this.storage.put(this.codeVerifierKey(this.clientId), verifier);\n }\n\n async codeVerifier(): Promise<string> {\n const codeVerifier = await this.storage.get<string>(\n this.codeVerifierKey(this.clientId)\n );\n if (!codeVerifier) {\n throw new Error(\"No code verifier found\");\n }\n return codeVerifier;\n }\n}\n"],"mappings":";AAgBO,IAAM,mCAAN,MAAsE;AAAA,EAK3E,YACS,SACA,YACA,iBACP;AAHO;AACA;AACA;AAAA,EACN;AAAA,EAEH,IAAI,iBAAsC;AACxC,WAAO;AAAA,MACL,aAAa,KAAK;AAAA,MAClB,YAAY;AAAA,MACZ,aAAa,CAAC,sBAAsB,eAAe;AAAA,MACnD,eAAe,CAAC,KAAK,WAAW;AAAA,MAChC,gBAAgB,CAAC,MAAM;AAAA,MACvB,4BAA4B;AAAA,IAC9B;AAAA,EACF;AAAA,EAEA,IAAI,cAAc;AAChB,WAAO,GAAG,KAAK,eAAe,IAAI,KAAK,QAAQ;AAAA,EACjD;AAAA,EAEA,IAAI,WAAW;AACb,QAAI,CAAC,KAAK,YAAY;AACpB,YAAM,IAAI,MAAM,6CAA6C;AAAA,IAC/D;AACA,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,WAAmB;AAC9B,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,IAAI,WAAW;AACb,QAAI,CAAC,KAAK,YAAY;AACpB,YAAM,IAAI,MAAM,6CAA6C;AAAA,IAC/D;AACA,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,IAAI,SAAS,WAAmB;AAC9B,SAAK,aAAa;AAAA,EACpB;AAAA,EAEA,UAAU,UAAkB;AAC1B,WAAO,IAAI,KAAK,UAAU,IAAI,KAAK,QAAQ,IAAI,QAAQ;AAAA,EACzD;AAAA,EAEA,cAAc,UAAkB;AAC9B,WAAO,GAAG,KAAK,UAAU,QAAQ,CAAC;AAAA,EACpC;AAAA,EAEA,MAAM,oBAAiE;AACrE,QAAI,CAAC,KAAK,YAAY;AACpB,aAAO;AAAA,IACT;AACA,WACG,MAAM,KAAK,QAAQ;AAAA,MAClB,KAAK,cAAc,KAAK,QAAQ;AAAA,IAClC,KAAM;AAAA,EAEV;AAAA,EAEA,MAAM,sBACJ,mBACe;AACf,UAAM,KAAK,QAAQ;AAAA,MACjB,KAAK,cAAc,kBAAkB,SAAS;AAAA,MAC9C;AAAA,IACF;AACA,SAAK,WAAW,kBAAkB;AAAA,EACpC;AAAA,EAEA,SAAS,UAAkB;AACzB,WAAO,GAAG,KAAK,UAAU,QAAQ,CAAC;AAAA,EACpC;AAAA,EAEA,MAAM,SAA2C;AAC/C,QAAI,CAAC,KAAK,YAAY;AACpB,aAAO;AAAA,IACT;AACA,WACG,MAAM,KAAK,QAAQ,IAAiB,KAAK,SAAS,KAAK,QAAQ,CAAC,KACjE;AAAA,EAEJ;AAAA,EAEA,MAAM,WAAW,QAAoC;AACnD,UAAM,KAAK,QAAQ,IAAI,KAAK,SAAS,KAAK,QAAQ,GAAG,MAAM;AAAA,EAC7D;AAAA,EAEA,IAAI,UAAU;AACZ,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,wBAAwB,SAA6B;AAGzD,UAAM,YAAY,QAAQ,aAAa,IAAI,WAAW;AACtD,QAAI,WAAW;AACb,cAAQ,aAAa,OAAO,SAAS,SAAS;AAAA,IAChD;AACA,SAAK,YAAY,QAAQ,SAAS;AAAA,EACpC;AAAA,EAEA,gBAAgB,UAAkB;AAChC,WAAO,GAAG,KAAK,UAAU,QAAQ,CAAC;AAAA,EACpC;AAAA,EAEA,MAAM,iBAAiB,UAAiC;AACtD,UAAM,KAAK,QAAQ,IAAI,KAAK,gBAAgB,KAAK,QAAQ,GAAG,QAAQ;AAAA,EACtE;AAAA,EAEA,MAAM,eAAgC;AACpC,UAAM,eAAe,MAAM,KAAK,QAAQ;AAAA,MACtC,KAAK,gBAAgB,KAAK,QAAQ;AAAA,IACpC;AACA,QAAI,CAAC,cAAc;AACjB,YAAM,IAAI,MAAM,wBAAwB;AAAA,IAC1C;AACA,WAAO;AAAA,EACT;AACF;","names":[]}
|
|
@@ -1,3 +1,15 @@
|
|
|
1
|
+
// src/mcp/client.ts
|
|
2
|
+
import { jsonSchema } from "ai";
|
|
3
|
+
import { nanoid } from "nanoid";
|
|
4
|
+
|
|
5
|
+
// src/mcp/client-connection.ts
|
|
6
|
+
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
7
|
+
import {
|
|
8
|
+
PromptListChangedNotificationSchema,
|
|
9
|
+
ResourceListChangedNotificationSchema,
|
|
10
|
+
ToolListChangedNotificationSchema
|
|
11
|
+
} from "@modelcontextprotocol/sdk/types.js";
|
|
12
|
+
|
|
1
13
|
// src/mcp/sse-edge.ts
|
|
2
14
|
import {
|
|
3
15
|
SSEClientTransport
|
|
@@ -46,14 +58,8 @@ var SSEEdgeClientTransport = class extends SSEClientTransport {
|
|
|
46
58
|
};
|
|
47
59
|
|
|
48
60
|
// src/mcp/client-connection.ts
|
|
49
|
-
import {
|
|
50
|
-
ToolListChangedNotificationSchema,
|
|
51
|
-
ResourceListChangedNotificationSchema,
|
|
52
|
-
PromptListChangedNotificationSchema
|
|
53
|
-
} from "@modelcontextprotocol/sdk/types.js";
|
|
54
|
-
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
55
61
|
var MCPClientConnection = class {
|
|
56
|
-
constructor(url, info, options = {
|
|
62
|
+
constructor(url, info, options = { client: {}, transport: {} }) {
|
|
57
63
|
this.url = url;
|
|
58
64
|
this.options = options;
|
|
59
65
|
this.connectionState = "connecting";
|
|
@@ -222,8 +228,6 @@ function capabilityErrorHandler(empty, method) {
|
|
|
222
228
|
}
|
|
223
229
|
|
|
224
230
|
// src/mcp/client.ts
|
|
225
|
-
import { jsonSchema } from "ai";
|
|
226
|
-
import { nanoid } from "nanoid";
|
|
227
231
|
var MCPClientManager = class {
|
|
228
232
|
/**
|
|
229
233
|
* @param _name Name of the MCP client
|
|
@@ -262,8 +266,8 @@ var MCPClientManager = class {
|
|
|
262
266
|
version: this._version
|
|
263
267
|
},
|
|
264
268
|
{
|
|
265
|
-
|
|
266
|
-
|
|
269
|
+
client: options.client ?? {},
|
|
270
|
+
transport: options.transport ?? {}
|
|
267
271
|
}
|
|
268
272
|
);
|
|
269
273
|
await this.mcpConnections[id].init(options.reconnect?.oauthCode);
|
|
@@ -273,9 +277,9 @@ var MCPClientManager = class {
|
|
|
273
277
|
options.transport.authProvider.redirectUrl.toString()
|
|
274
278
|
);
|
|
275
279
|
return {
|
|
276
|
-
id,
|
|
277
280
|
authUrl,
|
|
278
|
-
clientId: options.transport?.authProvider?.clientId
|
|
281
|
+
clientId: options.transport?.authProvider?.clientId,
|
|
282
|
+
id
|
|
279
283
|
};
|
|
280
284
|
}
|
|
281
285
|
return {
|
|
@@ -352,19 +356,19 @@ var MCPClientManager = class {
|
|
|
352
356
|
return [
|
|
353
357
|
`${tool.serverId}_${tool.name}`,
|
|
354
358
|
{
|
|
355
|
-
parameters: jsonSchema(tool.inputSchema),
|
|
356
359
|
description: tool.description,
|
|
357
360
|
execute: async (args) => {
|
|
358
361
|
const result = await this.callTool({
|
|
359
|
-
name: tool.name,
|
|
360
362
|
arguments: args,
|
|
363
|
+
name: tool.name,
|
|
361
364
|
serverId: tool.serverId
|
|
362
365
|
});
|
|
363
366
|
if (result.isError) {
|
|
364
367
|
throw new Error(result.content[0].text);
|
|
365
368
|
}
|
|
366
369
|
return result;
|
|
367
|
-
}
|
|
370
|
+
},
|
|
371
|
+
parameters: jsonSchema(tool.inputSchema)
|
|
368
372
|
}
|
|
369
373
|
];
|
|
370
374
|
})
|
|
@@ -444,7 +448,7 @@ var MCPClientManager = class {
|
|
|
444
448
|
};
|
|
445
449
|
function getNamespacedData(mcpClients, type) {
|
|
446
450
|
const sets = Object.entries(mcpClients).map(([name, conn]) => {
|
|
447
|
-
return {
|
|
451
|
+
return { data: conn[type], name };
|
|
448
452
|
});
|
|
449
453
|
const namespacedData = sets.flatMap(({ name: serverId, data }) => {
|
|
450
454
|
return data.map((item) => {
|
|
@@ -462,4 +466,4 @@ export {
|
|
|
462
466
|
MCPClientManager,
|
|
463
467
|
getNamespacedData
|
|
464
468
|
};
|
|
465
|
-
//# sourceMappingURL=chunk-
|
|
469
|
+
//# sourceMappingURL=chunk-E3LCYPCB.js.map
|