Agents: Avoid calling tools that haven't been explicitly enabled

meta-llama · Dec 16, 2024 · fd021c3 · fd021c3
1 parent 20383bf
commit fd021c3
Showing 1 changed file with 1 addition and 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
@@ -535,7 +535,7 @@ async def _run(
                     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