@springbrand/chat-client 0.1.3-alpha.25 → 0.1.3-alpha.28

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/chat-client",
-  "version": "0.1.3-alpha.25",
+  "version": "0.1.3-alpha.28",
   "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.31"
+    "@springbrand/agent-runtime": "0.2.0-alpha.36"
   },
   "scripts": {
     "typecheck": "tsc --noEmit"

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

@@ -194,6 +194,7 @@ type UniversalAgentChatContextValue = {
   chatId: string;
   connection: ChatConnectionOptions;
   agent: SessionAgentConnection;
+  chatAgent: SessionAgentConnection;
 };
 
 const UniversalAgentChatContext = createContext<UniversalAgentChatContextValue | null>(null);
@@ -224,9 +225,13 @@ export function UniversalAgentChatProvider({
       name: chatId,
     }],
   });
+  const chatAgent = useMemo(
+    () => createSafeUIMessageAgentConnection(agent),
+    [agent],
+  );
   const value = useMemo(
-    () => ({ chatId, connection, agent }),
-    [agent, chatId, connection],
+    () => ({ chatId, connection, agent, chatAgent }),
+    [agent, chatAgent, chatId, connection],
   );
   return createElement(
     UniversalAgentChatContext.Provider,
@@ -367,7 +372,7 @@ export function useUniversalAgentChat(): ChatRuntime {
   if (!context) {
     throw new Error("useUniversalAgentChat must be used within UniversalAgentChatProvider");
   }
-  const { chatId, connection, agent } = context;
+  const { chatId, connection, agent, chatAgent } = context;
   const [optimisticMessages, setOptimisticMessages] =
     useState<ChatMessage[]>([]);
   const [optimisticQueued, setOptimisticQueued] =
@@ -376,11 +381,6 @@ export function useUniversalAgentChat(): ChatRuntime {
   const [canRetry, setCanRetry] = useState(false);
   const failedDispatchRef = useRef<FailedDispatch | undefined>(undefined);
   const activeSubmissionIdRef = useRef<string | undefined>(undefined);
-  const chatAgent = useMemo(
-    () => createSafeUIMessageAgentConnection(agent),
-    [agent],
-  );
-
   // 轮询→推送:这条 chat 连接连的是 UniversalAgent facet,`agent.state` 即 facet 的
   // AgentState 广播(useAgent 内部 useState,收到 cf_agent_state 即 re-render)。待批项
   // 随它推来 —— 审批读侧复用这条已开的连接,不另开第二条 WebSocket。三态(原则 VII):