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.27

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.27
File	Size	Uploaded
openinference_instrumentation_vertexai-0.1.27.tar.gz	22.6 kB	Oct 1, 2026	Details

Built distribution (wheel)

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

Total release size: 40.7 kB

Release files / openinference_instrumentation_vertexai-0.1.27.tar.gz

Download URL	openinference_instrumentation_vertexai-0.1.27.tar.gz
Size	22.6 kB
Tags	Source
SHA-256 checksum How to use checksums	`9500493cbd65c0cbc941c084f857a7c3573a0c0575af11c451059be59900e95e`
BLAKE2b-256 checksum How to use checksums	`7ebdb44b85473276117021d4b95284982efb854aca74dcfb907622033d4509f9`
Upload date	Oct 1, 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/c7f15ec66b1374ee3e7aafbdba5a55959afacac7

Workflow

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

Artifact

Subject: openinference_instrumentation_vertexai-0.1.27.tar.gz
SHA-256 checksum: 9500493cbd65c0cbc941c084f857a7c3573a0c0575af11c451059be59900e95e Verifying attestations

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

Download URL	openinference_instrumentation_vertexai-0.1.27-py3-none-any.whl
Size	18.2 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`53174af7aa5c766b5bd9999d8f2fc35f26fb0ae7a7abb8c56ff184be4afae14b`
BLAKE2b-256 checksum How to use checksums	`c3e465439072b22347cd1103eab5ab1adf889706039fc713d4a3dd7e1c0dfbf6`
Upload date	Oct 1, 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/c7f15ec66b1374ee3e7aafbdba5a55959afacac7

Workflow

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

Artifact

Subject: openinference_instrumentation_vertexai-0.1.27-py3-none-any.whl
SHA-256 checksum: 53174af7aa5c766b5bd9999d8f2fc35f26fb0ae7a7abb8c56ff184be4afae14b Verifying attestations