@aomi-labs/react 0.2.2 → 0.2.4

package/dist/index.cjs

@@ -42,10 +42,12 @@ __export(index_exports, {
   ControlContextProvider: () => ControlContextProvider,
   EventContextProvider: () => EventContextProvider,
   NotificationContextProvider: () => NotificationContextProvider,
+  SUPPORTED_CHAINS: () => SUPPORTED_CHAINS,
   ThreadContextProvider: () => ThreadContextProvider,
   UserContextProvider: () => UserContextProvider,
   cn: () => cn,
   formatAddress: () => formatAddress,
+  getChainInfo: () => getChainInfo,
   getNetworkName: () => getNetworkName,
   initThreadControl: () => initThreadControl,
   useAomiRuntime: () => useAomiRuntime,
@@ -301,11 +303,14 @@ var BackendApi = class {
     }
     return await response.json();
   }
-  async postChatMessage(sessionId, message, namespace, publicKey, apiKey) {
+  async postChatMessage(sessionId, message, namespace, publicKey, apiKey, userState) {
     const payload = { message, namespace };
     if (publicKey) {
       payload.public_key = publicKey;
     }
+    if (userState) {
+      payload.user_state = JSON.stringify(userState);
+    }
     return postState(
       this.backendUrl,
       "/api/chat",
@@ -1391,6 +1396,15 @@ var getNetworkName = (chainId) => {
   }
 };
 var formatAddress = (addr) => addr ? `${addr.slice(0, 6)}...${addr.slice(-4)}` : "Connect Wallet";
+var SUPPORTED_CHAINS = [
+  { id: 1, name: "Ethereum", ticker: "ETH" },
+  { id: 137, name: "Polygon", ticker: "MATIC" },
+  { id: 42161, name: "Arbitrum", ticker: "ARB" },
+  { id: 8453, name: "Base", ticker: "BASE" },
+  { id: 10, name: "Optimism", ticker: "OP" },
+  { id: 11155111, name: "Sepolia", ticker: "SEP" }
+];
+var getChainInfo = (chainId) => chainId === void 0 ? void 0 : SUPPORTED_CHAINS.find((c) => c.id === chainId);
 
 // packages/react/src/state/backend-state.ts
 function createBackendState() {
@@ -1464,7 +1478,7 @@ var MessageController = class {
     this.getThreadContextApi().setThreadMessages(threadId, threadMessages);
   }
   async outbound(message, threadId) {
-    var _a, _b, _c, _d, _e, _f;
+    var _a, _b, _c, _d, _e, _f, _g, _h;
     const backendState = this.config.backendStateRef.current;
     const text = message.content.filter(
       (part) => part.type === "text"
@@ -1492,6 +1506,7 @@ var MessageController = class {
     const namespace = this.config.getNamespace();
     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);
     try {
       this.markRunning(threadId, true);
       const response = await this.config.backendApiRef.current.postChatMessage(
@@ -1499,12 +1514,13 @@ var MessageController = class {
         text,
         namespace,
         publicKey,
-        apiKey
+        apiKey,
+        userState
       );
       if (response == null ? void 0 : response.messages) {
         this.inbound(threadId, response.messages);
       }
-      if (((_f = response == null ? void 0 : response.system_events) == null ? void 0 : _f.length) && this.config.onSyncEvents) {
+      if (((_h = response == null ? void 0 : response.system_events) == null ? void 0 : _h.length) && this.config.onSyncEvents) {
         this.config.onSyncEvents(backendThreadId, response.system_events);
       }
       if (response == null ? void 0 : response.is_processing) {
@@ -1518,7 +1534,7 @@ var MessageController = class {
     }
   }
   async flushPendingChat(threadId) {
-    var _a, _b, _c, _d, _e;
+    var _a, _b, _c, _d, _e, _f, _g;
     const backendState = this.config.backendStateRef.current;
     const pending = dequeuePendingChat(backendState, threadId);
     if (!pending.length) return;
@@ -1526,6 +1542,7 @@ var MessageController = class {
     const namespace = this.config.getNamespace();
     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);
     for (const text of pending) {
       try {
         await this.config.backendApiRef.current.postChatMessage(
@@ -1533,7 +1550,8 @@ var MessageController = class {
           text,
           namespace,
           publicKey,
-          apiKey
+          apiKey,
+          userState
         );
       } catch (error) {
         console.error("Failed to send queued message:", error);
@@ -1688,6 +1706,7 @@ function useRuntimeOrchestrator(backendApi, options) {
       getPublicKey: options.getPublicKey,
       getNamespace: options.getNamespace,
       getApiKey: options.getApiKey,
+      getUserState: options.getUserState,
       onSyncEvents: options.onSyncEvents
     });
   }
@@ -1783,7 +1802,8 @@ function buildThreadListAdapter({
   userAddress,
   setIsRunning,
   getNamespace,
-  getApiKey
+  getApiKey,
+  getUserState
 }) {
   const backendState = backendStateRef.current;
   const { regularThreads, archivedThreads } = buildThreadLists(
@@ -1876,6 +1896,7 @@ function buildThreadListAdapter({
           backendState.pendingChat.delete(uiThreadId);
           const namespace = getNamespace();
           const apiKey = (_b = getApiKey == null ? void 0 : getApiKey()) != null ? _b : void 0;
+          const userState = getUserState == null ? void 0 : getUserState();
           for (const text of pendingMessages) {
             try {
               await backendApiRef.current.postChatMessage(
@@ -1883,7 +1904,8 @@ function buildThreadListAdapter({
                 text,
                 namespace,
                 userAddress,
-                apiKey
+                apiKey,
+                userState
               );
             } catch (error) {
               console.error("Failed to send queued message:", error);
@@ -2146,7 +2168,8 @@ function AomiRuntimeCore({
         var _a, _b;
         return (_b = (_a = getCurrentThreadControl().namespace) != null ? _a : getControlState().defaultNamespace) != null ? _b : "default";
       },
-      getApiKey: () => getControlState().apiKey
+      getApiKey: () => getControlState().apiKey,
+      getUserState
     }),
     [
       backendApiRef,
@@ -2157,7 +2180,8 @@ function AomiRuntimeCore({
       threadContext,
       threadContext.currentThreadId,
       threadContext.allThreadsMetadata,
-      getControlState
+      getControlState,
+      getUserState
     ]
   );
   (0, import_react8.useEffect)(() => {
@@ -2542,10 +2566,12 @@ function useNotificationHandler({
   ControlContextProvider,
   EventContextProvider,
   NotificationContextProvider,
+  SUPPORTED_CHAINS,
   ThreadContextProvider,
   UserContextProvider,
   cn,
   formatAddress,
+  getChainInfo,
   getNetworkName,
   initThreadControl,
   useAomiRuntime,