support select

MystenLabs · williamrobertson13 · Oct 24, 2023 · Oct 19, 2023 · Oct 19, 2023 · Oct 19, 2023
commit df6d37f5fc1fc70dca8ab26af48f4713c50d72b4
diff --git a/sdk/dapp-kit/src/hooks/useSuiClientInfiniteQuery.ts b/sdk/dapp-kit/src/hooks/useSuiClientInfiniteQuery.ts
@@ -37,12 +37,15 @@ export type SuiRpcPaginatedMethods = {
 		: never;
 };
 
-export type UseSuiClientInfiniteQueryOptions<T extends keyof SuiRpcPaginatedMethods> = PartialBy<
+export type UseSuiClientInfiniteQueryOptions<
+	T extends keyof SuiRpcPaginatedMethods,
+	TData,
+> = PartialBy<
 	Omit<
 		UseInfiniteQueryOptions<
 			SuiRpcPaginatedMethods[T]['result'],
 			Error,
-			SuiRpcPaginatedMethods[T]['result'],
+			TData,
 			SuiRpcPaginatedMethods[T]['result'],
 			unknown[]
 		>,
@@ -51,10 +54,17 @@ export type UseSuiClientInfiniteQueryOptions<T extends keyof SuiRpcPaginatedMeth
 	'queryKey'
 >;
 
-export function useSuiClientInfiniteQuery<T extends keyof SuiRpcPaginatedMethods>(
+export function useSuiClientInfiniteQuery<
+	T extends keyof SuiRpcPaginatedMethods,
+	TData = SuiRpcPaginatedMethods[T]['result'],
+>(
 	method: T,
 	params: SuiRpcPaginatedMethods[T]['params'],
-	{ queryKey = [], enabled = !!params, ...options }: UseSuiClientInfiniteQueryOptions<T> = {},
+	{
+		queryKey = [],
+		enabled = !!params,
+		...options
+	}: UseSuiClientInfiniteQueryOptions<T, TData> = {},
 ) {
 	const suiContext = useSuiClientContext();