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.
Files changed (41) hide show
  1. package/dist/ai-chat-agent.d.ts +49 -3
  2. package/dist/ai-chat-agent.js +129 -66
  3. package/dist/ai-chat-agent.js.map +1 -1
  4. package/dist/ai-react.d.ts +12 -0
  5. package/dist/ai-react.js +38 -24
  6. package/dist/ai-react.js.map +1 -1
  7. package/dist/ai-types.d.ts +5 -0
  8. package/dist/chunk-BZXOAZUX.js +106 -0
  9. package/dist/chunk-BZXOAZUX.js.map +1 -0
  10. package/dist/chunk-QSGN3REV.js +123 -0
  11. package/dist/chunk-QSGN3REV.js.map +1 -0
  12. package/dist/{chunk-X6BBKLSC.js → chunk-RIYR6FR6.js} +382 -173
  13. package/dist/chunk-RIYR6FR6.js.map +1 -0
  14. package/dist/chunk-Y67CHZBI.js +464 -0
  15. package/dist/chunk-Y67CHZBI.js.map +1 -0
  16. package/dist/client.d.ts +9 -1
  17. package/dist/client.js +6 -133
  18. package/dist/client.js.map +1 -1
  19. package/dist/index.d.ts +111 -13
  20. package/dist/index.js +6 -4
  21. package/dist/mcp/client.d.ts +783 -0
  22. package/dist/mcp/client.js +9 -0
  23. package/dist/mcp/do-oauth-client-provider.d.ts +41 -0
  24. package/dist/mcp/do-oauth-client-provider.js +7 -0
  25. package/dist/mcp/do-oauth-client-provider.js.map +1 -0
  26. package/dist/mcp/index.d.ts +84 -0
  27. package/dist/mcp/index.js +779 -0
  28. package/dist/mcp/index.js.map +1 -0
  29. package/dist/react.d.ts +14 -0
  30. package/dist/react.js +38 -29
  31. package/dist/react.js.map +1 -1
  32. package/dist/schedule.js +0 -2
  33. package/dist/schedule.js.map +1 -1
  34. package/package.json +36 -8
  35. package/src/index.ts +481 -126
  36. package/dist/chunk-HMLY7DHA.js +0 -16
  37. package/dist/chunk-X6BBKLSC.js.map +0 -1
  38. package/dist/mcp.d.ts +0 -58
  39. package/dist/mcp.js +0 -945
  40. package/dist/mcp.js.map +0 -1
  41. /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
- __privateAdd,
3
- __privateGet,
4
- __privateSet
5
- } from "./chunk-HMLY7DHA.js";
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
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/client.ts"],"sourcesContent":["import {\n PartySocket,\n type PartySocketOptions,\n type PartyFetchOptions,\n} from \"partysocket\";\nimport type { RPCRequest, RPCResponse } from \"./\";\n\n/**\n * Options for creating an AgentClient\n */\nexport type AgentClientOptions<State = unknown> = Omit<\n PartySocketOptions,\n \"party\" | \"room\"\n> & {\n /** Name of the agent to connect to */\n agent: string;\n /** Name of the specific Agent instance */\n name?: string;\n /** Called when the Agent's state is updated */\n onStateUpdate?: (state: State, source: \"server\" | \"client\") => void;\n};\n\n/**\n * Options for streaming RPC calls\n */\nexport type StreamOptions = {\n /** Called when a chunk of data is received */\n onChunk?: (chunk: unknown) => void;\n /** Called when the stream ends */\n onDone?: (finalChunk: unknown) => void;\n /** Called when an error occurs */\n onError?: (error: string) => void;\n};\n\n/**\n * Options for the agentFetch function\n */\nexport type AgentClientFetchOptions = Omit<\n PartyFetchOptions,\n \"party\" | \"room\"\n> & {\n /** Name of the agent to connect to */\n agent: string;\n /** Name of the specific Agent instance */\n name?: string;\n};\n\n/**\n * WebSocket client for connecting to an Agent\n */\nexport class AgentClient<State = unknown> extends PartySocket {\n /**\n * @deprecated Use agentFetch instead\n */\n static fetch(_opts: PartyFetchOptions): Promise<Response> {\n throw new Error(\n \"AgentClient.fetch is not implemented, use agentFetch instead\"\n );\n }\n agent: string;\n name: string;\n #options: AgentClientOptions<State>;\n #pendingCalls = new Map<\n string,\n {\n resolve: (value: unknown) => void;\n reject: (error: Error) => void;\n stream?: StreamOptions;\n type?: unknown;\n }\n >();\n\n constructor(options: AgentClientOptions<State>) {\n super({\n prefix: \"agents\",\n party: options.agent,\n room: options.name || \"default\",\n ...options,\n });\n this.agent = options.agent;\n this.name = options.name || \"default\";\n this.#options = options;\n\n // warn if agent or name isn't in lowercase\n if (this.agent !== this.agent.toLowerCase()) {\n console.warn(\n `Agent name: ${this.agent} should probably be in lowercase. Received: ${this.agent}`\n );\n }\n if (this.name !== this.name.toLowerCase()) {\n console.warn(\n `Agent instance name: ${this.name} should probably be in lowercase. Received: ${this.name}`\n );\n }\n\n this.addEventListener(\"message\", (event) => {\n if (typeof event.data === \"string\") {\n let parsedMessage: Record<string, unknown>;\n try {\n parsedMessage = JSON.parse(event.data);\n } catch (error) {\n // silently ignore invalid messages for now\n // TODO: log errors with log levels\n return;\n }\n if (parsedMessage.type === \"cf_agent_state\") {\n this.#options.onStateUpdate?.(parsedMessage.state as State, \"server\");\n return;\n }\n if (parsedMessage.type === \"rpc\") {\n const response = parsedMessage as RPCResponse;\n const pending = this.#pendingCalls.get(response.id);\n if (!pending) return;\n\n if (!response.success) {\n pending.reject(new Error(response.error));\n this.#pendingCalls.delete(response.id);\n pending.stream?.onError?.(response.error);\n return;\n }\n\n // Handle streaming responses\n if (\"done\" in response) {\n if (response.done) {\n pending.resolve(response.result);\n this.#pendingCalls.delete(response.id);\n pending.stream?.onDone?.(response.result);\n } else {\n pending.stream?.onChunk?.(response.result);\n }\n } else {\n // Non-streaming response\n pending.resolve(response.result);\n this.#pendingCalls.delete(response.id);\n }\n }\n }\n });\n }\n\n setState(state: State) {\n this.send(JSON.stringify({ type: \"cf_agent_state\", state }));\n this.#options.onStateUpdate?.(state, \"client\");\n }\n\n /**\n * Call a method on the Agent\n * @param method Name of the method to call\n * @param args Arguments to pass to the method\n * @param streamOptions Options for handling streaming responses\n * @returns Promise that resolves with the method's return value\n */\n async call<T = unknown>(\n method: string,\n args: unknown[] = [],\n streamOptions?: StreamOptions\n ): Promise<T> {\n return new Promise<T>((resolve, reject) => {\n const id = Math.random().toString(36).slice(2);\n this.#pendingCalls.set(id, {\n resolve: (value: unknown) => resolve(value as T),\n reject,\n stream: streamOptions,\n type: null as T,\n });\n\n const request: RPCRequest = {\n type: \"rpc\",\n id,\n method,\n args,\n };\n\n this.send(JSON.stringify(request));\n });\n }\n}\n\n/**\n * Make an HTTP request to an Agent\n * @param opts Connection options\n * @param init Request initialization options\n * @returns Promise resolving to a Response\n */\nexport function agentFetch(opts: AgentClientFetchOptions, init?: RequestInit) {\n // warn if agent or name isn't in lowercase\n if (opts.agent !== opts.agent.toLowerCase()) {\n console.warn(\n `Agent name: ${opts.agent} should probably be in lowercase. Received: ${opts.agent}`\n );\n }\n if (opts.name && opts.name !== opts.name.toLowerCase()) {\n console.warn(\n `Agent instance name: ${opts.name} should probably be in lowercase. Received: ${opts.name}`\n );\n }\n\n return PartySocket.fetch(\n {\n prefix: \"agents\",\n party: opts.agent,\n room: opts.name || \"default\",\n ...opts,\n },\n init\n );\n}\n"],"mappings":";;;;;;;AAAA;AAAA,EACE;AAAA,OAGK;AAJP;AAkDO,IAAM,cAAN,cAA2C,YAAY;AAAA,EAsB5D,YAAY,SAAoC;AAC9C,UAAM;AAAA,MACJ,QAAQ;AAAA,MACR,OAAO,QAAQ;AAAA,MACf,MAAM,QAAQ,QAAQ;AAAA,MACtB,GAAG;AAAA,IACL,CAAC;AAjBH;AACA,sCAAgB,oBAAI,IAQlB;AASA,SAAK,QAAQ,QAAQ;AACrB,SAAK,OAAO,QAAQ,QAAQ;AAC5B,uBAAK,UAAW;AAGhB,QAAI,KAAK,UAAU,KAAK,MAAM,YAAY,GAAG;AAC3C,cAAQ;AAAA,QACN,eAAe,KAAK,KAAK,+CAA+C,KAAK,KAAK;AAAA,MACpF;AAAA,IACF;AACA,QAAI,KAAK,SAAS,KAAK,KAAK,YAAY,GAAG;AACzC,cAAQ;AAAA,QACN,wBAAwB,KAAK,IAAI,+CAA+C,KAAK,IAAI;AAAA,MAC3F;AAAA,IACF;AAEA,SAAK,iBAAiB,WAAW,CAAC,UAAU;AAC1C,UAAI,OAAO,MAAM,SAAS,UAAU;AAClC,YAAI;AACJ,YAAI;AACF,0BAAgB,KAAK,MAAM,MAAM,IAAI;AAAA,QACvC,SAAS,OAAO;AAGd;AAAA,QACF;AACA,YAAI,cAAc,SAAS,kBAAkB;AAC3C,6BAAK,UAAS,gBAAgB,cAAc,OAAgB,QAAQ;AACpE;AAAA,QACF;AACA,YAAI,cAAc,SAAS,OAAO;AAChC,gBAAM,WAAW;AACjB,gBAAM,UAAU,mBAAK,eAAc,IAAI,SAAS,EAAE;AAClD,cAAI,CAAC,QAAS;AAEd,cAAI,CAAC,SAAS,SAAS;AACrB,oBAAQ,OAAO,IAAI,MAAM,SAAS,KAAK,CAAC;AACxC,+BAAK,eAAc,OAAO,SAAS,EAAE;AACrC,oBAAQ,QAAQ,UAAU,SAAS,KAAK;AACxC;AAAA,UACF;AAGA,cAAI,UAAU,UAAU;AACtB,gBAAI,SAAS,MAAM;AACjB,sBAAQ,QAAQ,SAAS,MAAM;AAC/B,iCAAK,eAAc,OAAO,SAAS,EAAE;AACrC,sBAAQ,QAAQ,SAAS,SAAS,MAAM;AAAA,YAC1C,OAAO;AACL,sBAAQ,QAAQ,UAAU,SAAS,MAAM;AAAA,YAC3C;AAAA,UACF,OAAO;AAEL,oBAAQ,QAAQ,SAAS,MAAM;AAC/B,+BAAK,eAAc,OAAO,SAAS,EAAE;AAAA,UACvC;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA,EApFA,OAAO,MAAM,OAA6C;AACxD,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAAA,EAkFA,SAAS,OAAc;AACrB,SAAK,KAAK,KAAK,UAAU,EAAE,MAAM,kBAAkB,MAAM,CAAC,CAAC;AAC3D,uBAAK,UAAS,gBAAgB,OAAO,QAAQ;AAAA,EAC/C;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,KACJ,QACA,OAAkB,CAAC,GACnB,eACY;AACZ,WAAO,IAAI,QAAW,CAAC,SAAS,WAAW;AACzC,YAAM,KAAK,KAAK,OAAO,EAAE,SAAS,EAAE,EAAE,MAAM,CAAC;AAC7C,yBAAK,eAAc,IAAI,IAAI;AAAA,QACzB,SAAS,CAAC,UAAmB,QAAQ,KAAU;AAAA,QAC/C;AAAA,QACA,QAAQ;AAAA,QACR,MAAM;AAAA,MACR,CAAC;AAED,YAAM,UAAsB;AAAA,QAC1B,MAAM;AAAA,QACN;AAAA,QACA;AAAA,QACA;AAAA,MACF;AAEA,WAAK,KAAK,KAAK,UAAU,OAAO,CAAC;AAAA,IACnC,CAAC;AAAA,EACH;AACF;AAnHE;AACA;AA0HK,SAAS,WAAW,MAA+B,MAAoB;AAE5E,MAAI,KAAK,UAAU,KAAK,MAAM,YAAY,GAAG;AAC3C,YAAQ;AAAA,MACN,eAAe,KAAK,KAAK,+CAA+C,KAAK,KAAK;AAAA,IACpF;AAAA,EACF;AACA,MAAI,KAAK,QAAQ,KAAK,SAAS,KAAK,KAAK,YAAY,GAAG;AACtD,YAAQ;AAAA,MACN,wBAAwB,KAAK,IAAI,+CAA+C,KAAK,IAAI;AAAA,IAC3F;AAAA,EACF;AAEA,SAAO,YAAY;AAAA,IACjB;AAAA,MACE,QAAQ;AAAA,MACR,OAAO,KAAK;AAAA,MACZ,MAAM,KAAK,QAAQ;AAAA,MACnB,GAAG;AAAA,IACL;AAAA,IACA;AAAA,EACF;AACF;","names":[]}
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 { WorkflowEntrypoint as WorkflowEntrypoint$1 } from "cloudflare:workers";
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
- #private;
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(): Promise<void>;
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
- #private;
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-X6BBKLSC.js";
10
- import "./chunk-HMLY7DHA.js";
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