MemoryKit

Python SDK

Official Python SDK for MemoryKit.

Installation

pip install memorykit

Quick setup

from memorykit import MemoryKit
 
mk = MemoryKit(api_key="ctx_...")

Memories

# Create
memory = mk.memories.create(
    content="User prefers dark mode.",
    tags=["preferences"],
    userId="user_123",
)
 
# List
memories = mk.memories.list(limit=20)
 
# Search
results = mk.memories.search(
    query="dark mode preferences",
)
 
# Query (RAG)
answer = mk.memories.query(
    query="What does the user prefer?",
    mode="balanced",
)
 
# Stream
for event in mk.memories.stream(
    query="Summarize preferences",
):
    if event["event"] == "text":
        print(event["data"]["content"], end="")
 
# Delete
mk.memories.delete("mem_abc123")

Chats

# Create a chat
chat = mk.chats.create(
    userId="user_123",
    title="Support",
)
 
# Send message
response = mk.chats.send_message(
    chat.id,
    message="How do I reset my password?",
)
 
# Stream message
for event in mk.chats.stream_message(
    chat.id,
    message="Tell me more",
):
    print(event["data"]["content"], end="")

Users

# Upsert user
mk.users.upsert(
    id="user_123",
    name="Alice",
    metadata={"plan": "pro"},
)
 
# Track event
mk.users.create_event(
    "user_123",
    type="page_view",
    data={"page": "/settings"},
)

On this page