import { trace, Span } from "@opentelemetry/api";
import type { ImageUploadCallback } from "./types";
import type { ImageGenerateParams, ImageEditParams, ImageCreateVariationParams, ImagesResponse } from "openai/resources/images";
export declare function setImageGenerationRequestAttributes(span: Span, params: ImageGenerateParams): void;
export declare function setImageEditRequestAttributes(span: Span, params: ImageEditParams, uploadCallback?: ImageUploadCallback): Promise<void>;
export declare function setImageVariationRequestAttributes(span: Span, params: ImageCreateVariationParams, uploadCallback?: ImageUploadCallback): Promise<void>;
export declare function setImageGenerationResponseAttributes(span: Span, response: ImagesResponse, uploadCallback?: ImageUploadCallback, instrumentationConfig?: {
    enrichTokens?: boolean;
}, params?: any): Promise<void>;
export declare function wrapImageGeneration(tracer: ReturnType<typeof trace.getTracer>, uploadCallback?: ImageUploadCallback, instrumentationConfig?: {
    enrichTokens?: boolean;
}): (original: (...args: any[]) => any) => (this: any, ...args: any[]) => any;
export declare function wrapImageEdit(tracer: ReturnType<typeof trace.getTracer>, uploadCallback?: ImageUploadCallback, instrumentationConfig?: {
    enrichTokens?: boolean;
}): (original: (...args: any[]) => any) => (this: any, ...args: any[]) => any;
export declare function wrapImageVariation(tracer: ReturnType<typeof trace.getTracer>, uploadCallback?: ImageUploadCallback, instrumentationConfig?: {
    enrichTokens?: boolean;
}): (original: (...args: any[]) => any) => (this: any, ...args: any[]) => any;
//# sourceMappingURL=image-wrappers.d.ts.map