From 3e62fcfcca4c11b57cfef374a0e9d00b5420e7d8 Mon Sep 17 00:00:00 2001
From: Cameron Pfiffer <cameron@pfiffer.org>
Date: Mon, 9 Jun 2025 17:43:32 -0700
Subject: [PATCH] Refactor user block argument classes for consistency

- Renamed AttachUserBlockArgs and DetachUserBlockArgs to AttachUserBlocksArgs and DetachUserBlocksArgs for uniformity.
- Updated corresponding tool classes to reflect the new naming conventions.

This change enhances clarity and maintains consistency across the codebase.
---
 tools/__init__.py |  6 +++---
 tools/blocks.py   | 12 ++++++------
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/tools/__init__.py b/tools/__init__.py
index a53214a..29d0e53 100644
--- a/tools/__init__.py
+++ b/tools/__init__.py
@@ -12,7 +12,7 @@ from .functions import (
 from .search import SearchArgs
 from .post import PostArgs
 from .feed import FeedArgs
-from .blocks import AttachUserBlockArgs, DetachUserBlockArgs, UpdateUserBlockArgs
+from .blocks import AttachUserBlocksArgs, DetachUserBlocksArgs, UpdateUserBlockArgs
 
 __all__ = [
     # Functions
@@ -26,7 +26,7 @@ __all__ = [
     "SearchArgs",
     "PostArgs",
     "FeedArgs",
-    "AttachUserBlockArgs",
-    "DetachUserBlockArgs",
+    "AttachUserBlocksArgs",
+    "DetachUserBlocksArgs",
     "UpdateUserBlockArgs",
 ]
\ No newline at end of file
diff --git a/tools/blocks.py b/tools/blocks.py
index 20f50d9..ef3e8a0 100644
--- a/tools/blocks.py
+++ b/tools/blocks.py
@@ -9,13 +9,13 @@ from letta_client import Letta
 logger = logging.getLogger(__name__)
 
 
-class AttachUserBlockArgs(BaseModel):
+class AttachUserBlocksArgs(BaseModel):
     handles: List[str] = Field(..., description="List of user Bluesky handles (e.g., ['user1.bsky.social', 'user2.bsky.social'])")
 
 
-class AttachUserBlockTool(BaseTool):
+class AttachUserBlocksTool(BaseTool):
     name: str = "attach_user_blocks"
-    args_schema: Type[BaseModel] = AttachUserBlockArgs
+    args_schema: Type[BaseModel] = AttachUserBlocksArgs
     description: str = "Attach user-specific memory blocks to the agent. Creates blocks if they don't exist."
     tags: List[str] = ["memory", "blocks", "user"]
 
@@ -77,13 +77,13 @@ class AttachUserBlockTool(BaseTool):
             raise e
 
 
-class DetachUserBlockArgs(BaseModel):
+class DetachUserBlocksArgs(BaseModel):
     handles: List[str] = Field(..., description="List of user Bluesky handles (e.g., ['user1.bsky.social', 'user2.bsky.social'])")
 
 
-class DetachUserBlockTool(BaseTool):
+class DetachUserBlocksTool(BaseTool):
     name: str = "detach_user_blocks"
-    args_schema: Type[BaseModel] = DetachUserBlockArgs
+    args_schema: Type[BaseModel] = DetachUserBlocksArgs
     description: str = "Detach user-specific memory blocks from the agent. Blocks are preserved for later use."
     tags: List[str] = ["memory", "blocks", "user"]
 
-- 
2.51.2