import { ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback } from 'ai';
import { ImagineUIDataParts, ImagineUIMessage } from '@imagine/ai-service/shared-types';
import { Subject, BehaviorSubject } from 'rxjs';
import { UseChatHelpers } from '@ai-sdk/react';
type ChatOnFinishCallbackArg = Parameters<ChatOnFinishCallback<ImagineUIMessage>>[0];
type ChatOnDataCallbackArg = Parameters<ChatOnDataCallback<ImagineUIMessage>>[0];
type ChatOnErrorCallbackArg = Parameters<ChatOnErrorCallback>[0];
export type AISDKChat = UseChatHelpers<ImagineUIMessage>;
type ChatState = {
    onData$: Subject<ChatOnDataCallbackArg>;
    onFinish$: Subject<ChatOnFinishCallbackArg>;
    onError$: Subject<ChatOnErrorCallbackArg>;
    status: AISDKChat["status"];
    isGenerating: boolean;
    todos: ImagineUIDataParts["plan"]["items"];
    scrollToBottom$: Subject<void>;
    scrollLocked$: BehaviorSubject<boolean>;
    history: Array<ImagineUIMessage> | null;
    submittedUserMessage: ImagineUIMessage | null;
    currentAssistantMessage: ImagineUIMessage | null;
    error: Error | undefined;
};
type ChatActions = {
    reset: () => void;
    setStatus: (status: AISDKChat["status"]) => void;
    setTodos: (todos: ImagineUIDataParts["plan"]["items"]) => void;
    resetTodos: () => void;
    setHistory: (history: Array<ImagineUIMessage> | null) => void;
    syncHistoryAfterGeneration: () => void;
    setSubmittedUserMessage: (submittedUserMessage: ImagineUIMessage | null) => void;
    setCurrentAssistantMessage: (currentAssistantMessage: ImagineUIMessage | null) => void;
    setError: (error: Error | undefined) => void;
    lockScroll: () => void;
    unlockScroll: () => void;
};
export type ChatSlice = ChatState & ChatActions;
export declare const chatSlice: import('@dhmk/zustand-lens').LensOpaqueType<ChatSlice, unknown>;
export {};
//# sourceMappingURL=chat.slice.d.ts.map