@economic/agents-react 1.3.0 → 1.3.3

package/README.md

@@ -196,4 +196,5 @@ The package also exports the handler types `AssistantActions`, `GetChatsHandler`
 ## Authentication
 
 Pass a JWT as `authToken`. It is sent both as an `Authorization: Bearer …` header and as a WebSocket subprotocol (`["bearer", token]`), matching the server-side [`getJwtAuthConfig`](../agents/README.md#authentication). When verification fails, the connection status becomes `"unauthorized"`.
+
 .

package/dist/index.d.mts

@@ -1,7 +1,3 @@
-import * as _$agents_client0 from "agents/client";
-import * as _$ai from "ai";
-import * as _$_ai_sdk_react0 from "@ai-sdk/react";
-
 //#region ../../node_modules/partysocket/ws-Cg2f-sDL.d.ts
 //#region src/type-helper.d.ts
 type TypedEventTarget<EventMap extends object> = {
@@ -257,20 +253,24 @@ interface UseAgentOptions {
   onClose?: (event: CloseEvent) => void;
   onError?: (event: ErrorEvent) => void;
 }
-declare function useAgent(options: UseAgentOptions): Omit<PartySocket, "path"> & {
-  agent: string;
-  name: string;
-  path: ReadonlyArray<{
+declare function useAgent(options: UseAgentOptions): {
+  agent: Omit<PartySocket, "path"> & {
     agent: string;
     name: string;
-  }>;
-  identified: boolean;
-  ready: Promise<void>;
-  state: AgentConnectionState | undefined;
-  setState: (state: AgentConnectionState) => void;
-  call: <T = unknown>(method: string, args?: unknown[], options?: _$agents_client0.CallOptions | _$agents_client0.StreamOptions) => Promise<T>;
-  stub: _$agents_client0.UntypedAgentStub;
-  getHttpUrl: () => string;
+    path: ReadonlyArray<{
+      agent: string;
+      name: string;
+    }>;
+    identified: boolean;
+    ready: Promise<void>;
+    state: AgentConnectionState | undefined;
+    setState: (state: AgentConnectionState) => void;
+    call: <T = unknown>(method: string, args?: unknown[], options?: import("agents/client").CallOptions | import("agents/client").StreamOptions) => Promise<T>;
+    stub: import("agents/client").UntypedAgentStub;
+    getHttpUrl: () => string;
+  };
+  status: AgentConnectionStatus;
+  type: AgentConnectionType | undefined;
 };
 //#endregion
 //#region src/useChat.d.ts
@@ -292,11 +292,11 @@ declare function useChat<ToolContext extends Record<string, unknown>>(options: U
     ready: Promise<void>;
     state: AgentConnectionState | undefined;
     setState: (state: AgentConnectionState) => void;
-    call: <T = unknown>(method: string, args?: unknown[], options?: _$agents_client0.CallOptions | _$agents_client0.StreamOptions) => Promise<T>;
-    stub: _$agents_client0.UntypedAgentStub;
+    call: <T = unknown>(method: string, args?: unknown[], options?: import("agents/client").CallOptions | import("agents/client").StreamOptions) => Promise<T>;
+    stub: import("agents/client").UntypedAgentStub;
     getHttpUrl: () => string;
   };
-  chat: Omit<_$_ai_sdk_react0.UseChatHelpers<_$ai.UIMessage<unknown, _$ai.UIDataTypes, _$ai.UITools>>, "addToolOutput"> & {
+  chat: Omit<import("@ai-sdk/react").UseChatHelpers<import("ai").UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>>, "addToolOutput"> & {
     clearHistory: () => void;
     addToolOutput: (opts: {
       toolCallId: string;
@@ -315,7 +315,9 @@ declare function useChat<ToolContext extends Record<string, unknown>>(options: U
 };
 //#endregion
 //#region src/useAssistant.d.ts
-interface UseAssistantOptions<ToolContext extends Record<string, unknown> = Record<string, unknown>> extends Omit<UseChatOptions<ToolContext>, "actorId" | "getInitialMessages"> {}
+interface UseAssistantOptions<ToolContext extends Record<string, unknown> = Record<string, unknown>> extends Omit<UseChatOptions<ToolContext>, "actorId" | "getInitialMessages"> {
+  initialChatId?: string;
+}
 declare function useAssistant<ToolContext extends Record<string, unknown>>(options: UseAssistantOptions<ToolContext>): {
   currentChatName: string | undefined;
   chats: ChatSummary[];
@@ -339,11 +341,11 @@ declare function useAssistant<ToolContext extends Record<string, unknown>>(optio
       ready: Promise<void>;
       state: AgentConnectionState | undefined;
       setState: (state: AgentConnectionState) => void;
-      call: <T = unknown>(method: string, args?: unknown[], options?: _$agents_client0.CallOptions | _$agents_client0.StreamOptions) => Promise<T>;
-      stub: _$agents_client0.UntypedAgentStub;
+      call: <T = unknown>(method: string, args?: unknown[], options?: import("agents/client").CallOptions | import("agents/client").StreamOptions) => Promise<T>;
+      stub: import("agents/client").UntypedAgentStub;
       getHttpUrl: () => string;
     };
-    chat: Omit<_$_ai_sdk_react0.UseChatHelpers<_$ai.UIMessage<unknown, _$ai.UIDataTypes, _$ai.UITools>>, "addToolOutput"> & {
+    chat: Omit<import("@ai-sdk/react").UseChatHelpers<import("ai").UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>>, "addToolOutput"> & {
       clearHistory: () => void;
       addToolOutput: (opts: {
         toolCallId: string;

package/dist/index.mjs

@@ -1,5 +1,5 @@
-import { useAgent as useAgent$1 } from "agents/react";
 import { useCallback, useEffect, useRef, useState } from "react";
+import { useAgent as useAgent$1 } from "agents/react";
 import { useAgentChat } from "@cloudflare/ai-chat/react";
 //#region src/util.ts
 function getTokenArgs(authToken) {
@@ -13,27 +13,49 @@ function getTokenArgs(authToken) {
 //#region src/useAgent.ts
 function useAgent(options) {
 	const { enabled, host, agentName, actorId, name, sub, authToken, onError, onOpen, onClose } = options;
-	return useAgent$1({
+	const [type, setType] = useState(void 0);
+	const [status, setStatus] = useState("connecting");
+	const agent = useAgent$1({
 		enabled,
 		host,
 		agent: agentName,
 		name: actorId ? `${actorId}:${name}` : name,
 		sub,
-		onOpen,
-		onClose,
-		onError,
+		onOpen: (event) => {
+			onOpen?.(event);
+		},
+		onClose: (event) => {
+			if (event.code >= 3e3) setStatus("unauthorized");
+			else setStatus("disconnected");
+			onClose?.(event);
+		},
+		onMessage: (event) => {
+			const msg = JSON.parse(event.data);
+			if (msg.type === "connection_status") {
+				if (msg.status === "unauthorized") agent.close(3e3);
+				setStatus(msg.status);
+			} else if (msg.type === "agent_type") setType(msg.agentType);
+		},
+		onError: (event) => {
+			setStatus("disconnected");
+			onError?.(event);
+		},
 		...getTokenArgs(authToken)
 	});
+	return {
+		agent,
+		status,
+		type
+	};
 }
 //#endregion
 //#region src/useChat.ts
 function useChat(options) {
 	const { toolContext, welcomeMessage, ...agentOptions } = options;
-	const agent = useAgent({
+	const { agent, status } = useAgent({
 		...agentOptions,
 		enabled: agentOptions.enabled ?? !!agentOptions.name
 	});
-	const status = agent.state?.status ?? "connecting";
 	const chat = useAgentChat({
 		agent,
 		body: toolContext ?? {},
@@ -78,10 +100,10 @@ function useChat(options) {
 //#endregion
 //#region src/useAssistant.ts
 function useAssistant(options) {
-	const { toolContext, welcomeMessage, ...agentOptions } = options;
-	const [chatId, setChatId] = useState();
+	const { toolContext, welcomeMessage, initialChatId, ...agentOptions } = options;
+	const [chatId, setChatId] = useState(initialChatId);
 	const [chats, setChats] = useState([]);
-	const assistant = useAgent(agentOptions);
+	const { agent: assistant, status: assistantStatus } = useAgent(agentOptions);
 	const refreshChats = useCallback(async () => {
 		setChats((await assistant.call("getChats")).map((chat) => {
 			const c = {
@@ -92,7 +114,6 @@ function useAssistant(options) {
 			return c;
 		}));
 	}, [assistant]);
-	const assistantStatus = assistant.state?.status;
 	useEffect(() => {
 		if (assistantStatus === "connected") refreshChats().catch(console.error);
 	}, [assistantStatus, refreshChats]);
@@ -114,12 +135,14 @@ function useAssistant(options) {
 		setChatId(void 0);
 		await refreshChats();
 	};
-	const isChatReady = !!chatId && assistant.state?.status === "connected";
+	const subAgentName = assistant.state?.subAgentName ?? "";
+	const isChatEnabled = !!chatId && !!subAgentName;
+	const isChatReady = isChatEnabled && assistant.state?.status === "connected";
 	const chat = useChat({
 		...agentOptions,
-		enabled: !!chatId,
-		sub: chatId ? [{
-			agent: assistant.state?.subAgentName ?? "",
+		enabled: isChatEnabled,
+		sub: isChatEnabled ? [{
+			agent: subAgentName,
 			name: chatId
 		}] : void 0,
 		getInitialMessages: isChatReady ? void 0 : null,
@@ -129,7 +152,7 @@ function useAssistant(options) {
 	return {
 		currentChatName: chatId,
 		chats,
-		status: (chatId ? chat.agent.state?.status : assistant.state?.status) ?? "connecting",
+		status: isChatEnabled && chat.status !== "connecting" ? chat.status : assistantStatus,
 		assistant: {
 			getChats,
 			createChat,

package/dist/v1.mjs

@@ -1,5 +1,5 @@
-import { useAgent } from "agents/react";
 import React from "react";
+import { useAgent } from "agents/react";
 import { useAgentChat } from "@cloudflare/ai-chat/react";
 //#region src/v1/index.ts
 function agentNameToKebab(name) {

package/package.json

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