Update test_amazon_bedrock_agent.py

fix tests
awslabs · Nov 20, 2024 · da9563f · da9563f
1 parent 8a9283f
commit da9563f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/src/tests/agents/test_amazon_bedrock_agent.py b/python/src/tests/agents/test_amazon_bedrock_agent.py
@@ -50,7 +50,8 @@ async def test_process_request_success(bedrock_agent):
         agentId='test_agent_id',
         agentAliasId='test_agent_alias_id',
         sessionId='test_session',
-        inputText='Test input'
+        inputText='Test input',
+        sessionState={}
     )
 
 @pytest.mark.asyncio
@@ -116,4 +117,4 @@ async def test_process_request_with_additional_params(bedrock_agent):
 
     assert isinstance(result, ConversationMessage)
     assert result.role == ParticipantRole.ASSISTANT.value
-    assert result.content == [{"text": "Response with additional params"}]
+    assert result.content == [{"text": "Response with additional params"}]