From 9690c09cec16be2254a2cf1415972710b66d960e Mon Sep 17 00:00:00 2001 From: cthomas Date: Sun, 2 Mar 2025 19:05:26 -0800 Subject: [PATCH] fix: re-add backwards compatibility for tool rules (#1180) --- letta/helpers/converters.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/letta/helpers/converters.py b/letta/helpers/converters.py index f65d2653..69e6f91f 100644 --- a/letta/helpers/converters.py +++ b/letta/helpers/converters.py @@ -78,11 +78,11 @@ def deserialize_tool_rule(data: Dict) -> Union[ChildToolRule, InitToolRule, Term """Deserialize a dictionary to the appropriate ToolRule subclass based on 'type'.""" rule_type = ToolRuleType(data.get("type")) - if rule_type == ToolRuleType.run_first: + if rule_type == ToolRuleType.run_first or rule_type == ToolRuleType.InitToolRule: return InitToolRule(**data) - elif rule_type == ToolRuleType.exit_loop: + elif rule_type == ToolRuleType.exit_loop or rule_type == ToolRuleType.TerminalToolRule: return TerminalToolRule(**data) - elif rule_type == ToolRuleType.constrain_child_tools: + elif rule_type == ToolRuleType.constrain_child_tools or rule_type == ToolRuleType.ToolRule: return ChildToolRule(**data) elif rule_type == ToolRuleType.conditional: return ConditionalToolRule(**data)