Skip to content

Commit 7df8315

Browse files
author
gitikavj
committed
revert: rename retrieved_memory tag back to user_context
1 parent 1bd22b7 commit 7df8315

3 files changed

Lines changed: 5 additions & 5 deletions

File tree

src/bedrock_agentcore/memory/integrations/strands/session_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -641,7 +641,7 @@ def retrieve_for_namespace(namespace: str, retrieval_config: RetrievalConfig):
641641
if all_context:
642642
context_text = "\n".join(all_context)
643643
event.agent.messages[-1]["content"].insert(
644-
0, {"text": f"<retrieved_memory>{context_text}</retrieved_memory>"}
644+
0, {"text": f"<user_context>{context_text}</user_context>"}
645645
)
646646
logger.info("Retrieved %s customer context items", len(all_context))
647647

tests/bedrock_agentcore/memory/integrations/strands/test_agentcore_memory_session_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1937,7 +1937,7 @@ def test_retrieve_customer_context_does_not_append_assistant_message(
19371937
# Memory prepended, original query remains last
19381938
content = mock_agent.messages[0]["content"]
19391939
assert len(content) == 2
1940-
assert "<retrieved_memory>" in content[0]["text"]
1940+
assert "<user_context>" in content[0]["text"]
19411941
assert content[1]["text"] == "What are my preferences?"
19421942

19431943
def test_retrieve_customer_context_no_assistant_message_multi_turn(
@@ -1982,5 +1982,5 @@ def test_retrieve_customer_context_no_assistant_message_multi_turn(
19821982
# Memory injected into last user message
19831983
content = mock_agent.messages[-1]["content"]
19841984
assert len(content) == 2
1985-
assert "<retrieved_memory>" in content[0]["text"]
1985+
assert "<user_context>" in content[0]["text"]
19861986
assert content[1]["text"] == "What do I like to eat?"

tests_integ/memory/integrations/test_session_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ def test_session_manager_with_retrieval_config_adds_context(self, test_memory_lt
151151
response2 = agent("What do I like to eat?")
152152
assert response2 is not None
153153
assert "sushi" in str(agent.messages)
154-
assert "<retrieved_memory>" in str(agent.messages)
154+
assert "<user_context>" in str(agent.messages)
155155

156156
def test_multiple_namespace_retrieval_config(self, test_memory_ltm):
157157
"""Test session manager with multiple namespace retrieval configurations."""
@@ -182,7 +182,7 @@ def test_multiple_namespace_retrieval_config(self, test_memory_ltm):
182182
response2 = agent("What do I like to eat?")
183183
assert response2 is not None
184184
assert "sushi" in str(agent.messages)
185-
assert "<retrieved_memory>" in str(agent.messages)
185+
assert "<user_context>" in str(agent.messages)
186186

187187
def test_session_manager_error_handling(self):
188188
"""Test session manager error handling with invalid configuration."""

0 commit comments

Comments
 (0)