import * as azure from '@azure/openai';
import { InstrumentationConfig, InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';

interface AzureOpenAIInstrumentationConfig extends InstrumentationConfig {
    /**
     * Whether to log prompts, completions and embeddings on traces.
     * @default true
     */
    traceContent?: boolean;
    /**
     * A custom logger to log any exceptions that happen during span creation.
     */
    exceptionLogger?: (e: Error) => void;
}

declare class AzureOpenAIInstrumentation extends InstrumentationBase {
    protected _config: AzureOpenAIInstrumentationConfig;
    constructor(config?: AzureOpenAIInstrumentationConfig);
    setConfig(config?: AzureOpenAIInstrumentationConfig): void;
    manuallyInstrument(module: typeof azure): void;
    protected init(): InstrumentationModuleDefinition;
    private patch;
    private unpatch;
    private patchOpenAI;
    private startSpan;
    private _wrapPromise;
    private _endSpan;
    private _shouldSendPrompts;
}

export { AzureOpenAIInstrumentation, type AzureOpenAIInstrumentationConfig };