@mocanetwork/airkit-connector 1.9.0-beta.1 → 1.9.0-beta.3

package/dist/airkitConnector.cjs.js

@@ -94,7 +94,7 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
   var e = new Error(message);
   return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
 };
-var version$2 = "1.9.0-beta.1";
+var version$2 = "1.9.0-beta.3";
 var airkitPackage = {
   version: version$2
 };
@@ -138,6 +138,10 @@ const AirAuthMessageTypes = {
   RECOVERY_RESPONSE: "air_auth_recovery_response",
   ISSUE_ON_BEHALF_NEW_USER_CONFIRMATION_REQUEST: "air_auth_issue_on_behalf_new_user_confirmation_request",
   ISSUE_ON_BEHALF_NEW_USER_CONFIRMATION_RESPONSE: "air_auth_issue_on_behalf_new_user_confirmation_response",
+  REMOVE_SIGNER_SIGNATURE_REQUEST: "air_auth_remove_signer_signature_request",
+  REMOVE_SIGNER_SIGNATURE_RESPONSE: "air_auth_remove_signer_signature_response",
+  RESET_WALLET_REQUEST: "air_auth_reset_wallet_request",
+  RESET_WALLET_RESPONSE: "air_auth_reset_wallet_response",
   EXPIRED_LOGOUT_REQUEST: "air_auth_expired_logout_request"
 };
 const AirCredentialMessageTypes = {
@@ -168,6 +172,12 @@ const AirRecoveryMessageTypes = {
   RECOVERY_IFRAME_VISIBILITY_REQUEST: "air_recovery_iframe_visibility_request",
   UPDATE_SESSION_CONFIG_REQUEST: "air_recovery_update_session_config_request",
   UPDATE_SESSION_CONFIG_RESPONSE: "air_recovery_update_session_config_response",
+  GET_AGENT_KEYS_REQUEST: "air_recovery_get_agent_keys_request",
+  GET_AGENT_KEYS_RESPONSE: "air_recovery_get_agent_keys_response",
+  REGISTER_AGENT_KEY_REQUEST: "air_recovery_register_agent_key_request",
+  REGISTER_AGENT_KEY_RESPONSE: "air_recovery_register_agent_key_response",
+  REMOVE_AGENT_KEY_REQUEST: "air_recovery_remove_agent_key_request",
+  REMOVE_AGENT_KEY_RESPONSE: "air_recovery_remove_agent_key_response",
   LOGOUT_REQUEST: "air_recovery_logout_request",
   LOGOUT_RESPONSE: "air_recovery_logout_response"
 };
@@ -1850,7 +1860,7 @@ class ProviderMessageService extends MessageServiceBase {
     });
   }
   async sendWalletProviderRequest(payload) {
-    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.RESPONSE), filter(msg => msg.payload.method === payload.method)));
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.RESPONSE), filter(msg => msg.payload.method === payload.method && msg.payload.requestId === payload.requestId)));
     await this.sendMessage({
       type: AirWalletProviderMessageTypes.REQUEST,
       payload
@@ -1862,7 +1872,7 @@ _a$4 = ProviderMessageService;
 _ProviderMessageService_instance = {
   value: void 0
 };
-var _AirWalletProvider_instances, _AirWalletProvider_providerMessageService, _AirWalletProvider_isWalletInitialized, _AirWalletProvider_getLoginResult, _AirWalletProvider_ensureWallet, _AirWalletProvider_eventListeners, _AirWalletProvider_emit;
+var _AirWalletProvider_instances, _AirWalletProvider_providerMessageService, _AirWalletProvider_isWalletInitialized, _AirWalletProvider_getLoginResult, _AirWalletProvider_ensureWallet, _AirWalletProvider_eventListeners, _AirWalletProvider_currentChainId, _AirWalletProvider_emit, _AirWalletProvider_getAddressForCurrentChain, _AirWalletProvider_parseChainId;
 class AirWalletProvider {
   constructor({
     ensureWallet,
@@ -1875,9 +1885,16 @@ class AirWalletProvider {
     _AirWalletProvider_getLoginResult.set(this, void 0);
     _AirWalletProvider_ensureWallet.set(this, void 0);
     _AirWalletProvider_eventListeners.set(this, void 0);
+    _AirWalletProvider_currentChainId.set(this, null);
     this.startEventMessageListening = async walletIframe => {
       await __classPrivateFieldGet$1(this, _AirWalletProvider_providerMessageService, "f").open(walletIframe);
       __classPrivateFieldGet$1(this, _AirWalletProvider_providerMessageService, "f").messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.EVENT)).subscribe(message => {
+        if (message.payload.event === "chainChanged") {
+          const parsedChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_parseChainId).call(this, message.payload.data);
+          if (parsedChainId !== null) {
+            __classPrivateFieldSet(this, _AirWalletProvider_currentChainId, parsedChainId);
+          }
+        }
         __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_emit).call(this, message.payload.event, ...[message.payload.data]);
       });
     };
@@ -1914,8 +1931,12 @@ class AirWalletProvider {
     if (!loginResult) {
       throw new UnauthorizedProviderError("User is not logged in");
     }
-    if ((method === "eth_accounts" || method === "eth_requestAccounts") && !__classPrivateFieldGet$1(this, _AirWalletProvider_isWalletInitialized, "f").call(this) && loginResult?.abstractAccountAddress) {
-      return [loginResult.abstractAccountAddress];
+    // Early return optimization: return cached address from login result if wallet not yet initialized
+    if ((method === "eth_accounts" || method === "eth_requestAccounts") && !__classPrivateFieldGet$1(this, _AirWalletProvider_isWalletInitialized, "f").call(this)) {
+      const address = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_getAddressForCurrentChain).call(this, loginResult);
+      if (address) {
+        return [address];
+      }
     }
     const skipWalletLoginMethods = ["eth_chainId"];
     try {
@@ -1937,6 +1958,12 @@ class AirWalletProvider {
     if (!response.success) {
       throw ensureProviderRpcError(response.payload);
     }
+    if (method === "eth_chainId") {
+      const parsedChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_parseChainId).call(this, response.payload.response);
+      if (parsedChainId !== null) {
+        __classPrivateFieldSet(this, _AirWalletProvider_currentChainId, parsedChainId);
+      }
+    }
     return response.payload.response;
   }
   on(eventName, listener) {
@@ -1960,7 +1987,7 @@ class AirWalletProvider {
     });
   }
 }
-_AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_isWalletInitialized = new WeakMap(), _AirWalletProvider_getLoginResult = new WeakMap(), _AirWalletProvider_ensureWallet = new WeakMap(), _AirWalletProvider_eventListeners = new WeakMap(), _AirWalletProvider_instances = new WeakSet(), _AirWalletProvider_emit = function _AirWalletProvider_emit(eventName, ...args) {
+_AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_isWalletInitialized = new WeakMap(), _AirWalletProvider_getLoginResult = new WeakMap(), _AirWalletProvider_ensureWallet = new WeakMap(), _AirWalletProvider_eventListeners = new WeakMap(), _AirWalletProvider_currentChainId = new WeakMap(), _AirWalletProvider_instances = new WeakSet(), _AirWalletProvider_emit = function _AirWalletProvider_emit(eventName, ...args) {
   (__classPrivateFieldGet$1(this, _AirWalletProvider_eventListeners, "f")[eventName] || []).forEach(listener => {
     try {
       return listener(...args);
@@ -1968,6 +1995,30 @@ _AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_is
       log$1.warn(error);
     }
   });
+}, _AirWalletProvider_getAddressForCurrentChain = function _AirWalletProvider_getAddressForCurrentChain(loginResult) {
+  const currentChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_currentChainId, "f");
+  // If we have the new addresses array and current chainId, look up by chain
+  if (loginResult.abstractAccountAddresses && currentChainId !== null) {
+    const chainIdStr = String(currentChainId);
+    const addressEntry = loginResult.abstractAccountAddresses.find(entry => entry.chainIds.includes(chainIdStr));
+    if (addressEntry) {
+      return addressEntry.address;
+    }
+  }
+  // Fallback to legacy single address (backward compatibility)
+  return loginResult.abstractAccountAddress;
+}, _AirWalletProvider_parseChainId = function _AirWalletProvider_parseChainId(chainId) {
+  if (typeof chainId === "number") {
+    return Number.isFinite(chainId) ? chainId : null;
+  }
+  if (typeof chainId === "bigint") {
+    return Number(chainId);
+  }
+  if (typeof chainId !== "string" || !chainId) {
+    return null;
+  }
+  const parsed = chainId.startsWith("0x") ? parseInt(chainId, 16) : parseInt(chainId, 10);
+  return Number.isNaN(parsed) ? null : parsed;
 };
 var _a$3, _AuthMessageService_instance;
 const ALLOWED_AUTH_MESSAGES = [AirAuthMessageTypes.INITIALIZATION_RESPONSE, AirAuthMessageTypes.LOGIN_RESPONSE, AirAuthMessageTypes.SETUP_WALLET_REQUEST, AirAuthMessageTypes.SETUP_RECOVERY_REQUEST, AirAuthMessageTypes.SETUP_CREDENTIAL_REQUEST, AirAuthMessageTypes.LOGOUT_RESPONSE, AirAuthMessageTypes.UPDATE_SESSION_CONFIG_RESPONSE, AirAuthMessageTypes.PARTNER_USER_INFO_RESPONSE, AirAuthMessageTypes.CROSS_PARTNER_TOKEN_RESPONSE, AirAuthMessageTypes.PARTNER_ACCESS_TOKEN_RESPONSE, AirAuthMessageTypes.IFRAME_VISIBILITY_REQUEST, AirAuthMessageTypes.START_RECOVERY_RESPONSE, AirAuthMessageTypes.EXPIRED_LOGOUT_REQUEST];
@@ -2275,6 +2326,29 @@ class RecoveryMessageService extends MessageServiceBase {
     });
     return response;
   }
+  async sendGetAgentKeysRequest() {
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.GET_AGENT_KEYS_RESPONSE)));
+    await this.sendMessage({
+      type: AirRecoveryMessageTypes.GET_AGENT_KEYS_REQUEST
+    });
+    return response;
+  }
+  async sendRegisterAgentKeyRequest(payload) {
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.REGISTER_AGENT_KEY_RESPONSE)));
+    await this.sendMessage({
+      type: AirRecoveryMessageTypes.REGISTER_AGENT_KEY_REQUEST,
+      payload
+    });
+    return response;
+  }
+  async sendRemoveAgentKeyRequest(payload) {
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.REMOVE_AGENT_KEY_RESPONSE)));
+    await this.sendMessage({
+      type: AirRecoveryMessageTypes.REMOVE_AGENT_KEY_REQUEST,
+      payload
+    });
+    return response;
+  }
   async open(element) {
     await this._open({
       window: element.contentWindow,
@@ -2799,9 +2873,6 @@ class AirService {
           case AirAuthMessageTypes.SETUP_CREDENTIAL_REQUEST:
             {
               try {
-                await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, {
-                  skipWalletLogin: true
-                });
                 await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
                 await __classPrivateFieldGet$1(this, _AirService_authMessagingService, "f").sendSetupCredentialSuccessResponse();
               } catch (err) {
@@ -2872,10 +2943,6 @@ class AirService {
     });
     if (payload.success === true) {
       __classPrivateFieldSet(this, _AirService_loginResult, __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_createLoginResult).call(this, payload));
-      if (payload.needsWalletRestart === true) {
-        await Promise.all([__classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_cleanUpWallet).call(this), __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_cleanUpCredential).call(this)]);
-        void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
-      }
       __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_triggerAirAuthLoggedIn).call(this);
       return __classPrivateFieldGet$1(this, _AirService_loginResult, "f");
     }
@@ -2995,6 +3062,43 @@ class AirService {
       throw new AirServiceError(result.payload.errorName, result.payload.errorMessage);
     }
   }
+  async getAgentKeys() {
+    await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+    const {
+      payload
+    } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendGetAgentKeysRequest();
+    if (payload.success === false) {
+      throw new AirServiceError(payload.errorName, payload.errorMessage);
+    }
+    return payload.agentKeys;
+  }
+  async registerAgentKey(publicKey) {
+    await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+    const {
+      payload
+    } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendRegisterAgentKeyRequest({
+      publicKey
+    });
+    if (payload.success === false) {
+      throw new AirServiceError(payload.errorName, payload.errorMessage);
+    }
+    return {
+      id: payload.id,
+      publicKey: payload.publicKey,
+      createdAt: payload.createdAt
+    };
+  }
+  async removeAgentKey(id) {
+    await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+    const {
+      payload
+    } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendRemoveAgentKeyRequest({
+      id
+    });
+    if (payload.success === false) {
+      throw new AirServiceError(payload.errorName, payload.errorMessage);
+    }
+  }
   async getUserInfo() {
     __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_assertLoggedIn).call(this);
     const info = await __classPrivateFieldGet$1(this, _AirService_authMessagingService, "f").sendPartnerUserInfoRequest();
@@ -3009,7 +3113,7 @@ class AirService {
         id: info.payload.user.id,
         abstractAccountAddress: info.payload.user.abstractAccountAddress,
         email: info.payload.user.email,
-        isMFASetup: info.payload.user.isMFASetup
+        isMFASetup: info.payload.user.activeMfaMethods.length > 0
       }
     };
   }
@@ -3128,9 +3232,12 @@ class AirService {
     authToken,
     programId,
     redirectUrl,
-    requestDisclosure,
+    fieldsToDisclose,
     offchain
   }) {
+    if (Array.isArray(fieldsToDisclose) && fieldsToDisclose.length === 0) {
+      throw new AirServiceError("CLIENT_ERROR", "fieldsToDisclose array must contain at least one field");
+    }
     await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
     const {
       payload
@@ -3138,7 +3245,7 @@ class AirService {
       partnerToken: authToken,
       programId,
       redirectUrl,
-      requestDisclosure,
+      fieldsToDisclose,
       offchain
     });
     if (payload.success === false) {
@@ -3154,7 +3261,9 @@ _AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _
   if (!this.isLoggedIn) throw new AirServiceError("NOT_LOGGED_IN", "User not logged in");
 }, _AirService_ensureCredential = async function _AirService_ensureCredential() {
   __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_assertInitialized).call(this);
-  void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this);
+  void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, {
+    skipWalletLogin: true
+  });
   try {
     if (!__classPrivateFieldGet$1(this, _AirService_credentialsInitialization, "f")) __classPrivateFieldSet(this, _AirService_credentialsInitialization, __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_initializeCredentials).call(this), "f");
     await __classPrivateFieldGet$1(this, _AirService_credentialsInitialization, "f");
@@ -3399,8 +3508,9 @@ _AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _
     isLoggedIn: true,
     id: payload.id,
     abstractAccountAddress: payload.abstractAccountAddress,
+    abstractAccountAddresses: payload.abstractAccountAddresses,
     token: payload.partnerAccessToken,
-    isMFASetup: payload.isMFASetup
+    isMFASetup: payload.activeMfaMethods.length > 0
   };
 }, _AirService_createWalletInitializedResult = function _AirService_createWalletInitializedResult(payload) {
   if ("isWalletLoggedIn" in payload && !payload.isWalletLoggedIn) {
@@ -3411,7 +3521,7 @@ _AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _
   }
   return {
     abstractAccountAddress: payload.addresses.aa,
-    isMFASetup: payload.isMFASetup
+    isMFASetup: payload.activeMfaMethods.length > 0
   };
 }, _AirService_cleanUpAuth = async function _AirService_cleanUpAuth() {
   // Logout auth session

package/dist/airkitConnector.esm.js

@@ -92,7 +92,7 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
   var e = new Error(message);
   return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
 };
-var version$2 = "1.9.0-beta.1";
+var version$2 = "1.9.0-beta.3";
 var airkitPackage = {
   version: version$2
 };
@@ -136,6 +136,10 @@ const AirAuthMessageTypes = {
   RECOVERY_RESPONSE: "air_auth_recovery_response",
   ISSUE_ON_BEHALF_NEW_USER_CONFIRMATION_REQUEST: "air_auth_issue_on_behalf_new_user_confirmation_request",
   ISSUE_ON_BEHALF_NEW_USER_CONFIRMATION_RESPONSE: "air_auth_issue_on_behalf_new_user_confirmation_response",
+  REMOVE_SIGNER_SIGNATURE_REQUEST: "air_auth_remove_signer_signature_request",
+  REMOVE_SIGNER_SIGNATURE_RESPONSE: "air_auth_remove_signer_signature_response",
+  RESET_WALLET_REQUEST: "air_auth_reset_wallet_request",
+  RESET_WALLET_RESPONSE: "air_auth_reset_wallet_response",
   EXPIRED_LOGOUT_REQUEST: "air_auth_expired_logout_request"
 };
 const AirCredentialMessageTypes = {
@@ -166,6 +170,12 @@ const AirRecoveryMessageTypes = {
   RECOVERY_IFRAME_VISIBILITY_REQUEST: "air_recovery_iframe_visibility_request",
   UPDATE_SESSION_CONFIG_REQUEST: "air_recovery_update_session_config_request",
   UPDATE_SESSION_CONFIG_RESPONSE: "air_recovery_update_session_config_response",
+  GET_AGENT_KEYS_REQUEST: "air_recovery_get_agent_keys_request",
+  GET_AGENT_KEYS_RESPONSE: "air_recovery_get_agent_keys_response",
+  REGISTER_AGENT_KEY_REQUEST: "air_recovery_register_agent_key_request",
+  REGISTER_AGENT_KEY_RESPONSE: "air_recovery_register_agent_key_response",
+  REMOVE_AGENT_KEY_REQUEST: "air_recovery_remove_agent_key_request",
+  REMOVE_AGENT_KEY_RESPONSE: "air_recovery_remove_agent_key_response",
   LOGOUT_REQUEST: "air_recovery_logout_request",
   LOGOUT_RESPONSE: "air_recovery_logout_response"
 };
@@ -1848,7 +1858,7 @@ class ProviderMessageService extends MessageServiceBase {
     });
   }
   async sendWalletProviderRequest(payload) {
-    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.RESPONSE), filter(msg => msg.payload.method === payload.method)));
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.RESPONSE), filter(msg => msg.payload.method === payload.method && msg.payload.requestId === payload.requestId)));
     await this.sendMessage({
       type: AirWalletProviderMessageTypes.REQUEST,
       payload
@@ -1860,7 +1870,7 @@ _a$4 = ProviderMessageService;
 _ProviderMessageService_instance = {
   value: void 0
 };
-var _AirWalletProvider_instances, _AirWalletProvider_providerMessageService, _AirWalletProvider_isWalletInitialized, _AirWalletProvider_getLoginResult, _AirWalletProvider_ensureWallet, _AirWalletProvider_eventListeners, _AirWalletProvider_emit;
+var _AirWalletProvider_instances, _AirWalletProvider_providerMessageService, _AirWalletProvider_isWalletInitialized, _AirWalletProvider_getLoginResult, _AirWalletProvider_ensureWallet, _AirWalletProvider_eventListeners, _AirWalletProvider_currentChainId, _AirWalletProvider_emit, _AirWalletProvider_getAddressForCurrentChain, _AirWalletProvider_parseChainId;
 class AirWalletProvider {
   constructor({
     ensureWallet,
@@ -1873,9 +1883,16 @@ class AirWalletProvider {
     _AirWalletProvider_getLoginResult.set(this, void 0);
     _AirWalletProvider_ensureWallet.set(this, void 0);
     _AirWalletProvider_eventListeners.set(this, void 0);
+    _AirWalletProvider_currentChainId.set(this, null);
     this.startEventMessageListening = async walletIframe => {
       await __classPrivateFieldGet$1(this, _AirWalletProvider_providerMessageService, "f").open(walletIframe);
       __classPrivateFieldGet$1(this, _AirWalletProvider_providerMessageService, "f").messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.EVENT)).subscribe(message => {
+        if (message.payload.event === "chainChanged") {
+          const parsedChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_parseChainId).call(this, message.payload.data);
+          if (parsedChainId !== null) {
+            __classPrivateFieldSet(this, _AirWalletProvider_currentChainId, parsedChainId);
+          }
+        }
         __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_emit).call(this, message.payload.event, ...[message.payload.data]);
       });
     };
@@ -1912,8 +1929,12 @@ class AirWalletProvider {
     if (!loginResult) {
       throw new UnauthorizedProviderError("User is not logged in");
     }
-    if ((method === "eth_accounts" || method === "eth_requestAccounts") && !__classPrivateFieldGet$1(this, _AirWalletProvider_isWalletInitialized, "f").call(this) && loginResult?.abstractAccountAddress) {
-      return [loginResult.abstractAccountAddress];
+    // Early return optimization: return cached address from login result if wallet not yet initialized
+    if ((method === "eth_accounts" || method === "eth_requestAccounts") && !__classPrivateFieldGet$1(this, _AirWalletProvider_isWalletInitialized, "f").call(this)) {
+      const address = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_getAddressForCurrentChain).call(this, loginResult);
+      if (address) {
+        return [address];
+      }
     }
     const skipWalletLoginMethods = ["eth_chainId"];
     try {
@@ -1935,6 +1956,12 @@ class AirWalletProvider {
     if (!response.success) {
       throw ensureProviderRpcError(response.payload);
     }
+    if (method === "eth_chainId") {
+      const parsedChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_parseChainId).call(this, response.payload.response);
+      if (parsedChainId !== null) {
+        __classPrivateFieldSet(this, _AirWalletProvider_currentChainId, parsedChainId);
+      }
+    }
     return response.payload.response;
   }
   on(eventName, listener) {
@@ -1958,7 +1985,7 @@ class AirWalletProvider {
     });
   }
 }
-_AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_isWalletInitialized = new WeakMap(), _AirWalletProvider_getLoginResult = new WeakMap(), _AirWalletProvider_ensureWallet = new WeakMap(), _AirWalletProvider_eventListeners = new WeakMap(), _AirWalletProvider_instances = new WeakSet(), _AirWalletProvider_emit = function _AirWalletProvider_emit(eventName, ...args) {
+_AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_isWalletInitialized = new WeakMap(), _AirWalletProvider_getLoginResult = new WeakMap(), _AirWalletProvider_ensureWallet = new WeakMap(), _AirWalletProvider_eventListeners = new WeakMap(), _AirWalletProvider_currentChainId = new WeakMap(), _AirWalletProvider_instances = new WeakSet(), _AirWalletProvider_emit = function _AirWalletProvider_emit(eventName, ...args) {
   (__classPrivateFieldGet$1(this, _AirWalletProvider_eventListeners, "f")[eventName] || []).forEach(listener => {
     try {
       return listener(...args);
@@ -1966,6 +1993,30 @@ _AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_is
       log$1.warn(error);
     }
   });
+}, _AirWalletProvider_getAddressForCurrentChain = function _AirWalletProvider_getAddressForCurrentChain(loginResult) {
+  const currentChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_currentChainId, "f");
+  // If we have the new addresses array and current chainId, look up by chain
+  if (loginResult.abstractAccountAddresses && currentChainId !== null) {
+    const chainIdStr = String(currentChainId);
+    const addressEntry = loginResult.abstractAccountAddresses.find(entry => entry.chainIds.includes(chainIdStr));
+    if (addressEntry) {
+      return addressEntry.address;
+    }
+  }
+  // Fallback to legacy single address (backward compatibility)
+  return loginResult.abstractAccountAddress;
+}, _AirWalletProvider_parseChainId = function _AirWalletProvider_parseChainId(chainId) {
+  if (typeof chainId === "number") {
+    return Number.isFinite(chainId) ? chainId : null;
+  }
+  if (typeof chainId === "bigint") {
+    return Number(chainId);
+  }
+  if (typeof chainId !== "string" || !chainId) {
+    return null;
+  }
+  const parsed = chainId.startsWith("0x") ? parseInt(chainId, 16) : parseInt(chainId, 10);
+  return Number.isNaN(parsed) ? null : parsed;
 };
 var _a$3, _AuthMessageService_instance;
 const ALLOWED_AUTH_MESSAGES = [AirAuthMessageTypes.INITIALIZATION_RESPONSE, AirAuthMessageTypes.LOGIN_RESPONSE, AirAuthMessageTypes.SETUP_WALLET_REQUEST, AirAuthMessageTypes.SETUP_RECOVERY_REQUEST, AirAuthMessageTypes.SETUP_CREDENTIAL_REQUEST, AirAuthMessageTypes.LOGOUT_RESPONSE, AirAuthMessageTypes.UPDATE_SESSION_CONFIG_RESPONSE, AirAuthMessageTypes.PARTNER_USER_INFO_RESPONSE, AirAuthMessageTypes.CROSS_PARTNER_TOKEN_RESPONSE, AirAuthMessageTypes.PARTNER_ACCESS_TOKEN_RESPONSE, AirAuthMessageTypes.IFRAME_VISIBILITY_REQUEST, AirAuthMessageTypes.START_RECOVERY_RESPONSE, AirAuthMessageTypes.EXPIRED_LOGOUT_REQUEST];
@@ -2273,6 +2324,29 @@ class RecoveryMessageService extends MessageServiceBase {
     });
     return response;
   }
+  async sendGetAgentKeysRequest() {
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.GET_AGENT_KEYS_RESPONSE)));
+    await this.sendMessage({
+      type: AirRecoveryMessageTypes.GET_AGENT_KEYS_REQUEST
+    });
+    return response;
+  }
+  async sendRegisterAgentKeyRequest(payload) {
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.REGISTER_AGENT_KEY_RESPONSE)));
+    await this.sendMessage({
+      type: AirRecoveryMessageTypes.REGISTER_AGENT_KEY_REQUEST,
+      payload
+    });
+    return response;
+  }
+  async sendRemoveAgentKeyRequest(payload) {
+    const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.REMOVE_AGENT_KEY_RESPONSE)));
+    await this.sendMessage({
+      type: AirRecoveryMessageTypes.REMOVE_AGENT_KEY_REQUEST,
+      payload
+    });
+    return response;
+  }
   async open(element) {
     await this._open({
       window: element.contentWindow,
@@ -2797,9 +2871,6 @@ class AirService {
           case AirAuthMessageTypes.SETUP_CREDENTIAL_REQUEST:
             {
               try {
-                await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, {
-                  skipWalletLogin: true
-                });
                 await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
                 await __classPrivateFieldGet$1(this, _AirService_authMessagingService, "f").sendSetupCredentialSuccessResponse();
               } catch (err) {
@@ -2870,10 +2941,6 @@ class AirService {
     });
     if (payload.success === true) {
       __classPrivateFieldSet(this, _AirService_loginResult, __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_createLoginResult).call(this, payload));
-      if (payload.needsWalletRestart === true) {
-        await Promise.all([__classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_cleanUpWallet).call(this), __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_cleanUpCredential).call(this)]);
-        void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
-      }
       __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_triggerAirAuthLoggedIn).call(this);
       return __classPrivateFieldGet$1(this, _AirService_loginResult, "f");
     }
@@ -2993,6 +3060,43 @@ class AirService {
       throw new AirServiceError(result.payload.errorName, result.payload.errorMessage);
     }
   }
+  async getAgentKeys() {
+    await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+    const {
+      payload
+    } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendGetAgentKeysRequest();
+    if (payload.success === false) {
+      throw new AirServiceError(payload.errorName, payload.errorMessage);
+    }
+    return payload.agentKeys;
+  }
+  async registerAgentKey(publicKey) {
+    await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+    const {
+      payload
+    } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendRegisterAgentKeyRequest({
+      publicKey
+    });
+    if (payload.success === false) {
+      throw new AirServiceError(payload.errorName, payload.errorMessage);
+    }
+    return {
+      id: payload.id,
+      publicKey: payload.publicKey,
+      createdAt: payload.createdAt
+    };
+  }
+  async removeAgentKey(id) {
+    await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+    const {
+      payload
+    } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendRemoveAgentKeyRequest({
+      id
+    });
+    if (payload.success === false) {
+      throw new AirServiceError(payload.errorName, payload.errorMessage);
+    }
+  }
   async getUserInfo() {
     __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_assertLoggedIn).call(this);
     const info = await __classPrivateFieldGet$1(this, _AirService_authMessagingService, "f").sendPartnerUserInfoRequest();
@@ -3007,7 +3111,7 @@ class AirService {
         id: info.payload.user.id,
         abstractAccountAddress: info.payload.user.abstractAccountAddress,
         email: info.payload.user.email,
-        isMFASetup: info.payload.user.isMFASetup
+        isMFASetup: info.payload.user.activeMfaMethods.length > 0
       }
     };
   }
@@ -3126,9 +3230,12 @@ class AirService {
     authToken,
     programId,
     redirectUrl,
-    requestDisclosure,
+    fieldsToDisclose,
     offchain
   }) {
+    if (Array.isArray(fieldsToDisclose) && fieldsToDisclose.length === 0) {
+      throw new AirServiceError("CLIENT_ERROR", "fieldsToDisclose array must contain at least one field");
+    }
     await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
     const {
       payload
@@ -3136,7 +3243,7 @@ class AirService {
       partnerToken: authToken,
       programId,
       redirectUrl,
-      requestDisclosure,
+      fieldsToDisclose,
       offchain
     });
     if (payload.success === false) {
@@ -3152,7 +3259,9 @@ _AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _
   if (!this.isLoggedIn) throw new AirServiceError("NOT_LOGGED_IN", "User not logged in");
 }, _AirService_ensureCredential = async function _AirService_ensureCredential() {
   __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_assertInitialized).call(this);
-  void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this);
+  void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, {
+    skipWalletLogin: true
+  });
   try {
     if (!__classPrivateFieldGet$1(this, _AirService_credentialsInitialization, "f")) __classPrivateFieldSet(this, _AirService_credentialsInitialization, __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_initializeCredentials).call(this), "f");
     await __classPrivateFieldGet$1(this, _AirService_credentialsInitialization, "f");
@@ -3397,8 +3506,9 @@ _AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _
     isLoggedIn: true,
     id: payload.id,
     abstractAccountAddress: payload.abstractAccountAddress,
+    abstractAccountAddresses: payload.abstractAccountAddresses,
     token: payload.partnerAccessToken,
-    isMFASetup: payload.isMFASetup
+    isMFASetup: payload.activeMfaMethods.length > 0
   };
 }, _AirService_createWalletInitializedResult = function _AirService_createWalletInitializedResult(payload) {
   if ("isWalletLoggedIn" in payload && !payload.isWalletLoggedIn) {
@@ -3409,7 +3519,7 @@ _AirService_loginResult = new WeakMap(), _AirService_buildEnv = new WeakMap(), _
   }
   return {
     abstractAccountAddress: payload.addresses.aa,
-    isMFASetup: payload.isMFASetup
+    isMFASetup: payload.activeMfaMethods.length > 0
   };
 }, _AirService_cleanUpAuth = async function _AirService_cleanUpAuth() {
   // Logout auth session

package/dist/airkitConnector.umd.js

@@ -98,7 +98,7 @@
     var e = new Error(message);
     return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
   };
-  var version$2 = "1.9.0-beta.1";
+  var version$2 = "1.9.0-beta.3";
   var airkitPackage = {
     version: version$2
   };
@@ -142,6 +142,10 @@
     RECOVERY_RESPONSE: "air_auth_recovery_response",
     ISSUE_ON_BEHALF_NEW_USER_CONFIRMATION_REQUEST: "air_auth_issue_on_behalf_new_user_confirmation_request",
     ISSUE_ON_BEHALF_NEW_USER_CONFIRMATION_RESPONSE: "air_auth_issue_on_behalf_new_user_confirmation_response",
+    REMOVE_SIGNER_SIGNATURE_REQUEST: "air_auth_remove_signer_signature_request",
+    REMOVE_SIGNER_SIGNATURE_RESPONSE: "air_auth_remove_signer_signature_response",
+    RESET_WALLET_REQUEST: "air_auth_reset_wallet_request",
+    RESET_WALLET_RESPONSE: "air_auth_reset_wallet_response",
     EXPIRED_LOGOUT_REQUEST: "air_auth_expired_logout_request"
   };
   const AirCredentialMessageTypes = {
@@ -172,6 +176,12 @@
     RECOVERY_IFRAME_VISIBILITY_REQUEST: "air_recovery_iframe_visibility_request",
     UPDATE_SESSION_CONFIG_REQUEST: "air_recovery_update_session_config_request",
     UPDATE_SESSION_CONFIG_RESPONSE: "air_recovery_update_session_config_response",
+    GET_AGENT_KEYS_REQUEST: "air_recovery_get_agent_keys_request",
+    GET_AGENT_KEYS_RESPONSE: "air_recovery_get_agent_keys_response",
+    REGISTER_AGENT_KEY_REQUEST: "air_recovery_register_agent_key_request",
+    REGISTER_AGENT_KEY_RESPONSE: "air_recovery_register_agent_key_response",
+    REMOVE_AGENT_KEY_REQUEST: "air_recovery_remove_agent_key_request",
+    REMOVE_AGENT_KEY_RESPONSE: "air_recovery_remove_agent_key_response",
     LOGOUT_REQUEST: "air_recovery_logout_request",
     LOGOUT_RESPONSE: "air_recovery_logout_response"
   };
@@ -1854,7 +1864,7 @@
       });
     }
     async sendWalletProviderRequest(payload) {
-      const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.RESPONSE), filter(msg => msg.payload.method === payload.method)));
+      const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.RESPONSE), filter(msg => msg.payload.method === payload.method && msg.payload.requestId === payload.requestId)));
       await this.sendMessage({
         type: AirWalletProviderMessageTypes.REQUEST,
         payload
@@ -1866,7 +1876,7 @@
   _ProviderMessageService_instance = {
     value: void 0
   };
-  var _AirWalletProvider_instances, _AirWalletProvider_providerMessageService, _AirWalletProvider_isWalletInitialized, _AirWalletProvider_getLoginResult, _AirWalletProvider_ensureWallet, _AirWalletProvider_eventListeners, _AirWalletProvider_emit;
+  var _AirWalletProvider_instances, _AirWalletProvider_providerMessageService, _AirWalletProvider_isWalletInitialized, _AirWalletProvider_getLoginResult, _AirWalletProvider_ensureWallet, _AirWalletProvider_eventListeners, _AirWalletProvider_currentChainId, _AirWalletProvider_emit, _AirWalletProvider_getAddressForCurrentChain, _AirWalletProvider_parseChainId;
   class AirWalletProvider {
     constructor({
       ensureWallet,
@@ -1879,9 +1889,16 @@
       _AirWalletProvider_getLoginResult.set(this, void 0);
       _AirWalletProvider_ensureWallet.set(this, void 0);
       _AirWalletProvider_eventListeners.set(this, void 0);
+      _AirWalletProvider_currentChainId.set(this, null);
       this.startEventMessageListening = async walletIframe => {
         await __classPrivateFieldGet$1(this, _AirWalletProvider_providerMessageService, "f").open(walletIframe);
         __classPrivateFieldGet$1(this, _AirWalletProvider_providerMessageService, "f").messages$.pipe(filter(msg => msg.type === AirWalletProviderMessageTypes.EVENT)).subscribe(message => {
+          if (message.payload.event === "chainChanged") {
+            const parsedChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_parseChainId).call(this, message.payload.data);
+            if (parsedChainId !== null) {
+              __classPrivateFieldSet(this, _AirWalletProvider_currentChainId, parsedChainId);
+            }
+          }
           __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_emit).call(this, message.payload.event, ...[message.payload.data]);
         });
       };
@@ -1918,8 +1935,12 @@
       if (!loginResult) {
         throw new UnauthorizedProviderError("User is not logged in");
       }
-      if ((method === "eth_accounts" || method === "eth_requestAccounts") && !__classPrivateFieldGet$1(this, _AirWalletProvider_isWalletInitialized, "f").call(this) && loginResult?.abstractAccountAddress) {
-        return [loginResult.abstractAccountAddress];
+      // Early return optimization: return cached address from login result if wallet not yet initialized
+      if ((method === "eth_accounts" || method === "eth_requestAccounts") && !__classPrivateFieldGet$1(this, _AirWalletProvider_isWalletInitialized, "f").call(this)) {
+        const address = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_getAddressForCurrentChain).call(this, loginResult);
+        if (address) {
+          return [address];
+        }
       }
       const skipWalletLoginMethods = ["eth_chainId"];
       try {
@@ -1941,6 +1962,12 @@
       if (!response.success) {
         throw ensureProviderRpcError(response.payload);
       }
+      if (method === "eth_chainId") {
+        const parsedChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_instances, "m", _AirWalletProvider_parseChainId).call(this, response.payload.response);
+        if (parsedChainId !== null) {
+          __classPrivateFieldSet(this, _AirWalletProvider_currentChainId, parsedChainId);
+        }
+      }
       return response.payload.response;
     }
     on(eventName, listener) {
@@ -1964,7 +1991,7 @@
       });
     }
   }
-  _AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_isWalletInitialized = new WeakMap(), _AirWalletProvider_getLoginResult = new WeakMap(), _AirWalletProvider_ensureWallet = new WeakMap(), _AirWalletProvider_eventListeners = new WeakMap(), _AirWalletProvider_instances = new WeakSet(), _AirWalletProvider_emit = function _AirWalletProvider_emit(eventName, ...args) {
+  _AirWalletProvider_providerMessageService = new WeakMap(), _AirWalletProvider_isWalletInitialized = new WeakMap(), _AirWalletProvider_getLoginResult = new WeakMap(), _AirWalletProvider_ensureWallet = new WeakMap(), _AirWalletProvider_eventListeners = new WeakMap(), _AirWalletProvider_currentChainId = new WeakMap(), _AirWalletProvider_instances = new WeakSet(), _AirWalletProvider_emit = function _AirWalletProvider_emit(eventName, ...args) {
     (__classPrivateFieldGet$1(this, _AirWalletProvider_eventListeners, "f")[eventName] || []).forEach(listener => {
       try {
         return listener(...args);
@@ -1972,6 +1999,30 @@
         log$1.warn(error);
       }
     });
+  }, _AirWalletProvider_getAddressForCurrentChain = function _AirWalletProvider_getAddressForCurrentChain(loginResult) {
+    const currentChainId = __classPrivateFieldGet$1(this, _AirWalletProvider_currentChainId, "f");
+    // If we have the new addresses array and current chainId, look up by chain
+    if (loginResult.abstractAccountAddresses && currentChainId !== null) {
+      const chainIdStr = String(currentChainId);
+      const addressEntry = loginResult.abstractAccountAddresses.find(entry => entry.chainIds.includes(chainIdStr));
+      if (addressEntry) {
+        return addressEntry.address;
+      }
+    }
+    // Fallback to legacy single address (backward compatibility)
+    return loginResult.abstractAccountAddress;
+  }, _AirWalletProvider_parseChainId = function _AirWalletProvider_parseChainId(chainId) {
+    if (typeof chainId === "number") {
+      return Number.isFinite(chainId) ? chainId : null;
+    }
+    if (typeof chainId === "bigint") {
+      return Number(chainId);
+    }
+    if (typeof chainId !== "string" || !chainId) {
+      return null;
+    }
+    const parsed = chainId.startsWith("0x") ? parseInt(chainId, 16) : parseInt(chainId, 10);
+    return Number.isNaN(parsed) ? null : parsed;
   };
   var _a$3, _AuthMessageService_instance;
   const ALLOWED_AUTH_MESSAGES = [AirAuthMessageTypes.INITIALIZATION_RESPONSE, AirAuthMessageTypes.LOGIN_RESPONSE, AirAuthMessageTypes.SETUP_WALLET_REQUEST, AirAuthMessageTypes.SETUP_RECOVERY_REQUEST, AirAuthMessageTypes.SETUP_CREDENTIAL_REQUEST, AirAuthMessageTypes.LOGOUT_RESPONSE, AirAuthMessageTypes.UPDATE_SESSION_CONFIG_RESPONSE, AirAuthMessageTypes.PARTNER_USER_INFO_RESPONSE, AirAuthMessageTypes.CROSS_PARTNER_TOKEN_RESPONSE, AirAuthMessageTypes.PARTNER_ACCESS_TOKEN_RESPONSE, AirAuthMessageTypes.IFRAME_VISIBILITY_REQUEST, AirAuthMessageTypes.START_RECOVERY_RESPONSE, AirAuthMessageTypes.EXPIRED_LOGOUT_REQUEST];
@@ -2279,6 +2330,29 @@
       });
       return response;
     }
+    async sendGetAgentKeysRequest() {
+      const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.GET_AGENT_KEYS_RESPONSE)));
+      await this.sendMessage({
+        type: AirRecoveryMessageTypes.GET_AGENT_KEYS_REQUEST
+      });
+      return response;
+    }
+    async sendRegisterAgentKeyRequest(payload) {
+      const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.REGISTER_AGENT_KEY_RESPONSE)));
+      await this.sendMessage({
+        type: AirRecoveryMessageTypes.REGISTER_AGENT_KEY_REQUEST,
+        payload
+      });
+      return response;
+    }
+    async sendRemoveAgentKeyRequest(payload) {
+      const response = firstValueFrom(this.messages$.pipe(filter(msg => msg.type === AirRecoveryMessageTypes.REMOVE_AGENT_KEY_RESPONSE)));
+      await this.sendMessage({
+        type: AirRecoveryMessageTypes.REMOVE_AGENT_KEY_REQUEST,
+        payload
+      });
+      return response;
+    }
     async open(element) {
       await this._open({
         window: element.contentWindow,
@@ -2803,9 +2877,6 @@
             case AirAuthMessageTypes.SETUP_CREDENTIAL_REQUEST:
               {
                 try {
-                  await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, {
-                    skipWalletLogin: true
-                  });
                   await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
                   await __classPrivateFieldGet$1(this, _AirService_authMessagingService, "f").sendSetupCredentialSuccessResponse();
                 } catch (err) {
@@ -2876,10 +2947,6 @@
       });
       if (payload.success === true) {
         __classPrivateFieldSet(this, _AirService_loginResult, __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_createLoginResult).call(this, payload));
-        if (payload.needsWalletRestart === true) {
-          await Promise.all([__classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_cleanUpWallet).call(this), __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_cleanUpCredential).call(this)]);
-          void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
-        }
         __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_triggerAirAuthLoggedIn).call(this);
         return __classPrivateFieldGet$1(this, _AirService_loginResult, "f");
       }
@@ -2999,6 +3066,43 @@
         throw new AirServiceError(result.payload.errorName, result.payload.errorMessage);
       }
     }
+    async getAgentKeys() {
+      await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+      const {
+        payload
+      } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendGetAgentKeysRequest();
+      if (payload.success === false) {
+        throw new AirServiceError(payload.errorName, payload.errorMessage);
+      }
+      return payload.agentKeys;
+    }
+    async registerAgentKey(publicKey) {
+      await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+      const {
+        payload
+      } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendRegisterAgentKeyRequest({
+        publicKey
+      });
+      if (payload.success === false) {
+        throw new AirServiceError(payload.errorName, payload.errorMessage);
+      }
+      return {
+        id: payload.id,
+        publicKey: payload.publicKey,
+        createdAt: payload.createdAt
+      };
+    }
+    async removeAgentKey(id) {
+      await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureRecovery).call(this);
+      const {
+        payload
+      } = await __classPrivateFieldGet$1(this, _AirService_recoveryMessagingService, "f").sendRemoveAgentKeyRequest({
+        id
+      });
+      if (payload.success === false) {
+        throw new AirServiceError(payload.errorName, payload.errorMessage);
+      }
+    }
     async getUserInfo() {
       __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_assertLoggedIn).call(this);
       const info = await __classPrivateFieldGet$1(this, _AirService_authMessagingService, "f").sendPartnerUserInfoRequest();
@@ -3013,7 +3117,7 @@
           id: info.payload.user.id,
           abstractAccountAddress: info.payload.user.abstractAccountAddress,
           email: info.payload.user.email,
-          isMFASetup: info.payload.user.isMFASetup
+          isMFASetup: info.payload.user.activeMfaMethods.length > 0
         }
       };
     }
@@ -3132,9 +3236,12 @@
       authToken,
       programId,
       redirectUrl,
-      requestDisclosure,
+      fieldsToDisclose,
       offchain
     }) {
+      if (Array.isArray(fieldsToDisclose) && fieldsToDisclose.length === 0) {
+        throw new AirServiceError("CLIENT_ERROR", "fieldsToDisclose array must contain at least one field");
+      }
       await __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureCredential).call(this);
       const {
         payload
@@ -3142,7 +3249,7 @@
         partnerToken: authToken,
         programId,
         redirectUrl,
-        requestDisclosure,
+        fieldsToDisclose,
         offchain
       });
       if (payload.success === false) {
@@ -3158,7 +3265,9 @@
     if (!this.isLoggedIn) throw new AirServiceError("NOT_LOGGED_IN", "User not logged in");
   }, _AirService_ensureCredential = async function _AirService_ensureCredential() {
     __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_assertInitialized).call(this);
-    void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this);
+    void __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_ensureWallet).call(this, {
+      skipWalletLogin: true
+    });
     try {
       if (!__classPrivateFieldGet$1(this, _AirService_credentialsInitialization, "f")) __classPrivateFieldSet(this, _AirService_credentialsInitialization, __classPrivateFieldGet$1(this, _AirService_instances, "m", _AirService_initializeCredentials).call(this), "f");
       await __classPrivateFieldGet$1(this, _AirService_credentialsInitialization, "f");
@@ -3403,8 +3512,9 @@
       isLoggedIn: true,
       id: payload.id,
       abstractAccountAddress: payload.abstractAccountAddress,
+      abstractAccountAddresses: payload.abstractAccountAddresses,
       token: payload.partnerAccessToken,
-      isMFASetup: payload.isMFASetup
+      isMFASetup: payload.activeMfaMethods.length > 0
     };
   }, _AirService_createWalletInitializedResult = function _AirService_createWalletInitializedResult(payload) {
     if ("isWalletLoggedIn" in payload && !payload.isWalletLoggedIn) {
@@ -3415,7 +3525,7 @@
     }
     return {
       abstractAccountAddress: payload.addresses.aa,
-      isMFASetup: payload.isMFASetup
+      isMFASetup: payload.activeMfaMethods.length > 0
     };
   }, _AirService_cleanUpAuth = async function _AirService_cleanUpAuth() {
     // Logout auth session

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mocanetwork/airkit-connector",
-  "version": "1.9.0-beta.1",
+  "version": "1.9.0-beta.3",
   "description": "AirKit wagmi connector to connect with the AirService",
   "main": "dist/airkitConnector.cjs.js",
   "unpkg": "dist/airkitConnector.umd.min.js",
@@ -24,7 +24,7 @@
     "check-circular": "madge -c --ts-config=tsconfig.json --extensions ts ./src"
   },
   "dependencies": {
-    "@mocanetwork/airkit": "^1.9.0-beta.1",
+    "@mocanetwork/airkit": "^1.9.0-beta.3",
     "@wagmi/core": "^2.x",
     "loglevel": "^1.8.1",
     "viem": "^2.x"
@@ -39,7 +39,7 @@
     "rollup": "^4.32.0",
     "ts-node": "^10.9.2",
     "tslib": "^2.6.2",
-    "typescript": "^5.3.3"
+    "typescript": "~5.8.2"
   },
   "peerDependencies": {
     "@wagmi/core": "^2.x",