@economic/agents-react 1.1.2 → 1.1.4

package/dist/index.d.mts

@@ -14,20 +14,20 @@ interface IntermediateEventTarget<EventMap> extends EventTarget {
   removeEventListener(type: string, callback: EventListenerOrEventListenerObject | null, options?: EventListenerOptions | boolean): void;
 } //#endregion
 //#region src/ws.d.ts
-declare class ErrorEvent extends Event {
+declare class ErrorEvent$1 extends Event {
   message: string;
   error: Error;
   constructor(error: Error, target: any);
 }
-declare class CloseEvent extends Event {
+declare class CloseEvent$1 extends Event {
   code: number;
   reason: string;
   wasClean: boolean;
   constructor(code: number | undefined, reason: string | undefined, target: any);
 }
 interface WebSocketEventMap {
-  close: CloseEvent;
-  error: ErrorEvent;
+  close: CloseEvent$1;
+  error: ErrorEvent$1;
   message: MessageEvent;
   open: Event;
 }
@@ -110,11 +110,11 @@ declare class ReconnectingWebSocket extends ReconnectingWebSocket_base {
   /**
    * An event listener to be called when the WebSocket connection's readyState changes to CLOSED
    */
-  onclose: ((event: CloseEvent) => void) | null;
+  onclose: ((event: CloseEvent$1) => void) | null;
   /**
    * An event listener to be called when an error occurs
    */
-  onerror: ((event: ErrorEvent) => void) | null;
+  onerror: ((event: ErrorEvent$1) => void) | null;
   /**
    * An event listener to be called when a message is received from the server
    */
@@ -224,9 +224,9 @@ interface UseAgentOptions {
     agent: string;
     name: string;
   }[];
-  onOpen?: () => void;
-  onClose?: () => void;
-  onError?: (error: Event) => void;
+  onOpen?: (event: Event) => void;
+  onClose?: (event: CloseEvent) => void;
+  onError?: (event: ErrorEvent) => void;
   authToken?: string;
   userId?: string;
   id: string;
@@ -250,6 +250,7 @@ declare function useAgent<T extends AgentConnectionState>(options: UseAgentOptio
 //#region src/useChatAgent.d.ts
 interface UseChatOptions<ToolContext extends Record<string, unknown>> extends UseAgentOptions {
   toolContext?: ToolContext;
+  welcomeMessage?: string;
 }
 declare function useChatAgent<ToolContext extends Record<string, unknown>>(options: UseChatOptions<ToolContext>): {
   status: AgentConnectionStatus;
@@ -281,12 +282,14 @@ declare function useChatAgent<ToolContext extends Record<string, unknown>>(optio
     isStreaming: boolean;
     isToolContinuation: boolean;
   };
+  rateMessage: (messageId: string, rating: number, comment?: string) => Promise<unknown>;
+  getMessageRatings: () => Promise<unknown>;
 };
 //#endregion
 //#region src/useAssistant.d.ts
-/** Conversation summary returned by the assistant's `getConversations` RPC. */
-type Conversation = {
-  durable_object_name: string;
+/** Chat summary returned by the assistant's `getChats` RPC. */
+type Chat = {
+  chatId: string;
   title?: string;
   summary?: string;
   created_at: number;
@@ -299,10 +302,10 @@ interface UseAssistantOptions<ToolContext extends Record<string, unknown> = Reco
 declare function useAssistant<ToolContext extends Record<string, unknown>>(options: UseAssistantOptions<ToolContext>): {
   chatId: string | undefined;
   status: AgentConnectionStatus;
-  getConversations: () => Promise<Conversation[]>;
-  createConversation: () => Promise<string>;
-  openConversation: (chatId: string) => void;
-  deleteConversation: (chatId: string) => Promise<void>;
+  getChats: () => Promise<Chat[]>;
+  createChat: () => Promise<string>;
+  openChat: (chatId: string) => void;
+  deleteChat: (chatId: string) => Promise<void>;
   agent: Omit<PartySocket, "path"> & {
     agent: string;
     name: string;
@@ -333,4 +336,4 @@ declare function useAssistant<ToolContext extends Record<string, unknown>>(optio
   }) | undefined;
 };
 //#endregion
-export { AgentConnectionState, AgentConnectionStatus, AgentConnectionType, Conversation, UseAgentOptions, UseAssistantOptions, UseChatOptions, useAgent, useAssistant, useChatAgent };
+export { AgentConnectionState, AgentConnectionStatus, AgentConnectionType, Chat, UseAgentOptions, UseAssistantOptions, UseChatOptions, useAgent, useAssistant, useChatAgent };

package/dist/index.mjs

@@ -1,6 +1,6 @@
 import { useAgent as useAgent$1 } from "agents/react";
 import { useAgentChat } from "@cloudflare/ai-chat/react";
-import { useState } from "react";
+import { useEffect, useRef, useState } from "react";
 //#region src/util.ts
 function getTokenArgs(authToken) {
 	if (!authToken) return {};
@@ -36,22 +36,29 @@ function useAssistant(options) {
 		id: userId,
 		authToken
 	});
-	const getConversations = async () => {
-		return await assistant.call("getConversations");
+	const getChats = async () => {
+		return (await assistant.call("getChats")).map((chat) => {
+			const c = {
+				...chat,
+				chatId: chat.durable_object_name
+			};
+			delete c.durable_object_name;
+			return c;
+		});
 	};
-	const createConversation = async () => {
-		const chatId = await assistant.call("createConversation");
+	const createChat = async () => {
+		const chatId = await assistant.call("createChat");
 		setChatId(chatId);
-		await getConversations();
+		await getChats();
 		return chatId;
 	};
-	const openConversation = (chatId) => {
+	const openChat = (chatId) => {
 		setChatId(chatId);
 	};
-	const deleteConversation = async (chatId) => {
-		await assistant.call("deleteConversation", [chatId]);
+	const deleteChat = async (chatId) => {
+		await assistant.call("deleteChat", [chatId]);
 		setChatId(void 0);
-		await getConversations();
+		await getChats();
 	};
 	const agent = useAgent({
 		enabled: !!chatId,
@@ -74,10 +81,10 @@ function useAssistant(options) {
 	return {
 		chatId,
 		status: (chatId ? agent.state?.status : assistant.state?.status) ?? "connecting",
-		getConversations,
-		createConversation,
-		openConversation,
-		deleteConversation,
+		getChats,
+		createChat,
+		openChat,
+		deleteChat,
 		agent,
 		chat: isChatReady ? chat : void 0
 	};
@@ -85,16 +92,47 @@ function useAssistant(options) {
 //#endregion
 //#region src/useChatAgent.ts
 function useChatAgent(options) {
-	const { toolContext, ...agentOptions } = options;
+	const { toolContext, welcomeMessage, ...agentOptions } = options;
 	const agent = useAgent(agentOptions);
 	const chat = useAgentChat({
 		agent,
 		body: toolContext ?? {}
 	});
+	const status = agent.state?.status ?? "connecting";
+	const hasSentWelcome = useRef(false);
+	useEffect(() => {
+		if (welcomeMessage && status === "connected" && chat.messages.length === 0 && !hasSentWelcome.current) {
+			hasSentWelcome.current = true;
+			chat.setMessages([{
+				id: "welcome-message",
+				role: "assistant",
+				parts: [{
+					type: "text",
+					text: welcomeMessage
+				}]
+			}]);
+		}
+	}, [
+		status,
+		chat.messages.length,
+		welcomeMessage
+	]);
+	async function rateMessage(messageId, rating, comment) {
+		return agent.call("rateMessage", [
+			messageId,
+			rating,
+			comment
+		]);
+	}
+	async function getMessageRatings() {
+		return agent.call("getMessageRatings");
+	}
 	return {
-		status: agent.state?.status ?? "connecting",
+		status,
 		agent,
-		chat
+		chat,
+		rateMessage,
+		getMessageRatings
 	};
 }
 //#endregion

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents-react",
-  "version": "1.1.2",
+  "version": "1.1.4",
   "license": "MIT",
   "files": [
     "dist"