#!/usr/bin/env python3
"""Debug script for mem0 operations."""

import os
import sys
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from memory_module.memory_integration import Mem0Integration
from config import MEM0_CONFIG


def test_mem0_operations():
    """Test each mem0 operation step by step."""
    print("=== Testing Mem0 Operations ===\n")
    
    # Initialize mem0
    mem = Mem0Integration(MEM0_CONFIG)
    user_id = "debug_user"
    
    # Test 1: Simple add
    print("Test 1: Adding a simple memory...")
    try:
        result = mem.memory.add(
            messages=[{"role": "user", "content": "I love pizza"}],
            user_id=user_id,
            metadata={"type": "test"}
        )
        print(f"Add result: {result}")
    except Exception as e:
        print(f"Error in add: {type(e).__name__}: {e}")
        import traceback
        traceback.print_exc()
    
    print("\n" + "="*50 + "\n")
    
    # Test 2: Get all memories
    print("Test 2: Getting all memories...")
    try:
        all_memories = mem.memory.get_all(user_id=user_id)
        print(f"All memories: {all_memories}")
    except Exception as e:
        print(f"Error in get_all: {type(e).__name__}: {e}")
        import traceback
        traceback.print_exc()
    
    print("\n" + "="*50 + "\n")
    
    # Test 3: Search memories
    print("Test 3: Searching memories...")
    try:
        search_results = mem.memory.search(
            query="pizza",
            user_id=user_id
        )
        print(f"Search results: {search_results}")
    except Exception as e:
        print(f"Error in search: {type(e).__name__}: {e}")
        import traceback
        traceback.print_exc()
    
    print("\n" + "="*50 + "\n")
    
    # Test 4: Add conversation-style memory
    print("Test 4: Adding conversation-style memory...")
    try:
        messages = [
            {"role": "user", "content": "Hi, my name is Alice"},
            {"role": "assistant", "content": "Nice to meet you Alice"}
        ]
        result = mem.memory.add(
            messages=messages,
            user_id=user_id,
            metadata={"type": "conversation_test"}
        )
        print(f"Conversation add result: {result}")
    except Exception as e:
        print(f"Error in conversation add: {type(e).__name__}: {e}")
        import traceback
        traceback.print_exc()


if __name__ == "__main__":
    test_mem0_operations()