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.
Files changed (41) hide show
  1. package/dist/ai-chat-agent.d.ts +49 -4
  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-KRBQHBPA.js → chunk-RIYR6FR6.js} +319 -141
  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 +109 -18
  20. package/dist/index.js +8 -8
  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 +383 -65
  36. package/dist/chunk-HMLY7DHA.js +0 -16
  37. package/dist/chunk-KRBQHBPA.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,7 +1,15 @@
1
1
  import { Server, Connection, PartyServerOptions } from "partyserver";
2
2
  export { Connection, ConnectionContext, WSMessage } from "partyserver";
3
- import { AsyncLocalStorage } from "node:async_hooks";
4
- 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";
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
- declare const unstable_context: AsyncLocalStorage<{
103
- agent: Agent<unknown>;
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
- #private;
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(): Promise<void>;
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
- #private;
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
- unstable_context
10
- } from "./chunk-KRBQHBPA.js";
11
- import "./chunk-HMLY7DHA.js";
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