UNPKG

@visactor/vmind

Version:

<div align="center"> <a href="https://github.com/VisActor#gh-light-mode-only" target="_blank"> <img alt="VisActor Logo" width="200" src="https://github.com/VisActor/.github/blob/main/profile/logo_500_200_light.svg"/> </a> <a href="https://githu

45 lines (44 loc) 2.35 kB
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; }