From db0b2a60c1f5038ea72f847fa7f56c7ff215d028 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Mon, 6 Jan 2025 14:49:55 -0800 Subject: [PATCH] remove breakpoints --- .../providers/inline/agents/meta_reference/agent_instance.py | 1 - tests/client-sdk/agents/test_agents.py | 3 +-- 2 files changed, 1 insertion(+), 3 deletions(-) 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 ac49a06ce..e4ebb3011 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -758,7 +758,6 @@ class ChatAgent(ShieldRunnerMixin): input_messages: List[Message], tool_defs: Dict[str, ToolDefinition], ) -> None: - breakpoint() memory_tool = tool_defs.get("memory", None) code_interpreter_tool = tool_defs.get(BuiltinTool.code_interpreter, None) if documents: diff --git a/tests/client-sdk/agents/test_agents.py b/tests/client-sdk/agents/test_agents.py index a8e06b7a2..ca39ca03f 100644 --- a/tests/client-sdk/agents/test_agents.py +++ b/tests/client-sdk/agents/test_agents.py @@ -211,7 +211,7 @@ def test_code_execution(llama_stack_client): tools=[ "code_interpreter", ], - tool_choice="required", + tool_choice="auto", input_shields=[], output_shields=[], enable_session_persistence=False, @@ -242,7 +242,6 @@ def test_code_execution(llama_stack_client): ) logs = [str(log) for log in EventLogger().log(response) if log is not None] logs_str = "".join(logs) - breakpoint() print(logs_str)