tokyo-brain 0.1.0

Universal memory infrastructure for AI agents

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

Feature	Description
🌍 50+ Languages	Multilingual embeddings — store in Japanese, recall in English
🏗️ 3-Tier Architecture	Hot (Redis) → Warm (Postgres+pgvector) → Cold (S3)
🔌 Plugin-First	Drop-in memory for LangChain, CrewAI, AutoGen, OpenClaw
🔒 Encryption at Rest	AES-256 encryption for all stored memories
📊 Tracks	Organise memories by track — history, preferences, knowledge
⚡ Sub-50ms Recall	Optimised vector search with HNSW indexing

API Reference

Brain(api_key, base_url, timeout)

Parameter	Type	Default	Description
api_key	str	required	Your API key (starts with tb-)
base_url	str	https://api.tokyobrain.ai	API endpoint
timeout	int	30	Request timeout in seconds

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

	Tokyo Brain	Mem0	Zep
Multilingual (50+ langs)	✅	❌	⚠️ Limited
3-tier storage	✅ Hot/Warm/Cold	❌ Single tier	⚠️ 2-tier
Plugin ecosystem	✅ LangChain, CrewAI, OpenClaw	⚠️ LangChain only	⚠️ LangChain only
Encryption at rest	✅ AES-256	❌	✅
Track-based organisation	✅	❌	❌
Sub-50ms recall	✅	❌	⚠️
Self-host option	✅	✅	✅
Pricing	Free tier + usage	Free tier + usage	Per-seat

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.