@aomi-labs/react 0.3.3 → 0.3.4

package/dist/index.cjs

@@ -276,6 +276,7 @@ function ControlContextProvider({
   var _a, _b;
   const [state, setStateInternal] = (0, import_react.useState)(() => ({
     apiKey: null,
+    clientId: null,
     availableModels: [],
     authorizedApps: [],
     defaultModel: null,
@@ -296,6 +297,11 @@ function ControlContextProvider({
   const callbacks = (0, import_react.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;
+  (0, import_react.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 }));
+  }, []);
   (0, import_react.useEffect)(() => {
     var _a2, _b2;
     try {
@@ -369,6 +375,24 @@ function ControlContextProvider({
       return next;
     });
   }, []);
+  const ingestSecrets = (0, import_react.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 = (0, import_react.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 = (0, import_react.useCallback)(async () => {
     try {
       const models = await aomiClientRef.current.getModels(
@@ -549,6 +573,8 @@ function ControlContextProvider({
       value: {
         state,
         setApiKey,
+        ingestSecrets,
+        clearSecrets,
         getAvailableModels,
         getAuthorizedApps,
         getCurrentThreadControl,
@@ -1070,7 +1096,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"
@@ -1093,18 +1119,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) {
@@ -1163,7 +1190,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(
@@ -1171,9 +1198,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);
@@ -1240,6 +1269,7 @@ function useRuntimeOrchestrator(aomiClient, options) {
       },
       onSyncEvents: options.onSyncEvents,
       getUserState: options.getUserState,
+      getClientId: options.getClientId,
       onStart: (threadId) => {
         if (threadContextRef.current.currentThreadId === threadId) {
           setIsRunning(true);
@@ -1262,29 +1292,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 = (0, import_react6.useCallback)(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);
         }
@@ -1617,7 +1650,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,
@@ -1631,7 +1664,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 = (0, import_react9.useCallback)(
     (nextUser) => ({
@@ -1858,8 +1895,9 @@ function AomiRuntimeCore({
   (0, import_react9.useEffect)(() => {
     return () => {
       polling.stopAll();
+      void clearSecrets();
     };
-  }, [polling]);
+  }, [polling, clearSecrets]);
   const userContext = useUser();
   const sendMessage = (0, import_react9.useCallback)(
     async (text) => {