Tokyo Brain

Give your AI agent a memory that never forgets.

Tokyo Brain is universal memory infrastructure for AI agents. Store, recall, and forget — with one API call.

Install

pip install tokyo-brain

Quickstart

from tokyo_brain import Brain

brain = Brain(api_key="tb-...")

# Store a memory
brain.store("User prefers dark mode and speaks Japanese", track="preferences")

# Recall relevant memories
result = brain.recall("What language does the user speak?")
for memory in result.memories:
    print(f"[{memory.score:.2f}] {memory.document}")

# Forget specific memories
brain.forget(ids=["mem_abc123"])

# Health check
health = brain.health()
print(f"Status: {health.status} | Version: {health.version}")

Features

API Reference

Brain(api_key, base_url, timeout)

brain.store(document, track, metadata)

Store a memory document. Returns StoreResult.

brain.recall(query, format, top_k)

Recall memories matching a natural-language query. Returns RecallResult with a list of Memory objects.

brain.forget(ids, where)

Delete memories by ID list or filter. Returns ForgetResult.

brain.health()

Check API health. Returns HealthResult.

Comparison

Self-Hosting

docker run -p 9500:9500 tokyobrain/server:latest

Then point the SDK at your local instance:

brain = Brain(api_key="tb-local", base_url="http://localhost:9500")

Documentation

Full documentation at tokyobrain.ai.

License

MIT — see LICENSE for details.