【AI】Claude APIのMemory Toolで記憶を永続化する方法

import anthropic
from anthropic.tools import BetaLocalFilesystemMemoryTool

client = anthropic.Anthropic()
memory = BetaLocalFilesystemMemoryTool(base_path="./memory")

runner = client.beta.messages.tool_runner(
    model="claude-sonnet-5",
    max_tokens=1024,
    messages=[
        {"role": "user", "content": "顧客Acme社はメール連絡を好むと覚えておいて"}
    ],
    tools=[memory],
)

final_message = runner.until_done()
print(final_message.content)

import anthropic

client = anthropic.Anthropic()

TOOLS = [{"type": "memory_20250818", "name": "memory"}]
messages = [{"role": "user", "content": "前回の続きから作業を進めて"}]

while True:
    message = client.messages.create(
        model="claude-sonnet-5",
        max_tokens=2048,
        messages=messages,
        tools=TOOLS,
    )

    if message.stop_reason != "tool_use":
        print(message.content)
        break

    tool_results = []
    for block in message.content:
        if block.type == "tool_use" and block.name == "memory":
            output = run_memory_command(block.input)
            tool_results.append({
                "type": "tool_result",
                "tool_use_id": block.id,
                "content": output,
            })

    messages.append({"role": "assistant", "content": message.content})
    messages.append({"role": "user", "content": tool_results})

from pathlib import Path

MEMORY_ROOT = Path("./memory_store").resolve()


def resolve_safe(path: str) -> Path:
    relative = path.removeprefix("/memories").lstrip("/")
    resolved = (MEMORY_ROOT / relative).resolve()
    if not resolved.is_relative_to(MEMORY_ROOT):
        raise ValueError(f"不正なパスが指定されました: {path}")
    return resolved


def run_memory_command(command: dict) -> str:
    action = command["command"]

    if action == "view":
        target = resolve_safe(command["path"])
        if target.is_dir():
            return "\n".join(sorted(str(p) for p in target.iterdir()))
        return target.read_text(encoding="utf-8")

    if action == "create":
        target = resolve_safe(command["path"])
        target.parent.mkdir(parents=True, exist_ok=True)
        target.write_text(command["file_text"], encoding="utf-8")
        return f"File created successfully at: {command['path']}"

    # str_replace / insert / delete / rename も同じ要領で実装する
    raise ValueError(f"未対応のコマンドです: {action}")

response = client.beta.messages.create(
    model="claude-sonnet-5",
    max_tokens=4096,
    messages=messages,
    tools=[{"type": "memory_20250818", "name": "memory"}],
    betas=["context-management-2025-06-27"],
    context_management={
        "edits": [
            {
                "type": "clear_tool_uses_20250919",
                "trigger": {"type": "input_tokens", "value": 30000},
                "keep": {"type": "tool_uses", "value": 3},
                "exclude_tools": ["memory"],
            }
        ]
    },
)

print(response.context_management)