OpenInference Google VertexAI Instrumentation

Python auto-instrumentation library for VertexAI SDK and the Google Cloud AI Platform. Traces are fully OpenTelemetry compatible and can be sent to any OpenTelemetry collector for viewing, such as Arize Phoenix or Arize AX.

Installation

pip install -Uqqq openinference-instrumentation-vertexai

Quickstart

Install openinference-instrumentation-vertexai and arize-phoenix.

pip install -U \
    openinference-instrumentation-vertexai \
    arize-phoenix \
    opentelemetry-sdk \
    opentelemetry-exporter-otlp \
    "opentelemetry-proto>=1.12.0"

Start the phoenix server so that it is ready to receive traces. The phoenix server runs entirely on your machine and does not send data over the internet.

python -m phoenix.server.main serve

Instrumenting vertexai is simple.

from openinference.instrumentation.vertexai import VertexAIInstrumentor
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import SimpleSpanProcessor

endpoint = "http://127.0.0.1:4317"
tracer_provider = TracerProvider()
tracer_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter(endpoint)))

VertexAIInstrumentor().instrument(tracer_provider=tracer_provider)

Now, all calls by generative_models are instrumented and can be viewed in the phoenix UI.

import vertexai
from vertexai.generative_models import GenerativeModel

vertexai.init(location="us-central1")
model = GenerativeModel("gemini-1.5-flash")

print(model.generate_content("Why is sky blue?"))

More Info

Metadata

Release files for openinference-instrumentation-vertexai 0.1.29

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for openinference-instrumentation-vertexai 0.1.29
File	Size	Uploaded
openinference_instrumentation_vertexai-0.1.29.tar.gz	23.2 kB	Oct 9, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for openinference-instrumentation-vertexai 0.1.29
File	Interpreter	ABI	Platform	Reset
openinference_instrumentation_vertexai-0.1.29-py3-none-any.whl 18.3 kB Oct 9, 2026	Python 3	none	any	Details

Total release size: 41.5 kB

Release files / openinference_instrumentation_vertexai-0.1.29.tar.gz

Download URL	openinference_instrumentation_vertexai-0.1.29.tar.gz
Size	23.2 kB
Tags	Source
SHA-256 checksum How to use checksums	`58f9ddd38e57bba58b8c73683284053ed3c14ecd7a1d930dc18bbc12a49f76f3`
BLAKE2b-256 checksum How to use checksums	`e947578310e3d9f20eef0aece68a175b6c2d03446582c71174a669957c427989`
Upload date	Oct 9, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/Arize-ai/openinference
Publishing commit: github.com/Arize-ai/openinference/tree/453da0e0b49acec7b78182ac3aacffbbde030f62

Workflow

Publishing configuration: github.com/Arize-ai/openinference/blob/453da0e0b49acec7b78182ac3aacffbbde030f62/.github/workflows/publish.yaml
Publishing logs: github.com/Arize-ai/openinference/actions/runs/37973512043/attempts/1

Artifact

Subject: openinference_instrumentation_vertexai-0.1.29.tar.gz
SHA-256 checksum: 58f9ddd38e57bba58b8c73683284053ed3c14ecd7a1d930dc18bbc12a49f76f3 Verifying attestations

Release files / openinference_instrumentation_vertexai-0.1.29-py3-none-any.whl

Download URL	openinference_instrumentation_vertexai-0.1.29-py3-none-any.whl
Size	18.3 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`aef22389eefb9fa139bbea580c60c9bfb9d4b5299d806c7bda45865c969f5f51`
BLAKE2b-256 checksum How to use checksums	`a570fed72d652f04ded8ec02b6824918aae20b960b2dde3e9c60e903b4d9cfff`
Upload date	Oct 9, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	Yes
Uploaded via	`twine/7.0.0 CPython/3.13.14`

Provenance

Identity

Publishing platform: GitHub Actions
Publishing repository: github.com/Arize-ai/openinference
Publishing commit: github.com/Arize-ai/openinference/tree/453da0e0b49acec7b78182ac3aacffbbde030f62

Workflow

Publishing configuration: github.com/Arize-ai/openinference/blob/453da0e0b49acec7b78182ac3aacffbbde030f62/.github/workflows/publish.yaml
Publishing logs: github.com/Arize-ai/openinference/actions/runs/37973512043/attempts/1

Artifact

Subject: openinference_instrumentation_vertexai-0.1.29-py3-none-any.whl
SHA-256 checksum: aef22389eefb9fa139bbea580c60c9bfb9d4b5299d806c7bda45865c969f5f51 Verifying attestations