ドキュメント要約ツール

def summarize(text: str, style: str = "bullet") -> str:
    prompt = f'''
次の文書を {style} 形式で日本語要約してください。

<doc>
{text}
</doc>

出力フォーマット:
<summary>
{{...}}
</summary>
'''
    res = client.messages.create(
        model="claude-sonnet-4-6",
        max_tokens=1024,
        messages=[{"role": "user", "content": prompt}],
    )
    return res.content[0].text

def chunk_text(text: str, max_chars: int = 4000) -> list[str]:
    paragraphs = text.split("\n\n")
    chunks, current = [], ""
    for p in paragraphs:
        if len(current) + len(p) > max_chars:
            chunks.append(current)
            current = p
        else:
            current += "\n\n" + p
    if current:
        chunks.append(current)
    return chunks

def map_reduce_summary(chunks: list[str]) -> str:
    partials = [summarize(c, style="bullet") for c in chunks]
    combined = "\n\n".join(f"## チャンク{i}\n{p}" for i, p in enumerate(partials))
    return summarize(combined, style="overview")