【AI】Claude APIのProgrammatic Tool Callingを使う方法

pip install anthropic

import anthropic

client = anthropic.Anthropic()

response = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=4096,
    messages=[
        {
            "role": "user",
            "content": "West、East、Centralの売上を調べて、売上が最も高いリージョンを教えて",
        }
    ],
    tools=[
        {"type": "code_execution_20260120", "name": "code_execution"},
        {
            "name": "query_database",
            "description": "売上DBにSQLを実行する。行のリストをJSON文字列で返す。",
            "input_schema": {
                "type": "object",
                "properties": {
                    "sql": {"type": "string", "description": "実行するSQL"}
                },
                "required": ["sql"],
            },
            "allowed_callers": ["code_execution_20260120"],
        },
    ],
)

print(response)

tool_use = next(b for b in response.content if b.type == "tool_use")
rows = run_sql(tool_use.input["sql"])  # 自前のDB実行関数

response = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=4096,
    container=response.container.id,  # 同じコンテナを再利用
    messages=[
        {"role": "user", "content": "West、East、Centralの売上を調べて、売上が最も高いリージョンを教えて"},
        {"role": "assistant", "content": response.content},
        {
            "role": "user",
            "content": [
                {
                    "type": "tool_result",
                    "tool_use_id": tool_use.id,
                    "content": rows,  # JSON文字列
                }
            ],
        },
    ],
    tools=tools,  # 初回とまったく同じtools配列を渡す
)