Spaces:
Runtime error
Runtime error
File size: 2,068 Bytes
015bec7 5c83bd9 ec9cdad 5c83bd9 015bec7 5c83bd9 c6421b9 015bec7 a590be9 015bec7 5c83bd9 015bec7 5c83bd9 015bec7 5c83bd9 015bec7 471c5c3 015bec7 5c83bd9 015bec7 c6421b9 015bec7 c6421b9 873e08b 5c83bd9 c6421b9 a590be9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | from typing import Any
# 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
from langgraph.store.base import BaseStore
from typing import Literal
# Correct import path
@tool
def search_sender_memory_tool(
query: str,
limit: int = 3,
runtime: ToolRuntime = None # Inject everything natively here
) -> 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 from graph state
active_user = runtime.state.get("user_id")
sender_email = runtime.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
results = runtime.store.search(
namespace=("email", active_user, "collection"),
query=query,
filter=metadata_filter,
limit=limit
)
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)
@tool
def give_previous_context(memory_summary: str) -> str:
"""
Args:
memory_summary: Structured summary containing sender identity,
past context, new facts stored, and suggested tone.
"""
return memory_summary
context_agent_tools=[search_sender_memory_tool,give_previous_context] |