import { Opik, Trace, Span } from 'opik';

type OpikParent = Trace | Span;
interface TrackOpikConfig {
    traceMetadata?: Record<string, unknown> & {
        tags?: string[];
    };
    client?: Opik;
    generationName?: string;
    parent?: OpikParent;
}
interface OpikExtension {
    flush: () => Promise<void>;
}

declare const trackOpenAI: <SDKType extends object>(sdk: SDKType, opikConfig?: TrackOpikConfig) => SDKType & OpikExtension;

export { trackOpenAI };