@traceloop/instrumentation-vertexai
Version:
Google's VertexAI Instrumentation
19 lines • 829 B
TypeScript
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