@aomi-labs/react 0.3.3 → 0.3.4

package/dist/index.d.cts

@@ -325,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 */
@@ -335,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

@@ -325,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 */
@@ -335,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,
@@ -1065,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"
@@ -1088,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) {
@@ -1158,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(
@@ -1166,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);
@@ -1235,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);
@@ -1257,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);
         }
@@ -1615,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,
@@ -1629,7 +1662,11 @@ 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) => ({
@@ -1856,8 +1893,9 @@ function AomiRuntimeCore({
   useEffect4(() => {
     return () => {
       polling.stopAll();
+      void clearSecrets();
     };
-  }, [polling]);
+  }, [polling, clearSecrets]);
   const userContext = useUser();
   const sendMessage = useCallback7(
     async (text) => {