loyal_companion/CLAUDE.md

CLAUDE.md

This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.

Build & Run Commands

# Install dependencies
pip install -r requirements.txt

# Install in development mode (required for testing)
pip install -e .

# Run the bot (requires .env with DISCORD_TOKEN and AI provider key)
python -m daemon_boyfriend

# Run with Docker (includes PostgreSQL)
docker-compose up -d

# Run database migrations
alembic upgrade head

# Syntax check all Python files
python -m py_compile src/daemon_boyfriend/**/*.py

Testing

# Install dev dependencies
pip install -e ".[dev]"

# Run all tests
python -m pytest tests/ -v

# Run tests with coverage
python -m pytest tests/ --cov=daemon_boyfriend --cov-report=term-missing

# Run specific test file
python -m pytest tests/test_models.py -v

# Run specific test class
python -m pytest tests/test_services.py::TestMoodService -v

The test suite uses:

• pytest with pytest-asyncio for async test support
• SQLite in-memory database for testing (via aiosqlite)
• Mock fixtures for Discord objects and AI providers in tests/conftest.py

Architecture

This is a Discord bot that responds to @mentions with AI-generated responses (multi-provider support). It features a "Living AI" system that gives the bot personality, mood, and relationship tracking.

Provider Pattern

The AI system uses a provider abstraction pattern:

• services/providers/base.py defines AIProvider abstract class with generate() method
• services/providers/openai.py, openrouter.py, anthropic.py, gemini.py implement the interface
• services/ai_service.py is the factory that creates the correct provider based on AI_PROVIDER env var
• OpenRouter uses OpenAI's client with a different base URL
• Gemini uses the google-genai SDK

Cog System

Discord functionality is in cogs/:

• ai_chat.py - @mention handler (responds when bot is mentioned)
• memory.py - Memory management commands (!setname, !remember, etc.)
• status.py - Bot health and status commands

Cogs are auto-loaded by bot.py from the cogs/ directory.

Database & Memory System

The bot uses PostgreSQL for persistent memory (optional, falls back to in-memory):

• models/ - SQLAlchemy models (User, UserFact, Conversation, Message, Guild, GuildMember)
• models/living_ai.py - Living AI models (BotState, BotOpinion, UserRelationship, etc.)
• services/database.py - Connection pool and async session management
• services/user_service.py - User CRUD, custom names, facts management
• services/persistent_conversation.py - Database-backed conversation history
• alembic/ - Database migrations

Key features:

• Custom names: Set preferred names for users so the bot knows "who is who"
• User facts: Bot remembers things about users (hobbies, preferences, etc.)
• Persistent conversations: Chat history survives restarts
• Conversation timeout: New conversation starts after 60 minutes of inactivity

Living AI System

The bot implements a "Living AI" system with emotional depth and relationship tracking:

Services (services/)

• mood_service.py - Valence-arousal mood model with time decay
• relationship_service.py - Relationship scoring (stranger to close friend)
• fact_extraction_service.py - Autonomous fact learning from conversations
• opinion_service.py - Bot develops opinions on topics over time
• self_awareness_service.py - Bot statistics and self-reflection
• communication_style_service.py - Learns user communication preferences
• proactive_service.py - Scheduled events (birthdays, follow-ups)
• association_service.py - Cross-user memory associations

Models (models/living_ai.py)

• BotState - Global mood state and statistics per guild
• BotOpinion - Topic sentiments and preferences
• UserRelationship - Per-user relationship scores and metrics
• UserCommunicationStyle - Learned communication preferences
• ScheduledEvent - Birthdays, follow-ups, reminders
• FactAssociation - Cross-user memory links
• MoodHistory - Mood changes over time

Mood System

Uses a valence-arousal model:

• Valence: -1 (sad) to +1 (happy)
• Arousal: -1 (calm) to +1 (excited)
• Labels: excited, happy, calm, neutral, bored, annoyed, curious
• Time decay: Mood gradually returns to neutral

Relationship Levels

• Stranger (0-20): Polite, formal
• Acquaintance (21-40): Friendly but reserved
• Friend (41-60): Casual, warm
• Good Friend (61-80): Personal, references past talks
• Close Friend (81-100): Very casual, inside jokes

Configuration

All config flows through config.py using pydantic-settings. The settings singleton is created at module load, so env vars must be set before importing.

Web Search

The bot can search the web for current information via SearXNG:

• services/searxng.py provides SearXNGService for web queries
• ai_chat.py uses a two-step approach: first asks AI if search is needed, then provides results as context
• Search is triggered automatically when the AI determines the query needs current information
• Configured via SEARXNG_URL, SEARXNG_ENABLED, and SEARXNG_MAX_RESULTS env vars

Key Design Decisions

• PersistentConversationManager stores conversations in PostgreSQL when DATABASE_URL is set
• ConversationManager is the in-memory fallback when database is not configured
• Long AI responses are split via split_message() in ai_chat.py to respect Discord's 2000 char limit
• The bot responds only to @mentions via on_message listener
• Web search uses AI to decide when to search, avoiding unnecessary API calls for general knowledge questions
• User context (custom name + known facts) is included in AI prompts for personalized responses
• PortableJSON type in models/base.py allows models to work with both PostgreSQL (JSONB) and SQLite (JSON)
• ensure_utc() helper handles timezone-naive datetimes from SQLite

Environment Variables

Required: DISCORD_TOKEN, plus one of OPENAI_API_KEY, OPENROUTER_API_KEY, ANTHROPIC_API_KEY, or GEMINI_API_KEY depending on AI_PROVIDER setting.

Optional:

• DATABASE_URL - PostgreSQL connection string (e.g., postgresql+asyncpg://user:pass@host:5432/db)
• POSTGRES_PASSWORD - Used by docker-compose for the PostgreSQL container
• SEARXNG_URL - SearXNG instance URL for web search capability

Living AI Configuration

• LIVING_AI_ENABLED - Master switch for Living AI features (default: true)
• MOOD_ENABLED - Enable mood system (default: true)
• RELATIONSHIP_ENABLED - Enable relationship tracking (default: true)
• FACT_EXTRACTION_ENABLED - Enable autonomous fact extraction (default: true)
• FACT_EXTRACTION_RATE - Probability of extracting facts (default: 0.3)
• PROACTIVE_ENABLED - Enable proactive messages (default: true)
• CROSS_USER_ENABLED - Enable cross-user memory associations (default: false)
• OPINION_FORMATION_ENABLED - Enable bot opinion formation (default: true)
• STYLE_LEARNING_ENABLED - Enable communication style learning (default: true)
• MOOD_DECAY_RATE - How fast mood returns to neutral per hour (default: 0.1)

Command Toggles

• COMMANDS_ENABLED - Master switch for all commands (default: true)
• CMD_RELATIONSHIP_ENABLED - Enable !relationship command
• CMD_MOOD_ENABLED - Enable !mood command
• CMD_BOTSTATS_ENABLED - Enable !botstats command
• CMD_OURHISTORY_ENABLED - Enable !ourhistory command
• CMD_BIRTHDAY_ENABLED - Enable !birthday command
• CMD_REMEMBER_ENABLED - Enable !remember command
• CMD_SETNAME_ENABLED - Enable !setname command
• CMD_WHATDOYOUKNOW_ENABLED - Enable !whatdoyouknow command
• CMD_FORGETME_ENABLED - Enable !forgetme command

Commands

User commands

• !setname <name> - Set your preferred name
• !clearname - Reset to Discord display name
• !remember <fact> - Tell the bot something about you
• !whatdoyouknow - See what the bot remembers about you
• !forgetme - Clear all facts about you
• !relationship - See your relationship level with the bot
• !mood - See the bot's current emotional state
• !botstats - Bot shares its self-awareness statistics
• !ourhistory - See your history with the bot
• !birthday <date> - Set your birthday for the bot to remember

Admin commands

• !setusername @user <name> - Set name for another user
• !teachbot @user <fact> - Add a fact about a user