Turn messy text into typed records

import os
from typing import Literal
from pydantic import BaseModel
from pydantic_ai import Agent
from pydantic_ai.models.openai import OpenAIChatModel
from pydantic_ai.providers.openai import OpenAIProvider

class Ticket(BaseModel):
    customer: str
    order_id: str | None
    category: Literal['shipping', 'billing', 'bug', 'other']
    amount: float | None
    urgent: bool

model = OpenAIChatModel(
    'pareto',
    provider=OpenAIProvider(
        base_url='https://api.unbiased.ai/v1',
        api_key=os.environ['UNBIASED_API_KEY'],
    ),
)
agent = Agent(
    model,
    output_type=Ticket,
    instructions='Extract one support ticket. Use null for anything the message does not say.',
)

inbox = [
    "hi its Priya Nair, order #A-10442 never showed up, tracking says delivered?? need it by friday for my sisters wedding",
    "Charged twice for my annual plan ($96 each). pls refund one. Tom Okafor",
    "app crashes every time i export a pdf on ipad. not urgent, just annoying. cheers, M. Lindqvist",
]
for text in inbox:
    ticket = agent.run_sync(text).output
    print(ticket.model_dump_json())

pip install "pydantic-ai-slim[openai]"
export UNBIASED_API_KEY=your-unbiased-api-key
python tickets.py

import os
from typing import Literal
from openai import OpenAI, pydantic_function_tool
from pydantic import BaseModel

class Ticket(BaseModel):
    customer: str
    order_id: str | None
    category: Literal['shipping', 'billing', 'bug', 'other']
    amount: float | None
    urgent: bool

client = OpenAI(
    base_url='https://api.unbiased.ai/v1',
    api_key=os.environ['UNBIASED_API_KEY'],
)

inbox = [
    "hi its Priya Nair, order #A-10442 never showed up, tracking says delivered?? need it by friday for my sisters wedding",
    "Charged twice for my annual plan ($96 each). pls refund one. Tom Okafor",
    "app crashes every time i export a pdf on ipad. not urgent, just annoying. cheers, M. Lindqvist",
]
for text in inbox:
    completion = client.chat.completions.parse(
        model='pareto',
        messages=[
            {'role': 'system', 'content': 'Extract one support ticket. Use null for anything the message does not say.'},
            {'role': 'user', 'content': text},
        ],
        tools=[pydantic_function_tool(Ticket)],
        tool_choice={'type': 'function', 'function': {'name': 'Ticket'}},
    )
    ticket = completion.choices[0].message.tool_calls[0].function.parsed_arguments
    print(ticket.model_dump_json())

pip install openai
export UNBIASED_API_KEY=your-unbiased-api-key
python tickets_openai.py