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: