From 9abe136646c81427c4268da6a39f027bc145e8b7 Mon Sep 17 00:00:00 2001 From: CREDO23 Date: Fri, 20 Feb 2026 19:22:22 +0200 Subject: [PATCH] register google drive tools in registry --- .../app/agents/new_chat/tools/registry.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/surfsense_backend/app/agents/new_chat/tools/registry.py b/surfsense_backend/app/agents/new_chat/tools/registry.py index 4c6345bc3..f2557eec0 100644 --- a/surfsense_backend/app/agents/new_chat/tools/registry.py +++ b/surfsense_backend/app/agents/new_chat/tools/registry.py @@ -55,6 +55,10 @@ from .linear import ( ) from .link_preview import create_link_preview_tool from .mcp_tool import load_mcp_tools +from .google_drive import ( + create_create_google_drive_file_tool, + create_trash_google_drive_file_tool, +) from .notion import ( create_create_notion_page_tool, create_delete_notion_page_tool, @@ -292,6 +296,29 @@ BUILTIN_TOOLS: list[ToolDefinition] = [ ), requires=["db_session", "search_space_id", "user_id"], ), + # ========================================================================= + # GOOGLE DRIVE TOOLS - create files, trash files + # ========================================================================= + ToolDefinition( + name="create_google_drive_file", + description="Create a new Google Doc or Google Sheet in Google Drive", + factory=lambda deps: create_create_google_drive_file_tool( + db_session=deps["db_session"], + search_space_id=deps["search_space_id"], + user_id=deps["user_id"], + ), + requires=["db_session", "search_space_id", "user_id"], + ), + ToolDefinition( + name="trash_google_drive_file", + description="Move an indexed Google Drive file to trash", + factory=lambda deps: create_trash_google_drive_file_tool( + db_session=deps["db_session"], + search_space_id=deps["search_space_id"], + user_id=deps["user_id"], + ), + requires=["db_session", "search_space_id", "user_id"], + ), ]