agents 0.0.0-7c40201 → 0.0.0-7f84d28
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 -4
- 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-KRBQHBPA.js → chunk-RIYR6FR6.js} +319 -141
- 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 +109 -18
- package/dist/index.js +8 -8
- 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 +383 -65
- package/dist/chunk-HMLY7DHA.js +0 -16
- package/dist/chunk-KRBQHBPA.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,7 +1,15 @@
|
|
|
1
1
|
import { Server, Connection, PartyServerOptions } from "partyserver";
|
|
2
2
|
export { Connection, ConnectionContext, WSMessage } from "partyserver";
|
|
3
|
-
import {
|
|
4
|
-
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";
|
|
5
13
|
|
|
6
14
|
/**
|
|
7
15
|
* RPC request message from client
|
|
@@ -60,10 +68,6 @@ declare function unstable_callable(
|
|
|
60
68
|
target: (this: This, ...args: Args) => Return,
|
|
61
69
|
context: ClassMethodDecoratorContext
|
|
62
70
|
) => (this: This, ...args: Args) => Return;
|
|
63
|
-
/**
|
|
64
|
-
* A class for creating workflow entry points that can be used with Cloudflare Workers
|
|
65
|
-
*/
|
|
66
|
-
declare class WorkflowEntrypoint extends WorkflowEntrypoint$1 {}
|
|
67
71
|
/**
|
|
68
72
|
* Represents a scheduled task within an Agent
|
|
69
73
|
* @template T Type of the payload data
|
|
@@ -99,18 +103,43 @@ type Schedule<T = string> = {
|
|
|
99
103
|
cron: string;
|
|
100
104
|
}
|
|
101
105
|
);
|
|
102
|
-
|
|
103
|
-
|
|
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;
|
|
104
131
|
connection: Connection | undefined;
|
|
105
|
-
request: Request | undefined;
|
|
106
|
-
}
|
|
132
|
+
request: Request<unknown, CfProperties<unknown>> | undefined;
|
|
133
|
+
};
|
|
107
134
|
/**
|
|
108
135
|
* Base class for creating Agent implementations
|
|
109
136
|
* @template Env Environment type containing bindings
|
|
110
137
|
* @template State State type to store within the Agent
|
|
111
138
|
*/
|
|
112
139
|
declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
113
|
-
|
|
140
|
+
private _state;
|
|
141
|
+
private _ParentClass;
|
|
142
|
+
mcp: MCPClientManager;
|
|
114
143
|
/**
|
|
115
144
|
* Initial state for the Agent
|
|
116
145
|
* Override to provide default state values
|
|
@@ -139,6 +168,7 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
139
168
|
...values: (string | number | boolean | null)[]
|
|
140
169
|
): T[];
|
|
141
170
|
constructor(ctx: AgentContext, env: Env);
|
|
171
|
+
private _setStateInternal;
|
|
142
172
|
/**
|
|
143
173
|
* Update the Agent's state
|
|
144
174
|
* @param state New state to set
|
|
@@ -155,6 +185,7 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
155
185
|
* @param email Email message to process
|
|
156
186
|
*/
|
|
157
187
|
onEmail(email: ForwardableEmailMessage): Promise<void>;
|
|
188
|
+
private _tryCatch;
|
|
158
189
|
onError(connection: Connection, error: unknown): void | Promise<void>;
|
|
159
190
|
onError(error: unknown): void | Promise<void>;
|
|
160
191
|
/**
|
|
@@ -188,7 +219,6 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
188
219
|
* @returns Array of matching Schedule objects
|
|
189
220
|
*/
|
|
190
221
|
getSchedules<T = string>(criteria?: {
|
|
191
|
-
description?: string;
|
|
192
222
|
id?: string;
|
|
193
223
|
type?: "scheduled" | "delayed" | "cron";
|
|
194
224
|
timeRange?: {
|
|
@@ -202,15 +232,72 @@ declare class Agent<Env, State = unknown> extends Server<Env> {
|
|
|
202
232
|
* @returns true if the task was cancelled, false otherwise
|
|
203
233
|
*/
|
|
204
234
|
cancelSchedule(id: string): Promise<boolean>;
|
|
235
|
+
private _scheduleNextAlarm;
|
|
205
236
|
/**
|
|
206
|
-
* Method called when an alarm fires
|
|
207
|
-
* 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/}
|
|
208
243
|
*/
|
|
209
|
-
alarm()
|
|
244
|
+
readonly alarm: () => Promise<void>;
|
|
210
245
|
/**
|
|
211
246
|
* Destroy the Agent, removing all state and scheduled tasks
|
|
212
247
|
*/
|
|
213
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;
|
|
214
301
|
}
|
|
215
302
|
/**
|
|
216
303
|
* Namespace for creating Agent instances
|
|
@@ -275,7 +362,9 @@ declare function getAgentByName<Env, T extends Agent<Env>>(
|
|
|
275
362
|
* A wrapper for streaming responses in callable methods
|
|
276
363
|
*/
|
|
277
364
|
declare class StreamingResponse {
|
|
278
|
-
|
|
365
|
+
private _connection;
|
|
366
|
+
private _id;
|
|
367
|
+
private _closed;
|
|
279
368
|
constructor(connection: Connection, id: string);
|
|
280
369
|
/**
|
|
281
370
|
* Send a chunk of data to the client
|
|
@@ -295,15 +384,17 @@ export {
|
|
|
295
384
|
type AgentNamespace,
|
|
296
385
|
type AgentOptions,
|
|
297
386
|
type CallableMetadata,
|
|
387
|
+
type MCPServer,
|
|
388
|
+
type MCPServerMessage,
|
|
389
|
+
type MCPServersState,
|
|
298
390
|
type RPCRequest,
|
|
299
391
|
type RPCResponse,
|
|
300
392
|
type Schedule,
|
|
301
393
|
type StateUpdateMessage,
|
|
302
394
|
StreamingResponse,
|
|
303
|
-
WorkflowEntrypoint,
|
|
304
395
|
getAgentByName,
|
|
396
|
+
getCurrentAgent,
|
|
305
397
|
routeAgentEmail,
|
|
306
398
|
routeAgentRequest,
|
|
307
399
|
unstable_callable,
|
|
308
|
-
unstable_context,
|
|
309
400
|
};
|
package/dist/index.js
CHANGED
|
@@ -1,22 +1,22 @@
|
|
|
1
1
|
import {
|
|
2
2
|
Agent,
|
|
3
3
|
StreamingResponse,
|
|
4
|
-
WorkflowEntrypoint,
|
|
5
4
|
getAgentByName,
|
|
5
|
+
getCurrentAgent,
|
|
6
6
|
routeAgentEmail,
|
|
7
7
|
routeAgentRequest,
|
|
8
|
-
unstable_callable
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
import "./chunk-
|
|
8
|
+
unstable_callable
|
|
9
|
+
} from "./chunk-RIYR6FR6.js";
|
|
10
|
+
import "./chunk-BZXOAZUX.js";
|
|
11
|
+
import "./chunk-QSGN3REV.js";
|
|
12
|
+
import "./chunk-Y67CHZBI.js";
|
|
12
13
|
export {
|
|
13
14
|
Agent,
|
|
14
15
|
StreamingResponse,
|
|
15
|
-
WorkflowEntrypoint,
|
|
16
16
|
getAgentByName,
|
|
17
|
+
getCurrentAgent,
|
|
17
18
|
routeAgentEmail,
|
|
18
19
|
routeAgentRequest,
|
|
19
|
-
unstable_callable
|
|
20
|
-
unstable_context
|
|
20
|
+
unstable_callable
|
|
21
21
|
};
|
|
22
22
|
//# sourceMappingURL=index.js.map
|