【AI】Claude APIのStructured Outputsを使う方法

pip install anthropic

export ANTHROPIC_API_KEY="your_api_key"

import json
import anthropic

client = anthropic.Anthropic()

schema = {
    "type": "object",
    "properties": {
        "title": {"type": "string"},
        "tags": {"type": "array", "items": {"type": "string"}},
        "priority": {"type": "integer"},
    },
    "required": ["title", "tags", "priority"],
    "additionalProperties": False,
}

response = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=1024,
    messages=[
        {"role": "user", "content": "次の問い合わせを分類して: ログイン画面でエラーが出て入れない"}
    ],
    output_config={
        "format": {"type": "json_schema", "schema": schema}
    },
)

text = next(block.text for block in response.content if block.type == "text")
data = json.loads(text)
print(data["title"], data["tags"], data["priority"])

from pydantic import BaseModel
import anthropic


class Ticket(BaseModel):
    title: str
    tags: list[str]
    priority: int


client = anthropic.Anthropic()

response = client.messages.parse(
    model="claude-sonnet-5",
    max_tokens=1024,
    messages=[
        {"role": "user", "content": "次の問い合わせを分類して: ログイン画面でエラーが出て入れない"}
    ],
    output_format=Ticket,
)

ticket = response.parsed_output
print(ticket.title, ticket.tags, ticket.priority)

tools = [
    {
        "name": "create_ticket",
        "strict": True,
        "description": "サポートチケットを登録する",
        "input_schema": {
            "type": "object",
            "properties": {
                "title": {"type": "string"},
                "priority": {"type": "integer"},
            },
            "required": ["title", "priority"],
            "additionalProperties": False,
        },
    }
]

response = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=1024,
    tools=tools,
    messages=[
        {"role": "user", "content": "緊急でログイン不具合のチケットを作って"}
    ],
)