import { InstrumentationConfig, InstrumentationBase, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
import * as bedrock from '@aws-sdk/client-bedrock-runtime';

declare enum BedrockVendor {
    AI21 = "ai21",
    AMAZON = "amazon",
    ANTHROPIC = "anthropic",
    COHERE = "cohere",
    META = "meta"
}
interface BedrockInstrumentationConfig 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 const bedrockFinishReasonMap: Record<string, string>;
declare class BedrockInstrumentation extends InstrumentationBase {
    protected _config: BedrockInstrumentationConfig;
    constructor(config?: BedrockInstrumentationConfig);
    setConfig(config?: BedrockInstrumentationConfig): void;
    protected init(): InstrumentationModuleDefinition;
    manuallyInstrument(module: typeof bedrock): void;
    private wrap;
    private unwrap;
    private wrapperMethod;
    private _wrapPromise;
    private _startSpan;
    private _endSpan;
    private _setRequestAttributes;
    private _setResponseAttributes;
    private _handleNovaStreamingMetadata;
    private _shouldSendPrompts;
    private _getOperationType;
    private _getStreamChunkContent;
    private _extractVendorAndModel;
}

export { BedrockInstrumentation, BedrockVendor, bedrockFinishReasonMap };
export type { BedrockInstrumentationConfig };