feat: add account service client and preflight service checks for build

operations and release 0.1.11

- Add AgentkitAccountClient for querying linked service status
- Add ServiceNotEnabledException for handling missing service errors
- Add PreflightMode enum (SKIP/WARN/FAIL/PROMPT) to control preflight behavior
- Add preflight check logic to BaseExecutor with _preflight_check() and _handle_preflight_result()
- Integrate preflight checks into BuildExecutor.execute() before strategy execution
- Add get_required_services() class

(cherry picked from commit f4901dcebb0dc0c6daa49d14a0dde6c85064e558)

Author: innsd

agentkit/client/base_agentkit_client.py

@@ -17,7 +17,7 @@
 Provides common initialization and API invocation logic.
 """
 
-from typing import Any, Dict, Union
+from typing import Any, Dict, Union, Optional
 
 from agentkit.client.base_service_client import BaseServiceClient, ApiConfig
 from agentkit.utils.ve_sign import get_volc_agentkit_host_info
@@ -47,6 +47,7 @@ def __init__(
         region: str = "",
         session_token: str = "",
         service_name: str = "",
+        header: Optional[Dict[str, Any]] = {"Accept": "application/json"},
     ) -> None:
         """
         Initialize the AgentKit client.
@@ -65,6 +66,7 @@ def __init__(
             session_token=session_token,
             service_name=service_name,
             credential_env_prefix='AGENTKIT',
+            header=header,
         )
 
     def _get_service_config(self) -> Dict[str, str]:

agentkit/client/base_service_client.py

@@ -72,6 +72,7 @@ def __init__(
         session_token: str = "",
         service_name: str = "",
         credential_env_prefix: str = "",
+        header: Optional[Dict[str, Any]] = {"Accept": "application/json"},
     ) -> None:
         """
         Initialize the service client.
@@ -110,7 +111,7 @@ def __init__(
         # Create ServiceInfo
         self.service_info = ServiceInfo(
             host=self.host,
-            header={'Accept': 'application/json'},
+            header=header,
             credentials=Credentials(
                 ak=self.access_key,
                 sk=self.secret_key,

agentkit/sdk/account/client.py

@@ -0,0 +1,110 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Auto-generated from API JSON definition
+# Do not edit manually
+
+from __future__ import annotations
+
+from typing import Dict, List, Optional
+from agentkit.client import BaseAgentkitClient
+from agentkit.sdk.account.types import (
+    ListAccountLinkedServicesRequest,
+    ListAccountLinkedServicesResponse,
+)
+
+
+class AgentkitAccountClient(BaseAgentkitClient):
+    """AgentKit Account Management Service"""
+    API_ACTIONS: Dict[str, str] = {
+        "ListAccountLinkedServices": "ListAccountLinkedServices",
+    }
+
+    def __init__(
+        self,
+        access_key: str = "",
+        secret_key: str = "",
+        region: str = "",
+        session_token: str = "",
+    ) -> None:
+        super().__init__(
+            access_key=access_key,
+            secret_key=secret_key,
+            region=region,
+            session_token=session_token,
+            service_name="account",
+        )
+
+
+    def list_account_linked_services(self, request: ListAccountLinkedServicesRequest) -> ListAccountLinkedServicesResponse:
+        return self._invoke_api(
+            api_action="ListAccountLinkedServices",
+            request=request,
+            response_type=ListAccountLinkedServicesResponse,
+        )
+
+    def get_service_status(self, service_name: str) -> Optional[str]:
+        """
+        Query the LinkServices status for a specific service.
+        
+        Args:
+            service_name: The name of the service to query (e.g., 'ark', 'vefaas')
+            
+        Returns:
+            The status string ('Enabled' or 'Disabled'), or None if service not found.
+        """
+        print(1)
+        response = self.list_account_linked_services(ListAccountLinkedServicesRequest())
+        print(2)
+        if not response.service_statuses:
+            return None
+        for svc in response.service_statuses:
+            if svc.service_name == service_name:
+                return svc.status
+        return None
+
+    def get_services_status(self, service_names: List[str]) -> Dict[str, Optional[str]]:
+        """
+        Query the LinkServices status for multiple services.
+        
+        Args:
+            service_names: List of service names to query (e.g., ['ark', 'vefaas', 'cr'])
+            
+        Returns:
+            A dict mapping service_name -> status ('Enabled'/'Disabled'/None if not found).
+        """
+        response = self.list_account_linked_services(ListAccountLinkedServicesRequest())
+        status_map: Dict[str, Optional[str]] = {name: None for name in service_names}
+        if response.service_statuses:
+            for svc in response.service_statuses:
+                if svc.service_name in status_map:
+                    status_map[svc.service_name] = svc.status
+        return status_map
+
+    def has_disabled_services(self) -> List[str]:
+        """
+        Check if any returned service has a disabled status.
+        
+        Returns:
+            A list of service names that are disabled (status != 'Enabled').
+            Returns empty list if all services are enabled.
+        """
+        response = self.list_account_linked_services(ListAccountLinkedServicesRequest())
+        disabled = []
+        if response.service_statuses:
+            for svc in response.service_statuses:
+                if svc.status != "Enabled":
+                    disabled.append(svc.service_name)
+        return disabled
+

agentkit/sdk/account/types.py

@@ -0,0 +1,45 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Auto-generated from API JSON definition
+# Do not edit manually
+
+from __future__ import annotations
+
+from typing import Optional
+from pydantic import BaseModel, Field
+
+class AccountBaseModel(BaseModel):
+    """AgentKit auto-generated base model"""
+    model_config = {
+        "populate_by_name": True,
+        "arbitrary_types_allowed": True
+    }
+
+
+# Data Types
+class ServiceStatusesForListAccountLinkedServices(AccountBaseModel):
+    service_name: Optional[str] = Field(default=None, alias="ServiceName")
+    status: Optional[str] = Field(default=None, alias="Status")
+
+
+# ListAccountLinkedServices - Request
+class ListAccountLinkedServicesRequest(AccountBaseModel):
+    pass
+
+
+# ListAccountLinkedServices - Response
+class ListAccountLinkedServicesResponse(AccountBaseModel):
+    service_statuses: Optional[list[ServiceStatusesForListAccountLinkedServices]] = Field(default=None, alias="ServiceStatuses")
+

agentkit/toolkit/builders/ve_pipeline.py

@@ -679,6 +679,8 @@ def _upload_to_tos(self, archive_path: str, config: VeCPCRBuilderConfig) -> str:
             return tos_url
 
         except Exception as e:
+            if "AccountDisable" in str(e):
+                raise Exception(f"Tos Service is not enabled, please enable it in the console. Enable services at: https://console.volcengine.com/agentkit/region:agentkit+cn-beijing/auth")
             raise Exception(f"Failed to upload to TOS: {str(e)}")
 
     def _prepare_cr_resources(self, config: VeCPCRBuilderConfig) -> CRServiceConfig:

agentkit/toolkit/executors/__init__.py

@@ -13,35 +13,38 @@
 # limitations under the License.
 
 """
-Executor 层 - 统一的配置管理和错误处理
+Executor layer - Unified configuration management and error handling.
 
-Executor 层职责：
-- 配置加载和验证
-- Strategy 选择和实例化  
-- Reporter 注入
-- 统一错误处理
-- 日志记录
+Responsibilities:
+- Configuration loading and validation
+- Strategy selection and instantiation
+- Reporter injection for progress reporting
+- Unified error handling and logging
 
-不做：
-- 结果转换（Strategy 直接返回标准 Result）
-- 进度报告（由 Strategy → Builder/Runner 处理）
+Design principle: Executors are thin orchestration layers. They delegate actual
+work to Strategies and return results directly without transformation.
 """
 
-from .base_executor import BaseExecutor
+from .base_executor import BaseExecutor, ServiceNotEnabledException
 from .build_executor import BuildExecutor, BuildOptions
 from .deploy_executor import DeployExecutor
 from .invoke_executor import InvokeExecutor
 from .status_executor import StatusExecutor
 from .lifecycle_executor import LifecycleExecutor
 from .init_executor import InitExecutor
 
+# Re-export PreflightMode from models for convenience
+from agentkit.toolkit.models import PreflightMode
+
 __all__ = [
     'BaseExecutor',
     'BuildExecutor',
-    'BuildOptions',  # 导出 BuildOptions 供 CLI 使用
+    'BuildOptions',
     'DeployExecutor',
     'InvokeExecutor',
     'StatusExecutor',
     'LifecycleExecutor',
-    'InitExecutor'
+    'InitExecutor',
+    'PreflightMode',
+    'ServiceNotEnabledException',
 ]