fix: Add generic search() method to Memory class for mem0 compatibility

- Added search() method that accepts query, user_id, agent_id, run_id, limit, rerank, and **kwargs parameters
- Method delegates to mem0 client when available or appropriate local search methods
- Fixes TypeError when Knowledge class calls memory.search() with rerank parameter
- Maintains backward compatibility with existing code
- Added test script to verify the fix

Fixes #717

Co-authored-by: Mervin Praison <[email protected]>

Author: claude[bot]

src/praisonai-agents/praisonaiagents/memory/memory.py

@@ -741,6 +741,54 @@ def search_user_memory(self, user_id: str, query: str, limit: int = 5, rerank: b
                     filtered.append(h)
             return filtered[:limit]
 
+    def search(self, query: str, user_id: Optional[str] = None, agent_id: Optional[str] = None, 
+               run_id: Optional[str] = None, limit: int = 5, rerank: bool = False, **kwargs) -> List[Dict[str, Any]]:
+        """
+        Generic search method that delegates to appropriate specific search methods.
+        Provides compatibility with mem0.Memory interface.
+        
+        Args:
+            query: The search query string
+            user_id: Optional user ID for user-specific search
+            agent_id: Optional agent ID for agent-specific search  
+            run_id: Optional run ID for run-specific search
+            limit: Maximum number of results to return
+            rerank: Whether to use advanced reranking
+            **kwargs: Additional search parameters
+            
+        Returns:
+            List of search results
+        """
+        # If using mem0, pass all parameters directly
+        if self.use_mem0 and hasattr(self, "mem0_client"):
+            search_params = {
+                "query": query,
+                "limit": limit,
+                "rerank": rerank
+            }
+            
+            # Add optional parameters if provided
+            if user_id is not None:
+                search_params["user_id"] = user_id
+            if agent_id is not None:
+                search_params["agent_id"] = agent_id
+            if run_id is not None:
+                search_params["run_id"] = run_id
+                
+            # Include any additional kwargs
+            search_params.update(kwargs)
+            
+            return self.mem0_client.search(**search_params)
+        
+        # For local memory, use specific search methods
+        if user_id:
+            # Use user-specific search
+            return self.search_user_memory(user_id, query, limit=limit, rerank=rerank, **kwargs)
+        else:
+            # Default to long-term memory search
+            # Note: agent_id and run_id filtering could be added to metadata filtering in the future
+            return self.search_long_term(query, limit=limit, rerank=rerank, **kwargs)
+
     def reset_user_memory(self):
         """
         Clear all user-based info. For simplicity, we do a full LTM reset. 

test_rag_fix.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python3
+"""Test script to verify the RAG Agent example fix"""
+
+import sys
+import os
+
+# Add the src directory to Python path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src', 'praisonai-agents'))
+
+try:
+    from praisonaiagents import Agent
+    
+    config = {
+        "vector_store": {
+            "provider": "chroma",
+            "config": {
+                "collection_name": "praison_test",
+                "path": ".praison_test",
+            }
+        }
+    }
+    
+    print("Creating Knowledge Agent...")
+    agent = Agent(
+        name="Knowledge Agent",
+        instructions="You answer questions based on the provided knowledge.",
+        knowledge=["test_document.txt"],  # Using a test file
+        knowledge_config=config,
+        user_id="user1"
+    )
+    
+    print("Testing search functionality...")
+    # This should now work without TypeError
+    result = agent.start("What is KAG in one line?")
+    print(f"Result: {result}")
+    
+    print("✅ Test passed! The RAG Agent example should now work correctly.")
+    
+except TypeError as e:
+    if "rerank" in str(e):
+        print(f"❌ Test failed: {e}")
+        print("The TypeError with 'rerank' parameter still occurs.")
+        sys.exit(1)
+    else:
+        print(f"❌ Test failed with different error: {e}")
+        sys.exit(1)
+except Exception as e:
+    print(f"⚠️  Test encountered different error: {type(e).__name__}: {e}")
+    print("This might be due to missing dependencies or configuration.")
+    sys.exit(0)  # Exit with 0 as this is not the specific error we're testing for