Skip to content

Commit 015ee2a

Browse files
committed
One more rename
1 parent 52dfed2 commit 015ee2a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

agent_memory_server/mcp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
from agent_memory_server.api import (
99
create_long_term_memory as core_create_long_term_memory,
10-
get_session_memory as core_get_session_memory,
10+
get_working_memory as core_get_working_memory,
1111
memory_prompt as core_memory_prompt,
1212
put_working_memory as core_put_working_memory,
1313
search_long_term_memory as core_search_long_term_memory,
@@ -741,4 +741,4 @@ async def get_working_memory(
741741
"""
742742
Get working memory for a session. This works like the GET /sessions/{id}/memory API endpoint.
743743
"""
744-
return await core_get_session_memory(session_id=session_id)
744+
return await core_get_working_memory(session_id=session_id)

tests/test_mcp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ async def test_create_long_term_memory(self, session, mcp_test_setup):
5656
)
5757
assert isinstance(results, CallToolResult)
5858
assert results.content[0].type == "text"
59-
assert results.content[0].text == '{"status": "ok"}'
59+
assert results.content[0].text == '{\n "status": "ok"\n}'
6060

6161
@pytest.mark.asyncio
6262
async def test_search_memory(self, session, mcp_test_setup):

0 commit comments

Comments
 (0)