From b0325a02b92dbc3207c9243c4edf8bd4e284525c Mon Sep 17 00:00:00 2001
From: William Fu-Hinthorn <13333726+hinthornw@users.noreply.github.com>
Date: Mon, 28 Oct 2024 11:18:06 -0700
Subject: [PATCH] Use dataclass

---
 src/memory_agent/graph.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/memory_agent/graph.py b/src/memory_agent/graph.py
index 09c9ff4..ce134e4 100644
--- a/src/memory_agent/graph.py
+++ b/src/memory_agent/graph.py
@@ -24,7 +24,7 @@ async def call_model(state: State, config: RunnableConfig, *, store: BaseStore)
 
     # Retrieve the most recent memories for context
     memories = await store.asearch(
-        ("memories", config["configurable"]["user_id"]), limit=10
+        ("memories", configurable.user_id), limit=10
     )
 
     # Format memories for inclusion in the prompt