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

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

openinference_instrumentation_vertexai-0.1.21.tar.gz (22.3 kB view details)

Uploaded Aug 27, 2026 Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

The dropdown lists show the available interpreters, ABIs, and platforms. Enable javascript to be able to filter the list of wheel files.

openinference_instrumentation_vertexai-0.1.21-py3-none-any.whl (17.6 kB view details)

Uploaded Aug 27, 2026 Python 3

File details

Details for the file openinference_instrumentation_vertexai-0.1.21.tar.gz.

File metadata

Download URL: openinference_instrumentation_vertexai-0.1.21.tar.gz
Upload date: Aug 27, 2026
Size: 22.3 kB
Tags: Source
Uploaded using Trusted Publishing? Yes
Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for openinference_instrumentation_vertexai-0.1.21.tar.gz
Algorithm	Hash digest
SHA256	`88b25d4c9364261ea149310690461529afe271b58d0607b455b230f8299e7b6b`
MD5	`ce420536fe5aacdc378fafdafb17e8d1`
BLAKE2b-256	`5c66c6958aa4ebbec9d8142d96f2dcfabc0048d7422a1a1a731e7f928edf864d`

See more details on using hashes here.

Provenance

The following attestation bundles were made for openinference_instrumentation_vertexai-0.1.21.tar.gz:

Publisher: publish.yaml on Arize-ai/openinference

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Statement:
- Statement type: https://in-toto.io/Statement/v1
- Predicate type: https://docs.pypi.org/attestations/publish/v1
- Subject name: openinference_instrumentation_vertexai-0.1.21.tar.gz
- Subject digest: 88b25d4c9364261ea149310690461529afe271b58d0607b455b230f8299e7b6b
- Sigstore transparency entry: 2612660419
- Sigstore integration time: Aug 27, 2026
Source repository:
- Permalink: Arize-ai/openinference@e6be63b86f80f09648ba778cce1b996879e3b8e5
- Branch / Tag: refs/heads/main
- Owner: https://github.com/Arize-ai
- Access: public
Publication detail:
- Token Issuer: https://token.actions.githubusercontent.com
- Runner Environment: github-hosted
- Publication workflow: publish.yaml@e6be63b86f80f09648ba778cce1b996879e3b8e5
- Trigger Event: workflow_dispatch