Update orchestrator.py

Fix case when USE_DEFAULT_AGENT_IF_NONE_IDENTIFIED is False and the AgentResponse.output didn't follow the correct structure
awslabs · Oct 8, 2024 · 87cb226 · 87cb226
1 parent 00d9624
commit 87cb226
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/src/multi_agent_orchestrator/orchestrator.py b/python/src/multi_agent_orchestrator/orchestrator.py
@@ -143,7 +143,8 @@ async def route_request(self,
                                                    user_id,
                                                    session_id,
                                                    additional_params),
-                    output= self.config.NO_SELECTED_AGENT_MESSAGE,
+                    output= ConversationMessage(role=ParticipantRole.ASSISTANT.value,
+                                                content=[{'text': self.config.NO_SELECTED_AGENT_MESSAGE}]),
                     streaming=False
                 )