Fix deleting session episodes v1 (MemMachine#532)

edwinyyyu · jealous · web-flow · commit dd9d649fd8b0 · 2025-11-17T13:19:28.000-08:00
* Fix deleting session episodes v1

Signed-off-by: Edwin Yu &lt;edwinyyyu@gmail.com&gt;

* Fix style

Signed-off-by: Edwin Yu &lt;edwinyyyu@gmail.com&gt;

---------

Signed-off-by: Edwin Yu &lt;edwinyyyu@gmail.com&gt;
Co-authored-by: Cedric Zhuang &lt;cedric.zhuang@memverge.com&gt;
diff --git a/src/memmachine/episodic_memory/long_term_memory/long_term_memory.py b/src/memmachine/episodic_memory/long_term_memory/long_term_memory.py
@@ -67,6 +67,7 @@ class LongTermMemory:
 
     def __init__(self, params: LongTermMemoryParams):
         self._group_id = params.group_id
+        self._session_id = params.session_id
         # Note: Things look a bit weird during refactor...
         # Internal session_id is used for external group_id. This is intentional.
         self._declarative_memory = DeclarativeMemory(
@@ -157,6 +158,6 @@ async def clear(self):
     async def forget_session(self):
         await self._declarative_memory.forget_filtered_episodes(
             property_filter={
-                "session_id": self._memory_context.session_id,
+                "session_id": self._session_id,
             }
         )

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,7 @@ class LongTermMemory:`
`67`	`67`
`68`	`68`	`def __init__(self, params: LongTermMemoryParams):`
`69`	`69`	`self._group_id = params.group_id`
	`70`	`+ self._session_id = params.session_id`
`70`	`71`	`# Note: Things look a bit weird during refactor...`
`71`	`72`	`# Internal session_id is used for external group_id. This is intentional.`
`72`	`73`	`self._declarative_memory = DeclarativeMemory(`
`@@ -157,6 +158,6 @@ async def clear(self):`
`157`	`158`	`async def forget_session(self):`
`158`	`159`	`await self._declarative_memory.forget_filtered_episodes(`
`159`	`160`	`property_filter={`
`160`		`- "session_id": self._memory_context.session_id,`
	`161`	`+ "session_id": self._session_id,`
`161`	`162`	`}`
`162`	`163`	`)`