import { CommonRequestOptions } from '@xsai/shared';

interface ChatProvider<T = string> {
    chat: (model: (string & {}) | T) => CommonRequestOptions;
}
interface ChatProviderWithExtraOptions<T = string, T2 = undefined> {
    chat: (model: (string & {}) | T, extraOptions?: T2) => CommonRequestOptions & Partial<T2>;
}
type CreateProviderOptions = Omit<CommonRequestOptions, 'model'>;
interface EmbedProvider<T = string> {
    embed: (model: (string & {}) | T) => CommonRequestOptions;
}
interface EmbedProviderWithExtraOptions<T = string, T2 = undefined> {
    embed: (model: (string & {}) | T, extraOptions?: T2) => CommonRequestOptions & Partial<T2>;
}
interface ImageProvider<T = string> {
    image: (model: (string & {}) | T) => CommonRequestOptions;
}
interface ModelProvider {
    model: () => CreateProviderOptions;
}
interface ModelProviderWithExtraOptions<T = undefined> {
    model: (options?: T) => CreateProviderOptions & Partial<T>;
}
interface SpeechProvider<T = string> {
    speech: (model: (string & {}) | T) => CommonRequestOptions;
}
interface SpeechProviderWithExtraOptions<T = string, T2 = undefined> {
    speech: (model: (string & {}) | T, extraOptions?: T2) => CommonRequestOptions & Partial<T2>;
}
interface TranscriptionProvider<T = string> {
    transcription: (model: (string & {}) | T) => CommonRequestOptions;
}
interface TranscriptionProviderWithExtraOptions<T = string, T2 = undefined> {
    transcription: (model: (string & {}) | T, extraOptions?: T2) => CommonRequestOptions & Partial<T2>;
}

export type { ChatProvider as C, EmbedProvider as E, ImageProvider as I, ModelProvider as M, SpeechProvider as S, TranscriptionProvider as T, ChatProviderWithExtraOptions as a, CreateProviderOptions as b, EmbedProviderWithExtraOptions as c, ModelProviderWithExtraOptions as d, SpeechProviderWithExtraOptions as e, TranscriptionProviderWithExtraOptions as f };