Use dataclass

This commit is contained in:
William Fu-Hinthorn
2024-10-28 11:18:06 -07:00
parent 8d62c797ce
commit b0325a02b9
+1 -1
View File
@@ -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