/**
 * The following code is modified based on
 * https://github.com/token-js/token.js/blob/main/src/chat/index.ts
 *
 * MIT License
 * Copyright (c) 2024 RPate97
 * https://github.com/token-js/token.js/blob/main/LICENSE
 */
import { ChatCompletionCreateParamsBase } from 'openai/resources/chat/completions';
import { models } from '../models.js';
import { CompletionResponse, ConfigOptions, StreamCompletionResponse } from '../userTypes/index.js';
export type OpenAIModel = (typeof models.openai.models)[number];
export type OpenAINonStreamingModel = (typeof models)['openai-non-streaming']['models'][number];
export type AI21Model = (typeof models.ai21.models)[number];
export type AnthropicModel = (typeof models.anthropic.models)[number];
export type GeminiModel = (typeof models.gemini.models)[number];
export type MistralModel = (typeof models.mistral.models)[number];
export type PerplexityModel = (typeof models.perplexity.models)[number];
export type GroqModel = (typeof models.groq.models)[number];
export type OpenRouterModel = string;
export type OpenAICompatibleModel = string;
export type AzureOpenAIModel = string;
export type LLMChatModel = OpenAIModel | OpenAINonStreamingModel | AI21Model | AnthropicModel | GeminiModel | MistralModel | PerplexityModel | GroqModel | OpenRouterModel | OpenAICompatibleModel;
export type LLMProvider = keyof typeof models;
type ProviderModelMap = {
    openai: OpenAIModel;
    'openai-non-streaming': OpenAINonStreamingModel;
    ai21: AI21Model;
    anthropic: AnthropicModel;
    gemini: GeminiModel;
    mistral: MistralModel;
    perplexity: PerplexityModel;
    groq: GroqModel;
    openrouter: OpenRouterModel;
    'openai-compatible': OpenAICompatibleModel;
    'azure-openai': AzureOpenAIModel;
};
type CompletionBase<P extends LLMProvider> = Pick<ChatCompletionCreateParamsBase, 'temperature' | 'top_p' | 'stop' | 'n' | 'messages' | 'max_tokens' | 'response_format' | 'tools' | 'tool_choice'> & {
    provider: P;
    model: ProviderModelMap[P];
};
export type CompletionStreaming<P extends LLMProvider> = CompletionBase<P> & {
    stream: true;
};
export type CompletionNonStreaming<P extends LLMProvider> = CompletionBase<P> & {
    stream?: false | null;
};
export type ProviderCompletionParams<P extends LLMProvider> = CompletionStreaming<P> | CompletionNonStreaming<P>;
export type CompletionParams = {
    [P in LLMProvider]: CompletionStreaming<P> | CompletionNonStreaming<P>;
}[LLMProvider];
export declare class LLMCompletions {
    private opts;
    constructor(opts: ConfigOptions);
    create<P extends LLMProvider>(body: CompletionNonStreaming<P>): Promise<CompletionResponse>;
    create<P extends LLMProvider>(body: CompletionStreaming<P>): Promise<StreamCompletionResponse>;
    create<P extends LLMProvider>(body: CompletionBase<P>): Promise<CompletionResponse | StreamCompletionResponse>;
}
export declare class LLMChat {
    completions: LLMCompletions;
    constructor(opts: ConfigOptions);
}
export {};
//# sourceMappingURL=index.d.ts.map