Pydantic AI

pip install "pydantic-ai-slim[openai]"

import os

from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

model = OpenAIChatModel(
    "auto",                                    # or any Mesh catalog id
    provider=OpenAIProvider(
        base_url="https://api.meshapi.ai/v1",
        api_key=os.environ["MESH_API_KEY"],    # rsk_...
    ),
)

agent = Agent(model, system_prompt="Be concise.")

@agent.tool_plain
def get_shipment_status(order_id: str) -> dict:
    """Look up the shipment status of an order."""
    book = {"A-1042": {"state": "in transit", "eta_days": 2}}
    if order_id in book:
        return {"status": "success", "order_id": order_id, **book[order_id]}
    return {"status": "error", "error_message": f"No order {order_id}."}


result = await agent.run("What's the shipment status of order A-1042?")
print(result.output)

from pydantic import BaseModel


class Shipment(BaseModel):
    order_id: str
    state: str
    eta_days: int


agent = Agent(model, output_type=Shipment)