phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 09:21:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Aidan Do 2024-12-21 22:40:29 +11:00
parent bae197c37e
commit 8119363e56
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/agents/meta_reference/agent_instance.py
View file

@ -539,7 +539,7 @@ class ChatAgent(ShieldRunnerMixin):
tool_call = message.tool_calls[0] tool_call = message.tool_calls[0]
name = tool_call.tool_name name = tool_call.tool_name
if not isinstance(name, BuiltinTool): if not isinstance(name, BuiltinTool) or name not in enabled_tools:
yield message yield message
return return