@@ -105,10 +105,10 @@ working_memory = WorkingMemory(
105105)
106106
107107# Store working memory
108- response = await client.put_session_memory (" user-session-123" , working_memory)
108+ response = await client.put_working_memory (" user-session-123" , working_memory)
109109
110110# Retrieve working memory
111- memory = await client.get_session_memory (" user-session-123" )
111+ memory = await client.get_working_memory (" user-session-123" )
112112
113113# Convenience method for data storage
114114await client.set_working_memory_data(
@@ -204,7 +204,7 @@ await client.update_working_memory_data(
204204 merge_strategy = " deep_merge" # "replace", "merge", or "deep_merge"
205205)
206206
207- # Append messages efficiently
207+ # Append messages
208208new_messages = [
209209 MemoryMessage(role = " user" , content = " What's the weather?" ),
210210 MemoryMessage(role = " assistant" , content = " It's sunny today!" )
@@ -225,7 +225,7 @@ from agent_memory_client.filters import (
225225)
226226from datetime import datetime, timezone
227227
228- # Complex search with multiple filters
228+ # Complex search with filters
229229results = await client.search_long_term_memory(
230230 text = " machine learning" ,
231231 session_id = SessionId(in_ = [" session-1" , " session-2" ]),
@@ -251,7 +251,7 @@ from agent_memory_client.exceptions import (
251251)
252252
253253try :
254- memory = await client.get_session_memory (" nonexistent-session" )
254+ memory = await client.get_working_memory (" nonexistent-session" )
255255except MemoryNotFoundError:
256256 print (" Session not found" )
257257except MemoryServerError as e:
@@ -286,12 +286,12 @@ pytest --cov=agent_memory_client
286286### Code Quality
287287
288288``` bash
289- # Format code
290- black agent_memory_client/
291-
292289# Lint code
293290ruff check agent_memory_client/
294291
292+ # Format code
293+ ruff format agent_memory_client/
294+
295295# Type checking
296296mypy agent_memory_client/
297297```
0 commit comments