chore(api/services): apply ruff reformatting (#7599)

Co-authored-by: -LAN- <laipz8200@outlook.com>

api/services/tools/builtin_tools_manage_service.py

@@ -20,21 +20,25 @@ logger = logging.getLogger(__name__)
 
 class BuiltinToolManageService:
     @staticmethod
-    def list_builtin_tool_provider_tools(
-        user_id: str, tenant_id: str, provider: str
-    ) -> list[UserTool]:
+    def list_builtin_tool_provider_tools(user_id: str, tenant_id: str, provider: str) -> list[UserTool]:
         """
-            list builtin tool provider tools
+        list builtin tool provider tools
         """
         provider_controller: ToolProviderController = ToolManager.get_builtin_provider(provider)
         tools = provider_controller.get_tools()
 
-        tool_provider_configurations = ToolConfigurationManager(tenant_id=tenant_id, provider_controller=provider_controller)
+        tool_provider_configurations = ToolConfigurationManager(
+            tenant_id=tenant_id, provider_controller=provider_controller
+        )
         # check if user has added the provider
-        builtin_provider: BuiltinToolProvider = db.session.query(BuiltinToolProvider).filter(
-            BuiltinToolProvider.tenant_id == tenant_id,
-            BuiltinToolProvider.provider == provider,
-        ).first()
+        builtin_provider: BuiltinToolProvider = (
+            db.session.query(BuiltinToolProvider)
+            .filter(
+                BuiltinToolProvider.tenant_id == tenant_id,
+                BuiltinToolProvider.provider == provider,
+            )
+            .first()
+        )
 
         credentials = {}
         if builtin_provider is not None:
@@ -44,47 +48,47 @@ class BuiltinToolManageService:
 
         result = []
         for tool in tools:
-            result.append(ToolTransformService.tool_to_user_tool(
-                tool=tool,
-                credentials=credentials,
-                tenant_id=tenant_id,
-                labels=ToolLabelManager.get_tool_labels(provider_controller)
-            ))
+            result.append(
+                ToolTransformService.tool_to_user_tool(
+                    tool=tool,
+                    credentials=credentials,
+                    tenant_id=tenant_id,
+                    labels=ToolLabelManager.get_tool_labels(provider_controller),
+                )
+            )
 
         return result
 
     @staticmethod
-    def list_builtin_provider_credentials_schema(
-        provider_name
-    ):
+    def list_builtin_provider_credentials_schema(provider_name):
         """
-            list builtin provider credentials schema
+        list builtin provider credentials schema
 
-            :return: the list of tool providers
+        :return: the list of tool providers
         """
         provider = ToolManager.get_builtin_provider(provider_name)
-        return jsonable_encoder([
-            v for _, v in (provider.credentials_schema or {}).items()
-        ])
+        return jsonable_encoder([v for _, v in (provider.credentials_schema or {}).items()])
 
     @staticmethod
-    def update_builtin_tool_provider(
-        user_id: str, tenant_id: str, provider_name: str, credentials: dict
-    ):
+    def update_builtin_tool_provider(user_id: str, tenant_id: str, provider_name: str, credentials: dict):
         """
-            update builtin tool provider
+        update builtin tool provider
         """
         # get if the provider exists
-        provider: BuiltinToolProvider = db.session.query(BuiltinToolProvider).filter(
-            BuiltinToolProvider.tenant_id == tenant_id,
-            BuiltinToolProvider.provider == provider_name,
-        ).first()
+        provider: BuiltinToolProvider = (
+            db.session.query(BuiltinToolProvider)
+            .filter(
+                BuiltinToolProvider.tenant_id == tenant_id,
+                BuiltinToolProvider.provider == provider_name,
+            )
+            .first()
+        )
 
         try:
             # get provider
             provider_controller = ToolManager.get_builtin_provider(provider_name)
             if not provider_controller.need_credentials:
-                raise ValueError(f'provider {provider_name} does not need credentials')
+                raise ValueError(f"provider {provider_name} does not need credentials")
             tool_configuration = ToolConfigurationManager(tenant_id=tenant_id, provider_controller=provider_controller)
             # get original credentials if exists
             if provider is not None:
@@ -121,19 +125,21 @@ class BuiltinToolManageService:
             # delete cache
             tool_configuration.delete_tool_credentials_cache()
 
-        return {'result': 'success'}
+        return {"result": "success"}
 
     @staticmethod
-    def get_builtin_tool_provider_credentials(
-        user_id: str, tenant_id: str, provider: str
-    ):
+    def get_builtin_tool_provider_credentials(user_id: str, tenant_id: str, provider: str):
         """
-            get builtin tool provider credentials
+        get builtin tool provider credentials
         """
-        provider: BuiltinToolProvider = db.session.query(BuiltinToolProvider).filter(
-            BuiltinToolProvider.tenant_id == tenant_id,
-            BuiltinToolProvider.provider == provider,
-        ).first()
+        provider: BuiltinToolProvider = (
+            db.session.query(BuiltinToolProvider)
+            .filter(
+                BuiltinToolProvider.tenant_id == tenant_id,
+                BuiltinToolProvider.provider == provider,
+            )
+            .first()
+        )
 
         if provider is None:
             return {}
@@ -145,19 +151,21 @@ class BuiltinToolManageService:
         return credentials
 
     @staticmethod
-    def delete_builtin_tool_provider(
-        user_id: str, tenant_id: str, provider_name: str
-    ):
+    def delete_builtin_tool_provider(user_id: str, tenant_id: str, provider_name: str):
         """
-            delete tool provider
+        delete tool provider
         """
-        provider: BuiltinToolProvider = db.session.query(BuiltinToolProvider).filter(
-            BuiltinToolProvider.tenant_id == tenant_id,
-            BuiltinToolProvider.provider == provider_name,
-        ).first()
+        provider: BuiltinToolProvider = (
+            db.session.query(BuiltinToolProvider)
+            .filter(
+                BuiltinToolProvider.tenant_id == tenant_id,
+                BuiltinToolProvider.provider == provider_name,
+            )
+            .first()
+        )
 
         if provider is None:
-            raise ValueError(f'you have not added provider {provider_name}')
+            raise ValueError(f"you have not added provider {provider_name}")
 
         db.session.delete(provider)
         db.session.commit()
@@ -167,38 +175,36 @@ class BuiltinToolManageService:
         tool_configuration = ToolConfigurationManager(tenant_id=tenant_id, provider_controller=provider_controller)
         tool_configuration.delete_tool_credentials_cache()
 
-        return {'result': 'success'}
+        return {"result": "success"}
 
     @staticmethod
-    def get_builtin_tool_provider_icon(
-        provider: str
-    ):
+    def get_builtin_tool_provider_icon(provider: str):
         """
-            get tool provider icon and it's mimetype
+        get tool provider icon and it's mimetype
         """
         icon_path, mime_type = ToolManager.get_builtin_provider_icon(provider)
-        with open(icon_path, 'rb') as f:
+        with open(icon_path, "rb") as f:
             icon_bytes = f.read()
 
         return icon_bytes, mime_type
 
     @staticmethod
-    def list_builtin_tools(
-        user_id: str, tenant_id: str
-    ) -> list[UserToolProvider]:
+    def list_builtin_tools(user_id: str, tenant_id: str) -> list[UserToolProvider]:
         """
-            list builtin tools
+        list builtin tools
         """
         # get all builtin providers
         provider_controllers = ToolManager.list_builtin_providers()
 
         # get all user added providers
-        db_providers: list[BuiltinToolProvider] = db.session.query(BuiltinToolProvider).filter(
-            BuiltinToolProvider.tenant_id == tenant_id
-        ).all() or []
+        db_providers: list[BuiltinToolProvider] = (
+            db.session.query(BuiltinToolProvider).filter(BuiltinToolProvider.tenant_id == tenant_id).all() or []
+        )
 
         # find provider
-        find_provider = lambda provider: next(filter(lambda db_provider: db_provider.provider == provider, db_providers), None)
+        find_provider = lambda provider: next(
+            filter(lambda db_provider: db_provider.provider == provider, db_providers), None
+        )
 
         result: list[UserToolProvider] = []
 
@@ -209,7 +215,7 @@ class BuiltinToolManageService:
                     include_set=dify_config.POSITION_TOOL_INCLUDES_SET,
                     exclude_set=dify_config.POSITION_TOOL_EXCLUDES_SET,
                     data=provider_controller,
-                    name_func=lambda x: x.identity.name
+                    name_func=lambda x: x.identity.name,
                 ):
                     continue
 
@@ -217,7 +223,7 @@ class BuiltinToolManageService:
                 user_builtin_provider = ToolTransformService.builtin_provider_to_user_provider(
                     provider_controller=provider_controller,
                     db_provider=find_provider(provider_controller.identity.name),
-                    decrypt_credentials=True
+                    decrypt_credentials=True,
                 )
 
                 # add icon
@@ -225,12 +231,14 @@ class BuiltinToolManageService:
 
                 tools = provider_controller.get_tools()
                 for tool in tools:
-                    user_builtin_provider.tools.append(ToolTransformService.tool_to_user_tool(
-                        tenant_id=tenant_id,
-                        tool=tool,
-                        credentials=user_builtin_provider.original_credentials,
-                        labels=ToolLabelManager.get_tool_labels(provider_controller)
-                    ))
+                    user_builtin_provider.tools.append(
+                        ToolTransformService.tool_to_user_tool(
+                            tenant_id=tenant_id,
+                            tool=tool,
+                            credentials=user_builtin_provider.original_credentials,
+                            labels=ToolLabelManager.get_tool_labels(provider_controller),
+                        )
+                    )
 
                 result.append(user_builtin_provider)
             except Exception as e: