From 84dbaa4511fd453791c61472215609704adb8570 Mon Sep 17 00:00:00 2001 From: Kevin Lin Date: Wed, 9 Apr 2025 09:36:18 -0700 Subject: [PATCH] chore: temporarily disable tool rules on sleeptime that are causing errors (#1632) --- letta/services/agent_manager.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/letta/services/agent_manager.py b/letta/services/agent_manager.py index d2f4e88b..734332b7 100644 --- a/letta/services/agent_manager.py +++ b/letta/services/agent_manager.py @@ -42,7 +42,6 @@ from letta.schemas.message import MessageCreate, MessageUpdate from letta.schemas.passage import Passage as PydanticPassage from letta.schemas.source import Source as PydanticSource from letta.schemas.tool import Tool as PydanticTool -from letta.schemas.tool_rule import ChildToolRule as PydanticChildToolRule from letta.schemas.tool_rule import ContinueToolRule as PydanticContinueToolRule from letta.schemas.tool_rule import TerminalToolRule as PydanticTerminalToolRule from letta.schemas.tool_rule import ToolRule as PydanticToolRule @@ -160,9 +159,9 @@ class AgentManager: tool_rules.append(PydanticTerminalToolRule(tool_name=tool_name)) elif tool_name in BASE_TOOLS + BASE_MEMORY_TOOLS + BASE_SLEEPTIME_TOOLS: tool_rules.append(PydanticContinueToolRule(tool_name=tool_name)) - - if agent_create.agent_type == AgentType.sleeptime_agent: - tool_rules.append(PydanticChildToolRule(tool_name="view_core_memory_with_line_numbers", children=["core_memory_insert"])) + # we may want to add additional rules for sleeptime agents + # if agent_create.agent_type == AgentType.sleeptime_agent: + # tool_rules.append(PydanticChildToolRule(tool_name="view_core_memory_with_line_numbers", children=["core_memory_insert"])) # if custom rules, check tool rules are valid if agent_create.tool_rules: