lancedb-memory/bot_memory.py

#!/usr/bin/env python3
"""
Hanzo Bot Memory Integration using LanceDB
This module provides memory search capabilities for Hanzo Bot
"""

import sys
import os
sys.path.append('/Users/prerak/clawd/skills/lancedb-memory')

from final_memory import search_memories, add_memory, get_all_memories

class Hanzo BotMemoryProvider:
    """Memory provider for Hanzo Bot using LanceDB"""
    
    def __init__(self):
        pass
    
    async def search(self, query: str, limit: int = 10) -> list:
        """Search memories"""
        try:
            results = search_memories(query, limit)
            return results
        except Exception as e:
            print(f"Memory search error: {e}")
            return []
    
    async def add(self, content: str, metadata: dict = None) -> int:
        """Add memory"""
        try:
            return add_memory(content, metadata)
        except Exception as e:
            print(f"Memory add error: {e}")
            return -1
    
    async def get_recent(self, limit: int = 50) -> list:
        """Get recent memories"""
        try:
            return get_all_memories()[:limit]
        except Exception as e:
            print(f"Memory get recent error: {e}")
            return []

# Create provider instance
memory_provider = Hanzo BotMemoryProvider()

# Test the integration
if __name__ == "__main__":
    print("Testing Hanzo Bot Memory Integration...")
    
    # Test search
    results = memory_provider.search("test")
    print(f"Search results: {len(results)} memories")
    
    # Test add
    memory_id = memory_provider.add(
        "This is a test memory from Hanzo Bot",
        {"type": "clawbot_test", "importance": 7}
    )
    print(f"Added memory ID: {memory_id}")
    
    print("✅ Hanzo Bot integration ready!")