diff --git a/packages/javascript/bh-shared-ui/src/hooks/useAvailableEnvironments/useAvailableEnvironments.tsx b/packages/javascript/bh-shared-ui/src/hooks/useAvailableEnvironments/useAvailableEnvironments.tsx index 03224166c..38c266421 100644 --- a/packages/javascript/bh-shared-ui/src/hooks/useAvailableEnvironments/useAvailableEnvironments.tsx +++ b/packages/javascript/bh-shared-ui/src/hooks/useAvailableEnvironments/useAvailableEnvironments.tsx @@ -25,18 +25,13 @@ export const availableEnvironmentKeys = { customKey?.length ? [...availableEnvironmentKeys.all, ...customKey] : availableEnvironmentKeys.all, }; -type QueryOptions = Omit< - UseQueryOptions, - 'queryKey' | 'queryFn' -> & { - appendQueryKey?: string[]; -}; +type QueryOptions = Omit, 'queryFn'>; export function useAvailableEnvironments(options?: QueryOptions) { - const { appendQueryKey, ...rest } = options ?? {}; + const { queryKey, ...rest } = options ?? {}; return useQuery({ - queryKey: availableEnvironmentKeys.getKey(appendQueryKey), + queryKey: availableEnvironmentKeys.getKey(queryKey), queryFn: ({ signal }) => apiClient.getAvailableEnvironments({ signal }).then((response) => response.data.data), ...rest, }); diff --git a/packages/javascript/bh-shared-ui/src/hooks/useFeatureFlags.tsx b/packages/javascript/bh-shared-ui/src/hooks/useFeatureFlags.tsx index e3585a766..950a5d464 100644 --- a/packages/javascript/bh-shared-ui/src/hooks/useFeatureFlags.tsx +++ b/packages/javascript/bh-shared-ui/src/hooks/useFeatureFlags.tsx @@ -43,13 +43,14 @@ export const toggleFeatureFlag = (flagId: string | number, options?: RequestOpti type QueryOptions = Omit, 'queryFn'>; -export function useFeatureFlags(queryOptions?: QueryOptions) { - const queryKey = featureFlagKeys.getKey(queryOptions?.queryKey); - return useQuery(queryKey, ({ signal }) => getFeatureFlags({ signal }), queryOptions); +export function useFeatureFlags(options?: QueryOptions) { + const { queryKey, ...rest } = options ?? {}; + + return useQuery(featureFlagKeys.getKey(options?.queryKey), ({ signal }) => getFeatureFlags({ signal }), rest); } -export function useFeatureFlag(flagKey: string, queryOptions?: Omit, 'select'>) { - return useFeatureFlags({ select: (data) => data.find((flag) => flag.key === flagKey), ...queryOptions }); +export function useFeatureFlag(flagKey: string, options?: Omit, 'select'>) { + return useFeatureFlags({ select: (data) => data.find((flag) => flag.key === flagKey), ...options }); } export function useToggleFeatureFlag() {