AI Memory Python SDK

Official Python SDK for AI Memory API — persistent memory for AI agents.

Installation

pip install aimemoryapi

Quick Start

from aimemoryapi import AiMemory

client = AiMemory()

# Register a user (starts 7-day free trial)
user = client.users.register(
    user_id="my_user_001",
    email="user@example.com",
)

# Create a project
project = client.projects.create(
    user_id="my_user_001",
    name="My AI Project",
)

# Start a session
session = client.sessions.start(
    user_id="my_user_001",
    project_id=project["id"],
)

# Add messages
client.messages.add(
    session_id=session["id"],
    role="user",
    content="Hello, AI!",
)

# Retrieve session context
full_session = client.sessions.get(session["id"])
print(f"Messages: {full_session['messageCount']}")

Resources

Error Handling

from aimemoryapi.errors import (
    AuthenticationError,
    InvalidRequestError,
    APIError,
)

try:
    client.users.register(user_id="user_1")
except AuthenticationError as e:
    print(f"Auth error: {e}")
except InvalidRequestError as e:
    print(f"Invalid request: {e}")
except APIError as e:
    print(f"Server error: {e}")

Requirements

• Python 3.8+
• requests >= 2.31.0
• pydantic >= 2.0.0

Links

• Website
• Documentation
• PyPI