From 98dbd5b7710123b69a39491ec5632be97f3101a2 Mon Sep 17 00:00:00 2001 From: Shubham Naik Date: Thu, 10 Apr 2025 11:11:27 -0700 Subject: [PATCH] Shub/let 1771 variables component (#1650) Co-authored-by: Shubham Naik --- letta/server/server.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/letta/server/server.py b/letta/server/server.py index 712a4f93..e6fd6ed5 100644 --- a/letta/server/server.py +++ b/letta/server/server.py @@ -1259,16 +1259,16 @@ class SyncServer(Server): if tool_source_type is not None and tool_source_type != "python": raise ValueError("Only Python source code is supported at this time") - # NOTE: we're creating a floating Tool object and NOT persisting to DB - tool = Tool( - name=tool_name, - source_code=tool_source, - args_json_schema=tool_args_json_schema, - ) - # If tools_json_schema is explicitly passed in, override it on the created Tool object if tool_json_schema: - tool.json_schema = tool_json_schema + tool = Tool(name=tool_name, source_code=tool_source, json_schema=tool_json_schema) + else: + # NOTE: we're creating a floating Tool object and NOT persisting to DB + tool = Tool( + name=tool_name, + source_code=tool_source, + args_json_schema=tool_args_json_schema, + ) assert tool.name is not None, "Failed to create tool object"