agents 0.0.0-4f3dfc7 → 0.0.0-54e1986
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 +49 -3
- package/dist/ai-chat-agent.js +129 -66
- package/dist/ai-chat-agent.js.map +1 -1
- package/dist/ai-react.d.ts +12 -0
- package/dist/ai-react.js +38 -24
- package/dist/ai-react.js.map +1 -1
- package/dist/ai-types.d.ts +5 -0
- package/dist/chunk-BZXOAZUX.js +106 -0
- package/dist/chunk-BZXOAZUX.js.map +1 -0
- package/dist/chunk-QSGN3REV.js +123 -0
- package/dist/chunk-QSGN3REV.js.map +1 -0
- package/dist/{chunk-X6BBKLSC.js → chunk-RIYR6FR6.js} +382 -173
- package/dist/chunk-RIYR6FR6.js.map +1 -0
- package/dist/chunk-Y67CHZBI.js +464 -0
- package/dist/chunk-Y67CHZBI.js.map +1 -0
- package/dist/client.d.ts +9 -1
- package/dist/client.js +6 -133
- package/dist/client.js.map +1 -1
- package/dist/index.d.ts +111 -13
- package/dist/index.js +6 -4
- package/dist/mcp/client.d.ts +783 -0
- package/dist/mcp/client.js +9 -0
- package/dist/mcp/do-oauth-client-provider.d.ts +41 -0
- package/dist/mcp/do-oauth-client-provider.js +7 -0
- package/dist/mcp/do-oauth-client-provider.js.map +1 -0
- package/dist/mcp/index.d.ts +84 -0
- package/dist/mcp/index.js +779 -0
- package/dist/mcp/index.js.map +1 -0
- package/dist/react.d.ts +14 -0
- package/dist/react.js +38 -29
- package/dist/react.js.map +1 -1
- package/dist/schedule.js +0 -2
- package/dist/schedule.js.map +1 -1
- package/package.json +36 -8
- package/src/index.ts +481 -126
- package/dist/chunk-HMLY7DHA.js +0 -16
- package/dist/chunk-X6BBKLSC.js.map +0 -1
- package/dist/mcp.d.ts +0 -58
- package/dist/mcp.js +0 -945
- package/dist/mcp.js.map +0 -1
- /package/dist/{chunk-HMLY7DHA.js.map → mcp/client.js.map} +0 -0
package/dist/client.d.ts
CHANGED
|
@@ -38,17 +38,24 @@ type AgentClientFetchOptions = Omit<PartyFetchOptions, "party" | "room"> & {
|
|
|
38
38
|
/** Name of the specific Agent instance */
|
|
39
39
|
name?: string;
|
|
40
40
|
};
|
|
41
|
+
/**
|
|
42
|
+
* Convert a camelCase string to a kebab-case string
|
|
43
|
+
* @param str The string to convert
|
|
44
|
+
* @returns The kebab-case string
|
|
45
|
+
*/
|
|
46
|
+
declare function camelCaseToKebabCase(str: string): string;
|
|
41
47
|
/**
|
|
42
48
|
* WebSocket client for connecting to an Agent
|
|
43
49
|
*/
|
|
44
50
|
declare class AgentClient<State = unknown> extends PartySocket {
|
|
45
|
-
#private;
|
|
46
51
|
/**
|
|
47
52
|
* @deprecated Use agentFetch instead
|
|
48
53
|
*/
|
|
49
54
|
static fetch(_opts: PartyFetchOptions): Promise<Response>;
|
|
50
55
|
agent: string;
|
|
51
56
|
name: string;
|
|
57
|
+
private options;
|
|
58
|
+
private _pendingCalls;
|
|
52
59
|
constructor(options: AgentClientOptions<State>);
|
|
53
60
|
setState(state: State): void;
|
|
54
61
|
/**
|
|
@@ -81,4 +88,5 @@ export {
|
|
|
81
88
|
type AgentClientOptions,
|
|
82
89
|
type StreamOptions,
|
|
83
90
|
agentFetch,
|
|
91
|
+
camelCaseToKebabCase,
|
|
84
92
|
};
|
package/dist/client.js
CHANGED
|
@@ -1,138 +1,11 @@
|
|
|
1
1
|
import {
|
|
2
|
-
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
} from "./chunk-
|
|
6
|
-
|
|
7
|
-
// src/client.ts
|
|
8
|
-
import {
|
|
9
|
-
PartySocket
|
|
10
|
-
} from "partysocket";
|
|
11
|
-
var _options, _pendingCalls;
|
|
12
|
-
var AgentClient = class extends PartySocket {
|
|
13
|
-
constructor(options) {
|
|
14
|
-
super({
|
|
15
|
-
prefix: "agents",
|
|
16
|
-
party: options.agent,
|
|
17
|
-
room: options.name || "default",
|
|
18
|
-
...options
|
|
19
|
-
});
|
|
20
|
-
__privateAdd(this, _options);
|
|
21
|
-
__privateAdd(this, _pendingCalls, /* @__PURE__ */ new Map());
|
|
22
|
-
this.agent = options.agent;
|
|
23
|
-
this.name = options.name || "default";
|
|
24
|
-
__privateSet(this, _options, options);
|
|
25
|
-
if (this.agent !== this.agent.toLowerCase()) {
|
|
26
|
-
console.warn(
|
|
27
|
-
`Agent name: ${this.agent} should probably be in lowercase. Received: ${this.agent}`
|
|
28
|
-
);
|
|
29
|
-
}
|
|
30
|
-
if (this.name !== this.name.toLowerCase()) {
|
|
31
|
-
console.warn(
|
|
32
|
-
`Agent instance name: ${this.name} should probably be in lowercase. Received: ${this.name}`
|
|
33
|
-
);
|
|
34
|
-
}
|
|
35
|
-
this.addEventListener("message", (event) => {
|
|
36
|
-
if (typeof event.data === "string") {
|
|
37
|
-
let parsedMessage;
|
|
38
|
-
try {
|
|
39
|
-
parsedMessage = JSON.parse(event.data);
|
|
40
|
-
} catch (error) {
|
|
41
|
-
return;
|
|
42
|
-
}
|
|
43
|
-
if (parsedMessage.type === "cf_agent_state") {
|
|
44
|
-
__privateGet(this, _options).onStateUpdate?.(parsedMessage.state, "server");
|
|
45
|
-
return;
|
|
46
|
-
}
|
|
47
|
-
if (parsedMessage.type === "rpc") {
|
|
48
|
-
const response = parsedMessage;
|
|
49
|
-
const pending = __privateGet(this, _pendingCalls).get(response.id);
|
|
50
|
-
if (!pending) return;
|
|
51
|
-
if (!response.success) {
|
|
52
|
-
pending.reject(new Error(response.error));
|
|
53
|
-
__privateGet(this, _pendingCalls).delete(response.id);
|
|
54
|
-
pending.stream?.onError?.(response.error);
|
|
55
|
-
return;
|
|
56
|
-
}
|
|
57
|
-
if ("done" in response) {
|
|
58
|
-
if (response.done) {
|
|
59
|
-
pending.resolve(response.result);
|
|
60
|
-
__privateGet(this, _pendingCalls).delete(response.id);
|
|
61
|
-
pending.stream?.onDone?.(response.result);
|
|
62
|
-
} else {
|
|
63
|
-
pending.stream?.onChunk?.(response.result);
|
|
64
|
-
}
|
|
65
|
-
} else {
|
|
66
|
-
pending.resolve(response.result);
|
|
67
|
-
__privateGet(this, _pendingCalls).delete(response.id);
|
|
68
|
-
}
|
|
69
|
-
}
|
|
70
|
-
}
|
|
71
|
-
});
|
|
72
|
-
}
|
|
73
|
-
/**
|
|
74
|
-
* @deprecated Use agentFetch instead
|
|
75
|
-
*/
|
|
76
|
-
static fetch(_opts) {
|
|
77
|
-
throw new Error(
|
|
78
|
-
"AgentClient.fetch is not implemented, use agentFetch instead"
|
|
79
|
-
);
|
|
80
|
-
}
|
|
81
|
-
setState(state) {
|
|
82
|
-
this.send(JSON.stringify({ type: "cf_agent_state", state }));
|
|
83
|
-
__privateGet(this, _options).onStateUpdate?.(state, "client");
|
|
84
|
-
}
|
|
85
|
-
/**
|
|
86
|
-
* Call a method on the Agent
|
|
87
|
-
* @param method Name of the method to call
|
|
88
|
-
* @param args Arguments to pass to the method
|
|
89
|
-
* @param streamOptions Options for handling streaming responses
|
|
90
|
-
* @returns Promise that resolves with the method's return value
|
|
91
|
-
*/
|
|
92
|
-
async call(method, args = [], streamOptions) {
|
|
93
|
-
return new Promise((resolve, reject) => {
|
|
94
|
-
const id = Math.random().toString(36).slice(2);
|
|
95
|
-
__privateGet(this, _pendingCalls).set(id, {
|
|
96
|
-
resolve: (value) => resolve(value),
|
|
97
|
-
reject,
|
|
98
|
-
stream: streamOptions,
|
|
99
|
-
type: null
|
|
100
|
-
});
|
|
101
|
-
const request = {
|
|
102
|
-
type: "rpc",
|
|
103
|
-
id,
|
|
104
|
-
method,
|
|
105
|
-
args
|
|
106
|
-
};
|
|
107
|
-
this.send(JSON.stringify(request));
|
|
108
|
-
});
|
|
109
|
-
}
|
|
110
|
-
};
|
|
111
|
-
_options = new WeakMap();
|
|
112
|
-
_pendingCalls = new WeakMap();
|
|
113
|
-
function agentFetch(opts, init) {
|
|
114
|
-
if (opts.agent !== opts.agent.toLowerCase()) {
|
|
115
|
-
console.warn(
|
|
116
|
-
`Agent name: ${opts.agent} should probably be in lowercase. Received: ${opts.agent}`
|
|
117
|
-
);
|
|
118
|
-
}
|
|
119
|
-
if (opts.name && opts.name !== opts.name.toLowerCase()) {
|
|
120
|
-
console.warn(
|
|
121
|
-
`Agent instance name: ${opts.name} should probably be in lowercase. Received: ${opts.name}`
|
|
122
|
-
);
|
|
123
|
-
}
|
|
124
|
-
return PartySocket.fetch(
|
|
125
|
-
{
|
|
126
|
-
prefix: "agents",
|
|
127
|
-
party: opts.agent,
|
|
128
|
-
room: opts.name || "default",
|
|
129
|
-
...opts
|
|
130
|
-
},
|
|
131
|
-
init
|
|
132
|
-
);
|
|
133
|
-
}
|
|
2
|
+
AgentClient,
|
|
3
|
+
agentFetch,
|
|
4
|
+
camelCaseToKebabCase
|
|
5
|
+
} from "./chunk-QSGN3REV.js";
|
|
134
6
|
export {
|
|
135
7
|
AgentClient,
|
|
136
|
-
agentFetch
|
|
8
|
+
agentFetch,
|
|
9
|
+
camelCaseToKebabCase
|
|
137
10
|
};
|
|
138
11
|
//# sourceMappingURL=client.js.map
|
package/dist/client.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"sources":[
|
|
1
|
+
{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}
|
package/dist/index.d.ts
CHANGED
|
@@ -1,6 +1,15 @@
|
|
|
1
1
|
import { Server, Connection, PartyServerOptions } from "partyserver";
|
|
2
2
|
export { Connection, ConnectionContext, WSMessage } from "partyserver";
|
|
3
|
-
import {
|
|
3
|
+
import { MCPClientManager } from "./mcp/client.js";
|
|
4
|
+
import { Tool, Prompt, Resource } from "@modelcontextprotocol/sdk/types.js";
|
|
5
|
+
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
6
|
+
import "zod";
|
|
7
|
+
import "@modelcontextprotocol/sdk/client/sse.js";
|
|
8
|
+
import "./mcp/do-oauth-client-provider.js";
|
|
9
|
+
import "@modelcontextprotocol/sdk/client/auth.js";
|
|
10
|
+
import "@modelcontextprotocol/sdk/shared/auth.js";
|
|
11
|
+
import "@modelcontextprotocol/sdk/shared/protocol.js";
|
|
12
|
+
import "ai";
|
|
4
13
|
|
|
5
14
|
/**
|
|
6
15
|
* RPC request message from client
|
|
@@ -59,10 +68,6 @@ declare function unstable_callable(
|
|
|
59
68
|
target: (this: This, ...args: Args) => Return,
|
|
60
69
|
context: ClassMethodDecoratorContext
|
|
61
70
|
) => (this: This, ...args: Args) => Return;
|
|
62
|
-
/**
|
|
63
|
-
* A class for creating workflow entry points that can be used with Cloudflare Workers
|
|
64
|
-
*/
|
|
65
|
-
declare class WorkflowEntrypoint extends WorkflowEntrypoint$1 {}
|
|
66
71
|
/**
|
|
67
72
|
* Represents a scheduled task within an Agent
|
|
68
73
|
* @template T Type of the payload data
|
|
@@ -98,13 +103,43 @@ type Schedule<T = string> = {
|
|
|
98
103
|
cron: string;
|
|
99
104
|
}
|
|
100
105
|
);
|
|
106
|
+
/**
|
|
107
|
+
* MCP Server state update message from server -> Client
|
|
108
|
+
*/
|
|
109
|
+
type MCPServerMessage = {
|
|
110
|
+
type: "cf_agent_mcp_servers";
|
|
111
|
+
mcp: MCPServersState;
|
|
112
|
+
};
|
|
113
|
+
type MCPServersState = {
|
|
114
|
+
servers: {
|
|
115
|
+
[id: string]: MCPServer;
|
|
116
|
+
};
|
|
117
|
+
tools: Tool[];
|
|
118
|
+
prompts: Prompt[];
|
|
119
|
+
resources: Resource[];
|
|
120
|
+
};
|
|
121
|
+
type MCPServer = {
|
|
122
|
+
name: string;
|
|
123
|
+
server_url: string;
|
|
124
|
+
auth_url: string | null;
|
|
125
|
+
state: "authenticating" | "connecting" | "ready" | "discovering" | "failed";
|
|
126
|
+
};
|
|
127
|
+
declare function getCurrentAgent<
|
|
128
|
+
T extends Agent<unknown, unknown> = Agent<unknown, unknown>,
|
|
129
|
+
>(): {
|
|
130
|
+
agent: T | undefined;
|
|
131
|
+
connection: Connection | undefined;
|
|
132
|
+
request: Request<unknown, CfProperties<unknown>> | undefined;
|
|
133
|
+
};
|
|
101
134
|
/**
|
|
102
135
|
* Base class for creating Agent implementations
|
|
103
136
|
* @template Env Environment type containing bindings
|
|
104
137
|
* @template State State type to store within the Agent
|
|
105
138
|
*/
|
|
106
139
|
declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
107
|
-
|
|
140
|
+
private _state;
|
|
141
|
+
private _ParentClass;
|
|
142
|
+
mcp: MCPClientManager;
|
|
108
143
|
/**
|
|
109
144
|
* Initial state for the Agent
|
|
110
145
|
* Override to provide default state values
|
|
@@ -133,6 +168,7 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
133
168
|
...values: (string | number | boolean | null)[]
|
|
134
169
|
): T[];
|
|
135
170
|
constructor(ctx: AgentContext, env: Env);
|
|
171
|
+
private _setStateInternal;
|
|
136
172
|
/**
|
|
137
173
|
* Update the Agent's state
|
|
138
174
|
* @param state New state to set
|
|
@@ -148,7 +184,8 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
148
184
|
* Called when the Agent receives an email
|
|
149
185
|
* @param email Email message to process
|
|
150
186
|
*/
|
|
151
|
-
onEmail(email: ForwardableEmailMessage): void
|
|
187
|
+
onEmail(email: ForwardableEmailMessage): Promise<void>;
|
|
188
|
+
private _tryCatch;
|
|
152
189
|
onError(connection: Connection, error: unknown): void | Promise<void>;
|
|
153
190
|
onError(error: unknown): void | Promise<void>;
|
|
154
191
|
/**
|
|
@@ -182,7 +219,6 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
182
219
|
* @returns Array of matching Schedule objects
|
|
183
220
|
*/
|
|
184
221
|
getSchedules<T = string>(criteria?: {
|
|
185
|
-
description?: string;
|
|
186
222
|
id?: string;
|
|
187
223
|
type?: "scheduled" | "delayed" | "cron";
|
|
188
224
|
timeRange?: {
|
|
@@ -196,15 +232,72 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
196
232
|
* @returns true if the task was cancelled, false otherwise
|
|
197
233
|
*/
|
|
198
234
|
cancelSchedule(id: string): Promise<boolean>;
|
|
235
|
+
private _scheduleNextAlarm;
|
|
199
236
|
/**
|
|
200
|
-
* Method called when an alarm fires
|
|
201
|
-
* Executes any scheduled tasks that are due
|
|
237
|
+
* Method called when an alarm fires.
|
|
238
|
+
* Executes any scheduled tasks that are due.
|
|
239
|
+
*
|
|
240
|
+
* @remarks
|
|
241
|
+
* To schedule a task, please use the `this.schedule` method instead.
|
|
242
|
+
* See {@link https://developers.cloudflare.com/agents/api-reference/schedule-tasks/}
|
|
202
243
|
*/
|
|
203
|
-
alarm()
|
|
244
|
+
readonly alarm: () => Promise<void>;
|
|
204
245
|
/**
|
|
205
246
|
* Destroy the Agent, removing all state and scheduled tasks
|
|
206
247
|
*/
|
|
207
248
|
destroy(): Promise<void>;
|
|
249
|
+
private _isCallable;
|
|
250
|
+
/**
|
|
251
|
+
* Connect to a new MCP Server
|
|
252
|
+
*
|
|
253
|
+
* @param url MCP Server SSE URL
|
|
254
|
+
* @param callbackHost Base host for the agent, used for the redirect URI.
|
|
255
|
+
* @param agentsPrefix agents routing prefix if not using `agents`
|
|
256
|
+
* @param options MCP client and transport (header) options
|
|
257
|
+
* @returns authUrl
|
|
258
|
+
*/
|
|
259
|
+
addMcpServer(
|
|
260
|
+
serverName: string,
|
|
261
|
+
url: string,
|
|
262
|
+
callbackHost: string,
|
|
263
|
+
agentsPrefix?: string,
|
|
264
|
+
options?: {
|
|
265
|
+
client?: ConstructorParameters<typeof Client>[1];
|
|
266
|
+
transport?: {
|
|
267
|
+
headers: HeadersInit;
|
|
268
|
+
};
|
|
269
|
+
}
|
|
270
|
+
): Promise<{
|
|
271
|
+
id: string;
|
|
272
|
+
authUrl: string | undefined;
|
|
273
|
+
}>;
|
|
274
|
+
_connectToMcpServerInternal(
|
|
275
|
+
serverName: string,
|
|
276
|
+
url: string,
|
|
277
|
+
callbackUrl: string,
|
|
278
|
+
options?: {
|
|
279
|
+
client?: ConstructorParameters<typeof Client>[1];
|
|
280
|
+
/**
|
|
281
|
+
* We don't expose the normal set of transport options because:
|
|
282
|
+
* 1) we can't serialize things like the auth provider or a fetch function into the DB for reconnection purposes
|
|
283
|
+
* 2) We probably want these options to be agnostic to the transport type (SSE vs Streamable)
|
|
284
|
+
*
|
|
285
|
+
* 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).
|
|
286
|
+
*/
|
|
287
|
+
transport?: {
|
|
288
|
+
headers?: HeadersInit;
|
|
289
|
+
};
|
|
290
|
+
},
|
|
291
|
+
reconnect?: {
|
|
292
|
+
id: string;
|
|
293
|
+
oauthClientId?: string;
|
|
294
|
+
}
|
|
295
|
+
): Promise<{
|
|
296
|
+
id: string;
|
|
297
|
+
authUrl: string | undefined;
|
|
298
|
+
}>;
|
|
299
|
+
removeMcpServer(id: string): Promise<void>;
|
|
300
|
+
private _getMcpServerStateInternal;
|
|
208
301
|
}
|
|
209
302
|
/**
|
|
210
303
|
* Namespace for creating Agent instances
|
|
@@ -269,7 +362,9 @@ declare function getAgentByName<Env, T extends Agent<Env>>(
|
|
|
269
362
|
* A wrapper for streaming responses in callable methods
|
|
270
363
|
*/
|
|
271
364
|
declare class StreamingResponse {
|
|
272
|
-
|
|
365
|
+
private _connection;
|
|
366
|
+
private _id;
|
|
367
|
+
private _closed;
|
|
273
368
|
constructor(connection: Connection, id: string);
|
|
274
369
|
/**
|
|
275
370
|
* Send a chunk of data to the client
|
|
@@ -289,13 +384,16 @@ export {
|
|
|
289
384
|
type AgentNamespace,
|
|
290
385
|
type AgentOptions,
|
|
291
386
|
type CallableMetadata,
|
|
387
|
+
type MCPServer,
|
|
388
|
+
type MCPServerMessage,
|
|
389
|
+
type MCPServersState,
|
|
292
390
|
type RPCRequest,
|
|
293
391
|
type RPCResponse,
|
|
294
392
|
type Schedule,
|
|
295
393
|
type StateUpdateMessage,
|
|
296
394
|
StreamingResponse,
|
|
297
|
-
WorkflowEntrypoint,
|
|
298
395
|
getAgentByName,
|
|
396
|
+
getCurrentAgent,
|
|
299
397
|
routeAgentEmail,
|
|
300
398
|
routeAgentRequest,
|
|
301
399
|
unstable_callable,
|
package/dist/index.js
CHANGED
|
@@ -1,18 +1,20 @@
|
|
|
1
1
|
import {
|
|
2
2
|
Agent,
|
|
3
3
|
StreamingResponse,
|
|
4
|
-
WorkflowEntrypoint,
|
|
5
4
|
getAgentByName,
|
|
5
|
+
getCurrentAgent,
|
|
6
6
|
routeAgentEmail,
|
|
7
7
|
routeAgentRequest,
|
|
8
8
|
unstable_callable
|
|
9
|
-
} from "./chunk-
|
|
10
|
-
import "./chunk-
|
|
9
|
+
} from "./chunk-RIYR6FR6.js";
|
|
10
|
+
import "./chunk-BZXOAZUX.js";
|
|
11
|
+
import "./chunk-QSGN3REV.js";
|
|
12
|
+
import "./chunk-Y67CHZBI.js";
|
|
11
13
|
export {
|
|
12
14
|
Agent,
|
|
13
15
|
StreamingResponse,
|
|
14
|
-
WorkflowEntrypoint,
|
|
15
16
|
getAgentByName,
|
|
17
|
+
getCurrentAgent,
|
|
16
18
|
routeAgentEmail,
|
|
17
19
|
routeAgentRequest,
|
|
18
20
|
unstable_callable
|