chore; linting

surfsense_backend/app/agents/new_chat/tools/google_drive/create_file.py

@@ -58,7 +58,9 @@ def create_create_google_drive_file_tool(
             - "Create a Google Doc called 'Meeting Notes'"
             - "Create a spreadsheet named 'Budget 2026' with some sample data"
         """
-        logger.info(f"create_google_drive_file called: name='{name}', type='{file_type}'")
+        logger.info(
+            f"create_google_drive_file called: name='{name}', type='{file_type}'"
+        )
 
         if db_session is None or search_space_id is None or user_id is None:
             return {
@@ -74,7 +76,9 @@ def create_create_google_drive_file_tool(
 
         try:
             metadata_service = GoogleDriveToolMetadataService(db_session)
-            context = await metadata_service.get_creation_context(search_space_id, user_id)
+            context = await metadata_service.get_creation_context(
+                search_space_id, user_id
+            )
 
             if "error" in context:
                 logger.error(f"Failed to fetch creation context: {context['error']}")
@@ -100,8 +104,12 @@ def create_create_google_drive_file_tool(
                 }
             )
 
-            decisions_raw = approval.get("decisions", []) if isinstance(approval, dict) else []
-            decisions = decisions_raw if isinstance(decisions_raw, list) else [decisions_raw]
+            decisions_raw = (
+                approval.get("decisions", []) if isinstance(approval, dict) else []
+            )
+            decisions = (
+                decisions_raw if isinstance(decisions_raw, list) else [decisions_raw]
+            )
             decisions = [d for d in decisions if isinstance(d, dict)]
             if not decisions:
                 logger.warning("No approval decision received")
@@ -183,7 +191,9 @@ def create_create_google_drive_file_tool(
             logger.info(
                 f"Creating Google Drive file: name='{final_name}', type='{final_file_type}', connector={actual_connector_id}"
             )
-            client = GoogleDriveClient(session=db_session, connector_id=actual_connector_id)
+            client = GoogleDriveClient(
+                session=db_session, connector_id=actual_connector_id
+            )
             try:
                 created = await client.create_file(
                     name=final_name,
@@ -203,7 +213,9 @@ def create_create_google_drive_file_tool(
                     }
                 raise
 
-            logger.info(f"Google Drive file created: id={created.get('id')}, name={created.get('name')}")
+            logger.info(
+                f"Google Drive file created: id={created.get('id')}, name={created.get('name')}"
+            )
             return {
                 "status": "success",
                 "file_id": created.get("id"),

surfsense_backend/app/agents/new_chat/tools/google_drive/trash_file.py

@@ -52,7 +52,9 @@ def create_delete_google_drive_file_tool(
             - "Delete the 'Meeting Notes' file from Google Drive"
             - "Trash the 'Old Budget' spreadsheet"
         """
-        logger.info(f"delete_google_drive_file called: file_name='{file_name}', delete_from_kb={delete_from_kb}")
+        logger.info(
+            f"delete_google_drive_file called: file_name='{file_name}', delete_from_kb={delete_from_kb}"
+        )
 
         if db_session is None or search_space_id is None or user_id is None:
             return {
@@ -103,8 +105,12 @@ def create_delete_google_drive_file_tool(
                 }
             )
 
-            decisions_raw = approval.get("decisions", []) if isinstance(approval, dict) else []
-            decisions = decisions_raw if isinstance(decisions_raw, list) else [decisions_raw]
+            decisions_raw = (
+                approval.get("decisions", []) if isinstance(approval, dict) else []
+            )
+            decisions = (
+                decisions_raw if isinstance(decisions_raw, list) else [decisions_raw]
+            )
             decisions = [d for d in decisions if isinstance(d, dict)]
             if not decisions:
                 logger.warning("No approval decision received")
@@ -130,11 +136,16 @@ def create_delete_google_drive_file_tool(
                 final_params = decision["args"]
 
             final_file_id = final_params.get("file_id", file_id)
-            final_connector_id = final_params.get("connector_id", connector_id_from_context)
+            final_connector_id = final_params.get(
+                "connector_id", connector_id_from_context
+            )
             final_delete_from_kb = final_params.get("delete_from_kb", delete_from_kb)
 
             if not final_connector_id:
-                return {"status": "error", "message": "No connector found for this file."}
+                return {
+                    "status": "error",
+                    "message": "No connector found for this file.",
+                }
 
             from sqlalchemy.future import select
 
@@ -174,7 +185,9 @@ def create_delete_google_drive_file_tool(
                     }
                 raise
 
-            logger.info(f"Google Drive file deleted (moved to trash): file_id={final_file_id}")
+            logger.info(
+                f"Google Drive file deleted (moved to trash): file_id={final_file_id}"
+            )
 
             trash_result: dict[str, Any] = {
                 "status": "success",
@@ -195,7 +208,9 @@ def create_delete_google_drive_file_tool(
                         await db_session.delete(document)
                         await db_session.commit()
                         deleted_from_kb = True
-                        logger.info(f"Deleted document {document_id} from knowledge base")
+                        logger.info(
+                            f"Deleted document {document_id} from knowledge base"
+                        )
                     else:
                         logger.warning(f"Document {document_id} not found in KB")
                 except Exception as e:

surfsense_backend/app/agents/new_chat/tools/registry.py

@@ -47,6 +47,10 @@ from app.db import ChatVisibility
 
 from .display_image import create_display_image_tool
 from .generate_image import create_generate_image_tool
+from .google_drive import (
+    create_create_google_drive_file_tool,
+    create_delete_google_drive_file_tool,
+)
 from .knowledge_base import create_search_knowledge_base_tool
 from .linear import (
     create_create_linear_issue_tool,
@@ -55,10 +59,6 @@ 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_delete_google_drive_file_tool,
-)
 from .notion import (
     create_create_notion_page_tool,
     create_delete_notion_page_tool,