From 0dd56fe7ec9551e1c97fd2132549ca47a53a50da Mon Sep 17 00:00:00 2001 From: Aidan Do Date: Mon, 16 Dec 2024 11:55:50 +1100 Subject: [PATCH] Approach #2 --- .../providers/inline/agents/meta_reference/agent_instance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py index b403b9203..0f252205a 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -535,7 +535,7 @@ class ChatAgent(ShieldRunnerMixin): tool_call = message.tool_calls[0] name = tool_call.tool_name - if not isinstance(name, BuiltinTool): + if not isinstance(name, BuiltinTool) or name not in enabled_tools: yield message return