import { AIModel, CompletionStream, Driver, EmbeddingsResult, ExecutionOptions, ExecutionResponse, PromptSegment, TrainingJob } from "@llumiverse/core";
export * from "./TestErrorCompletionStream.js";
export * from "./TestValidationErrorCompletionStream.js";
export declare enum TestDriverModels {
    executionError = "execution-error",
    validationError = "validation-error"
}
export declare class TestDriver implements Driver<PromptSegment[]> {
    provider: string;
    createTrainingPrompt(): Promise<string>;
    startTraining(): Promise<TrainingJob>;
    cancelTraining(): Promise<TrainingJob>;
    getTrainingJob(_jobId: string): Promise<TrainingJob>;
    createPrompt(segments: PromptSegment[], _opts: ExecutionOptions): Promise<PromptSegment[]>;
    execute(segments: PromptSegment[], options: ExecutionOptions): Promise<ExecutionResponse<PromptSegment[]>>;
    stream(segments: PromptSegment[], options: ExecutionOptions): Promise<CompletionStream<PromptSegment[]>>;
    listTrainableModels(): Promise<AIModel<string>[]>;
    listModels(): Promise<AIModel<string>[]>;
    validateConnection(): Promise<boolean>;
    generateEmbeddings(): Promise<EmbeddingsResult>;
    executeError(segments: PromptSegment[], _options: ExecutionOptions): Promise<ExecutionResponse<PromptSegment[]>>;
    executeValidationError(segments: PromptSegment[], _options: ExecutionOptions): Promise<ExecutionResponse<PromptSegment[]>>;
}
//# sourceMappingURL=index.d.ts.map