Pydantic AI

1
import os
2
3
# Route LLM calls through Respan (gateway pattern — only RESPAN_API_KEY needed)
4
respan_api_key = os.environ["RESPAN_API_KEY"]
5
respan_base_url = os.getenv("RESPAN_BASE_URL", "https://api.respan.ai/api")
6
os.environ["OPENAI_BASE_URL"] = respan_base_url
7
os.environ["OPENAI_API_KEY"] = respan_api_key
8
9
from pydantic_ai import Agent
10
from respan_tracing import RespanTelemetry
11
from respan_exporter_pydantic_ai import instrument_pydantic_ai
12
13
def main():
14
    telemetry = RespanTelemetry(
15
        app_name="pydantic-ai-hello-world",
16
        api_key=respan_api_key,
17
        base_url=respan_base_url,
18
    )
19
    instrument_pydantic_ai()
20
21
    agent = Agent(
22
        model="openai:gpt-4o",
23
        system_prompt="You are a helpful assistant.",
24
    )
25
    result = agent.run_sync("What is the capital of France?")
26
    print("Agent Output:", result.output)
27
28
    telemetry.flush()
29
30
if __name__ == "__main__":
31
    main()

1
import os
2
3
respan_api_key = os.environ["RESPAN_API_KEY"]
4
respan_base_url = os.getenv("RESPAN_BASE_URL", "https://api.respan.ai/api")
5
os.environ["OPENAI_BASE_URL"] = respan_base_url
6
os.environ["OPENAI_API_KEY"] = respan_api_key
7
8
from pydantic_ai import Agent
9
from respan_tracing import RespanTelemetry
10
from respan_exporter_pydantic_ai import instrument_pydantic_ai
11
12
def main():
13
    telemetry = RespanTelemetry(
14
        app_name="pydantic-ai-gateway",
15
        api_key=respan_api_key,
16
        base_url=respan_base_url,
17
    )
18
    instrument_pydantic_ai(
19
        include_content=True,
20
        include_binary_content=True,
21
    )
22
23
    agent = Agent(
24
        model="openai:gpt-4o",
25
        system_prompt="You are a helpful assistant.",
26
    )
27
    result = agent.run_sync("What is the capital of France?")
28
    print("Agent Output:", result.output)
29
30
    telemetry.flush()
31
32
if __name__ == "__main__":
33
    main()

1
import os
2
3
respan_api_key = os.environ["RESPAN_API_KEY"]
4
respan_base_url = os.getenv("RESPAN_BASE_URL", "https://api.respan.ai/api")
5
os.environ["OPENAI_BASE_URL"] = respan_base_url
6
os.environ["OPENAI_API_KEY"] = respan_api_key
7
8
from pydantic_ai import Agent
9
from respan_tracing import RespanTelemetry
10
from respan_tracing.decorators import workflow, task
11
from respan_exporter_pydantic_ai import instrument_pydantic_ai
12
13
agent = Agent("openai:gpt-4o", system_prompt="You are a helpful travel assistant.")
14
15
@task(name="fetch_destination_info")
16
def fetch_destination_info(destination: str) -> str:
17
    result = agent.run_sync(f"Give me a one-sentence summary of {destination}.")
18
    return result.output
19
20
@workflow(name="travel_planning_workflow")
21
def travel_planning_workflow(destination: str):
22
    info = fetch_destination_info(destination)
23
    return info
24
25
def main():
26
    telemetry = RespanTelemetry(
27
        app_name="pydantic-ai-tracing",
28
        api_key=respan_api_key,
29
        base_url=respan_base_url,
30
    )
31
    instrument_pydantic_ai()
32
    output = travel_planning_workflow("Paris")
33
    print("Workflow Output:", output)
34
    telemetry.flush()
35
36
if __name__ == "__main__":
37
    main()

1
import os
2
3
respan_api_key = os.environ["RESPAN_API_KEY"]
4
respan_base_url = os.getenv("RESPAN_BASE_URL", "https://api.respan.ai/api")
5
os.environ["OPENAI_BASE_URL"] = respan_base_url
6
os.environ["OPENAI_API_KEY"] = respan_api_key
7
8
from pydantic_ai import Agent
9
from respan_tracing import RespanTelemetry, get_client
10
from respan_tracing.decorators import task
11
from respan_exporter_pydantic_ai import instrument_pydantic_ai
12
13
agent = Agent("openai:gpt-4o")
14
15
@task(name="customer_query")
16
def customer_query(query: str):
17
    client = get_client()
18
    if client:
19
        client.update_current_span(
20
            respan_params={
21
                "customer_identifier": "user_12345",
22
                "metadata": {"plan": "premium", "session_id": "abc-987"},
23
                "custom_tags": ["pydantic-ai", "test-run"],
24
            }
25
        )
26
    result = agent.run_sync(query)
27
    return result.output
28
29
def main():
30
    telemetry = RespanTelemetry(
31
        app_name="pydantic-ai-params",
32
        api_key=respan_api_key,
33
        base_url=respan_base_url,
34
    )
35
    instrument_pydantic_ai()
36
    output = customer_query("Hello, who are you?")
37
    print("Output:", output)
38
    telemetry.flush()
39
40
if __name__ == "__main__":
41
    main()

1
import os
2
3
respan_api_key = os.environ["RESPAN_API_KEY"]
4
respan_base_url = os.getenv("RESPAN_BASE_URL", "https://api.respan.ai/api")
5
os.environ["OPENAI_BASE_URL"] = respan_base_url
6
os.environ["OPENAI_API_KEY"] = respan_api_key
7
8
from pydantic_ai import Agent
9
from respan_tracing import RespanTelemetry
10
from respan_tracing.decorators import workflow
11
from respan_exporter_pydantic_ai import instrument_pydantic_ai
12
13
agent = Agent(
14
    "openai:gpt-4o",
15
    system_prompt=(
16
        "You are a calculator assistant. You must use the provided tools for any arithmetic. "
17
        "Never compute numbers yourself; always call the add tool when asked to add numbers."
18
    ),
19
)
20
21
@agent.tool_plain
22
def add(a: int, b: int) -> int:
23
    """Add two numbers together."""
24
    return a + b
25
26
@workflow(name="calculator_agent_run")
27
def run_calculator_agent(prompt: str):
28
    result = agent.run_sync(prompt)
29
    return result.output
30
31
def main():
32
    telemetry = RespanTelemetry(
33
        app_name="pydantic-ai-tool-use",
34
        api_key=respan_api_key,
35
        base_url=respan_base_url,
36
    )
37
    instrument_pydantic_ai()
38
    output = run_calculator_agent("Use your add tool to compute 15 + 27, then reply with the result.")
39
    print("Agent Output:", output)
40
    telemetry.flush()
41
42
if __name__ == "__main__":
43
    main()

1
agent = Agent("openai:gpt-4o")
2
instrument_pydantic_ai(agent=agent)