agents 0.0.0-df716f2 → 0.0.0-e03246e

package/dist/ai-chat-agent.d.ts

@@ -1,31 +1,36 @@
-import { Agent, AgentContext } from './index.js';
-import { Message, StreamTextOnFinishCallback, ToolSet } from 'ai';
-import { Connection, WSMessage } from 'partyserver';
-import 'node:async_hooks';
-import 'cloudflare:workers';
+import { Agent, AgentContext } from "./index.js";
+import { Message, StreamTextOnFinishCallback, ToolSet } from "ai";
+import { Connection, WSMessage } from "partyserver";
+import "node:async_hooks";
+import "cloudflare:workers";
 
 /**
  * Extension of Agent with built-in chat capabilities
  * @template Env Environment type containing bindings
  */
-declare class AIChatAgent<Env = unknown, State = unknown> extends Agent<Env, State> {
-    #private;
-    /** Array of chat messages for the current conversation */
-    messages: Message[];
-    constructor(ctx: AgentContext, env: Env);
-    onMessage(connection: Connection, message: WSMessage): Promise<void>;
-    onRequest(request: Request): Promise<Response>;
-    /**
-     * Handle incoming chat messages and generate a response
-     * @param onFinish Callback to be called when the response is finished
-     * @returns Response to send to the client or undefined
-     */
-    onChatMessage(onFinish: StreamTextOnFinishCallback<ToolSet>): Promise<Response | undefined>;
-    /**
-     * Save messages on the server side and trigger AI response
-     * @param messages Chat messages to save
-     */
-    saveMessages(messages: Message[]): Promise<void>;
+declare class AIChatAgent<Env = unknown, State = unknown> extends Agent<
+  Env,
+  State
+> {
+  #private;
+  /** Array of chat messages for the current conversation */
+  messages: Message[];
+  constructor(ctx: AgentContext, env: Env);
+  onMessage(connection: Connection, message: WSMessage): Promise<void>;
+  onRequest(request: Request): Promise<Response>;
+  /**
+   * Handle incoming chat messages and generate a response
+   * @param onFinish Callback to be called when the response is finished
+   * @returns Response to send to the client or undefined
+   */
+  onChatMessage(
+    onFinish: StreamTextOnFinishCallback<ToolSet>
+  ): Promise<Response | undefined>;
+  /**
+   * Save messages on the server side and trigger AI response
+   * @param messages Chat messages to save
+   */
+  saveMessages(messages: Message[]): Promise<void>;
 }
 
 export { AIChatAgent };

package/dist/ai-chat-agent.js

@@ -0,0 +1,168 @@
+import {
+  Agent
+} from "./chunk-YMUU7QHV.js";
+import {
+  __privateAdd,
+  __privateMethod
+} from "./chunk-HMLY7DHA.js";
+
+// src/ai-chat-agent.ts
+import { appendResponseMessages } from "ai";
+var decoder = new TextDecoder();
+var _AIChatAgent_instances, broadcastChatMessage_fn, tryCatch_fn, persistMessages_fn, reply_fn;
+var AIChatAgent = class extends Agent {
+  constructor(ctx, env) {
+    super(ctx, env);
+    __privateAdd(this, _AIChatAgent_instances);
+    this.sql`create table if not exists cf_ai_chat_agent_messages (
+      id text primary key,
+      message text not null,
+      created_at datetime default current_timestamp
+    )`;
+    this.messages = (this.sql`select * from cf_ai_chat_agent_messages` || []).map((row) => {
+      return JSON.parse(row.message);
+    });
+  }
+  async onMessage(connection, message) {
+    if (typeof message === "string") {
+      let data;
+      try {
+        data = JSON.parse(message);
+      } catch (error) {
+        return;
+      }
+      if (data.type === "cf_agent_use_chat_request" && data.init.method === "POST") {
+        const {
+          method,
+          keepalive,
+          headers,
+          body,
+          // we're reading this
+          redirect,
+          integrity,
+          credentials,
+          mode,
+          referrer,
+          referrerPolicy,
+          window
+          // dispatcher,
+          // duplex
+        } = data.init;
+        const { messages } = JSON.parse(body);
+        __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
+          type: "cf_agent_chat_messages",
+          messages
+        }, [connection.id]);
+        await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, messages, [connection.id]);
+        return __privateMethod(this, _AIChatAgent_instances, tryCatch_fn).call(this, async () => {
+          const response = await this.onChatMessage(async ({ response: response2 }) => {
+            const finalMessages = appendResponseMessages({
+              messages,
+              responseMessages: response2.messages
+            });
+            await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, finalMessages, [connection.id]);
+          });
+          if (response) {
+            await __privateMethod(this, _AIChatAgent_instances, reply_fn).call(this, data.id, response);
+          }
+        });
+      }
+      if (data.type === "cf_agent_chat_clear") {
+        this.sql`delete from cf_ai_chat_agent_messages`;
+        this.messages = [];
+        __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
+          type: "cf_agent_chat_clear"
+        }, [connection.id]);
+      } else if (data.type === "cf_agent_chat_messages") {
+        await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, data.messages, [connection.id]);
+      }
+    }
+  }
+  async onRequest(request) {
+    return __privateMethod(this, _AIChatAgent_instances, tryCatch_fn).call(this, () => {
+      const url = new URL(request.url);
+      if (url.pathname.endsWith("/get-messages")) {
+        const messages = (this.sql`select * from cf_ai_chat_agent_messages` || []).map((row) => {
+          return JSON.parse(row.message);
+        });
+        return Response.json(messages);
+      }
+      return super.onRequest(request);
+    });
+  }
+  /**
+   * Handle incoming chat messages and generate a response
+   * @param onFinish Callback to be called when the response is finished
+   * @returns Response to send to the client or undefined
+   */
+  async onChatMessage(onFinish) {
+    throw new Error(
+      "recieved a chat message, override onChatMessage and return a Response to send to the client"
+    );
+  }
+  /**
+   * Save messages on the server side and trigger AI response
+   * @param messages Chat messages to save
+   */
+  async saveMessages(messages) {
+    await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, messages);
+    const response = await this.onChatMessage(async ({ response: response2 }) => {
+      const finalMessages = appendResponseMessages({
+        messages,
+        responseMessages: response2.messages
+      });
+      await __privateMethod(this, _AIChatAgent_instances, persistMessages_fn).call(this, finalMessages, []);
+    });
+    if (response) {
+      for await (const chunk of response.body) {
+        decoder.decode(chunk);
+      }
+      response.body?.cancel();
+    }
+  }
+};
+_AIChatAgent_instances = new WeakSet();
+broadcastChatMessage_fn = function(message, exclude) {
+  this.broadcast(JSON.stringify(message), exclude);
+};
+tryCatch_fn = async function(fn) {
+  try {
+    return await fn();
+  } catch (e) {
+    throw this.onError(e);
+  }
+};
+persistMessages_fn = async function(messages, excludeBroadcastIds = []) {
+  this.sql`delete from cf_ai_chat_agent_messages`;
+  for (const message of messages) {
+    this.sql`insert into cf_ai_chat_agent_messages (id, message) values (${message.id},${JSON.stringify(message)})`;
+  }
+  this.messages = messages;
+  __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
+    type: "cf_agent_chat_messages",
+    messages
+  }, excludeBroadcastIds);
+};
+reply_fn = async function(id, response) {
+  return __privateMethod(this, _AIChatAgent_instances, tryCatch_fn).call(this, async () => {
+    for await (const chunk of response.body) {
+      const body = decoder.decode(chunk);
+      __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
+        id,
+        type: "cf_agent_use_chat_response",
+        body,
+        done: false
+      });
+    }
+    __privateMethod(this, _AIChatAgent_instances, broadcastChatMessage_fn).call(this, {
+      id,
+      type: "cf_agent_use_chat_response",
+      body: "",
+      done: true
+    });
+  });
+};
+export {
+  AIChatAgent
+};
+//# sourceMappingURL=ai-chat-agent.js.map

package/dist/ai-chat-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ai-chat-agent.ts"],"sourcesContent":["import { Agent, type AgentContext, type Connection, type WSMessage } from \"./\";\nimport type {\n  Message as ChatMessage,\n  StreamTextOnFinishCallback,\n  ToolSet,\n} from \"ai\";\nimport { appendResponseMessages } from \"ai\";\nimport type { OutgoingMessage, IncomingMessage } from \"./ai-types\";\nconst decoder = new TextDecoder();\n\n/**\n * Extension of Agent with built-in chat capabilities\n * @template Env Environment type containing bindings\n */\nexport class AIChatAgent<Env = unknown, State = unknown> extends Agent<\n  Env,\n  State\n> {\n  /** Array of chat messages for the current conversation */\n  messages: ChatMessage[];\n  constructor(ctx: AgentContext, env: Env) {\n    super(ctx, env);\n    this.sql`create table if not exists cf_ai_chat_agent_messages (\n      id text primary key,\n      message text not null,\n      created_at datetime default current_timestamp\n    )`;\n    this.messages = (\n      this.sql`select * from cf_ai_chat_agent_messages` || []\n    ).map((row) => {\n      return JSON.parse(row.message as string);\n    });\n  }\n\n  #broadcastChatMessage(message: OutgoingMessage, exclude?: string[]) {\n    this.broadcast(JSON.stringify(message), exclude);\n  }\n\n  override async onMessage(connection: Connection, message: WSMessage) {\n    if (typeof message === \"string\") {\n      let data: IncomingMessage;\n      try {\n        data = JSON.parse(message) as IncomingMessage;\n      } catch (error) {\n        // silently ignore invalid messages for now\n        // TODO: log errors with log levels\n        return;\n      }\n      if (\n        data.type === \"cf_agent_use_chat_request\" &&\n        data.init.method === \"POST\"\n      ) {\n        const {\n          method,\n          keepalive,\n          headers,\n          body, // we're reading this\n          redirect,\n          integrity,\n          credentials,\n          mode,\n          referrer,\n          referrerPolicy,\n          window,\n          // dispatcher,\n          // duplex\n        } = data.init;\n        const { messages } = JSON.parse(body as string);\n        this.#broadcastChatMessage(\n          {\n            type: \"cf_agent_chat_messages\",\n            messages,\n          },\n          [connection.id]\n        );\n        await this.#persistMessages(messages, [connection.id]);\n        return this.#tryCatch(async () => {\n          const response = await this.onChatMessage(async ({ response }) => {\n            const finalMessages = appendResponseMessages({\n              messages,\n              responseMessages: response.messages,\n            });\n\n            await this.#persistMessages(finalMessages, [connection.id]);\n          });\n          if (response) {\n            await this.#reply(data.id, response);\n          }\n        });\n      }\n      if (data.type === \"cf_agent_chat_clear\") {\n        this.sql`delete from cf_ai_chat_agent_messages`;\n        this.messages = [];\n        this.#broadcastChatMessage(\n          {\n            type: \"cf_agent_chat_clear\",\n          },\n          [connection.id]\n        );\n      } else if (data.type === \"cf_agent_chat_messages\") {\n        // replace the messages with the new ones\n        await this.#persistMessages(data.messages, [connection.id]);\n      }\n    }\n  }\n\n  override async onRequest(request: Request): Promise<Response> {\n    return this.#tryCatch(() => {\n      const url = new URL(request.url);\n      if (url.pathname.endsWith(\"/get-messages\")) {\n        const messages = (\n          this.sql`select * from cf_ai_chat_agent_messages` || []\n        ).map((row) => {\n          return JSON.parse(row.message as string);\n        });\n        return Response.json(messages);\n      }\n      return super.onRequest(request);\n    });\n  }\n\n  async #tryCatch<T>(fn: () => T | Promise<T>) {\n    try {\n      return await fn();\n    } catch (e) {\n      throw this.onError(e);\n    }\n  }\n\n  /**\n   * Handle incoming chat messages and generate a response\n   * @param onFinish Callback to be called when the response is finished\n   * @returns Response to send to the client or undefined\n   */\n  async onChatMessage(\n    onFinish: StreamTextOnFinishCallback<ToolSet>\n  ): Promise<Response | undefined> {\n    throw new Error(\n      \"recieved a chat message, override onChatMessage and return a Response to send to the client\"\n    );\n  }\n\n  /**\n   * Save messages on the server side and trigger AI response\n   * @param messages Chat messages to save\n   */\n  async saveMessages(messages: ChatMessage[]) {\n    await this.#persistMessages(messages);\n    const response = await this.onChatMessage(async ({ response }) => {\n      const finalMessages = appendResponseMessages({\n        messages,\n        responseMessages: response.messages,\n      });\n\n      await this.#persistMessages(finalMessages, []);\n    });\n    if (response) {\n      // we're just going to drain the body\n      // @ts-ignore TODO: fix this type error\n      for await (const chunk of response.body!) {\n        decoder.decode(chunk);\n      }\n      response.body?.cancel();\n    }\n  }\n\n  async #persistMessages(\n    messages: ChatMessage[],\n    excludeBroadcastIds: string[] = []\n  ) {\n    this.sql`delete from cf_ai_chat_agent_messages`;\n    for (const message of messages) {\n      this.sql`insert into cf_ai_chat_agent_messages (id, message) values (${\n        message.id\n      },${JSON.stringify(message)})`;\n    }\n    this.messages = messages;\n    this.#broadcastChatMessage(\n      {\n        type: \"cf_agent_chat_messages\",\n        messages: messages,\n      },\n      excludeBroadcastIds\n    );\n  }\n\n  async #reply(id: string, response: Response) {\n    // now take chunks out from dataStreamResponse and send them to the client\n    return this.#tryCatch(async () => {\n      // @ts-expect-error TODO: fix this type error\n      for await (const chunk of response.body!) {\n        const body = decoder.decode(chunk);\n\n        this.#broadcastChatMessage({\n          id,\n          type: \"cf_agent_use_chat_response\",\n          body,\n          done: false,\n        });\n      }\n\n      this.#broadcastChatMessage({\n        id,\n        type: \"cf_agent_use_chat_response\",\n        body: \"\",\n        done: true,\n      });\n    });\n  }\n}\n"],"mappings":";;;;;;;;;AAMA,SAAS,8BAA8B;AAEvC,IAAM,UAAU,IAAI,YAAY;AARhC;AAcO,IAAM,cAAN,cAA0D,MAG/D;AAAA,EAGA,YAAY,KAAmB,KAAU;AACvC,UAAM,KAAK,GAAG;AAPX;AAQH,SAAK;AAAA;AAAA;AAAA;AAAA;AAKL,SAAK,YACH,KAAK,gDAAgD,CAAC,GACtD,IAAI,CAAC,QAAQ;AACb,aAAO,KAAK,MAAM,IAAI,OAAiB;AAAA,IACzC,CAAC;AAAA,EACH;AAAA,EAMA,MAAe,UAAU,YAAwB,SAAoB;AACnE,QAAI,OAAO,YAAY,UAAU;AAC/B,UAAI;AACJ,UAAI;AACF,eAAO,KAAK,MAAM,OAAO;AAAA,MAC3B,SAAS,OAAO;AAGd;AAAA,MACF;AACA,UACE,KAAK,SAAS,+BACd,KAAK,KAAK,WAAW,QACrB;AACA,cAAM;AAAA,UACJ;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA;AAAA;AAAA,QAGF,IAAI,KAAK;AACT,cAAM,EAAE,SAAS,IAAI,KAAK,MAAM,IAAc;AAC9C,8BAAK,iDAAL,WACE;AAAA,UACE,MAAM;AAAA,UACN;AAAA,QACF,GACA,CAAC,WAAW,EAAE;AAEhB,cAAM,sBAAK,4CAAL,WAAsB,UAAU,CAAC,WAAW,EAAE;AACpD,eAAO,sBAAK,qCAAL,WAAe,YAAY;AAChC,gBAAM,WAAW,MAAM,KAAK,cAAc,OAAO,EAAE,UAAAA,UAAS,MAAM;AAChE,kBAAM,gBAAgB,uBAAuB;AAAA,cAC3C;AAAA,cACA,kBAAkBA,UAAS;AAAA,YAC7B,CAAC;AAED,kBAAM,sBAAK,4CAAL,WAAsB,eAAe,CAAC,WAAW,EAAE;AAAA,UAC3D,CAAC;AACD,cAAI,UAAU;AACZ,kBAAM,sBAAK,kCAAL,WAAY,KAAK,IAAI;AAAA,UAC7B;AAAA,QACF;AAAA,MACF;AACA,UAAI,KAAK,SAAS,uBAAuB;AACvC,aAAK;AACL,aAAK,WAAW,CAAC;AACjB,8BAAK,iDAAL,WACE;AAAA,UACE,MAAM;AAAA,QACR,GACA,CAAC,WAAW,EAAE;AAAA,MAElB,WAAW,KAAK,SAAS,0BAA0B;AAEjD,cAAM,sBAAK,4CAAL,WAAsB,KAAK,UAAU,CAAC,WAAW,EAAE;AAAA,MAC3D;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAe,UAAU,SAAqC;AAC5D,WAAO,sBAAK,qCAAL,WAAe,MAAM;AAC1B,YAAM,MAAM,IAAI,IAAI,QAAQ,GAAG;AAC/B,UAAI,IAAI,SAAS,SAAS,eAAe,GAAG;AAC1C,cAAM,YACJ,KAAK,gDAAgD,CAAC,GACtD,IAAI,CAAC,QAAQ;AACb,iBAAO,KAAK,MAAM,IAAI,OAAiB;AAAA,QACzC,CAAC;AACD,eAAO,SAAS,KAAK,QAAQ;AAAA,MAC/B;AACA,aAAO,MAAM,UAAU,OAAO;AAAA,IAChC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAeA,MAAM,cACJ,UAC+B;AAC/B,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,aAAa,UAAyB;AAC1C,UAAM,sBAAK,4CAAL,WAAsB;AAC5B,UAAM,WAAW,MAAM,KAAK,cAAc,OAAO,EAAE,UAAAA,UAAS,MAAM;AAChE,YAAM,gBAAgB,uBAAuB;AAAA,QAC3C;AAAA,QACA,kBAAkBA,UAAS;AAAA,MAC7B,CAAC;AAED,YAAM,sBAAK,4CAAL,WAAsB,eAAe,CAAC;AAAA,IAC9C,CAAC;AACD,QAAI,UAAU;AAGZ,uBAAiB,SAAS,SAAS,MAAO;AACxC,gBAAQ,OAAO,KAAK;AAAA,MACtB;AACA,eAAS,MAAM,OAAO;AAAA,IACxB;AAAA,EACF;AA6CF;AAnMO;AAoBL,0BAAqB,SAAC,SAA0B,SAAoB;AAClE,OAAK,UAAU,KAAK,UAAU,OAAO,GAAG,OAAO;AACjD;AAqFM,cAAY,eAAC,IAA0B;AAC3C,MAAI;AACF,WAAO,MAAM,GAAG;AAAA,EAClB,SAAS,GAAG;AACV,UAAM,KAAK,QAAQ,CAAC;AAAA,EACtB;AACF;AAuCM,qBAAgB,eACpB,UACA,sBAAgC,CAAC,GACjC;AACA,OAAK;AACL,aAAW,WAAW,UAAU;AAC9B,SAAK,kEACH,QAAQ,EACV,IAAI,KAAK,UAAU,OAAO,CAAC;AAAA,EAC7B;AACA,OAAK,WAAW;AAChB,wBAAK,iDAAL,WACE;AAAA,IACE,MAAM;AAAA,IACN;AAAA,EACF,GACA;AAEJ;AAEM,WAAM,eAAC,IAAY,UAAoB;AAE3C,SAAO,sBAAK,qCAAL,WAAe,YAAY;AAEhC,qBAAiB,SAAS,SAAS,MAAO;AACxC,YAAM,OAAO,QAAQ,OAAO,KAAK;AAEjC,4BAAK,iDAAL,WAA2B;AAAA,QACzB;AAAA,QACA,MAAM;AAAA,QACN;AAAA,QACA,MAAM;AAAA,MACR;AAAA,IACF;AAEA,0BAAK,iDAAL,WAA2B;AAAA,MACzB;AAAA,MACA,MAAM;AAAA,MACN,MAAM;AAAA,MACN,MAAM;AAAA,IACR;AAAA,EACF;AACF;","names":["response"]}

package/dist/ai-react.d.ts

@@ -1,60 +1,88 @@
-import * as ai from 'ai';
-import { Message } from 'ai';
-import { useChat } from '@ai-sdk/react';
-import { useAgent } from './react.js';
-import 'partysocket';
-import 'partysocket/react';
-import './client.js';
+import * as ai from "ai";
+import { Message } from "ai";
+import { useChat } from "@ai-sdk/react";
+import { useAgent } from "./react.js";
+import "partysocket";
+import "partysocket/react";
+import "./client.js";
 
 type GetInitialMessagesOptions = {
-    agent: string;
-    name: string;
-    url: string;
+  agent: string;
+  name: string;
+  url: string;
 };
 /**
  * Options for the useAgentChat hook
  */
-type UseAgentChatOptions<State> = Omit<Parameters<typeof useChat>[0] & {
+type UseAgentChatOptions<State> = Omit<
+  Parameters<typeof useChat>[0] & {
     /** Agent connection from useAgent */
     agent: ReturnType<typeof useAgent<State>>;
-    getInitialMessages?: undefined | null | ((options: GetInitialMessagesOptions) => Promise<Message[]>);
-}, "fetch">;
+    getInitialMessages?:
+      | undefined
+      | null
+      | ((options: GetInitialMessagesOptions) => Promise<Message[]>);
+  },
+  "fetch"
+>;
 /**
  * React hook for building AI chat interfaces using an Agent
  * @param options Chat options including the agent connection
  * @returns Chat interface controls and state with added clearHistory method
  */
-declare function useAgentChat<State = unknown>(options: UseAgentChatOptions<State>): {
-    /**
-     * Set the chat messages and synchronize with the Agent
-     * @param messages New messages to set
-     */
-    setMessages: (messages: Message[]) => void;
-    /**
-     * Clear chat history on both client and Agent
-     */
-    clearHistory: () => void;
-    messages: ai.UIMessage[];
-    error: undefined | Error;
-    append: (message: Message | ai.CreateMessage, chatRequestOptions?: ai.ChatRequestOptions) => Promise<string | null | undefined>;
-    reload: (chatRequestOptions?: ai.ChatRequestOptions) => Promise<string | null | undefined>;
-    stop: () => void;
-    input: string;
-    setInput: React.Dispatch<React.SetStateAction<string>>;
-    handleInputChange: (e: React.ChangeEvent<HTMLInputElement> | React.ChangeEvent<HTMLTextAreaElement>) => void;
-    handleSubmit: (event?: {
-        preventDefault?: () => void;
-    }, chatRequestOptions?: ai.ChatRequestOptions) => void;
-    metadata?: Object;
-    isLoading: boolean;
-    status: "submitted" | "streaming" | "ready" | "error";
-    data?: ai.JSONValue[];
-    setData: (data: ai.JSONValue[] | undefined | ((data: ai.JSONValue[] | undefined) => ai.JSONValue[] | undefined)) => void;
-    id: string;
-    addToolResult: ({ toolCallId, result, }: {
-        toolCallId: string;
-        result: any;
-    }) => void;
+declare function useAgentChat<State = unknown>(
+  options: UseAgentChatOptions<State>
+): {
+  /**
+   * Set the chat messages and synchronize with the Agent
+   * @param messages New messages to set
+   */
+  setMessages: (messages: Message[]) => void;
+  /**
+   * Clear chat history on both client and Agent
+   */
+  clearHistory: () => void;
+  messages: ai.UIMessage[];
+  error: undefined | Error;
+  append: (
+    message: Message | ai.CreateMessage,
+    chatRequestOptions?: ai.ChatRequestOptions
+  ) => Promise<string | null | undefined>;
+  reload: (
+    chatRequestOptions?: ai.ChatRequestOptions
+  ) => Promise<string | null | undefined>;
+  stop: () => void;
+  input: string;
+  setInput: React.Dispatch<React.SetStateAction<string>>;
+  handleInputChange: (
+    e:
+      | React.ChangeEvent<HTMLInputElement>
+      | React.ChangeEvent<HTMLTextAreaElement>
+  ) => void;
+  handleSubmit: (
+    event?: {
+      preventDefault?: () => void;
+    },
+    chatRequestOptions?: ai.ChatRequestOptions
+  ) => void;
+  metadata?: Object;
+  isLoading: boolean;
+  status: "submitted" | "streaming" | "ready" | "error";
+  data?: ai.JSONValue[];
+  setData: (
+    data:
+      | ai.JSONValue[]
+      | undefined
+      | ((data: ai.JSONValue[] | undefined) => ai.JSONValue[] | undefined)
+  ) => void;
+  id: string;
+  addToolResult: ({
+    toolCallId,
+    result,
+  }: {
+    toolCallId: string;
+    result: any;
+  }) => void;
 };
 
 export { useAgentChat };

package/dist/ai-react.js

@@ -0,0 +1,196 @@
+import "./chunk-HMLY7DHA.js";
+
+// src/ai-react.tsx
+import { useChat } from "@ai-sdk/react";
+import { useEffect, use } from "react";
+var requestCache = /* @__PURE__ */ new Map();
+function useAgentChat(options) {
+  const { agent, getInitialMessages, ...rest } = options;
+  const agentUrl = new URL(
+    `${// @ts-expect-error we're using a protected _url property that includes query params
+    (agent._url || agent._pkurl)?.replace("ws://", "http://").replace("wss://", "https://")}`
+  );
+  agentUrl.searchParams.delete("_pk");
+  const agentUrlString = agentUrl.toString();
+  async function defaultGetInitialMessagesFetch({
+    url
+  }) {
+    const getMessagesUrl = new URL(url);
+    getMessagesUrl.pathname += "/get-messages";
+    const response = await fetch(getMessagesUrl.toString(), {
+      headers: options.headers,
+      credentials: options.credentials
+    });
+    return response.json();
+  }
+  const getInitialMessagesFetch = getInitialMessages || defaultGetInitialMessagesFetch;
+  function doGetInitialMessages(getInitialMessagesOptions) {
+    if (requestCache.has(agentUrlString)) {
+      return requestCache.get(agentUrlString);
+    }
+    const promise = getInitialMessagesFetch(getInitialMessagesOptions);
+    requestCache.set(agentUrlString, promise);
+    return promise;
+  }
+  const initialMessages = getInitialMessages !== null ? use(
+    doGetInitialMessages({
+      agent: agent.agent,
+      name: agent.name,
+      url: agentUrlString
+    })
+  ) : rest.initialMessages;
+  useEffect(() => {
+    return () => {
+      requestCache.delete(agentUrlString);
+    };
+  }, [agentUrlString]);
+  async function aiFetch(request, options2 = {}) {
+    const {
+      method,
+      keepalive,
+      headers,
+      body,
+      redirect,
+      integrity,
+      signal,
+      credentials,
+      mode,
+      referrer,
+      referrerPolicy,
+      window
+      //  dispatcher, duplex
+    } = options2;
+    const id = crypto.randomUUID();
+    const abortController = new AbortController();
+    signal?.addEventListener("abort", () => {
+      abortController.abort();
+    });
+    agent.addEventListener(
+      "message",
+      (event) => {
+        let data;
+        try {
+          data = JSON.parse(event.data);
+        } catch (error) {
+          return;
+        }
+        if (data.type === "cf_agent_use_chat_response") {
+          if (data.id === id) {
+            controller.enqueue(new TextEncoder().encode(data.body));
+            if (data.done) {
+              controller.close();
+              abortController.abort();
+            }
+          }
+        }
+      },
+      { signal: abortController.signal }
+    );
+    let controller;
+    const stream = new ReadableStream({
+      start(c) {
+        controller = c;
+      }
+    });
+    agent.send(
+      JSON.stringify({
+        type: "cf_agent_use_chat_request",
+        id,
+        url: request.toString(),
+        init: {
+          method,
+          keepalive,
+          headers,
+          body,
+          redirect,
+          integrity,
+          credentials,
+          mode,
+          referrer,
+          referrerPolicy,
+          window
+          // dispatcher,
+          // duplex
+        }
+      })
+    );
+    return new Response(stream);
+  }
+  const useChatHelpers = useChat({
+    initialMessages,
+    sendExtraMessageFields: true,
+    fetch: aiFetch,
+    ...rest
+  });
+  useEffect(() => {
+    function onClearHistory(event) {
+      if (typeof event.data !== "string") {
+        return;
+      }
+      let data;
+      try {
+        data = JSON.parse(event.data);
+      } catch (error) {
+        return;
+      }
+      if (data.type === "cf_agent_chat_clear") {
+        useChatHelpers.setMessages([]);
+      }
+    }
+    function onMessages(event) {
+      if (typeof event.data !== "string") {
+        return;
+      }
+      let data;
+      try {
+        data = JSON.parse(event.data);
+      } catch (error) {
+        return;
+      }
+      if (data.type === "cf_agent_chat_messages") {
+        useChatHelpers.setMessages(data.messages);
+      }
+    }
+    agent.addEventListener("message", onClearHistory);
+    agent.addEventListener("message", onMessages);
+    return () => {
+      agent.removeEventListener("message", onClearHistory);
+      agent.removeEventListener("message", onMessages);
+    };
+  }, [
+    agent.addEventListener,
+    agent.removeEventListener,
+    useChatHelpers.setMessages
+  ]);
+  return {
+    ...useChatHelpers,
+    /**
+     * Set the chat messages and synchronize with the Agent
+     * @param messages New messages to set
+     */
+    setMessages: (messages) => {
+      useChatHelpers.setMessages(messages);
+      agent.send(
+        JSON.stringify({
+          type: "cf_agent_chat_messages",
+          messages
+        })
+      );
+    },
+    /**
+     * Clear chat history on both client and Agent
+     */
+    clearHistory: () => {
+      useChatHelpers.setMessages([]);
+      agent.send(
+        JSON.stringify({
+          type: "cf_agent_chat_clear"
+        })
+      );
+    }
+  };
+}
+export {
+  useAgentChat
+};
+//# sourceMappingURL=ai-react.js.map

package/dist/ai-react.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ai-react.tsx"],"sourcesContent":["import { useChat } from \"@ai-sdk/react\";\nimport type { Message } from \"ai\";\nimport type { useAgent } from \"./react\";\nimport { useEffect, use } from \"react\";\nimport type { OutgoingMessage } from \"./ai-types\";\n\ntype GetInitialMessagesOptions = {\n  agent: string;\n  name: string;\n  url: string;\n};\n\n/**\n * Options for the useAgentChat hook\n */\ntype UseAgentChatOptions<State> = Omit<\n  Parameters<typeof useChat>[0] & {\n    /** Agent connection from useAgent */\n    agent: ReturnType<typeof useAgent<State>>;\n    getInitialMessages?:\n      | undefined\n      | null\n      // | (() => Message[])\n      | ((options: GetInitialMessagesOptions) => Promise<Message[]>);\n  },\n  \"fetch\"\n>;\n\n// TODO: clear cache when the agent is unmounted?\nconst requestCache = new Map<string, Promise<Message[]>>();\n\n/**\n * React hook for building AI chat interfaces using an Agent\n * @param options Chat options including the agent connection\n * @returns Chat interface controls and state with added clearHistory method\n */\nexport function useAgentChat<State = unknown>(\n  options: UseAgentChatOptions<State>\n) {\n  const { agent, getInitialMessages, ...rest } = options;\n\n  const agentUrl = new URL(\n    `${// @ts-expect-error we're using a protected _url property that includes query params\n    ((agent._url as string | null) || agent._pkurl)\n      ?.replace(\"ws://\", \"http://\")\n      .replace(\"wss://\", \"https://\")}`\n  );\n\n  // delete the _pk query param\n  agentUrl.searchParams.delete(\"_pk\");\n  const agentUrlString = agentUrl.toString();\n\n  async function defaultGetInitialMessagesFetch({\n    url,\n  }: GetInitialMessagesOptions) {\n    const getMessagesUrl = new URL(url);\n    getMessagesUrl.pathname += \"/get-messages\";\n    const response = await fetch(getMessagesUrl.toString(), {\n      headers: options.headers,\n      credentials: options.credentials,\n    });\n    return response.json<Message[]>();\n  }\n\n  const getInitialMessagesFetch =\n    getInitialMessages || defaultGetInitialMessagesFetch;\n\n  function doGetInitialMessages(\n    getInitialMessagesOptions: GetInitialMessagesOptions\n  ) {\n    if (requestCache.has(agentUrlString)) {\n      return requestCache.get(agentUrlString)!;\n    }\n    const promise = getInitialMessagesFetch(getInitialMessagesOptions);\n    requestCache.set(agentUrlString, promise);\n    return promise;\n  }\n\n  const initialMessages =\n    getInitialMessages !== null\n      ? use(\n          doGetInitialMessages({\n            agent: agent.agent,\n            name: agent.name,\n            url: agentUrlString,\n          })\n        )\n      : rest.initialMessages;\n\n  useEffect(() => {\n    return () => {\n      requestCache.delete(agentUrlString);\n    };\n  }, [agentUrlString]);\n\n  async function aiFetch(\n    request: RequestInfo | URL,\n    options: RequestInit = {}\n  ) {\n    // we're going to use a websocket to do the actual \"fetching\"\n    // but still satisfy the type signature of the fetch function\n    // so we'll return a promise that resolves to a response\n\n    const {\n      method,\n      keepalive,\n      headers,\n      body,\n      redirect,\n      integrity,\n      signal,\n      credentials,\n      mode,\n      referrer,\n      referrerPolicy,\n      window,\n      //  dispatcher, duplex\n    } = options;\n    const id = crypto.randomUUID();\n    const abortController = new AbortController();\n\n    signal?.addEventListener(\"abort\", () => {\n      abortController.abort();\n    });\n\n    agent.addEventListener(\n      \"message\",\n      (event) => {\n        let data: OutgoingMessage;\n        try {\n          data = JSON.parse(event.data) as OutgoingMessage;\n        } catch (error) {\n          // silently ignore invalid messages for now\n          // TODO: log errors with log levels\n          return;\n        }\n        if (data.type === \"cf_agent_use_chat_response\") {\n          if (data.id === id) {\n            controller.enqueue(new TextEncoder().encode(data.body));\n            if (data.done) {\n              controller.close();\n              abortController.abort();\n            }\n          }\n        }\n      },\n      { signal: abortController.signal }\n    );\n\n    let controller: ReadableStreamDefaultController;\n\n    const stream = new ReadableStream({\n      start(c) {\n        controller = c;\n      },\n    });\n\n    agent.send(\n      JSON.stringify({\n        type: \"cf_agent_use_chat_request\",\n        id,\n        url: request.toString(),\n        init: {\n          method,\n          keepalive,\n          headers,\n          body,\n          redirect,\n          integrity,\n          credentials,\n          mode,\n          referrer,\n          referrerPolicy,\n          window,\n          // dispatcher,\n          // duplex\n        },\n      })\n    );\n\n    return new Response(stream);\n  }\n  const useChatHelpers = useChat({\n    initialMessages,\n    sendExtraMessageFields: true,\n    fetch: aiFetch,\n    ...rest,\n  });\n\n  useEffect(() => {\n    function onClearHistory(event: MessageEvent) {\n      if (typeof event.data !== \"string\") {\n        return;\n      }\n      let data: OutgoingMessage;\n      try {\n        data = JSON.parse(event.data) as OutgoingMessage;\n      } catch (error) {\n        // silently ignore invalid messages for now\n        // TODO: log errors with log levels\n        return;\n      }\n      if (data.type === \"cf_agent_chat_clear\") {\n        useChatHelpers.setMessages([]);\n      }\n    }\n\n    function onMessages(event: MessageEvent) {\n      if (typeof event.data !== \"string\") {\n        return;\n      }\n      let data: OutgoingMessage;\n      try {\n        data = JSON.parse(event.data) as OutgoingMessage;\n      } catch (error) {\n        // silently ignore invalid messages for now\n        // TODO: log errors with log levels\n        return;\n      }\n      if (data.type === \"cf_agent_chat_messages\") {\n        useChatHelpers.setMessages(data.messages);\n      }\n    }\n\n    agent.addEventListener(\"message\", onClearHistory);\n    agent.addEventListener(\"message\", onMessages);\n\n    return () => {\n      agent.removeEventListener(\"message\", onClearHistory);\n      agent.removeEventListener(\"message\", onMessages);\n    };\n  }, [\n    agent.addEventListener,\n    agent.removeEventListener,\n    useChatHelpers.setMessages,\n  ]);\n\n  return {\n    ...useChatHelpers,\n    /**\n     * Set the chat messages and synchronize with the Agent\n     * @param messages New messages to set\n     */\n    setMessages: (messages: Message[]) => {\n      useChatHelpers.setMessages(messages);\n      agent.send(\n        JSON.stringify({\n          type: \"cf_agent_chat_messages\",\n          messages,\n        })\n      );\n    },\n    /**\n     * Clear chat history on both client and Agent\n     */\n    clearHistory: () => {\n      useChatHelpers.setMessages([]);\n      agent.send(\n        JSON.stringify({\n          type: \"cf_agent_chat_clear\",\n        })\n      );\n    },\n  };\n}\n"],"mappings":";;;AAAA,SAAS,eAAe;AAGxB,SAAS,WAAW,WAAW;AA0B/B,IAAM,eAAe,oBAAI,IAAgC;AAOlD,SAAS,aACd,SACA;AACA,QAAM,EAAE,OAAO,oBAAoB,GAAG,KAAK,IAAI;AAE/C,QAAM,WAAW,IAAI;AAAA,IACnB;AAAA,KACE,MAAM,QAA0B,MAAM,SACpC,QAAQ,SAAS,SAAS,EAC3B,QAAQ,UAAU,UAAU,CAAC;AAAA,EAClC;AAGA,WAAS,aAAa,OAAO,KAAK;AAClC,QAAM,iBAAiB,SAAS,SAAS;AAEzC,iBAAe,+BAA+B;AAAA,IAC5C;AAAA,EACF,GAA8B;AAC5B,UAAM,iBAAiB,IAAI,IAAI,GAAG;AAClC,mBAAe,YAAY;AAC3B,UAAM,WAAW,MAAM,MAAM,eAAe,SAAS,GAAG;AAAA,MACtD,SAAS,QAAQ;AAAA,MACjB,aAAa,QAAQ;AAAA,IACvB,CAAC;AACD,WAAO,SAAS,KAAgB;AAAA,EAClC;AAEA,QAAM,0BACJ,sBAAsB;AAExB,WAAS,qBACP,2BACA;AACA,QAAI,aAAa,IAAI,cAAc,GAAG;AACpC,aAAO,aAAa,IAAI,cAAc;AAAA,IACxC;AACA,UAAM,UAAU,wBAAwB,yBAAyB;AACjE,iBAAa,IAAI,gBAAgB,OAAO;AACxC,WAAO;AAAA,EACT;AAEA,QAAM,kBACJ,uBAAuB,OACnB;AAAA,IACE,qBAAqB;AAAA,MACnB,OAAO,MAAM;AAAA,MACb,MAAM,MAAM;AAAA,MACZ,KAAK;AAAA,IACP,CAAC;AAAA,EACH,IACA,KAAK;AAEX,YAAU,MAAM;AACd,WAAO,MAAM;AACX,mBAAa,OAAO,cAAc;AAAA,IACpC;AAAA,EACF,GAAG,CAAC,cAAc,CAAC;AAEnB,iBAAe,QACb,SACAA,WAAuB,CAAC,GACxB;AAKA,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA;AAAA,IAEF,IAAIA;AACJ,UAAM,KAAK,OAAO,WAAW;AAC7B,UAAM,kBAAkB,IAAI,gBAAgB;AAE5C,YAAQ,iBAAiB,SAAS,MAAM;AACtC,sBAAgB,MAAM;AAAA,IACxB,CAAC;AAED,UAAM;AAAA,MACJ;AAAA,MACA,CAAC,UAAU;AACT,YAAI;AACJ,YAAI;AACF,iBAAO,KAAK,MAAM,MAAM,IAAI;AAAA,QAC9B,SAAS,OAAO;AAGd;AAAA,QACF;AACA,YAAI,KAAK,SAAS,8BAA8B;AAC9C,cAAI,KAAK,OAAO,IAAI;AAClB,uBAAW,QAAQ,IAAI,YAAY,EAAE,OAAO,KAAK,IAAI,CAAC;AACtD,gBAAI,KAAK,MAAM;AACb,yBAAW,MAAM;AACjB,8BAAgB,MAAM;AAAA,YACxB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,MACA,EAAE,QAAQ,gBAAgB,OAAO;AAAA,IACnC;AAEA,QAAI;AAEJ,UAAM,SAAS,IAAI,eAAe;AAAA,MAChC,MAAM,GAAG;AACP,qBAAa;AAAA,MACf;AAAA,IACF,CAAC;AAED,UAAM;AAAA,MACJ,KAAK,UAAU;AAAA,QACb,MAAM;AAAA,QACN;AAAA,QACA,KAAK,QAAQ,SAAS;AAAA,QACtB,MAAM;AAAA,UACJ;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA;AAAA;AAAA,QAGF;AAAA,MACF,CAAC;AAAA,IACH;AAEA,WAAO,IAAI,SAAS,MAAM;AAAA,EAC5B;AACA,QAAM,iBAAiB,QAAQ;AAAA,IAC7B;AAAA,IACA,wBAAwB;AAAA,IACxB,OAAO;AAAA,IACP,GAAG;AAAA,EACL,CAAC;AAED,YAAU,MAAM;AACd,aAAS,eAAe,OAAqB;AAC3C,UAAI,OAAO,MAAM,SAAS,UAAU;AAClC;AAAA,MACF;AACA,UAAI;AACJ,UAAI;AACF,eAAO,KAAK,MAAM,MAAM,IAAI;AAAA,MAC9B,SAAS,OAAO;AAGd;AAAA,MACF;AACA,UAAI,KAAK,SAAS,uBAAuB;AACvC,uBAAe,YAAY,CAAC,CAAC;AAAA,MAC/B;AAAA,IACF;AAEA,aAAS,WAAW,OAAqB;AACvC,UAAI,OAAO,MAAM,SAAS,UAAU;AAClC;AAAA,MACF;AACA,UAAI;AACJ,UAAI;AACF,eAAO,KAAK,MAAM,MAAM,IAAI;AAAA,MAC9B,SAAS,OAAO;AAGd;AAAA,MACF;AACA,UAAI,KAAK,SAAS,0BAA0B;AAC1C,uBAAe,YAAY,KAAK,QAAQ;AAAA,MAC1C;AAAA,IACF;AAEA,UAAM,iBAAiB,WAAW,cAAc;AAChD,UAAM,iBAAiB,WAAW,UAAU;AAE5C,WAAO,MAAM;AACX,YAAM,oBAAoB,WAAW,cAAc;AACnD,YAAM,oBAAoB,WAAW,UAAU;AAAA,IACjD;AAAA,EACF,GAAG;AAAA,IACD,MAAM;AAAA,IACN,MAAM;AAAA,IACN,eAAe;AAAA,EACjB,CAAC;AAED,SAAO;AAAA,IACL,GAAG;AAAA;AAAA;AAAA;AAAA;AAAA,IAKH,aAAa,CAAC,aAAwB;AACpC,qBAAe,YAAY,QAAQ;AACnC,YAAM;AAAA,QACJ,KAAK,UAAU;AAAA,UACb,MAAM;AAAA,UACN;AAAA,QACF,CAAC;AAAA,MACH;AAAA,IACF;AAAA;AAAA;AAAA;AAAA,IAIA,cAAc,MAAM;AAClB,qBAAe,YAAY,CAAC,CAAC;AAC7B,YAAM;AAAA,QACJ,KAAK,UAAU;AAAA,UACb,MAAM;AAAA,QACR,CAAC;AAAA,MACH;AAAA,IACF;AAAA,EACF;AACF;","names":["options"]}