agents 0.0.0-dd6a9e3 → 0.0.0-de1d9aa

package/dist/ai-types-D5YoPrBZ.d.ts

@@ -0,0 +1,95 @@
+import { UIMessage } from "ai";
+
+//#region src/ai-types.d.ts
+
+/**
+ * Enum for message types to improve type safety and maintainability
+ */
+declare enum MessageType {
+  CF_AGENT_CHAT_MESSAGES = "cf_agent_chat_messages",
+  CF_AGENT_USE_CHAT_REQUEST = "cf_agent_use_chat_request",
+  CF_AGENT_USE_CHAT_RESPONSE = "cf_agent_use_chat_response",
+  CF_AGENT_CHAT_CLEAR = "cf_agent_chat_clear",
+  CF_AGENT_CHAT_REQUEST_CANCEL = "cf_agent_chat_request_cancel",
+  CF_AGENT_MCP_SERVERS = "cf_agent_mcp_servers",
+  CF_MCP_AGENT_EVENT = "cf_mcp_agent_event",
+  CF_AGENT_STATE = "cf_agent_state",
+  RPC = "rpc"
+}
+/**
+ * Types of messages sent from the Agent to clients
+ */
+type OutgoingMessage<ChatMessage extends UIMessage = UIMessage> =
+  | {
+      /** Indicates this message is a command to clear chat history */
+      type: MessageType.CF_AGENT_CHAT_CLEAR;
+    }
+  | {
+      /** Indicates this message contains updated chat messages */
+      type: MessageType.CF_AGENT_CHAT_MESSAGES;
+      /** Array of chat messages */
+      messages: ChatMessage[];
+    }
+  | {
+      /** Indicates this message is a response to a chat request */
+      type: MessageType.CF_AGENT_USE_CHAT_RESPONSE;
+      /** Unique ID of the request this response corresponds to */
+      id: string;
+      /** Content body of the response */
+      body: string;
+      /** Whether this is the final chunk of the response */
+      done: boolean;
+      /** Whether this response contains an error */
+      error?: boolean;
+    }
+  | {
+      /** Indicates this message is a command to clear chat history */
+      type: MessageType.CF_AGENT_CHAT_CLEAR;
+    };
+/**
+ * Types of messages sent from clients to the Agent
+ */
+type IncomingMessage<ChatMessage extends UIMessage = UIMessage> =
+  | {
+      /** Indicates this message is a command to clear chat history */
+      type: MessageType.CF_AGENT_CHAT_CLEAR;
+    }
+  | {
+      /** Indicates this message is a request to the chat API */
+      type: MessageType.CF_AGENT_USE_CHAT_REQUEST;
+      /** Unique ID for this request */
+      id: string;
+      /** Request initialization options */
+      init: Pick<
+        RequestInit,
+        | "method"
+        | "keepalive"
+        | "headers"
+        | "body"
+        | "redirect"
+        | "integrity"
+        | "credentials"
+        | "mode"
+        | "referrer"
+        | "referrerPolicy"
+        | "window"
+      >;
+    }
+  | {
+      /** Indicates this message is a command to clear chat history */
+      type: MessageType.CF_AGENT_CHAT_CLEAR;
+    }
+  | {
+      /** Indicates this message contains updated chat messages */
+      type: MessageType.CF_AGENT_CHAT_MESSAGES;
+      /** Array of chat messages */
+      messages: ChatMessage[];
+    }
+  | {
+      /** Indicates the user wants to stop generation of this message */
+      type: MessageType.CF_AGENT_CHAT_REQUEST_CANCEL;
+      id: string;
+    };
+//#endregion
+export { MessageType as n, OutgoingMessage as r, IncomingMessage as t };
+//# sourceMappingURL=ai-types-D5YoPrBZ.d.ts.map

package/dist/ai-types.d.ts

@@ -1,69 +1,6 @@
-import { Message } from "ai";
-
-/**
- * Types of messages sent from the Agent to clients
- */
-type OutgoingMessage =
-  | {
-      /** Indicates this message contains updated chat messages */
-      type: "cf_agent_chat_messages";
-      /** Array of chat messages */
-      messages: Message[];
-    }
-  | {
-      /** Indicates this message is a response to a chat request */
-      type: "cf_agent_use_chat_response";
-      /** Unique ID of the request this response corresponds to */
-      id: string;
-      /** Content body of the response */
-      body: string;
-      /** Whether this is the final chunk of the response */
-      done: boolean;
-    }
-  | {
-      /** Indicates this message contains updated chat messages */
-      type: "cf_agent_chat_messages";
-      /** Array of chat messages */
-      messages: Message[];
-    }
-  | {
-      /** Indicates this message is a command to clear chat history */
-      type: "cf_agent_chat_clear";
-    };
-/**
- * Types of messages sent from clients to the Agent
- */
-type IncomingMessage =
-  | {
-      /** Indicates this message is a request to the chat API */
-      type: "cf_agent_use_chat_request";
-      /** Unique ID for this request */
-      id: string;
-      /** Request initialization options */
-      init: Pick<
-        RequestInit,
-        | "method"
-        | "keepalive"
-        | "headers"
-        | "body"
-        | "redirect"
-        | "integrity"
-        | "credentials"
-        | "mode"
-        | "referrer"
-        | "referrerPolicy"
-        | "window"
-      >;
-    }
-  | {
-      /** Indicates this message is a command to clear chat history */
-      type: "cf_agent_chat_clear";
-    }
-  | {
-      /** Indicates this message contains updated chat messages */
-      type: "cf_agent_chat_messages";
-      /** Array of chat messages */
-      messages: Message[];
-    };
-
-export type { IncomingMessage, OutgoingMessage };
+import {
+  n as MessageType,
+  r as OutgoingMessage,
+  t as IncomingMessage
+} from "./ai-types-D5YoPrBZ.js";
+export { IncomingMessage, MessageType, OutgoingMessage };

package/dist/ai-types.js

@@ -1 +1,3 @@
-//# sourceMappingURL=ai-types.js.map
+import { t as MessageType } from "./ai-types-B3aQaFv3.js";
+
+export { MessageType };

package/dist/client-BfiZ3HQd.js

@@ -0,0 +1,117 @@
+import { t as MessageType } from "./ai-types-B3aQaFv3.js";
+import { PartySocket } from "partysocket";
+
+//#region src/client.ts
+/**
+* Convert a camelCase string to a kebab-case string
+* @param str The string to convert
+* @returns The kebab-case string
+*/
+function camelCaseToKebabCase(str) {
+	if (str === str.toUpperCase() && str !== str.toLowerCase()) return str.toLowerCase().replace(/_/g, "-");
+	let kebabified = str.replace(/[A-Z]/g, (letter) => `-${letter.toLowerCase()}`);
+	kebabified = kebabified.startsWith("-") ? kebabified.slice(1) : kebabified;
+	return kebabified.replace(/_/g, "-").replace(/-$/, "");
+}
+/**
+* WebSocket client for connecting to an Agent
+*/
+var AgentClient = class extends PartySocket {
+	/**
+	* @deprecated Use agentFetch instead
+	*/
+	static fetch(_opts) {
+		throw new Error("AgentClient.fetch is not implemented, use agentFetch instead");
+	}
+	constructor(options) {
+		const agentNamespace = camelCaseToKebabCase(options.agent);
+		super({
+			party: agentNamespace,
+			prefix: "agents",
+			room: options.name || "default",
+			...options
+		});
+		this._pendingCalls = /* @__PURE__ */ new Map();
+		this.agent = agentNamespace;
+		this.name = options.name || "default";
+		this.options = options;
+		this.addEventListener("message", (event) => {
+			if (typeof event.data === "string") {
+				let parsedMessage;
+				try {
+					parsedMessage = JSON.parse(event.data);
+				} catch (_error) {
+					return;
+				}
+				if (parsedMessage.type === MessageType.CF_AGENT_STATE) {
+					this.options.onStateUpdate?.(parsedMessage.state, "server");
+					return;
+				}
+				if (parsedMessage.type === MessageType.RPC) {
+					const response = parsedMessage;
+					const pending = this._pendingCalls.get(response.id);
+					if (!pending) return;
+					if (!response.success) {
+						pending.reject(new Error(response.error));
+						this._pendingCalls.delete(response.id);
+						pending.stream?.onError?.(response.error);
+						return;
+					}
+					if ("done" in response) if (response.done) {
+						pending.resolve(response.result);
+						this._pendingCalls.delete(response.id);
+						pending.stream?.onDone?.(response.result);
+					} else pending.stream?.onChunk?.(response.result);
+					else {
+						pending.resolve(response.result);
+						this._pendingCalls.delete(response.id);
+					}
+				}
+			}
+		});
+	}
+	setState(state) {
+		this.send(JSON.stringify({
+			state,
+			type: MessageType.CF_AGENT_STATE
+		}));
+		this.options.onStateUpdate?.(state, "client");
+	}
+	async call(method, args = [], streamOptions) {
+		return new Promise((resolve, reject) => {
+			const id = Math.random().toString(36).slice(2);
+			this._pendingCalls.set(id, {
+				reject,
+				resolve: (value) => resolve(value),
+				stream: streamOptions,
+				type: null
+			});
+			const request = {
+				args,
+				id,
+				method,
+				type: MessageType.RPC
+			};
+			this.send(JSON.stringify(request));
+		});
+	}
+};
+/**
+* Make an HTTP request to an Agent
+* @param opts Connection options
+* @param init Request initialization options
+* @returns Promise resolving to a Response
+*/
+function agentFetch(opts, init) {
+	const agentNamespace = camelCaseToKebabCase(opts.agent);
+	return PartySocket.fetch({
+		party: agentNamespace,
+		prefix: "agents",
+		room: opts.name || "default",
+		...opts
+	}, init);
+}
+
+//#endregion
+export { agentFetch as n, camelCaseToKebabCase as r, AgentClient as t };
+//# sourceMappingURL=client-BfiZ3HQd.js.map

package/dist/client-BfiZ3HQd.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client-BfiZ3HQd.js","names":["parsedMessage: Record<string, unknown>","request: RPCRequest"],"sources":["../src/client.ts"],"sourcesContent":["import {\n  type PartyFetchOptions,\n  PartySocket,\n  type PartySocketOptions\n} from \"partysocket\";\nimport type { RPCRequest, RPCResponse } from \"./\";\nimport type {\n  SerializableReturnValue,\n  SerializableValue\n} from \"./serializable\";\nimport { MessageType } from \"./ai-types\";\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 * Convert a camelCase string to a kebab-case string\n * @param str The string to convert\n * @returns The kebab-case string\n */\nexport function camelCaseToKebabCase(str: string): string {\n  // If string is all uppercase, convert to lowercase\n  if (str === str.toUpperCase() && str !== str.toLowerCase()) {\n    return str.toLowerCase().replace(/_/g, \"-\");\n  }\n\n  // Otherwise handle camelCase to kebab-case\n  let kebabified = str.replace(\n    /[A-Z]/g,\n    (letter) => `-${letter.toLowerCase()}`\n  );\n  kebabified = kebabified.startsWith(\"-\") ? kebabified.slice(1) : kebabified;\n  // Convert any remaining underscores to hyphens and remove trailing -'s\n  return kebabified.replace(/_/g, \"-\").replace(/-$/, \"\");\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  private options: AgentClientOptions<State>;\n  private _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    const agentNamespace = camelCaseToKebabCase(options.agent);\n    super({\n      party: agentNamespace,\n      prefix: \"agents\",\n      room: options.name || \"default\",\n      ...options\n    });\n    this.agent = agentNamespace;\n    this.name = options.name || \"default\";\n    this.options = options;\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 === MessageType.CF_AGENT_STATE) {\n          this.options.onStateUpdate?.(parsedMessage.state as State, \"server\");\n          return;\n        }\n        if (parsedMessage.type === MessageType.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({ state, type: MessageType.CF_AGENT_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  call<T extends SerializableReturnValue>(\n    method: string,\n    args?: SerializableValue[],\n    streamOptions?: StreamOptions\n  ): Promise<T>;\n  call<T = unknown>(\n    method: string,\n    args?: unknown[],\n    streamOptions?: StreamOptions\n  ): Promise<T>;\n  async call<T>(\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        reject,\n        resolve: (value: unknown) => resolve(value as T),\n        stream: streamOptions,\n        type: null as T\n      });\n\n      const request: RPCRequest = {\n        args,\n        id,\n        method,\n        type: MessageType.RPC\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  const agentNamespace = camelCaseToKebabCase(opts.agent);\n\n  return PartySocket.fetch(\n    {\n      party: agentNamespace,\n      prefix: \"agents\",\n      room: opts.name || \"default\",\n      ...opts\n    },\n    init\n  );\n}\n"],"mappings":";;;;;;;;;AAyDA,SAAgB,qBAAqB,KAAqB;AAExD,KAAI,QAAQ,IAAI,aAAa,IAAI,QAAQ,IAAI,aAAa,CACxD,QAAO,IAAI,aAAa,CAAC,QAAQ,MAAM,IAAI;CAI7C,IAAI,aAAa,IAAI,QACnB,WACC,WAAW,IAAI,OAAO,aAAa,GACrC;AACD,cAAa,WAAW,WAAW,IAAI,GAAG,WAAW,MAAM,EAAE,GAAG;AAEhE,QAAO,WAAW,QAAQ,MAAM,IAAI,CAAC,QAAQ,MAAM,GAAG;;;;;AAMxD,IAAa,cAAb,cAAkD,YAAY;;;;CAI5D,OAAO,MAAM,OAA6C;AACxD,QAAM,IAAI,MACR,+DACD;;CAeH,YAAY,SAAoC;EAC9C,MAAM,iBAAiB,qBAAqB,QAAQ,MAAM;AAC1D,QAAM;GACJ,OAAO;GACP,QAAQ;GACR,MAAM,QAAQ,QAAQ;GACtB,GAAG;GACJ,CAAC;uCAjBoB,IAAI,KAQzB;AAUD,OAAK,QAAQ;AACb,OAAK,OAAO,QAAQ,QAAQ;AAC5B,OAAK,UAAU;AAEf,OAAK,iBAAiB,YAAY,UAAU;AAC1C,OAAI,OAAO,MAAM,SAAS,UAAU;IAClC,IAAIA;AACJ,QAAI;AACF,qBAAgB,KAAK,MAAM,MAAM,KAAK;aAC/B,QAAQ;AAGf;;AAEF,QAAI,cAAc,SAAS,YAAY,gBAAgB;AACrD,UAAK,QAAQ,gBAAgB,cAAc,OAAgB,SAAS;AACpE;;AAEF,QAAI,cAAc,SAAS,YAAY,KAAK;KAC1C,MAAM,WAAW;KACjB,MAAM,UAAU,KAAK,cAAc,IAAI,SAAS,GAAG;AACnD,SAAI,CAAC,QAAS;AAEd,SAAI,CAAC,SAAS,SAAS;AACrB,cAAQ,OAAO,IAAI,MAAM,SAAS,MAAM,CAAC;AACzC,WAAK,cAAc,OAAO,SAAS,GAAG;AACtC,cAAQ,QAAQ,UAAU,SAAS,MAAM;AACzC;;AAIF,SAAI,UAAU,SACZ,KAAI,SAAS,MAAM;AACjB,cAAQ,QAAQ,SAAS,OAAO;AAChC,WAAK,cAAc,OAAO,SAAS,GAAG;AACtC,cAAQ,QAAQ,SAAS,SAAS,OAAO;WAEzC,SAAQ,QAAQ,UAAU,SAAS,OAAO;UAEvC;AAEL,cAAQ,QAAQ,SAAS,OAAO;AAChC,WAAK,cAAc,OAAO,SAAS,GAAG;;;;IAI5C;;CAGJ,SAAS,OAAc;AACrB,OAAK,KAAK,KAAK,UAAU;GAAE;GAAO,MAAM,YAAY;GAAgB,CAAC,CAAC;AACtE,OAAK,QAAQ,gBAAgB,OAAO,SAAS;;CAoB/C,MAAM,KACJ,QACA,OAAkB,EAAE,EACpB,eACY;AACZ,SAAO,IAAI,SAAY,SAAS,WAAW;GACzC,MAAM,KAAK,KAAK,QAAQ,CAAC,SAAS,GAAG,CAAC,MAAM,EAAE;AAC9C,QAAK,cAAc,IAAI,IAAI;IACzB;IACA,UAAU,UAAmB,QAAQ,MAAW;IAChD,QAAQ;IACR,MAAM;IACP,CAAC;GAEF,MAAMC,UAAsB;IAC1B;IACA;IACA;IACA,MAAM,YAAY;IACnB;AAED,QAAK,KAAK,KAAK,UAAU,QAAQ,CAAC;IAClC;;;;;;;;;AAUN,SAAgB,WAAW,MAA+B,MAAoB;CAC5E,MAAM,iBAAiB,qBAAqB,KAAK,MAAM;AAEvD,QAAO,YAAY,MACjB;EACE,OAAO;EACP,QAAQ;EACR,MAAM,KAAK,QAAQ;EACnB,GAAG;EACJ,EACD,KACD"}