diff --git a/src/bedrock_agentcore/memory/integrations/strands/session_manager.py b/src/bedrock_agentcore/memory/integrations/strands/session_manager.py index 1e2ef14..b51cc90 100644 --- a/src/bedrock_agentcore/memory/integrations/strands/session_manager.py +++ b/src/bedrock_agentcore/memory/integrations/strands/session_manager.py @@ -641,7 +641,7 @@ def retrieve_for_namespace(namespace: str, retrieval_config: RetrievalConfig): if all_context: context_text = "\n".join(all_context) event.agent.messages[-1]["content"].insert( - 0, {"text": f"{context_text}"} + 0, {"text": f"{context_text}"} ) logger.info("Retrieved %s customer context items", len(all_context)) diff --git a/tests/bedrock_agentcore/memory/integrations/strands/test_agentcore_memory_session_manager.py b/tests/bedrock_agentcore/memory/integrations/strands/test_agentcore_memory_session_manager.py index 1eafd2d..ba00153 100644 --- a/tests/bedrock_agentcore/memory/integrations/strands/test_agentcore_memory_session_manager.py +++ b/tests/bedrock_agentcore/memory/integrations/strands/test_agentcore_memory_session_manager.py @@ -1937,7 +1937,7 @@ def test_retrieve_customer_context_does_not_append_assistant_message( # Memory prepended, original query remains last content = mock_agent.messages[0]["content"] assert len(content) == 2 - assert "" in content[0]["text"] + assert "" in content[0]["text"] assert content[1]["text"] == "What are my preferences?" def test_retrieve_customer_context_no_assistant_message_multi_turn( @@ -1982,5 +1982,5 @@ def test_retrieve_customer_context_no_assistant_message_multi_turn( # Memory injected into last user message content = mock_agent.messages[-1]["content"] assert len(content) == 2 - assert "" in content[0]["text"] + assert "" in content[0]["text"] assert content[1]["text"] == "What do I like to eat?" diff --git a/tests_integ/memory/integrations/test_session_manager.py b/tests_integ/memory/integrations/test_session_manager.py index d6ce766..3a93d12 100644 --- a/tests_integ/memory/integrations/test_session_manager.py +++ b/tests_integ/memory/integrations/test_session_manager.py @@ -151,7 +151,7 @@ def test_session_manager_with_retrieval_config_adds_context(self, test_memory_lt response2 = agent("What do I like to eat?") assert response2 is not None assert "sushi" in str(agent.messages) - assert "" in str(agent.messages) + assert "" in str(agent.messages) def test_multiple_namespace_retrieval_config(self, test_memory_ltm): """Test session manager with multiple namespace retrieval configurations.""" @@ -182,7 +182,7 @@ def test_multiple_namespace_retrieval_config(self, test_memory_ltm): response2 = agent("What do I like to eat?") assert response2 is not None assert "sushi" in str(agent.messages) - assert "" in str(agent.messages) + assert "" in str(agent.messages) def test_session_manager_error_handling(self): """Test session manager error handling with invalid configuration."""