@springbrand/chat-client 0.3.0-alpha.1 → 0.3.0-alpha.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/chat-client",
-  "version": "0.3.0-alpha.1",
+  "version": "0.3.0-alpha.3",
   "type": "module",
   "files": [
     "src",
@@ -13,8 +13,8 @@
     ".": "./src/index.ts"
   },
   "peerDependencies": {
-    "@cloudflare/ai-chat": "^0.11.0",
-    "agents": "^0.22.0",
+    "@cloudflare/ai-chat": "^0.12.0",
+    "agents": "^0.23.0",
     "ai": "^7.0.0",
     "react": "^19.0.0"
   },
@@ -22,12 +22,12 @@
     "nanoid": "^5.1.16"
   },
   "devDependencies": {
-    "@cloudflare/ai-chat": "0.11.0",
+    "@cloudflare/ai-chat": "0.12.0",
     "@types/react": "^19.2.18",
     "@types/react-dom": "^19.2.5",
     "react-dom": "^19.2.8",
     "typescript": "^7.0.2",
-    "@springbrand/agent-runtime": "0.3.0-alpha.1"
+    "@springbrand/agent-runtime": "0.3.0-alpha.4"
   },
   "scripts": {
     "typecheck": "tsc --noEmit"

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

@@ -180,8 +180,9 @@ type SharedChatConnectionOptions = {
 };
 
 export type ChatConnectionOptions = SharedChatConnectionOptions & (
-  | { inboxName: string; basePath?: never }
-  | { inboxName?: never; basePath?: string }
+  | { inboxName: string; basePath?: never; sessionBasePath?: never }
+  | { inboxName?: never; basePath?: string; sessionBasePath?: never }
+  | { inboxName?: never; basePath?: never; sessionBasePath: string }
 );
 
 type SessionAgentConnection = ReturnType<typeof useAgent<RuntimeState>>;
@@ -206,19 +207,24 @@ export function UniversalAgentChatProvider({
   children: ReactNode;
 }) {
   const inboxAgent = connection.inboxAgent ?? "Inbox";
+  const routed = connection.sessionBasePath !== undefined;
   const agent = useAgent<RuntimeState>({
-    agent: inboxAgent,
-    ...(connection.inboxName === undefined
-      ? { basePath: connection.basePath ?? CURRENT_INBOX_AGENT_PATH }
-      : { name: connection.inboxName }),
+    agent: routed ? connection.sessionAgent ?? "UniversalAgent" : inboxAgent,
+    ...(routed
+      ? { basePath: connection.sessionBasePath, name: chatId }
+      : connection.inboxName === undefined
+        ? { basePath: connection.basePath ?? CURRENT_INBOX_AGENT_PATH }
+        : { 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",
-      name: chatId,
-    }],
+    ...(routed ? {} : {
+      sub: [{
+        agent: connection.sessionAgent ?? "UniversalAgent",
+        name: chatId,
+      }],
+    }),
   });
   const chatAgent = useMemo(
     () => createSafeUIMessageAgentConnection(agent),
@@ -597,6 +603,13 @@ export function useUniversalAgentChat(): ChatRuntime {
     throw new Error("useUniversalAgentChat must be used within UniversalAgentChatProvider");
   }
   const { chatId, connection, agent, chatAgent } = context;
+  const routedHistoryUrl = connection.sessionBasePath !== undefined &&
+      connection.host !== undefined
+    ? new URL(
+        `/${connection.sessionBasePath.replace(/^\/+/, "")}`,
+        connection.host,
+      ).toString()
+    : undefined;
   const [optimisticMessages, setOptimisticMessages] =
     useState<ChatMessage[]>([]);
   const [optimisticQueued, setOptimisticQueued] =
@@ -632,7 +645,10 @@ export function useUniversalAgentChat(): ChatRuntime {
     agent: chatAgent,
     getInitialMessages: connection.skipInitialMessages
       ? null
-      : ({ url }) => loadInitialMessages(url, connection.credentials),
+      : ({ url }) => loadInitialMessages(
+          routedHistoryUrl ?? url,
+          connection.credentials,
+        ),
     ...(connection.credentials === undefined
       ? {}
       : { credentials: connection.credentials }),