OpenInference PydanticAI
Python auto-instrumentation library for PydanticAI. These traces are fully OpenTelemetry compatible and can be sent to an OpenTelemetry collector for viewing, such as Arize Phoenix.
Installation
pip install openinference-instrumentation-pydantic-ai
Quickstart
This quickstart shows you how to instrument your PydanticAI agents.
Install required packages.
pip install pydantic-ai arize-phoenix opentelemetry-sdk opentelemetry-exporter-otlp
Start Phoenix in the background as a collector. By default, it listens on http://localhost:6006. You can visit the app via a browser at the same address. (Phoenix does not send data over the internet. It only operates locally on your machine.)
phoenix serve
Here's a simple example that demonstrates how to use PydanticAI with OpenInference instrumentation:
import os
from pydantic import BaseModel
from pydantic_ai import Agent
from pydantic_ai.models.instrumented import InstrumentationSettings
from pydantic_ai.models.openai import OpenAIModel
from pydantic_ai.providers.openai import OpenAIProvider
from opentelemetry import trace
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
from opentelemetry.sdk.trace import TracerProvider
from openinference.instrumentation.pydantic_ai import OpenInferenceSpanProcessor
from opentelemetry.sdk.trace.export import SimpleSpanProcessor
# Set your OpenAI API key
os.environ["OPENAI_API_KEY"] = "YOUR_OPENAI_API_KEY"
# Set up the tracer provider
tracer_provider = TracerProvider()
trace.set_tracer_provider(tracer_provider)
# Add the OpenInference span processor
endpoint = "http://127.0.0.1:6006/v1/traces"
exporter = OTLPSpanExporter(endpoint=endpoint)
tracer_provider.add_span_processor(OpenInferenceSpanProcessor())
tracer_provider.add_span_processor(SimpleSpanProcessor(exporter))
# Define your Pydantic model
class LocationModel(BaseModel):
city: str
country: str
instrumentation = InstrumentationSettings(version=2)
# Create and configure the agent
model = OpenAIModel("gpt-4", provider=OpenAIProvider())
agent = Agent(model, output_type=LocationModel, instrument=instrumentation)
# Run the agent
result = agent.run_sync("The windy city in the US of A.")
print(result)
This example:
- Sets up OpenTelemetry tracing with Phoenix
- Defines a simple Pydantic model for location data
- Creates a PydanticAI agent with instrumentation enabled
- Runs a query and gets structured output
The traces will be visible in the Phoenix UI at http://localhost:6006.
More Info
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file openinference_instrumentation_pydantic_ai-0.1.17.tar.gz.
File metadata
- Download URL: openinference_instrumentation_pydantic_ai-0.1.17.tar.gz
- Upload date:
- Size: 22.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
b7d5e66e27acdaf0606de48cdd298ae9491f578dcf4ed3f50217c165e9fd1fc4
|
|
| MD5 |
0bcdc376b0187d3b373618c7a3458f3e
|
|
| BLAKE2b-256 |
f3ba0f4cd6f87c55d51574bca067709676e10b4a7d321400e94afcad639829ee
|
Provenance
The following attestation bundles were made for openinference_instrumentation_pydantic_ai-0.1.17.tar.gz:
Publisher:
publish.yaml on Arize-ai/openinference
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
openinference_instrumentation_pydantic_ai-0.1.17.tar.gz -
Subject digest:
b7d5e66e27acdaf0606de48cdd298ae9491f578dcf4ed3f50217c165e9fd1fc4 - Sigstore transparency entry: 2023504899
- Sigstore integration time:
-
Permalink:
Arize-ai/openinference@77c67edc2c7bdcd3002ba3cbaa3aa608c5aa8830 -
Branch / Tag:
refs/heads/main - Owner: https://github.com/Arize-ai
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yaml@77c67edc2c7bdcd3002ba3cbaa3aa608c5aa8830 -
Trigger Event:
workflow_dispatch
-
Statement type:
File details
Details for the file openinference_instrumentation_pydantic_ai-0.1.17-py3-none-any.whl.
File metadata
- Download URL: openinference_instrumentation_pydantic_ai-0.1.17-py3-none-any.whl
- Upload date:
- Size: 16.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/6.1.0 CPython/3.13.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
23aef1844d545664d32ccd26268468a4abde8ca076870c604166022a809948e5
|
|
| MD5 |
2a83427dd0e47a714d85cefaf99fa73b
|
|
| BLAKE2b-256 |
d0f5fd6f39ff69bd821fb6347e2a16b353c5175632ef3b2f968ac28427e87702
|
Provenance
The following attestation bundles were made for openinference_instrumentation_pydantic_ai-0.1.17-py3-none-any.whl:
Publisher:
publish.yaml on Arize-ai/openinference
-
Statement:
-
Statement type:
https://in-toto.io/Statement/v1 -
Predicate type:
https://docs.pypi.org/attestations/publish/v1 -
Subject name:
openinference_instrumentation_pydantic_ai-0.1.17-py3-none-any.whl -
Subject digest:
23aef1844d545664d32ccd26268468a4abde8ca076870c604166022a809948e5 - Sigstore transparency entry: 2023505094
- Sigstore integration time:
-
Permalink:
Arize-ai/openinference@77c67edc2c7bdcd3002ba3cbaa3aa608c5aa8830 -
Branch / Tag:
refs/heads/main - Owner: https://github.com/Arize-ai
-
Access:
public
-
Token Issuer:
https://token.actions.githubusercontent.com -
Runner Environment:
github-hosted -
Publication workflow:
publish.yaml@77c67edc2c7bdcd3002ba3cbaa3aa608c5aa8830 -
Trigger Event:
workflow_dispatch
-
Statement type:
