From e77d0e69ee3278cf3d911a3c796759421e7cd647 Mon Sep 17 00:00:00 2001 From: Hassan El Mghari Date: Mon, 23 Sep 2024 07:46:08 -0700 Subject: [PATCH] Update agent_instance.py --- .../providers/impls/meta_reference/agents/agent_instance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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