Update bedrock_llm_agent.py

implemented is_streaming_enabled for BedrockLLMAgent
awslabs · Oct 8, 2024 · 697226d · 697226d
1 parent 7683903
commit 697226d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/src/multi_agent_orchestrator/agents/bedrock_llm_agent.py b/python/src/multi_agent_orchestrator/agents/bedrock_llm_agent.py
@@ -90,6 +90,9 @@ def __init__(self, options: BedrockLLMAgentOptions):
                 options.custom_system_prompt.get('variables')
             )
 
+    def is_streaming_enabled(self) -> bool:
+        return self.streaming is True
+
     async def process_request(
         self,
         input_text: str,