import type { AtomName, BaseContext, DataExtractionCtx, DataCleanCtx, DataQueryCtx, ChartGeneratorCtx, ChartCommandCtx, DataInsightCtx, MultipleDataCleanCtx, MultipleChartCommandsCtx, ChartQAExtractionCtx, VChartSpecCtx, SpecInsightCtx, BaseOptions, DataExtractionOptions, DataCleanOptions, SpecInsightOptions, ChartCommandOptions, CustomPromptOptions } from './atom';
import type { ChartGeneratorOptions } from '../atom/chartGenerator/type';
import type { DataInsightOptions } from '../atom/dataInsight/type';
import type { ImageReaderCtx } from '../atom/imageReader/interface';
export interface Tasks {
    shouldRun: boolean;
    query: string;
}
export type TaskMapping = Partial<Record<AtomName, Tasks>>;
export type AtomTypeMap = {
    [AtomName.BASE]: BaseContext;
    [AtomName.DATA_EXTRACT]: DataExtractionCtx;
    [AtomName.DATA_CLEAN]: DataCleanCtx;
    [AtomName.MULTIPLE_DATA_CLEAN]: MultipleDataCleanCtx;
    [AtomName.DATA_QUERY]: DataQueryCtx;
    [AtomName.DATA_INSIGHT]: DataInsightCtx;
    [AtomName.SPEC_INSIGHT]: SpecInsightCtx;
    [AtomName.CHART_COMMAND]: ChartCommandCtx;
    [AtomName.MULTIPLE_CHART_COMMAND]: MultipleChartCommandsCtx;
    [AtomName.CHART_GENERATE]: ChartGeneratorCtx;
    [AtomName.CHART_QA_EXTRACTION]: ChartQAExtractionCtx;
    [AtomName.CUSTOM_PROMPT]: any;
    [AtomName.VCHART_SPEC]: VChartSpecCtx;
    [AtomName.IMAGE_READER]: ImageReaderCtx;
};
export type MapAtomTypes<T extends (keyof AtomTypeMap)[]> = {
    [K in keyof T]: T[K] extends keyof AtomTypeMap ? AtomTypeMap[T[K]] : never;
};
export type CombineAll<T extends any[]> = T extends [infer First, ...infer Rest] ? First & CombineAll<Rest> : Record<string, unknown>;
export interface ScheduleOptions {
    [AtomName.BASE]?: BaseOptions;
    [AtomName.DATA_EXTRACT]?: DataExtractionOptions;
    [AtomName.DATA_CLEAN]?: DataCleanOptions;
    [AtomName.MULTIPLE_DATA_CLEAN]?: DataCleanOptions;
    [AtomName.DATA_QUERY]?: BaseOptions;
    [AtomName.DATA_INSIGHT]?: DataInsightOptions;
    [AtomName.SPEC_INSIGHT]?: SpecInsightOptions;
    [AtomName.CHART_GENERATE]?: ChartGeneratorOptions;
    [AtomName.CHART_COMMAND]?: ChartCommandOptions;
    [AtomName.MULTIPLE_CHART_COMMAND]?: ChartCommandOptions;
    [AtomName.CHART_QA_EXTRACTION]?: BaseOptions;
    [AtomName.CUSTOM_PROMPT]?: CustomPromptOptions;
    [AtomName.VCHART_SPEC]?: BaseOptions;
}