diff --git a/packages/extension-query/src/QueryClient.ts b/packages/extension-query/src/QueryClient.ts index 5f0a2c0..08e01ca 100644 --- a/packages/extension-query/src/QueryClient.ts +++ b/packages/extension-query/src/QueryClient.ts @@ -29,7 +29,7 @@ export interface ServiceResult extends Context.Ta QueryClient > { readonly Live: Layer.Layer< - Self, + Self | EH extends QueryErrorHandler.DefaultQueryErrorHandler ? EH : never, never, EH extends QueryErrorHandler.DefaultQueryErrorHandler ? never : EH > @@ -53,7 +53,12 @@ export const Service = () => ( EH extends QueryErrorHandler.DefaultQueryErrorHandler ? never : EH > ) - )) + )).pipe( + Layer.provideMerge((props?.ErrorHandler + ? Layer.empty + : QueryErrorHandler.DefaultQueryErrorHandler.Live + ) as Layer.Layer) + ) return TagClass }