fix: query keys for account public hooks

AbstractSDK · Feb 22, 2024 · 6275682 · 6275682
1 parent 5f301a1
commit 6275682
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/.changeset/polite-trainers-shave.md b/.changeset/polite-trainers-shave.md
@@ -0,0 +1,5 @@
+---
+"@abstract-money/react": patch
+---
+
+Fix the query key for account public hooks.
diff --git a/packages/react/src/hooks/account/public/use-account-base-addresses-from-api.ts b/packages/react/src/hooks/account/public/use-account-base-addresses-from-api.ts
@@ -17,6 +17,7 @@ type QueryData = QueryFnData
 type QueryKey = readonly [
   'accountBaseAddresses',
   AccountPublicClient | undefined,
+  AccountId | undefined,
 ]
 
 type QueryOptions = Omit<
@@ -46,8 +47,8 @@ export function useAccountBaseAddressesFromApi(
     chainName: accountId?.chainName,
   })
   const queryKey = React.useMemo(
-    () => ['accountBaseAddresses', accountPublicClient] as const,
-    [accountPublicClient],
+    () => ['accountBaseAddresses', accountPublicClient, accountId] as const,
+    [accountPublicClient, accountId],
   )
 
   const enabled = React.useMemo(

diff --git a/packages/react/src/hooks/account/public/use-module-instantiate2-address-from-api.ts b/packages/react/src/hooks/account/public/use-module-instantiate2-address-from-api.ts
@@ -20,6 +20,7 @@ type QueryData = QueryFnData
 type QueryKey = readonly [
   'moduleInstantiate2AddressFromApi',
   AccountPublicClient | undefined,
+  AccountId | undefined,
   UseModuleInstantiate2AddressFromApiArgs['args'],
 ]
 
@@ -68,8 +69,13 @@ export function useModuleInstantiate2AddressFromApi(
   })
   const queryKey = React.useMemo(
     () =>
-      ['moduleInstantiate2AddressFromApi', accountPublicClient, args] as const,
-    [accountPublicClient],
+      [
+        'moduleInstantiate2AddressFromApi',
+        accountPublicClient,
+        accountId,
+        args,
+      ] as const,
+    [accountPublicClient, accountId, args],
   )
 
   const enabled = React.useMemo(