diff --git a/src/api/nitro-query/createNitroQuery.ts b/src/api/nitro-query/createNitroQuery.ts index 44a570f..bb21b29 100644 --- a/src/api/nitro-query/createNitroQuery.ts +++ b/src/api/nitro-query/createNitroQuery.ts @@ -1,6 +1,6 @@ -import { GetCommunication, IMessageEvent, MessageEvent } from '@nitrots/nitro-renderer'; +import { GetCommunication, IMessageComposer, IMessageEvent, MessageEvent } from '@nitrots/nitro-renderer'; import { QueryKey, useQuery, UseQueryOptions, UseQueryResult } from '@tanstack/react-query'; -import { SendMessageComposer } from '../SendMessageComposer'; +import { SendMessageComposer } from '../nitro/SendMessageComposer'; export interface NitroQueryConfig { @@ -14,7 +14,7 @@ export interface NitroQueryConfig * `null` skips sending (useful when the server pushes the event * unprompted — you only want subscription, not a request). */ - request: (() => unknown) | null; + request: (() => IMessageComposer) | null; /** * The parser class to listen for as the response. */ @@ -64,7 +64,7 @@ export const useNitroQuery = ( const options: UseQueryOptions = { queryKey: key, - queryFn: () => awaitNitroResponse({ key, request, parser, select, accept, timeoutMs }), + queryFn: () => awaitNitroResponse({ request, parser, select, accept, timeoutMs }), enabled, staleTime, refetchOnMount @@ -105,7 +105,7 @@ export const awaitNitroResponse = ( try { - resolve(select ? select(event) : (event)); + resolve(select ? select(event) : (event as unknown as TData)); } catch(err) {