fix handleQueryState

ocadotechnology · Aug 15, 2024 · f04cf95 · f04cf95
1 parent 1f925cb
commit f04cf95
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/utils/api.tsx b/src/utils/api.tsx
@@ -2,6 +2,7 @@ import { CircularProgress } from "@mui/material"
 import type {
   FetchBaseQueryError,
   TypedUseQueryHookResult,
+  TypedUseQueryStateResult,
 } from "@reduxjs/toolkit/query/react"
 import { type ReactNode } from "react"
 
@@ -260,8 +261,10 @@ export type HandleQueryStateOptions = Partial<{
 }>
 
 export function handleQueryState<QueryArg, ResultType>(
-  result: TypedUseQueryHookResult<ResultType, QueryArg, any>,
-  children: (data: ResultType) => ReactNode,
+  result:
+    | TypedUseQueryHookResult<ResultType, QueryArg, any>
+    | TypedUseQueryStateResult<ResultType, QueryArg, any>,
+  children: (data: NonNullable<ResultType>) => ReactNode,
   options?: HandleQueryStateOptions,
 ): ReactNode {
   const { data, isLoading, isSuccess, error } = result