@economic/agents-react 1.1.8 → 1.1.10

package/dist/index.d.mts

@@ -228,6 +228,18 @@ type MessageFeedback = {
   created_at: number;
   updated_at: number;
 };
+type MessageFeedbackHandler = (messageId: string, rating: number, comment?: string) => Promise<void>;
+type GetMessageFeedbackHandler = () => Promise<Record<string, MessageFeedback>>;
+type GetChatsHandler = () => Promise<ChatSummary[]>;
+type CreateChatHandler = () => Promise<string>;
+type OpenChatHandler = (chatId: string) => void;
+type DeleteChatHandler = (chatId: string) => Promise<void>;
+interface AssistantActions {
+  getChats: GetChatsHandler;
+  createChat: CreateChatHandler;
+  openChat: OpenChatHandler;
+  deleteChat: DeleteChatHandler;
+}
 //#endregion
 //#region src/useAgent.d.ts
 interface UseAgentOptions {
@@ -297,14 +309,15 @@ declare function useChat<ToolContext extends Record<string, unknown>>(options: U
     isStreaming: boolean;
     isToolContinuation: boolean;
   };
-  submitMessageFeedback: (messageId: string, rating: number, comment?: string) => Promise<unknown>;
-  getMessageFeedback: () => Promise<Record<string, MessageFeedback>>;
+  submitMessageFeedback: MessageFeedbackHandler;
+  getMessageFeedback: GetMessageFeedbackHandler;
 };
 //#endregion
 //#region src/useAssistant.d.ts
 interface UseAssistantOptions<ToolContext extends Record<string, unknown> = Record<string, unknown>> extends Omit<UseChatOptions<ToolContext>, "actorId" | "getInitialMessages"> {}
 declare function useAssistant<ToolContext extends Record<string, unknown>>(options: UseAssistantOptions<ToolContext>): {
   currentChatName: string | undefined;
+  chats: ChatSummary[];
   status: AgentConnectionStatus;
   assistant: {
     getChats: () => Promise<ChatSummary[]>;
@@ -342,9 +355,9 @@ declare function useAssistant<ToolContext extends Record<string, unknown>>(optio
       isStreaming: boolean;
       isToolContinuation: boolean;
     };
-    submitMessageFeedback: (messageId: string, rating: number, comment?: string) => Promise<unknown>;
-    getMessageFeedback: () => Promise<Record<string, MessageFeedback>>;
+    submitMessageFeedback: MessageFeedbackHandler;
+    getMessageFeedback: GetMessageFeedbackHandler;
   };
 };
 //#endregion
-export { type AgentConnectionState, type AgentConnectionStatus, type AgentConnectionType, type ChatSummary, type MessageFeedback, useAgent, useAssistant, useChat };
+export { type AgentConnectionState, type AgentConnectionStatus, type AgentConnectionType, type AssistantActions, type ChatSummary, type CreateChatHandler, type DeleteChatHandler, type GetChatsHandler, type GetMessageFeedbackHandler, type MessageFeedback, type MessageFeedbackHandler, type OpenChatHandler, useAgent, useAssistant, useChat };

package/dist/index.mjs

@@ -1,5 +1,5 @@
 import { useAgent as useAgent$1 } from "agents/react";
-import { useEffect, useRef, useState } from "react";
+import { useCallback, useEffect, useRef, useState } from "react";
 import { useAgentChat } from "@cloudflare/ai-chat/react";
 //#region src/util.ts
 function getTokenArgs(authToken) {
@@ -57,16 +57,16 @@ function useChat(options) {
 		chat.messages.length,
 		welcomeMessage
 	]);
-	async function submitMessageFeedback(messageId, rating, comment) {
-		return agent.call("submitMessageFeedback", [
+	const submitMessageFeedback = async (messageId, rating, comment) => {
+		await agent.call("submitMessageFeedback", [
 			messageId,
 			rating,
 			comment
 		]);
-	}
-	async function getMessageFeedback() {
+	};
+	const getMessageFeedback = async () => {
 		return await agent.call("getMessageFeedback");
-	}
+	};
 	return {
 		status,
 		agent,
@@ -80,22 +80,31 @@ function useChat(options) {
 function useAssistant(options) {
 	const { toolContext, welcomeMessage, ...agentOptions } = options;
 	const [chatId, setChatId] = useState();
+	const [chats, setChats] = useState([]);
 	const assistant = useAgent(agentOptions);
-	const getChats = async () => {
-		return (await assistant.call("getChats")).map((chat) => {
+	const refreshChats = useCallback(async () => {
+		setChats((await assistant.call("getChats")).map((chat) => {
 			const c = {
 				...chat,
 				name: chat.durable_object_name
 			};
 			delete c.durable_object_name;
 			return c;
-		});
+		}));
+	}, [assistant]);
+	const assistantStatus = assistant.state?.status;
+	useEffect(() => {
+		if (assistantStatus === "connected") refreshChats().catch(console.error);
+	}, [assistantStatus, refreshChats]);
+	const getChats = async () => {
+		await refreshChats();
+		return chats;
 	};
 	const createChat = async () => {
-		const chatId = await assistant.call("createChat");
-		setChatId(chatId);
-		await getChats();
-		return chatId;
+		const newChatId = await assistant.call("createChat");
+		setChatId(newChatId);
+		await refreshChats();
+		return newChatId;
 	};
 	const openChat = (chatId) => {
 		setChatId(chatId);
@@ -103,7 +112,7 @@ function useAssistant(options) {
 	const deleteChat = async (chatId) => {
 		await assistant.call("deleteChat", [chatId]);
 		setChatId(void 0);
-		await getChats();
+		await refreshChats();
 	};
 	const isChatReady = !!chatId && assistant.state?.status === "connected";
 	const chat = useChat({
@@ -119,6 +128,7 @@ function useAssistant(options) {
 	});
 	return {
 		currentChatName: chatId,
+		chats,
 		status: (chatId ? chat.agent.state?.status : assistant.state?.status) ?? "connecting",
 		assistant: {
 			getChats,

package/package.json

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