From c64babf80677c684a9a6ae01340a22ea90ee8cd3 Mon Sep 17 00:00:00 2001
From: Anthony Bernabeu <64135631+brnaba-aws@users.noreply.github.com>
Date: Tue, 8 Oct 2024 14:00:44 +0200
Subject: [PATCH] Update orchestrator.py

return streaming based on selected agent streaming configuration
---
 python/src/multi_agent_orchestrator/orchestrator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/src/multi_agent_orchestrator/orchestrator.py b/python/src/multi_agent_orchestrator/orchestrator.py
index 72b8114..063a749 100644
--- 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: