Update orchestrator.py

return streaming based on selected agent streaming configuration
awslabs · Oct 8, 2024 · c64babf · c64babf
1 parent 697226d
commit c64babf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/src/multi_agent_orchestrator/orchestrator.py b/python/src/multi_agent_orchestrator/orchestrator.py
@@ -185,7 +185,7 @@ async def route_request(self,
             return AgentResponse(
                     metadata=metadata,
                     output=agent_response,
-                    streaming=False
+                    streaming=classifier_result.selected_agent.is_streaming_enabled()
                 )
 
         except Exception as error: