diff --git a/llama_stack/providers/impls/meta_reference/agents/agent_instance.py b/llama_stack/providers/impls/meta_reference/agents/agent_instance.py index 51ee8621f..e27f30d88 100644 --- a/llama_stack/providers/impls/meta_reference/agents/agent_instance.py +++ b/llama_stack/providers/impls/meta_reference/agents/agent_instance.py @@ -392,7 +392,7 @@ class ChatAgent(ShieldRunnerMixin): input_messages, tools=self._get_tools(), tool_prompt_format=self.agent_config.tool_prompt_format, - stream=True, + stream=stream, sampling_params=sampling_params, ): event = chunk.event