From ba242c04cccabb5d06043b9fa5027b68c10932ee Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Tue, 7 Jan 2025 10:06:28 -0800 Subject: [PATCH] remove memory from available tools to agent --- .../inline/agents/meta_reference/agent_instance.py | 6 +++++- 1 file changed, 5 insertions(+), 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 cea4146e9..ceb764ffe 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -482,7 +482,11 @@ class ChatAgent(ShieldRunnerMixin): async for chunk in await self.inference_api.chat_completion( self.agent_config.model, input_messages, - tools=[tool for tool in tool_defs.values()], + tools=[ + tool + for tool in tool_defs.values() + if tool.tool_name != "memory" + ], tool_prompt_format=self.agent_config.tool_prompt_format, stream=True, sampling_params=sampling_params,