/**
 * The following code is modified based on
 * https://github.com/token-js/token.js/blob/main/src/userTypes/index.ts
 *
 * MIT License
 * Copyright (c) 2024 RPate97
 * https://github.com/token-js/token.js/blob/main/LICENSE
 */
/**
 * These types are explicitly intended to be imported by the user. We keep them separate for clarity
 * and so that they can be easily imported and used alongside the primary LLM class.
 */
import { ClientOptions } from 'openai';
import { ChatCompletion, ChatCompletionChunk, ChatCompletionTool as OpenAIChatCompletionTool, ChatCompletionMessageParam as OpenAICompletionMessageParam } from 'openai/resources/index';
export type ConfigOptions = Pick<ClientOptions, 'apiKey' | 'baseURL'> & {
    defaultHeaders?: Record<string, string>;
    azure?: {
        endpoint?: string;
        apiVersion?: string;
        azureADTokenProvider?: any;
    };
};
export type ChatCompletionChoice = Omit<ChatCompletion.Choice, 'finish_reason'> & {
    finish_reason: ChatCompletion.Choice['finish_reason'] | 'unknown';
};
export type ChatCompletionChunkChoice = Omit<ChatCompletionChunk.Choice, 'finish_reason'> & {
    finish_reason: ChatCompletionChunk.Choice['finish_reason'] | 'unknown';
};
type CompletionResponseFields = 'created' | 'model' | 'usage' | 'object';
export type CompletionResponse = Pick<ChatCompletion, CompletionResponseFields> & {
    id: string | null;
    choices: Array<ChatCompletionChoice>;
};
export type CompletionResponseChunk = Pick<ChatCompletionChunk, CompletionResponseFields> & {
    id: string | null;
    choices: Array<ChatCompletionChunkChoice>;
};
export type StreamCompletionResponse = AsyncIterable<CompletionResponseChunk>;
export type ChatCompletionMessageParam = OpenAICompletionMessageParam;
export type ChatCompletionTool = OpenAIChatCompletionTool;
export {};
//# sourceMappingURL=index.d.ts.map