#!/usr/bin/env python3
"""Test MCP server initialization with checkpoint."""

from salience_os_seed.runtime.mcp_server import MONIKAMCPServer
from salience_os_seed.conversation.session import ConversationConfig

print("Initializing MCP server with checkpoint...")
config = ConversationConfig()
config.lm.checkpoint_path = "storage/proto_lm/checkpoint.pt"

try:
    server = MONIKAMCPServer(conversation_config=config)
    print("✅ MCP Server initialized successfully!")
    print(f"   Proto-LM step: {server.proto_lm.step}")
    print(f"   Vocab size: {server.proto_lm.vocab.size()}")
except Exception as e:
    print(f"❌ Failed to initialize: {e}")
    import traceback
    traceback.print_exc()