@economic/agents-react 1.1.2 → 1.1.3

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,6 +282,8 @@ 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

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 {};
@@ -85,16 +85,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.3",
   "license": "MIT",
   "files": [
     "dist"