@springbrand/chat-client 0.1.3-alpha.21 → 0.1.3-alpha.23

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/chat-client",
-  "version": "0.1.3-alpha.21",
+  "version": "0.1.3-alpha.23",
   "type": "module",
   "files": [
     "src",
@@ -25,7 +25,7 @@
     "@types/react-dom": "^19.2.3",
     "react-dom": "^19.2.7",
     "typescript": "^7.0.2",
-    "@springbrand/agent-runtime": "0.2.0-alpha.28"
+    "@springbrand/agent-runtime": "0.2.0-alpha.30"
   },
   "scripts": {
     "typecheck": "tsc --noEmit"

package/src/use-universal-agent-chat.ts

@@ -33,6 +33,7 @@ import { CURRENT_INBOX_AGENT_PATH } from "./chat-sessions";
 type ChatMessageMetadata = Record<string, unknown> & {
   error?: string;
   requestedCapabilities?: readonly RequestedCapability[];
+  turnId?: string;
   turnStatus?: string;
 };
 type ChatMessage = UIMessage<ChatMessageMetadata>;
@@ -174,6 +175,7 @@ type SharedChatConnectionOptions = {
   host?: string;
   credentials?: RequestCredentials;
   protocols?: string | string[];
+  onMessage?: (message: MessageEvent) => void;
   inboxAgent?: string;
   sessionAgent?: string;
   /** Host projection used only until the selected Session sends its first state. */
@@ -213,6 +215,7 @@ export function UniversalAgentChatProvider({
       : { name: connection.inboxName }),
     ...(connection.host === undefined ? {} : { host: connection.host }),
     ...(connection.protocols === undefined ? {} : { protocols: connection.protocols }),
+    ...(connection.onMessage === undefined ? {} : { onMessage: connection.onMessage }),
     connectionTimeout: 6_000,
     sub: [{
       agent: connection.sessionAgent ?? "UniversalAgent",
@@ -569,11 +572,25 @@ export function useUniversalAgentChat(): ChatRuntime {
   const dispatchMessage = useCallback(async (
     message: ChatMessage,
     delivery: MessageDelivery,
-  ): Promise<MessageDispatchReceipt> =>
-    agentRef.current.call<MessageDispatchReceipt>(
+  ): Promise<MessageDispatchReceipt> => {
+    const dispatch = () => agentRef.current.call<MessageDispatchReceipt>(
       "dispatchMessage",
       [message, delivery],
-    ), []);
+    );
+    try {
+      return await dispatch();
+    } catch (error) {
+      // The server deduplicates dispatches by message.id, so replay only the
+      // transport failures that mean admission may already have succeeded.
+      const retryable = error instanceof Error && (
+        /^RPC call to dispatchMessage timed out after \d+ms$/u.test(error.message) ||
+        (error.message === "Connection closed" && agentRef.current.shouldReconnect)
+      );
+      if (!retryable) throw error;
+      await agentRef.current.ready;
+      return dispatch();
+    }
+  }, []);
 
   const submitMessage = useCallback(async (
     message: ChatMessage,