Spaces:
Running
Running
Download src/agent/nodes/memory_agent.py from saibalajiomg/customercore: direct link, hf CLI and curl.
- Browser
- Download file 636 Bytes
-
https://huggingface.co/spaces/saibalajiomg/customercore/resolve/main/src/agent/nodes/memory_agent.py
- Command line
-
hf download hf://spaces/saibalajiomg/customercore/src/agent/nodes/memory_agent.py
-
curl -L -o memory_agent.py https://huggingface.co/spaces/saibalajiomg/customercore/resolve/main/src/agent/nodes/memory_agent.py
636 Bytes
| from src.agent.state import AgentState | |
| from src.agent.memory import recall_memories | |
| import structlog | |
| log = structlog.get_logger() | |
| def memory_agent_node(state: AgentState) -> AgentState: | |
| ticket = state["ticket"] | |
| customer_id = ticket["customer_id"] | |
| tenant_id = ticket["tenant_id"] | |
| body = ticket["body"] | |
| # Recall up to 3 relevant long-term memories | |
| memories = recall_memories(customer_id, tenant_id, query=body, limit=3) | |
| log.info("memory_agent_done", customer_id=customer_id, memories_recalled=len(memories)) | |
| return { | |
| "recalled_memories": memories, | |
| "current_step": "memory_agent", | |
| } | |