added support for sessionState in Agents for Amazon Bedrock

awslabs · Nov 20, 2024 · 3156653 · 3156653
1 parent f93fa68
commit 3156653
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/python/src/multi_agent_orchestrator/agents/amazon_bedrock_agent.py b/python/src/multi_agent_orchestrator/agents/amazon_bedrock_agent.py
@@ -59,12 +59,18 @@ async def process_request(
         :param additional_params: Optional additional parameters as key-value pairs.
         :return: A ConversationMessage object containing the agent's response.
         """
+
+        session_state = {}
+        if (additional_params and 'sessionState' in additional_params):
+            session_state = additional_params['sessionState']
         try:
+            print(session_state)
             response = self.client.invoke_agent(
                 agentId=self.agent_id,
                 agentAliasId=self.agent_alias_id,
                 sessionId=session_id,
-                inputText=input_text
+                inputText=input_text,
+                sessionState=session_state
             )
 
             completion = ""