@economic/agents-react 1.3.3 → 1.3.4

package/dist/index.d.mts

@@ -205,11 +205,6 @@ declare class PartySocket extends ReconnectingWebSocket {
 //#region src/types.d.ts
 type AgentConnectionStatus = "connecting" | "connected" | "disconnected" | "unauthorized";
 type AgentConnectionType = "agent" | "chat" | "assistant";
-type AgentConnectionState = {
-  status: AgentConnectionStatus;
-  type: AgentConnectionType;
-  subAgentName?: string;
-};
 type ChatSummary = {
   name: string;
   title?: string;
@@ -231,6 +226,7 @@ type CreateChatHandler = () => Promise<string>;
 type OpenChatHandler = (chatId: string) => void;
 type DeleteChatHandler = (chatId: string) => Promise<void>;
 interface AssistantActions {
+  state: unknown;
   getChats: GetChatsHandler;
   createChat: CreateChatHandler;
   openChat: OpenChatHandler;
@@ -252,6 +248,7 @@ interface UseAgentOptions {
   onOpen?: (event: Event) => void;
   onClose?: (event: CloseEvent) => void;
   onError?: (event: ErrorEvent) => void;
+  onMessage?: (event: MessageEvent) => void;
 }
 declare function useAgent(options: UseAgentOptions): {
   agent: Omit<PartySocket, "path"> & {
@@ -263,8 +260,8 @@ declare function useAgent(options: UseAgentOptions): {
     }>;
     identified: boolean;
     ready: Promise<void>;
-    state: AgentConnectionState | undefined;
-    setState: (state: AgentConnectionState) => void;
+    state: unknown;
+    setState: (state: unknown) => 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;
@@ -290,8 +287,8 @@ declare function useChat<ToolContext extends Record<string, unknown>>(options: U
     }>;
     identified: boolean;
     ready: Promise<void>;
-    state: AgentConnectionState | undefined;
-    setState: (state: AgentConnectionState) => void;
+    state: unknown;
+    setState: (state: unknown) => 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;
@@ -323,6 +320,7 @@ declare function useAssistant<ToolContext extends Record<string, unknown>>(optio
   chats: ChatSummary[];
   status: AgentConnectionStatus;
   assistant: {
+    state: unknown;
     getChats: () => Promise<ChatSummary[]>;
     createChat: () => Promise<string>;
     openChat: (chatId: string) => void;
@@ -339,8 +337,8 @@ declare function useAssistant<ToolContext extends Record<string, unknown>>(optio
       }>;
       identified: boolean;
       ready: Promise<void>;
-      state: AgentConnectionState | undefined;
-      setState: (state: AgentConnectionState) => void;
+      state: unknown;
+      setState: (state: unknown) => 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;
@@ -364,4 +362,4 @@ declare function useAssistant<ToolContext extends Record<string, unknown>>(optio
   };
 };
 //#endregion
-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 };
+export { 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

@@ -12,7 +12,7 @@ function getTokenArgs(authToken) {
 //#endregion
 //#region src/useAgent.ts
 function useAgent(options) {
-	const { enabled, host, agentName, actorId, name, sub, authToken, onError, onOpen, onClose } = options;
+	const { enabled, host, agentName, actorId, name, sub, authToken, onError, onOpen, onClose, onMessage } = options;
 	const [type, setType] = useState(void 0);
 	const [status, setStatus] = useState("connecting");
 	const agent = useAgent$1({
@@ -35,6 +35,7 @@ function useAgent(options) {
 				if (msg.status === "unauthorized") agent.close(3e3);
 				setStatus(msg.status);
 			} else if (msg.type === "agent_type") setType(msg.agentType);
+			onMessage?.(event);
 		},
 		onError: (event) => {
 			setStatus("disconnected");
@@ -103,7 +104,14 @@ function useAssistant(options) {
 	const { toolContext, welcomeMessage, initialChatId, ...agentOptions } = options;
 	const [chatId, setChatId] = useState(initialChatId);
 	const [chats, setChats] = useState([]);
-	const { agent: assistant, status: assistantStatus } = useAgent(agentOptions);
+	const [subAgentName, setSubAgentName] = useState(void 0);
+	const { agent: assistant, status: assistantStatus } = useAgent({
+		...agentOptions,
+		onMessage: (event) => {
+			const msg = JSON.parse(event.data);
+			if (msg.type === "sub_agent_name") setSubAgentName(msg.subAgentName);
+		}
+	});
 	const refreshChats = useCallback(async () => {
 		setChats((await assistant.call("getChats")).map((chat) => {
 			const c = {
@@ -135,9 +143,8 @@ function useAssistant(options) {
 		setChatId(void 0);
 		await refreshChats();
 	};
-	const subAgentName = assistant.state?.subAgentName ?? "";
 	const isChatEnabled = !!chatId && !!subAgentName;
-	const isChatReady = isChatEnabled && assistant.state?.status === "connected";
+	const isChatReady = isChatEnabled && assistantStatus === "connected";
 	const chat = useChat({
 		...agentOptions,
 		enabled: isChatEnabled,
@@ -154,6 +161,7 @@ function useAssistant(options) {
 		chats,
 		status: isChatEnabled && chat.status !== "connecting" ? chat.status : assistantStatus,
 		assistant: {
+			state: assistant.state,
 			getChats,
 			createChat,
 			openChat,

package/package.json

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