【AI】Claude APIの拡張思考をPythonで実装する方法

import anthropic

client = anthropic.Anthropic()

response = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=16000,
    thinking={"type": "adaptive"},
    messages=[
        {"role": "user", "content": "1から100までの素数の個数を、手順を踏んで求めてください。"}
    ],
)

for block in response.content:
    if block.type == "thinking":
        print("思考:", block.thinking)
    elif block.type == "text":
        print("回答:", block.text)

response = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=16000,
    thinking={"type": "adaptive"},
    output_config={"effort": "low"},
    messages=[
        {"role": "user", "content": "配列を昇順にソートする関数をPythonで書いてください。"}
    ],
)

response = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=16000,
    thinking={"type": "adaptive", "display": "summarized"},
    messages=[
        {"role": "user", "content": "在庫管理システムのテーブル設計を提案してください。"}
    ],
)

for block in response.content:
    if block.type == "thinking":
        print(block.thinking)
    elif block.type == "text":
        print(block.text)

with client.messages.stream(
    model="claude-sonnet-5",
    max_tokens=16000,
    thinking={"type": "adaptive", "display": "summarized"},
    messages=[
        {"role": "user", "content": "クイックソートの実装と計算量を説明してください。"}
    ],
) as stream:
    for event in stream:
        if event.type == "content_block_delta" and event.delta.type == "thinking_delta":
            print(event.delta.thinking, end="", flush=True)
        elif event.type == "content_block_delta" and event.delta.type == "text_delta":
            print(event.delta.text, end="", flush=True)

messages = [
    {"role": "user", "content": "二分探索をPythonで実装してください。"}
]

first = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=16000,
    thinking={"type": "adaptive"},
    messages=messages,
)

messages.append({"role": "assistant", "content": first.content})
messages.append({"role": "user", "content": "計算量も教えてください。"})

second = client.messages.create(
    model="claude-sonnet-5",
    max_tokens=16000,
    thinking={"type": "adaptive"},
    messages=messages,
)