@aomi-labs/react 0.3.2 → 0.3.4

package/dist/index.d.cts

@@ -16,10 +16,13 @@ type UserState = {
     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;
 };
@@ -166,6 +169,10 @@ type AomiRuntimeApi = {
     getUserState: () => UserState;
     /** Update user state (partial updates merged with existing state) */
     setUser: (data: Partial<UserState>) => void;
+    /** Add or overwrite a value in user_state.ext */
+    addExtValue: (key: string, value: unknown) => void;
+    /** Remove a value from user_state.ext */
+    removeExtValue: (key: string) => void;
     /** Subscribe to user state changes. Returns unsubscribe function. */
     onUserStateChange: (callback: (user: UserState) => void) => () => void;
     /** ID of the currently active thread */
@@ -318,6 +325,8 @@ declare const getChainInfo: (chainId: number | undefined) => ChainInfo | undefin
 type ControlState = {
     /** API key for authenticated requests */
     apiKey: string | null;
+    /** Stable client identifier for this browser tab (associates sessions with secrets) */
+    clientId: string | null;
     /** Available models fetched from backend */
     availableModels: string[];
     /** Authorized apps fetched from backend */
@@ -328,10 +337,14 @@ type ControlState = {
     defaultApp: string | null;
 };
 type ControlContextApi = {
-    /** Global state (apiKey, available models/apps) */
+    /** Global state (apiKey, clientId, available models/apps) */
     state: ControlState;
     /** Update global state (apiKey only) */
     setApiKey: (apiKey: string | null) => void;
+    /** Ingest secrets into the backend vault, returns opaque handles */
+    ingestSecrets: (secrets: Record<string, string>) => Promise<Record<string, string>>;
+    /** Clear all secrets from the backend vault */
+    clearSecrets: () => Promise<void>;
     /** Fetch available models from backend */
     getAvailableModels: () => Promise<string[]>;
     /** Fetch authorized apps from backend */

package/dist/index.d.ts

@@ -16,10 +16,13 @@ type UserState = {
     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;
 };
@@ -166,6 +169,10 @@ type AomiRuntimeApi = {
     getUserState: () => UserState;
     /** Update user state (partial updates merged with existing state) */
     setUser: (data: Partial<UserState>) => void;
+    /** Add or overwrite a value in user_state.ext */
+    addExtValue: (key: string, value: unknown) => void;
+    /** Remove a value from user_state.ext */
+    removeExtValue: (key: string) => void;
     /** Subscribe to user state changes. Returns unsubscribe function. */
     onUserStateChange: (callback: (user: UserState) => void) => () => void;
     /** ID of the currently active thread */
@@ -318,6 +325,8 @@ declare const getChainInfo: (chainId: number | undefined) => ChainInfo | undefin
 type ControlState = {
     /** API key for authenticated requests */
     apiKey: string | null;
+    /** Stable client identifier for this browser tab (associates sessions with secrets) */
+    clientId: string | null;
     /** Available models fetched from backend */
     availableModels: string[];
     /** Authorized apps fetched from backend */
@@ -328,10 +337,14 @@ type ControlState = {
     defaultApp: string | null;
 };
 type ControlContextApi = {
-    /** Global state (apiKey, available models/apps) */
+    /** Global state (apiKey, clientId, available models/apps) */
     state: ControlState;
     /** Update global state (apiKey only) */
     setApiKey: (apiKey: string | null) => void;
+    /** Ingest secrets into the backend vault, returns opaque handles */
+    ingestSecrets: (secrets: Record<string, string>) => Promise<Record<string, string>>;
+    /** Clear all secrets from the backend vault */
+    clearSecrets: () => Promise<void>;
     /** Fetch available models from backend */
     getAvailableModels: () => Promise<string[]>;
     /** Fetch authorized apps from backend */

package/dist/index.js

@@ -239,6 +239,7 @@ function ControlContextProvider({
   var _a, _b;
   const [state, setStateInternal] = useState(() => ({
     apiKey: null,
+    clientId: null,
     availableModels: [],
     authorizedApps: [],
     defaultModel: null,
@@ -259,6 +260,11 @@ function ControlContextProvider({
   const callbacks = useRef(/* @__PURE__ */ new Set());
   const currentThreadMetadata = getThreadMetadata(sessionId);
   const isProcessing = (_b = (_a = currentThreadMetadata == null ? void 0 : currentThreadMetadata.control) == null ? void 0 : _a.isProcessing) != null ? _b : false;
+  useEffect(() => {
+    var _a2, _b2, _c;
+    const clientId = (_c = (_b2 = (_a2 = globalThis.crypto) == null ? void 0 : _a2.randomUUID) == null ? void 0 : _b2.call(_a2)) != null ? _c : `client-${Date.now()}`;
+    setStateInternal((prev) => __spreadProps(__spreadValues({}, prev), { clientId }));
+  }, []);
   useEffect(() => {
     var _a2, _b2;
     try {
@@ -332,6 +338,24 @@ function ControlContextProvider({
       return next;
     });
   }, []);
+  const ingestSecrets = useCallback(
+    async (secrets) => {
+      const clientId = stateRef.current.clientId;
+      if (!clientId) throw new Error("clientId not initialized");
+      const { handles } = await aomiClientRef.current.ingestSecrets(
+        clientId,
+        secrets
+      );
+      return handles;
+    },
+    []
+  );
+  const clearSecrets = useCallback(async () => {
+    var _a2, _b2;
+    const clientId = stateRef.current.clientId;
+    if (!clientId) return;
+    await ((_b2 = (_a2 = aomiClientRef.current).clearSecrets) == null ? void 0 : _b2.call(_a2, clientId));
+  }, []);
   const getAvailableModels = useCallback(async () => {
     try {
       const models = await aomiClientRef.current.getModels(
@@ -512,6 +536,8 @@ function ControlContextProvider({
       value: {
         state,
         setApiKey,
+        ingestSecrets,
+        clearSecrets,
         getAvailableModels,
         getAuthorizedApps,
         getCurrentThreadControl,
@@ -832,6 +858,8 @@ function useUser() {
   return {
     user: context.user,
     setUser: context.setUser,
+    addExtValue: context.addExtValue,
+    removeExtValue: context.removeExtValue,
     getUserState: context.getUserState,
     onUserStateChange: context.onUserStateChange
   };
@@ -841,7 +869,8 @@ function UserContextProvider({ children }) {
     isConnected: false,
     address: void 0,
     chainId: void 0,
-    ensName: void 0
+    ensName: void 0,
+    ext: void 0
   });
   const userRef = useRef4(user);
   userRef.current = user;
@@ -857,6 +886,36 @@ function UserContextProvider({ children }) {
       return next;
     });
   }, []);
+  const addExtValue = useCallback4((key, value) => {
+    setUserState((prev) => {
+      var _a;
+      const next = __spreadProps(__spreadValues({}, prev), {
+        ext: __spreadProps(__spreadValues({}, (_a = prev.ext) != null ? _a : {}), {
+          [key]: value
+        })
+      });
+      StateChangeCallbacks.current.forEach((callback) => {
+        callback(next);
+      });
+      return next;
+    });
+  }, []);
+  const removeExtValue = useCallback4((key) => {
+    setUserState((prev) => {
+      if (!prev.ext || !(key in prev.ext)) {
+        return prev;
+      }
+      const nextExt = __spreadValues({}, prev.ext);
+      delete nextExt[key];
+      const next = __spreadProps(__spreadValues({}, prev), {
+        ext: Object.keys(nextExt).length > 0 ? nextExt : void 0
+      });
+      StateChangeCallbacks.current.forEach((callback) => {
+        callback(next);
+      });
+      return next;
+    });
+  }, []);
   const getUserState = useCallback4(() => userRef.current, []);
   const onUserStateChange = useCallback4(
     (callback) => {
@@ -873,6 +932,8 @@ function UserContextProvider({ children }) {
       value: {
         user,
         setUser,
+        addExtValue,
+        removeExtValue,
         getUserState,
         onUserStateChange
       },
@@ -1030,7 +1091,7 @@ var MessageController = class {
     this.getThreadContextApi().setThreadMessages(threadId, threadMessages);
   }
   async outbound(message, threadId) {
-    var _a, _b, _c, _d, _e, _f, _g, _h;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     const backendState = this.config.backendStateRef.current;
     const text = message.content.filter(
       (part) => part.type === "text"
@@ -1053,18 +1114,19 @@ var MessageController = class {
     const app = this.config.getApp();
     const publicKey = (_b = (_a = this.config).getPublicKey) == null ? void 0 : _b.call(_a);
     const apiKey = (_e = (_d = (_c = this.config).getApiKey) == null ? void 0 : _d.call(_c)) != null ? _e : void 0;
-    const userState = (_g = (_f = this.config).getUserState) == null ? void 0 : _g.call(_f);
+    const clientId = (_g = (_f = this.config).getClientId) == null ? void 0 : _g.call(_f);
+    const userState = (_i = (_h = this.config).getUserState) == null ? void 0 : _i.call(_h);
     try {
       this.markRunning(threadId, true);
       const response = await this.config.aomiClientRef.current.sendMessage(
         backendThreadId,
         text,
-        { app, publicKey, apiKey, userState }
+        { app, publicKey, apiKey, userState, clientId }
       );
       if (response == null ? void 0 : response.messages) {
         this.inbound(threadId, response.messages);
       }
-      if (((_h = response == null ? void 0 : response.system_events) == null ? void 0 : _h.length) && this.config.onSyncEvents) {
+      if (((_j = response == null ? void 0 : response.system_events) == null ? void 0 : _j.length) && this.config.onSyncEvents) {
         this.config.onSyncEvents(backendThreadId, response.system_events);
       }
       if (response == null ? void 0 : response.is_processing) {
@@ -1123,7 +1185,7 @@ var PollingController = class {
     const backendThreadId = resolveThreadId(backendState, threadId);
     setThreadRunning(backendState, threadId, true);
     const tick = async () => {
-      var _a2, _b2;
+      var _a2, _b2, _c, _d;
       if (!this.intervals.has(threadId)) return;
       try {
         console.log(
@@ -1131,9 +1193,11 @@ var PollingController = class {
           threadId
         );
         const userState = (_b2 = (_a2 = this.config).getUserState) == null ? void 0 : _b2.call(_a2);
+        const clientId = (_d = (_c = this.config).getClientId) == null ? void 0 : _d.call(_c);
         const state = await this.config.aomiClientRef.current.fetchState(
           backendThreadId,
-          userState
+          userState,
+          clientId
         );
         if (!this.intervals.has(threadId)) return;
         this.handleState(threadId, state);
@@ -1200,6 +1264,7 @@ function useRuntimeOrchestrator(aomiClient, options) {
       },
       onSyncEvents: options.onSyncEvents,
       getUserState: options.getUserState,
+      getClientId: options.getClientId,
       onStart: (threadId) => {
         if (threadContextRef.current.currentThreadId === threadId) {
           setIsRunning(true);
@@ -1222,29 +1287,32 @@ function useRuntimeOrchestrator(aomiClient, options) {
       getPublicKey: options.getPublicKey,
       getApp: options.getApp,
       getApiKey: options.getApiKey,
+      getClientId: options.getClientId,
       getUserState: options.getUserState,
       onSyncEvents: options.onSyncEvents
     });
   }
   const ensureInitialState = useCallback5(async (threadId) => {
-    var _a, _b, _c, _d;
+    var _a, _b, _c, _d, _e;
     if (pendingFetches.current.has(threadId)) return;
     const backendThreadId = resolveThreadId(backendStateRef.current, threadId);
     pendingFetches.current.add(threadId);
     try {
       const userState = (_a = options.getUserState) == null ? void 0 : _a.call(options);
+      const clientId = (_b = options.getClientId) == null ? void 0 : _b.call(options);
       const state = await aomiClientRef.current.fetchState(
         backendThreadId,
-        userState
+        userState,
+        clientId
       );
-      (_b = messageControllerRef.current) == null ? void 0 : _b.inbound(threadId, state.messages);
-      if (((_c = state.system_events) == null ? void 0 : _c.length) && options.onSyncEvents) {
+      (_c = messageControllerRef.current) == null ? void 0 : _c.inbound(threadId, state.messages);
+      if (((_d = state.system_events) == null ? void 0 : _d.length) && options.onSyncEvents) {
         options.onSyncEvents(backendThreadId, state.system_events);
       }
       if (threadContextRef.current.currentThreadId === threadId) {
         if (state.is_processing) {
           setIsRunning(true);
-          (_d = pollingRef.current) == null ? void 0 : _d.start(threadId);
+          (_e = pollingRef.current) == null ? void 0 : _e.start(threadId);
         } else {
           setIsRunning(false);
         }
@@ -1580,7 +1648,7 @@ function AomiRuntimeCore({
   const notificationContext = useNotification();
   const { dispatchInboundSystem: dispatchSystemEvents } = eventContext;
   const { user, onUserStateChange, getUserState } = useUser();
-  const { getControlState, getCurrentThreadApp } = useControl();
+  const { getControlState, getCurrentThreadApp, clearSecrets } = useControl();
   const {
     backendStateRef,
     polling,
@@ -1594,24 +1662,46 @@ function AomiRuntimeCore({
     getPublicKey: () => getUserState().address,
     getUserState,
     getApp: getCurrentThreadApp,
-    getApiKey: () => getControlState().apiKey
+    getApiKey: () => getControlState().apiKey,
+    getClientId: () => {
+      var _a;
+      return (_a = getControlState().clientId) != null ? _a : void 0;
+    }
   });
+  const walletSnapshot = useCallback7(
+    (nextUser) => ({
+      address: nextUser.address,
+      chainId: nextUser.chainId,
+      isConnected: nextUser.isConnected,
+      ensName: nextUser.ensName
+    }),
+    [getUserState]
+  );
+  const lastWalletStateRef = useRef7(walletSnapshot(getUserState()));
   useEffect4(() => {
+    lastWalletStateRef.current = walletSnapshot(getUserState());
     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) {
+        return;
+      }
+      lastWalletStateRef.current = nextWalletState;
       const sessionId = threadContext.currentThreadId;
       const message = JSON.stringify({
         type: "wallet:state_changed",
-        payload: {
-          address: newUser.address,
-          chainId: newUser.chainId,
-          isConnected: newUser.isConnected,
-          ensName: newUser.ensName
-        }
+        payload: nextWalletState
       });
       await aomiClientRef.current.sendSystemMessage(sessionId, message);
     });
     return unsubscribe;
-  }, [onUserStateChange, aomiClientRef, threadContext.currentThreadId]);
+  }, [
+    onUserStateChange,
+    aomiClientRef,
+    threadContext.currentThreadId,
+    getUserState,
+    walletSnapshot
+  ]);
   const threadContextRef = useRef7(threadContext);
   threadContextRef.current = threadContext;
   const currentThreadIdRef = useRef7(threadContext.currentThreadId);
@@ -1803,8 +1893,9 @@ function AomiRuntimeCore({
   useEffect4(() => {
     return () => {
       polling.stopAll();
+      void clearSecrets();
     };
-  }, [polling]);
+  }, [polling, clearSecrets]);
   const userContext = useUser();
   const sendMessage = useCallback7(
     async (text) => {
@@ -1867,6 +1958,8 @@ function AomiRuntimeCore({
       user: userContext.user,
       getUserState: userContext.getUserState,
       setUser: userContext.setUser,
+      addExtValue: userContext.addExtValue,
+      removeExtValue: userContext.removeExtValue,
       onUserStateChange: userContext.onUserStateChange,
       // Thread API
       currentThreadId: threadContext.currentThreadId,