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

meta-llama · Dec 21, 2024 · 8119363 · 8119363
1 parent bae197c
commit 8119363
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
@@ -539,7 +539,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