@aomi-labs/react 0.3.9 → 0.3.11

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { AomiClient, Session, WalletRequest } from '@aomi-labs/client';
-export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, WalletEip712Payload, WalletRequest, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
+import { AomiClient, Session, WalletRequest, UserState } from '@aomi-labs/client';
+export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, UserState, WalletEip712Payload, WalletRequest, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ReactNode, SetStateAction } from 'react';
 import { ThreadMessageLike } from '@assistant-ui/react';
@@ -11,25 +11,6 @@ type AomiRuntimeProviderProps = {
 };
 declare function AomiRuntimeProvider({ children, backendUrl, }: Readonly<AomiRuntimeProviderProps>): react_jsx_runtime.JSX.Element;
 
-type UserState = {
-    address?: string;
-    chainId?: number;
-    isConnected: boolean;
-    ensName?: string;
-    ext?: Record<string, unknown>;
-};
-declare function useUser(): {
-    user: UserState;
-    setUser: (data: Partial<UserState>) => void;
-    addExtValue: (key: string, value: unknown) => void;
-    removeExtValue: (key: string) => void;
-    getUserState: () => UserState;
-    onUserStateChange: (callback: (user: UserState) => void) => () => void;
-};
-declare function UserContextProvider({ children }: {
-    children: ReactNode;
-}): react_jsx_runtime.JSX.Element;
-
 type ThreadContext = {
     currentThreadId: string;
     setCurrentThreadId: (id: string) => void;
@@ -152,12 +133,14 @@ type WalletHandlerConfig = {
 type WalletHandlerApi = {
     /** All queued wallet requests (tx + eip712) */
     pendingRequests: WalletRequest[];
-    /** Enqueue a wallet request (called by orchestrator on ClientSession events) */
-    enqueueRequest: (request: WalletRequest) => void;
+    /** Replace pending requests with the session's authoritative snapshot. */
+    setRequests: (requests: WalletRequest[]) => void;
+    /** Mark a request as in-flight so it is not replayed while awaiting backend ack. */
+    startRequest: (id: string) => void;
     /** Complete a request successfully — sends response to backend via ClientSession */
-    resolveRequest: (id: string, result: WalletRequestResult) => void;
+    resolveRequest: (id: string, result: WalletRequestResult) => Promise<void>;
     /** Fail a request — sends error to backend via ClientSession */
-    rejectRequest: (id: string, error?: string) => void;
+    rejectRequest: (id: string, error?: string) => Promise<void>;
 };
 declare function useWalletHandler({ getSession, }: WalletHandlerConfig): WalletHandlerApi;
 
@@ -212,10 +195,10 @@ type AomiRuntimeApi = {
     pendingWalletRequests: WalletRequest[];
     /** Mark a wallet request as being processed */
     startWalletRequest: (id: string) => void;
-    /** Complete a wallet request — dequeues + sends response to backend */
-    resolveWalletRequest: (id: string, result: WalletRequestResult) => void;
-    /** Fail a wallet request — dequeues + sends error to backend */
-    rejectWalletRequest: (id: string, error?: string) => void;
+    /** Complete a wallet request after the backend acknowledges the response */
+    resolveWalletRequest: (id: string, result: WalletRequestResult) => Promise<void>;
+    /** Fail a wallet request after the backend acknowledges the error */
+    rejectWalletRequest: (id: string, error?: string) => Promise<void>;
     /** Subscribe to inbound events by type. Returns unsubscribe function. */
     subscribe: (type: string, callback: EventSubscriber) => () => void;
     /** Send a system command to the backend */
@@ -280,6 +263,18 @@ type NotificationApi = {
 };
 declare function useNotificationHandler({ onNotification, }?: NotificationHandlerConfig): NotificationApi;
 
+declare function useUser(): {
+    user: UserState;
+    setUser: (data: Partial<UserState>) => void;
+    addExtValue: (key: string, value: unknown) => void;
+    removeExtValue: (key: string) => void;
+    getUserState: () => UserState;
+    onUserStateChange: (callback: (user: UserState) => void) => () => void;
+};
+declare function UserContextProvider({ children }: {
+    children: ReactNode;
+}): react_jsx_runtime.JSX.Element;
+
 /**
  * Utility function to merge Tailwind CSS classes with conflict resolution.
  * Combines clsx for conditional classes and tailwind-merge for deduplication.
@@ -385,4 +380,4 @@ type ControlContextProviderProps = {
 };
 declare function ControlContextProvider({ children, aomiClient, sessionId, publicKey, getThreadMetadata, updateThreadMetadata, }: ControlContextProviderProps): react_jsx_runtime.JSX.Element;
 
-export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type StoredProviderKey, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type UserState, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };
+export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type StoredProviderKey, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { AomiClient, Session, WalletRequest } from '@aomi-labs/client';
-export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, WalletEip712Payload, WalletRequest, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
+import { AomiClient, Session, WalletRequest, UserState } from '@aomi-labs/client';
+export { AomiChatResponse, AomiClient, AomiClientOptions, AomiCreateThreadResponse, AomiInterruptResponse, AomiMessage, AomiSSEEvent, AomiStateResponse, AomiSystemEvent, AomiSystemResponse, AomiThread, UserState, WalletEip712Payload, WalletRequest, WalletTxPayload, toViemSignTypedDataArgs } from '@aomi-labs/client';
 import * as react_jsx_runtime from 'react/jsx-runtime';
 import { ReactNode, SetStateAction } from 'react';
 import { ThreadMessageLike } from '@assistant-ui/react';
@@ -11,25 +11,6 @@ type AomiRuntimeProviderProps = {
 };
 declare function AomiRuntimeProvider({ children, backendUrl, }: Readonly<AomiRuntimeProviderProps>): react_jsx_runtime.JSX.Element;
 
-type UserState = {
-    address?: string;
-    chainId?: number;
-    isConnected: boolean;
-    ensName?: string;
-    ext?: Record<string, unknown>;
-};
-declare function useUser(): {
-    user: UserState;
-    setUser: (data: Partial<UserState>) => void;
-    addExtValue: (key: string, value: unknown) => void;
-    removeExtValue: (key: string) => void;
-    getUserState: () => UserState;
-    onUserStateChange: (callback: (user: UserState) => void) => () => void;
-};
-declare function UserContextProvider({ children }: {
-    children: ReactNode;
-}): react_jsx_runtime.JSX.Element;
-
 type ThreadContext = {
     currentThreadId: string;
     setCurrentThreadId: (id: string) => void;
@@ -152,12 +133,14 @@ type WalletHandlerConfig = {
 type WalletHandlerApi = {
     /** All queued wallet requests (tx + eip712) */
     pendingRequests: WalletRequest[];
-    /** Enqueue a wallet request (called by orchestrator on ClientSession events) */
-    enqueueRequest: (request: WalletRequest) => void;
+    /** Replace pending requests with the session's authoritative snapshot. */
+    setRequests: (requests: WalletRequest[]) => void;
+    /** Mark a request as in-flight so it is not replayed while awaiting backend ack. */
+    startRequest: (id: string) => void;
     /** Complete a request successfully — sends response to backend via ClientSession */
-    resolveRequest: (id: string, result: WalletRequestResult) => void;
+    resolveRequest: (id: string, result: WalletRequestResult) => Promise<void>;
     /** Fail a request — sends error to backend via ClientSession */
-    rejectRequest: (id: string, error?: string) => void;
+    rejectRequest: (id: string, error?: string) => Promise<void>;
 };
 declare function useWalletHandler({ getSession, }: WalletHandlerConfig): WalletHandlerApi;
 
@@ -212,10 +195,10 @@ type AomiRuntimeApi = {
     pendingWalletRequests: WalletRequest[];
     /** Mark a wallet request as being processed */
     startWalletRequest: (id: string) => void;
-    /** Complete a wallet request — dequeues + sends response to backend */
-    resolveWalletRequest: (id: string, result: WalletRequestResult) => void;
-    /** Fail a wallet request — dequeues + sends error to backend */
-    rejectWalletRequest: (id: string, error?: string) => void;
+    /** Complete a wallet request after the backend acknowledges the response */
+    resolveWalletRequest: (id: string, result: WalletRequestResult) => Promise<void>;
+    /** Fail a wallet request after the backend acknowledges the error */
+    rejectWalletRequest: (id: string, error?: string) => Promise<void>;
     /** Subscribe to inbound events by type. Returns unsubscribe function. */
     subscribe: (type: string, callback: EventSubscriber) => () => void;
     /** Send a system command to the backend */
@@ -280,6 +263,18 @@ type NotificationApi = {
 };
 declare function useNotificationHandler({ onNotification, }?: NotificationHandlerConfig): NotificationApi;
 
+declare function useUser(): {
+    user: UserState;
+    setUser: (data: Partial<UserState>) => void;
+    addExtValue: (key: string, value: unknown) => void;
+    removeExtValue: (key: string) => void;
+    getUserState: () => UserState;
+    onUserStateChange: (callback: (user: UserState) => void) => () => void;
+};
+declare function UserContextProvider({ children }: {
+    children: ReactNode;
+}): react_jsx_runtime.JSX.Element;
+
 /**
  * Utility function to merge Tailwind CSS classes with conflict resolution.
  * Combines clsx for conditional classes and tailwind-merge for deduplication.
@@ -385,4 +380,4 @@ type ControlContextProviderProps = {
 };
 declare function ControlContextProvider({ children, aomiClient, sessionId, publicKey, getThreadMetadata, updateThreadMetadata, }: ControlContextProviderProps): react_jsx_runtime.JSX.Element;
 
-export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type StoredProviderKey, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type UserState, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };
+export { type AomiRuntimeApi, AomiRuntimeProvider, type AomiRuntimeProviderProps, type ChainInfo, type ControlContextApi, ControlContextProvider, type ControlContextProviderProps, type ControlState, type EventContext, EventContextProvider, type EventContextProviderProps, type EventSubscriber, type InboundEvent, type Notification$1 as Notification, type NotificationApi, NotificationContextProvider, type NotificationContextProviderProps, type NotificationContextApi as NotificationContextValue, type NotificationHandlerConfig, type NotificationType, type SSEStatus, SUPPORTED_CHAINS, type NotificationData as ShowNotificationParams, type StoredProviderKey, type ThreadContext, ThreadContextProvider, type ThreadControlState, type ThreadMetadata, type UserConfig, UserContextProvider, type WalletHandlerApi, type WalletHandlerConfig, type WalletRequestKind, type WalletRequestResult, type WalletRequestStatus, cn, formatAddress, getChainInfo, getNetworkName, initThreadControl, useAomiRuntime, useControl, useCurrentThreadMessages, useCurrentThreadMetadata, useEventContext, useNotification, useNotificationHandler, useThreadContext, useUser, useWalletHandler };

package/dist/index.js

@@ -37,7 +37,7 @@ import { toViemSignTypedDataArgs } from "@aomi-labs/client";
 
 // packages/react/src/runtime/aomi-runtime.tsx
 import { useMemo as useMemo3 } from "react";
-import { AomiClient } from "@aomi-labs/client";
+import { AomiClient, UserState as UserState4 } from "@aomi-labs/client";
 
 // packages/react/src/contexts/control-context.tsx
 import {
@@ -880,18 +880,10 @@ import {
   useRef as useRef4,
   useState as useState3
 } from "react";
+import { UserState } from "@aomi-labs/client";
+import { UserState as UserState2 } from "@aomi-labs/client";
 import { jsx as jsx5 } from "react/jsx-runtime";
 var UserContext = createContext5(void 0);
-function normalizeUserState(next, data) {
-  if (data.isConnected === false) {
-    return __spreadProps(__spreadValues({}, next), {
-      address: void 0,
-      chainId: void 0,
-      ensName: void 0
-    });
-  }
-  return next;
-}
 function useUser() {
   const context = useContext5(UserContext);
   if (!context) {
@@ -908,10 +900,10 @@ function useUser() {
 }
 function UserContextProvider({ children }) {
   const [user, setUserState] = useState3({
-    isConnected: false,
     address: void 0,
-    chainId: void 0,
-    ensName: void 0,
+    chain_id: void 0,
+    is_connected: false,
+    ens_name: void 0,
     ext: void 0
   });
   const userRef = useRef4(user);
@@ -921,7 +913,13 @@ function UserContextProvider({ children }) {
   );
   const setUser = useCallback4((data) => {
     setUserState((prev) => {
-      const next = normalizeUserState(__spreadValues(__spreadValues({}, prev), data), data);
+      var _a, _b, _c;
+      const normalizedData = (_a = UserState.normalize(data)) != null ? _a : {};
+      const next = normalizedData.is_connected === false ? __spreadProps(__spreadValues({}, (_b = UserState.normalize(__spreadValues(__spreadValues({}, prev), normalizedData))) != null ? _b : prev), {
+        address: void 0,
+        chain_id: void 0,
+        ens_name: void 0
+      }) : (_c = UserState.normalize(__spreadValues(__spreadValues({}, prev), normalizedData))) != null ? _c : prev;
       StateChangeCallbacks.current.forEach((callback) => {
         callback(next);
       });
@@ -930,12 +928,7 @@ function UserContextProvider({ children }) {
   }, []);
   const addExtValue = useCallback4((key, value) => {
     setUserState((prev) => {
-      var _a;
-      const next = __spreadProps(__spreadValues({}, prev), {
-        ext: __spreadProps(__spreadValues({}, (_a = prev.ext) != null ? _a : {}), {
-          [key]: value
-        })
-      });
+      const next = UserState.withExt(prev, key, value);
       StateChangeCallbacks.current.forEach((callback) => {
         callback(next);
       });
@@ -944,10 +937,11 @@ function UserContextProvider({ children }) {
   }, []);
   const removeExtValue = useCallback4((key) => {
     setUserState((prev) => {
-      if (!prev.ext || !(key in prev.ext)) {
+      const ext = prev.ext;
+      if (typeof ext !== "object" || ext === null || Array.isArray(ext) || !(key in ext)) {
         return prev;
       }
-      const nextExt = __spreadValues({}, prev.ext);
+      const nextExt = __spreadValues({}, ext);
       delete nextExt[key];
       const next = __spreadProps(__spreadValues({}, prev), {
         ext: Object.keys(nextExt).length > 0 ? nextExt : void 0
@@ -990,6 +984,7 @@ import {
   AssistantRuntimeProvider,
   useExternalStoreRuntime
 } from "@assistant-ui/react";
+import { UserState as UserState3 } from "@aomi-labs/client";
 
 // packages/react/src/runtime/orchestrator.ts
 import { useCallback as useCallback5, useEffect as useEffect2, useRef as useRef5, useState as useState4 } from "react";
@@ -1183,16 +1178,13 @@ function useRuntimeOrchestrator(aomiClient, options) {
         })
       );
       cleanups.push(
-        session.on("wallet_tx_request", (req) => {
-          var _a2;
-          return (_a2 = options.onWalletRequest) == null ? void 0 : _a2.call(options, req);
-        })
-      );
-      cleanups.push(
-        session.on("wallet_eip712_request", (req) => {
-          var _a2;
-          return (_a2 = options.onWalletRequest) == null ? void 0 : _a2.call(options, req);
-        })
+        session.on(
+          "wallet_requests_changed",
+          (requests) => {
+            var _a2;
+            return (_a2 = options.onPendingRequestsChange) == null ? void 0 : _a2.call(options, requests);
+          }
+        )
       );
       cleanups.push(
         session.on("title_changed", ({ title }) => {
@@ -1218,7 +1210,7 @@ function useRuntimeOrchestrator(aomiClient, options) {
   );
   const ensureInitialState = useCallback5(
     async (threadId) => {
-      var _a;
+      var _a, _b;
       if (pendingFetches.current.has(threadId)) return;
       pendingFetches.current.add(threadId);
       try {
@@ -1226,6 +1218,7 @@ function useRuntimeOrchestrator(aomiClient, options) {
         const userState = (_a = options.getUserState) == null ? void 0 : _a.call(options);
         if (userState) session.resolveUserState(userState);
         await session.fetchCurrentState();
+        (_b = options.onPendingRequestsChange) == null ? void 0 : _b.call(options, session.getPendingRequests());
         if (threadContextRef.current.currentThreadId === threadId) {
           setIsRunning(session.getIsProcessing());
         }
@@ -1242,7 +1235,7 @@ function useRuntimeOrchestrator(aomiClient, options) {
   );
   const sendMessage = useCallback5(
     async (text, threadId) => {
-      var _a;
+      var _a, _b;
       const session = getSession(threadId);
       const userState = (_a = options.getUserState) == null ? void 0 : _a.call(options);
       if (userState) session.resolveUserState(userState);
@@ -1260,6 +1253,7 @@ function useRuntimeOrchestrator(aomiClient, options) {
         lastActiveAt: (/* @__PURE__ */ new Date()).toISOString()
       });
       await session.sendAsync(text);
+      (_b = options.onPendingRequestsChange) == null ? void 0 : _b.call(options, session.getPendingRequests());
     },
     [getSession]
   );
@@ -1447,44 +1441,85 @@ function useWalletHandler({
   getSession
 }) {
   const [pendingRequests, setPendingRequests] = useState5([]);
-  const requestsRef = useRef6([]);
-  const enqueueRequest = useCallback6((request) => {
-    requestsRef.current = [...requestsRef.current, request];
-    setPendingRequests(requestsRef.current);
+  const requestsRef = useRef6(pendingRequests);
+  const inFlightRequestSetRef = useRef6(/* @__PURE__ */ new Set());
+  const suppressedRequestSetRef = useRef6(/* @__PURE__ */ new Set());
+  const syncVisibleRequests = useCallback6(() => {
+    setPendingRequests(
+      requestsRef.current.filter(
+        (request) => !suppressedRequestSetRef.current.has(request.id)
+      )
+    );
   }, []);
+  const setRequests = useCallback6((requests) => {
+    const incomingIds = new Set(requests.map((request) => request.id));
+    for (const id of suppressedRequestSetRef.current) {
+      if (!incomingIds.has(id) && !inFlightRequestSetRef.current.has(id)) {
+        suppressedRequestSetRef.current.delete(id);
+      }
+    }
+    const preservedInFlight = requestsRef.current.filter(
+      (request) => inFlightRequestSetRef.current.has(request.id) && !incomingIds.has(request.id)
+    );
+    requestsRef.current = [...requests, ...preservedInFlight];
+    syncVisibleRequests();
+  }, [syncVisibleRequests]);
+  const startRequest = useCallback6((id) => {
+    if (!requestsRef.current.some((request) => request.id === id)) {
+      return;
+    }
+    inFlightRequestSetRef.current.add(id);
+    suppressedRequestSetRef.current.add(id);
+    syncVisibleRequests();
+  }, [syncVisibleRequests]);
   const resolveRequest = useCallback6(
-    (id, result) => {
+    async (id, result) => {
       const session = getSession();
       if (!session) {
         console.error("[wallet-handler] No session available to resolve request");
         return;
       }
-      requestsRef.current = requestsRef.current.filter((r) => r.id !== id);
-      setPendingRequests(requestsRef.current);
-      void session.resolve(id, result).catch((err) => {
+      startRequest(id);
+      try {
+        await session.resolve(id, result);
+      } catch (err) {
         console.error("[wallet-handler] Failed to resolve request:", err);
-      });
+      } finally {
+        requestsRef.current = requestsRef.current.filter(
+          (request) => request.id !== id
+        );
+        inFlightRequestSetRef.current.delete(id);
+        syncVisibleRequests();
+      }
     },
-    [getSession]
+    [getSession, startRequest, syncVisibleRequests]
   );
   const rejectRequest = useCallback6(
-    (id, error) => {
+    async (id, error) => {
       const session = getSession();
       if (!session) {
         console.error("[wallet-handler] No session available to reject request");
         return;
       }
-      requestsRef.current = requestsRef.current.filter((r) => r.id !== id);
-      setPendingRequests(requestsRef.current);
-      void session.reject(id, error).catch((err) => {
+      startRequest(id);
+      try {
+        await session.reject(id, error);
+      } catch (err) {
         console.error("[wallet-handler] Failed to reject request:", err);
-      });
+      } finally {
+        requestsRef.current = requestsRef.current.filter(
+          (request) => request.id !== id
+        );
+        inFlightRequestSetRef.current.delete(id);
+        syncVisibleRequests();
+      }
     },
-    [getSession]
+    [getSession, startRequest, syncVisibleRequests]
   );
   return {
     pendingRequests,
-    enqueueRequest,
+    setRequests,
+    startRequest,
     resolveRequest,
     rejectRequest
   };
@@ -1518,11 +1553,7 @@ function AomiRuntimeCore({
     cancelGeneration: orchestratorCancel,
     aomiClientRef
   } = useRuntimeOrchestrator(aomiClient, {
-    getPublicKey: () => {
-      var _a;
-      const userState = getUserState();
-      return userState.isConnected ? (_a = userState.address) != null ? _a : void 0 : void 0;
-    },
+    getPublicKey: () => UserState3.isConnected(getUserState()) ? UserState3.address(getUserState()) : void 0,
     getUserState,
     getApp: getCurrentThreadApp,
     getApiKey: () => getControlState().apiKey,
@@ -1530,17 +1561,20 @@ function AomiRuntimeCore({
       var _a;
       return (_a = getControlState().clientId) != null ? _a : void 0;
     },
-    onWalletRequest: (request) => walletHandler.enqueueRequest(request),
+    onPendingRequestsChange: walletHandler.setRequests,
     onEvent: (event) => eventContext.dispatch(event)
   });
   sessionManagerRef.current = sessionManager;
   const walletSnapshot = useCallback7(
-    (nextUser) => ({
-      address: nextUser.address,
-      chainId: nextUser.chainId,
-      isConnected: nextUser.isConnected,
-      ensName: nextUser.ensName
-    }),
+    (nextUser) => {
+      var _a;
+      return {
+        address: UserState3.address(nextUser),
+        chain_id: UserState3.chainId(nextUser),
+        is_connected: (_a = UserState3.isConnected(nextUser)) != null ? _a : false,
+        ens_name: typeof nextUser.ens_name === "string" ? nextUser.ens_name : void 0
+      };
+    },
     [getUserState]
   );
   const lastWalletStateRef = useRef7(walletSnapshot(getUserState()));
@@ -1549,7 +1583,7 @@ function AomiRuntimeCore({
     const unsubscribe = onUserStateChange(async (newUser) => {
       const nextWalletState = walletSnapshot(newUser);
       const prevWalletState = lastWalletStateRef.current;
-      if (prevWalletState.address === nextWalletState.address && prevWalletState.chainId === nextWalletState.chainId && prevWalletState.isConnected === nextWalletState.isConnected && prevWalletState.ensName === nextWalletState.ensName) {
+      if (prevWalletState.address === nextWalletState.address && prevWalletState.chain_id === nextWalletState.chain_id && prevWalletState.is_connected === nextWalletState.is_connected && prevWalletState.ens_name === nextWalletState.ens_name) {
         return;
       }
       lastWalletStateRef.current = nextWalletState;
@@ -1570,26 +1604,50 @@ function AomiRuntimeCore({
   ]);
   const threadContextRef = useRef7(threadContext);
   threadContextRef.current = threadContext;
-  const currentThreadIdRef = useRef7(threadContext.currentThreadId);
-  useEffect3(() => {
-    currentThreadIdRef.current = threadContext.currentThreadId;
-  }, [threadContext.currentThreadId]);
+  const remoteThreadIdsRef = useRef7(/* @__PURE__ */ new Set());
+  const warmedThreadIdsRef = useRef7(/* @__PURE__ */ new Set());
+  const warmThread = useCallback7(
+    async (threadId) => {
+      if (!remoteThreadIdsRef.current.has(threadId) || warmedThreadIdsRef.current.has(threadId)) {
+        return;
+      }
+      const userState = getUserState();
+      await aomiClientRef.current.createThread(
+        threadId,
+        UserState3.isConnected(userState) ? UserState3.address(userState) : void 0
+      );
+      warmedThreadIdsRef.current.add(threadId);
+    },
+    [aomiClientRef, getUserState]
+  );
   useEffect3(() => {
     const unsubscribe = eventContext.subscribe(
       "user_state_request",
       () => {
+        var _a, _b, _c;
+        const session = (_b = (_a = sessionManagerRef.current) == null ? void 0 : _a.get(threadContext.currentThreadId)) != null ? _b : getSession(threadContext.currentThreadId);
         eventContext.sendOutboundSystem({
           type: "user_state_response",
           sessionId: threadContext.currentThreadId,
-          payload: getUserState()
+          payload: (_c = session.getUserState()) != null ? _c : getUserState()
         });
       }
     );
     return unsubscribe;
-  }, [eventContext, threadContext.currentThreadId, getUserState]);
+  }, [eventContext, threadContext.currentThreadId, getSession, getUserState]);
   useEffect3(() => {
-    void ensureInitialState(threadContext.currentThreadId);
-  }, [ensureInitialState, threadContext.currentThreadId]);
+    const threadId = threadContext.currentThreadId;
+    let cancelled = false;
+    void (async () => {
+      await warmThread(threadId);
+      if (!cancelled) {
+        await ensureInitialState(threadId);
+      }
+    })();
+    return () => {
+      cancelled = true;
+    };
+  }, [ensureInitialState, threadContext.currentThreadId, warmThread]);
   useEffect3(() => {
     const threadId = threadContext.currentThreadId;
     const currentMeta = threadContext.getThreadMetadata(threadId);
@@ -1605,16 +1663,22 @@ function AomiRuntimeCore({
     threadContext.currentThreadId
   );
   useEffect3(() => {
-    const userAddress = user.isConnected ? user.address : void 0;
-    if (!userAddress) return;
+    const userAddress = UserState3.isConnected(user) ? UserState3.address(user) : void 0;
+    if (!userAddress) {
+      remoteThreadIdsRef.current.clear();
+      warmedThreadIdsRef.current.clear();
+      return;
+    }
     const fetchThreadList = async () => {
       var _a, _b, _c;
       try {
         const threadList = await aomiClientRef.current.listThreads(userAddress);
         const currentContext = threadContextRef.current;
+        const remoteThreadIds = /* @__PURE__ */ new Set();
         const newMetadata = new Map(currentContext.allThreadsMetadata);
         let maxChatNum = currentContext.threadCnt;
         for (const thread of threadList) {
+          remoteThreadIds.add(thread.session_id);
           const rawTitle = (_a = thread.title) != null ? _a : "";
           const title = isPlaceholderTitle(rawTitle) ? "" : rawTitle;
           const lastActive = ((_b = newMetadata.get(thread.session_id)) == null ? void 0 : _b.lastActiveAt) || (/* @__PURE__ */ new Date()).toISOString();
@@ -1633,16 +1697,26 @@ function AomiRuntimeCore({
             }
           }
         }
+        remoteThreadIdsRef.current = remoteThreadIds;
+        warmedThreadIdsRef.current = new Set(
+          Array.from(warmedThreadIdsRef.current).filter(
+            (threadId) => remoteThreadIds.has(threadId)
+          )
+        );
         currentContext.setThreadMetadata(newMetadata);
         if (maxChatNum > currentContext.threadCnt) {
           currentContext.setThreadCnt(maxChatNum);
         }
+        if (remoteThreadIds.has(currentContext.currentThreadId)) {
+          await warmThread(currentContext.currentThreadId);
+          await ensureInitialState(currentContext.currentThreadId);
+        }
       } catch (error) {
         console.error("Failed to fetch thread list:", error);
       }
     };
     void fetchThreadList();
-  }, [user.address, user.isConnected, aomiClientRef]);
+  }, [user, aomiClientRef, ensureInitialState, warmThread]);
   const threadListAdapter = useMemo2(
     () => buildThreadListAdapter({
       aomiClientRef,
@@ -1791,9 +1865,7 @@ function AomiRuntimeCore({
       clearAllNotifications: notificationContext.clearAll,
       // Wallet API
       pendingWalletRequests: walletHandler.pendingRequests,
-      startWalletRequest: () => {
-      },
-      // No-op: ClientSession manages processing state
+      startWalletRequest: walletHandler.startRequest,
       resolveWalletRequest: walletHandler.resolveRequest,
       rejectWalletRequest: walletHandler.rejectRequest,
       // Event API
@@ -1845,7 +1917,7 @@ function AomiRuntimeInner({
     {
       aomiClient,
       sessionId: threadContext.currentThreadId,
-      publicKey: user.isConnected ? (_a = user.address) != null ? _a : void 0 : void 0,
+      publicKey: UserState4.isConnected(user) ? (_a = UserState4.address(user)) != null ? _a : void 0 : void 0,
       getThreadMetadata: threadContext.getThreadMetadata,
       updateThreadMetadata: threadContext.updateThreadMetadata,
       children: /* @__PURE__ */ jsx7(