import { AvailableModels } from "../types/AvailableModels.js";
import { ChatMessage } from "../types/ChatMessages.js";
import { Model } from "../types/Model.js";
import { AbstractLLMInferenceEngine } from "./AbstractLLMInferenceEngine.js";
import { AxiosInstance } from "axios";
export type OpenAIAzureModel = Model | 'openai-gpt-4o' | 'openai-gpt-4o-mini' | 'azure-gpt-4o' | 'azure-gpt-4o-mini';
declare class ChatMessageBody {
    role: string;
    content: string;
    constructor(role: string, content: string);
}
declare class ChatInferenceRequest {
    messages: ChatMessageBody[];
    model: OpenAIAzureModel;
    constructor(messages: ChatMessageBody[], model: OpenAIAzureModel);
}
declare class CompletionInferenceRequest {
    prompt: string;
    model: OpenAIAzureModel;
    constructor(prompt: string, model: OpenAIAzureModel);
}
export declare class OpenAIAzureInferenceEngine extends AbstractLLMInferenceEngine {
    availableModels: AvailableModels;
    url: string;
    apiKey: string;
    httpClient: AxiosInstance;
    model: OpenAIAzureModel;
    endpoints: {
        chat: {
            path: string;
            method: string;
            type: typeof ChatInferenceRequest;
        };
        completion: {
            path: string;
            method: string;
            type: typeof CompletionInferenceRequest;
        };
    };
    constructor(model: OpenAIAzureModel, url: string, apiKey: string);
    runChatInference(chat: ChatMessage[]): Promise<ChatMessage[]>;
    runCompletionInference(prompt: string): Promise<string>;
    progressCallback(initProgress: any): void;
}
export {};