@traceloop/instrumentation-vertexai
Version:
Google's VertexAI Instrumentation
18 lines • 817 B
TypeScript
import { InstrumentationBase, InstrumentationModuleDefinition } from "@opentelemetry/instrumentation";
import { AIPlatformInstrumentationConfig } from "./types";
import type * as aiplatform from "@google-cloud/aiplatform";
export declare class AIPlatformInstrumentation extends InstrumentationBase {
protected _config: AIPlatformInstrumentationConfig;
constructor(config?: AIPlatformInstrumentationConfig);
setConfig(config?: AIPlatformInstrumentationConfig): void;
protected init(): InstrumentationModuleDefinition;
manuallyInstrument(module: typeof aiplatform): void;
private wrap;
private unwrap;
private wrapperMethod;
private _startSpan;
private _wrapPromise;
private _endSpan;
private _shouldSendPrompts;
}
//# sourceMappingURL=aiplatform-instrumentation.d.ts.map