changes to serach memory tool

app/tools/context_agent_tools.py

@@ -1,5 +1,5 @@
 from typing import Any
-# from langmem import create_search_memory_tool
+from langmem import create_search_memory_tool
 from typing import Dict, Any, Optional,Annotated
 from langchain.tools import tool,ToolRuntime
 from langgraph.prebuilt import InjectedState
@@ -7,56 +7,13 @@ from langgraph.store.base import BaseStore
 from typing import Literal
   # Correct import path
 
-@tool
-def search_sender_memory_tool(
-    query: str,
-    limit: int = 3,
-    state: InjectedState = None,  # Inject the LangGraph state here
-    runtime: ToolRuntime = None   # Retain your runtime for the store
-) -> str:
-    """Accepts a SINGLE string query to search the sender's history. 
-    Execute this tool multiple times if you need to search for different facts.
-    """
-    
-    # 1. Pull values directly from the injected graph state safely
-    if not state:
-        return "Error: Graph state injection failed entirely."
-        
-    active_user = state.get("user_id")
-    sender_email = state.get("sender_email_id") 
-    
-    if not sender_email:
-        return "Error: Cannot isolate history. Active sender_email_id is missing from state context."
-
-    metadata_filter = {
-        "content.receiver_email_id": sender_email
-    }
-    
-    # 2. Access the BaseStore directly through runtime.store
-    if not runtime or not runtime.store:
-        return "Error: Runtime store context is missing."
-        
-    results = runtime.store.search(
-        namespace=("email", active_user, "collection"),
-        query=query,
-        filter=metadata_filter,
-        limit=limit
+search_memory_tool = create_search_memory_tool(
+    namespace=(
+        "emails", 
+        "{user_id}", 
+        "{sender_email_id}" 
     )
-    
-    if not results:
-        return f"No prior email context found specifically for sender: {sender_email}."
-        
-    formatted_memories = []
-    for item in results:
-        val = item.value
-        formatted_memories.append(
-            f"--- Past Interaction Summary ---\n"
-            f"Sender: {val.get('user_email_id')}\n"
-            f"Receiver: {val.get('receiver_email_id')}\n"
-            f"Context Summary: {val.get('summary')}\n"
-        )
-        
-    return "\n".join(formatted_memories)
+)
 
 
 
@@ -70,4 +27,4 @@ def give_previous_context(memory_summary: str) -> str:
     """
     return memory_summary
 
-context_agent_tools=[search_sender_memory_tool,give_previous_context]
+context_agent_tools=[search_memory_tool,give_previous_context]