From 6b9a6dfa7325e98a6aa806ed644aec2ba4395b88 Mon Sep 17 00:00:00 2001 From: Cameron Pfiffer Date: Mon, 9 Jun 2025 17:43:41 -0700 Subject: [PATCH] Update user block argument classes for consistency - Renamed AttachUserBlockArgs and DetachUserBlockArgs to AttachUserBlocksArgs and DetachUserBlocksArgs in register_tools.py. - Updated the TOOL_CONFIGS to reflect the new argument class names. This change improves clarity and maintains uniformity across the codebase. --- register_tools.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/register_tools.py b/register_tools.py index dde353c..94a8a41 100755 --- a/register_tools.py +++ b/register_tools.py @@ -23,7 +23,7 @@ from tools.functions import ( from tools.search import SearchArgs from tools.post import PostArgs from tools.feed import FeedArgs -from tools.blocks import AttachUserBlockArgs, DetachUserBlockArgs #, UpdateUserBlockArgs +from tools.blocks import AttachUserBlocksArgs, DetachUserBlocksArgs #, UpdateUserBlockArgs load_dotenv() logging.basicConfig(level=logging.INFO) @@ -53,13 +53,13 @@ TOOL_CONFIGS = [ }, { "func": attach_user_blocks, - "args_schema": AttachUserBlockArgs, + "args_schema": AttachUserBlocksArgs, "description": "Attach user-specific memory blocks to the agent. Creates blocks if they don't exist.", "tags": ["memory", "blocks", "user"] }, { "func": detach_user_blocks, - "args_schema": DetachUserBlockArgs, + "args_schema": DetachUserBlocksArgs, "description": "Detach user-specific memory blocks from the agent. Blocks are preserved for later use.", "tags": ["memory", "blocks", "user"] }, -- 2.51.2