UNPKG

@traceloop/instrumentation-vertexai

Version:
19 lines 829 B
import { InstrumentationBase, InstrumentationModuleDefinition } from "@opentelemetry/instrumentation"; import { VertexAIInstrumentationConfig } from "./types"; import type * as vertexAI from "@google-cloud/vertexai"; export declare class VertexAIInstrumentation extends InstrumentationBase { protected _config: VertexAIInstrumentationConfig; constructor(config?: VertexAIInstrumentationConfig); setConfig(config?: VertexAIInstrumentationConfig): void; protected init(): InstrumentationModuleDefinition; manuallyInstrument(module: typeof vertexAI): void; private wrap; private unwrap; private wrapperMethod; private _startSpan; private _wrapPromise; private _endSpan; private _formatPartsData; private _shouldSendPrompts; } //# sourceMappingURL=vertexai-instrumentation.d.ts.map