alibaba-cloud-ops-mcp-server 0.8.9__tar.gz → 0.9.1__tar.gz

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alibaba-cloud-ops-mcp-server
-Version: 0.8.9
+Version: 0.9.1
 Summary: A MCP server for Alibaba Cloud
 Author-email: Zheng Dayu <dayu.zdy@alibaba-inc.com>
 License-File: LICENSE
@@ -11,7 +11,7 @@ Requires-Dist: alibabacloud-ecs20140526>=6.1.0
 Requires-Dist: alibabacloud-oos20190601>=3.4.1
 Requires-Dist: alibabacloud-oss-v2>=1.1.0
 Requires-Dist: click>=8.1.8
-Requires-Dist: mcp[cli]>=1.9.0
+Requires-Dist: fastmcp>=2.8.0
 Description-Content-Type: text/markdown
 
 # Alibaba Cloud Ops MCP Server
@@ -55,6 +55,8 @@ To use `alibaba-cloud-ops-mcp-server` MCP Server with any other MCP Client, you
 }
 ```
 
+[For detailed parameter description, see MCP startup parameter document](./README_mcp_args.md)
+
 ## MCP Maketplace Integration
 
 * [Cline](https://cline.bot/mcp-marketplace)

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/README.md

@@ -39,6 +39,8 @@ To use `alibaba-cloud-ops-mcp-server` MCP Server with any other MCP Client, you
 }
 ```
 
+[For detailed parameter description, see MCP startup parameter document](./README_mcp_args.md)
+
 ## MCP Maketplace Integration
 
 * [Cline](https://cline.bot/mcp-marketplace)

alibaba_cloud_ops_mcp_server-0.9.1/README_mcp_args.md

@@ -0,0 +1,22 @@
+# MCP Startup Parameters Guide
+
+This document provides a detailed introduction to the available parameters for starting the Alibaba Cloud MCP Server, helping users configure the server according to their needs.
+
+## Parameter Table
+
+|   Parameter    | Required |  Type  |  Default   | Description                                                                                                                                                                                                                                                                                                                                                                                                                           |
+|:--------------:|:--------:|:------:|:----------:|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| `--transport`  |    No    | string |  `stdio`   | Transport protocol for MCP Server communication.<br>Options:<br>&nbsp;&nbsp;&nbsp;&nbsp;• `stdio` <br>&nbsp;&nbsp;&nbsp;&nbsp;• `sse` <br>&nbsp;&nbsp;&nbsp;&nbsp;• `streamable-http`                                                                                                                                                                                                                                                 |
+| `--port`       |    No    |  int   |  `8000`    | Specifies the port number MCP Server listens on. Make sure the port is not occupied.                                                                                                                                                                                                                                                                                                                                                  |
+| `--host`       |    No    | string | `127.0.0.1`| Specifies the host address MCP Server listens on. `0.0.0.0` means listening on all network interfaces.                                                                                                                                                                                                                                                                                                                                |
+| `--services`   |    No    | string |   None     | Comma-separated services, e.g., `ecs,vpc`.<br>Supported services:<br>&nbsp;&nbsp;&nbsp;&nbsp;• `ecs`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `oos`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `rds`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `vpc`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `slb`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `ess`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `ros`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `cbn`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `dds`<br>&nbsp;&nbsp;&nbsp;&nbsp;• `r-kvstore` |
+
+## Usage Example
+
+```bash
+uv run src/alibaba_cloud_ops_mcp_server/server.py --transport sse --port 8080 --host 0.0.0.0 --services ecs,vpc
+```
+
+---
+
+For more help, please refer to the main project documentation or contact the maintainer. 

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/README_zh.md

@@ -39,6 +39,8 @@ curl -LsSf https://astral.sh/uv/install.sh | sh
 }
 ```
 
+[详细参数说明见 MCP 启动参数文档](./README_mcp_args.md)
+
 ## MCP市场集成
 
 * [Cline](https://cline.bot/mcp-marketplace)

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "alibaba-cloud-ops-mcp-server"
-version = "0.8.9"
+version = "0.9.1"
 description = "A MCP server for Alibaba Cloud"
 readme = "README.md"
 authors = [
@@ -14,7 +14,7 @@ dependencies = [
     "alibabacloud_oss_v2>=1.1.0",
     "alibabacloud-credentials>=1.0.0",
     "click>=8.1.8",
-    "mcp[cli]>=1.9.0",
+    "fastmcp>=2.8.0"
 ]
 
 [build-system]
@@ -34,4 +34,4 @@ dev = [
 ]
 
 [project.scripts]
-alibaba-cloud-ops-mcp-server = "alibaba_cloud_ops_mcp_server:main"
+alibaba-cloud-ops-mcp-server = "alibaba_cloud_ops_mcp_server:main"

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/src/alibaba_cloud_ops_mcp_server/alibabacloud/api_meta_client.py

@@ -47,9 +47,16 @@ class ApiMetaClient:
     @classmethod
     def get_service_version(cls, service):
         data = cls.get_response_from_pop_api(cls.GET_PRODUCT_LIST)
-        version = next((item.get(DEFAULT_VERSION) for item in data if item.get(CODE).lower() == service), None)
+        version = next((item.get(DEFAULT_VERSION) for item in data if item.get(CODE).lower() == service.lower()), None)
         return version
 
+    @classmethod
+    def get_all_service_info(cls):
+        data = cls.get_response_from_pop_api(cls.GET_PRODUCT_LIST)
+        filtered_data = [{"code": item["code"], "name": item["name"]} for item in data]
+
+        return filtered_data
+
     @classmethod
     def get_service_style(cls, service):
         data = cls.get_response_from_pop_api(cls.GET_PRODUCT_LIST)
@@ -147,7 +154,8 @@ class ApiMetaClient:
         return combined_params
 
     @classmethod
-    def get_apis_in_service(cls, service, version):
+    def get_apis_in_service(cls, service):
+        version = cls.get_service_version(service)
         data = cls.get_response_from_pop_api(cls.GET_API_OVERVIEW, service=service, version=version)
         apis = list(data[APIS].keys())
         return apis

alibaba_cloud_ops_mcp_server-0.9.1/src/alibaba_cloud_ops_mcp_server/alibabacloud/static/PROMPT_UNDERSTANDING.md

@@ -0,0 +1,134 @@
+## Optimized Prompt
+
+When a user submits a request, analyze their needs and check if matching tools exist. If yes, use them directly. If not, proceed to the retrieval phase.
+
+---
+
+## Request Flow
+
+1. **Analysis & Selection**
+   - Analyze user intent
+   - Choose between specific tools or common API flow
+   - Verify service support
+
+2. **API Flow** (if no specific tool)
+   - Identify service
+   - Select API via `ListAPIs`
+   - Get params via `GetAPIInfo`
+   - Execute via `CommonAPICaller`
+
+3. **Error Handling**
+   - Service not supported: "Unfortunately, we currently do not support this service"
+   - API failures: Check error code, params, permissions
+   - Param validation: Verify types and formats
+
+---
+
+### Retrieval Phase
+
+1. **Service Selection**
+
+   Supported Services:
+   - ecs: Elastic Compute Service (ECS)
+   - oos: Operations Orchestration Service (OOS)
+   - rds: Relational Database Service (RDS)
+   - vpc: Virtual Private Cloud (VPC)
+   - slb: Server Load Balancer (SLB)
+   - ess: Elastic Scaling (ESS)
+   - ros: Resource Orchestration Service (ROS)
+   - cbn: Cloud Enterprise Network (CBN)
+   - dds: MongoDB Database Service (DDS)
+   - r-kvstore: Cloud database Tair (compatible with Redis) (R-KVStore)
+
+2. **API Process**
+   - Use `ListAPIs` for available APIs
+   - Use `GetAPIInfo` for API details
+   - Use `CommonAPICaller` to execute
+
+---
+
+### Notes
+- Filter for most appropriate result
+- Choose based on user context and common usage
+- Validate parameters before calls
+- Handle errors gracefully
+
+---
+
+### Common Scenarios
+
+1. **Instance Management**
+   ```
+   User: "Start ECS instance i-1234567890abcdef0"
+   Action: Use OOS_StartInstances
+   ```
+
+2. **Monitoring**
+   ```
+   User: "Check ECS CPU usage"
+   Action: Use CMS_GetCpuUsageData
+   ```
+
+3. **Custom API**
+   ```
+   User: "Create VPC in cn-hangzhou"
+   Action: ListAPIs → GetAPIInfo → CommonAPICaller
+   ```
+
+---
+
+## Available Tools
+
+### ECS (OOS/API)
+- RunCommand: Execute commands on instances
+- StartInstances: Start ECS instances
+- StopInstances: Stop ECS instances
+- RebootInstances: Reboot ECS instances
+- DescribeInstances: List instance details
+- DescribeRegions: List available regions
+- DescribeZones: List available zones
+- DescribeAvailableResource: Check resource inventory
+- DescribeImages: List available images
+- DescribeSecurityGroups: List security groups
+- RunInstances: Create new instances
+- DeleteInstances: Delete instances
+- ResetPassword: Change instance password
+- ReplaceSystemDisk: Change instance OS
+
+### VPC (API)
+- DescribeVpcs: List VPCs
+- DescribeVSwitches: List VSwitches
+
+### RDS (OOS/API)
+- DescribeDBInstances: List database instances
+- StartDBInstances: Start RDS instances
+- StopDBInstances: Stop RDS instances
+- RestartDBInstances: Restart RDS instances
+
+### OSS (API)
+- ListBuckets: List OSS buckets
+- PutBucket: Create bucket
+- DeleteBucket: Delete bucket
+- ListObjects: List bucket contents
+
+### CloudMonitor (API)
+- GetCpuUsageData: Get instance CPU usage
+- GetCpuLoadavgData: Get 1m CPU load
+- GetCpuloadavg5mData: Get 5m CPU load
+- GetCpuloadavg15mData: Get 15m CPU load
+- GetMemUsedData: Get memory usage
+- GetMemUsageData: Get memory utilization
+- GetDiskUsageData: Get disk utilization
+- GetDiskTotalData: Get total disk space
+- GetDiskUsedData: Get used disk space
+
+Note: (OOS) = Operations Orchestration Service, (API) = Direct API call
+
+---
+
+### Best Practices
+- Use pre-defined tools when possible
+- Follow API rate limits
+- Implement proper error handling
+- Validate all parameters
+- Use appropriate endpoints

alibaba_cloud_ops_mcp_server-0.9.1/src/alibaba_cloud_ops_mcp_server/alibabacloud/static/__init__.py

@@ -0,0 +1,8 @@
+from importlib import resources
+
+with (
+    resources.files('alibaba_cloud_ops_mcp_server.alibabacloud.static')
+    .joinpath('PROMPT_UNDERSTANDING.md')
+    .open('r', encoding='utf-8') as f
+):
+    PROMPT_UNDERSTANDING = f.read()

alibaba_cloud_ops_mcp_server-0.9.1/src/alibaba_cloud_ops_mcp_server/alibabacloud/utils.py

@@ -0,0 +1,46 @@
+import logging
+
+from alibabacloud_credentials.client import Client as CredClient
+from alibabacloud_tea_openapi.models import Config
+from fastmcp.server.dependencies import get_http_request
+
+logger = logging.getLogger(__name__)
+
+
+def get_credentials_from_header():
+    credentials = None
+    try:
+        request = get_http_request()
+        headers = request.headers
+        access_key_id = headers.get('x-acs-accesskey-id', None)
+        access_key_secret = headers.get('x-acs-accesskey-secret', None)
+        token = headers.get('x-acs-security-token', None)
+
+        if access_key_id:
+            credentials = {
+                'AccessKeyId': access_key_id,
+                'AccessKeySecret': access_key_secret,
+                'SecurityToken': token
+            }
+
+    except Exception as e:
+        logger.info(f'get_credentials_from_header error: {e}')
+    return credentials
+
+
+def create_config():
+    credentials = get_credentials_from_header()
+    if credentials:
+        access_key_id = credentials.get('AccessKeyId', None)
+        access_key_secret = credentials.get('AccessKeySecret', None)
+        token = credentials.get('SecurityToken', None)
+        config = Config(
+            access_key_id=access_key_id,
+            access_key_secret=access_key_secret,
+            security_token=token
+        )
+    else:
+        credentialsClient = CredClient()
+        config = Config(credential=credentialsClient)
+    config.user_agent = 'alibaba-cloud-ops-mcp-server'
+    return config

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/src/alibaba_cloud_ops_mcp_server/config.py

@@ -21,7 +21,7 @@ config = {
     ],
     'rds': [
         'DescribeDBInstances'
-    ],
+    ]
 }
 
 

alibaba_cloud_ops_mcp_server-0.9.1/src/alibaba_cloud_ops_mcp_server/server.py

@@ -0,0 +1,78 @@
+from fastmcp import FastMCP
+import click
+import logging
+
+from alibaba_cloud_ops_mcp_server.tools.common_api_tools import set_custom_service_list
+from alibaba_cloud_ops_mcp_server.config import config
+from alibaba_cloud_ops_mcp_server.tools import cms_tools, oos_tools, oss_tools, api_tools, common_api_tools
+
+logger = logging.getLogger(__name__)
+
+SUPPORTED_SERVICES_MAP = {
+    "ecs": "Elastic Compute Service (ECS)",
+    "oos": "Operations Orchestration Service (OOS)",
+    "rds": "Relational Database Service (RDS)",
+    "vpc": "Virtual Private Cloud (VPC)",
+    "slb": "Server Load Balancer (SLB)",
+    "ess": "Elastic Scaling (ESS)",
+    "ros": "Resource Orchestration Service (ROS)",
+    "cbn": "Cloud Enterprise Network (CBN)",
+    "dds": "MongoDB Database Service (DDS)",
+    "r-kvstore": "Cloud database Tair (compatible with Redis) (R-KVStore)"
+}
+
+
+@click.command()
+@click.option(
+    "--transport",
+    type=click.Choice(["stdio", "sse", "streamable-http"]),
+    default="stdio",
+    help="Transport type",
+)
+@click.option(
+    "--port",
+    type=int,
+    default=8000,
+    help="Port number",
+)
+@click.option(
+    "--host",
+    type=str,
+    default="127.0.0.1",
+    help="Host",
+)
+@click.option(
+    "--services",
+    type=str,
+    default=None,
+    help="Comma-separated list of supported services, e.g., 'ecs,vpc,rds'",
+)
+def main(transport: str, port: int, host: str, services: str):
+    # Create an MCP server
+    mcp = FastMCP(
+        name="alibaba-cloud-ops-mcp-server",
+        port=port,
+        host=host
+    )
+
+    if services:
+        service_keys = [s.strip().lower() for s in services.split(",")]
+        service_list = [(key, SUPPORTED_SERVICES_MAP.get(key, key)) for key in service_keys]
+        set_custom_service_list(service_list)
+        for tool in common_api_tools.tools:
+            mcp.tool(tool)
+    for tool in oos_tools.tools:
+        mcp.tool(tool)
+    for tool in cms_tools.tools:
+        mcp.tool(tool)
+    for tool in oss_tools.tools:
+        mcp.tool(tool)
+    api_tools.create_api_tools(mcp, config)
+
+    # Initialize and run the server
+    logger.debug(f'mcp server is running on {transport} mode.')
+    mcp.run(transport=transport)
+
+
+if __name__ == "__main__":
+    main()

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/src/alibaba_cloud_ops_mcp_server/tools/api_tools.py

@@ -24,12 +24,38 @@ type_map = {
     'number': float
 }
 
+REGION_ENDPOINT_SERVICE = ['ecs', 'oos', 'vpc', 'slb']
+
+DOUBLE_ENDPOINT_SERVICE = {
+    'rds': ['cn-qingdao', 'cn-beijing', 'cn-hangzhou', 'cn-shanghai', 'cn-shenzhen', 'cn-heyuan', 'cn-guangzhou', 'cn-hongkong'],
+    'ess': ['cn-qingdao', 'cn-beijing', 'cn-hangzhou', 'cn-shanghai', 'cn-nanjing', 'cn-shenzhen'],
+    'ros': ['cn-qingdao'],
+    'dds': ['cn-qingdao', 'cn-beijing', 'cn-wulanchabu', 'cn-hangzhou', 'cn-shanghai', 'cn-shenzhen', 'cn-heyuan', 'cn-guangzhou'],
+    'r-kvstore': ['cn-qingdao', 'cn-beijing', 'cn-wulanchabu', 'cn-hangzhou', 'cn-shanghai', 'cn-shenzhen', 'cn-heyuan']
+}
+
+CENTRAL_ENDPOINTS_SERVICE = ['cbn']
+
+
+def _get_service_endpoint(service: str, region_id: str):
+    region_id = region_id.lower()
+    use_region_endpoint = service in REGION_ENDPOINT_SERVICE or (
+            service in DOUBLE_ENDPOINT_SERVICE and region_id in DOUBLE_ENDPOINT_SERVICE[service]
+    )
+
+    if use_region_endpoint:
+        return f'{service}.{region_id}.aliyuncs.com'
+    elif service in CENTRAL_ENDPOINTS_SERVICE or service in DOUBLE_ENDPOINT_SERVICE:
+        return f'{service}.aliyuncs.com'
+    else:
+        return f'{service}.{region_id}.aliyuncs.com'
+
 
 def create_client(service: str, region_id: str) -> OpenApiClient:
     config = create_config()
     if isinstance(service, str):
         service = service.lower()
-    config.endpoint = f'{service}.{region_id}.aliyuncs.com'
+    config.endpoint = _get_service_endpoint(service, region_id.lower())
     return OpenApiClient(config)
 
 
@@ -84,24 +110,37 @@ def _create_function_schemas(service, api, api_meta):
     schemas = {}
     schemas[api] = {}
     parameters = api_meta.get('parameters', [])
+
+    required_params = []
+    optional_params = []
+
     for parameter in parameters:
         name = parameter.get('name')
         # TODO 目前忽略了带'.'的参数
         if '.' in name:
             continue
         schema = parameter.get('schema', '')
+        required = schema.get('required', False)
+
+        if required:
+            required_params.append(parameter)
+        else:
+            optional_params.append(parameter)
+
+    def process_parameter(parameter):
+        name = parameter.get('name')
+        schema = parameter.get('schema', '')
         description = schema.get('description', '')
         example = schema.get('example', '')
         type_ = schema.get('type', '')
         description = f'{description} 参数类型: {type_},参数示例：{example}'
         required = schema.get('required', False)
-        
-        # 只有在service为ecs时，才对特定参数进行特殊处理
+
         if service.lower() == 'ecs' and name in ECS_LIST_PARAMETERS and type_ == 'string':
             python_type = list
         else:
             python_type = type_map.get(type_, str)
-            
+
         field_info = (
             python_type,
             field(
@@ -109,7 +148,16 @@ def _create_function_schemas(service, api, api_meta):
                 metadata={'description': description, 'required': required}
             )
         )
+        return name, field_info
+
+    for parameter in required_params:
+        name, field_info = process_parameter(parameter)
         schemas[api][name] = field_info
+
+    for parameter in optional_params:
+        name, field_info = process_parameter(parameter)
+        schemas[api][name] = field_info
+
     if 'RegionId' not in schemas[api]:
         schemas[api]['RegionId'] = (
             str,

{alibaba_cloud_ops_mcp_server-0.8.9 → alibaba_cloud_ops_mcp_server-0.9.1}/src/alibaba_cloud_ops_mcp_server/tools/cms_tools.py

@@ -37,8 +37,8 @@ def _get_cms_metric_data(region_id: str, instance_ids: List[str], metric_name: s
 
 @tools.append
 def CMS_GetCpuUsageData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取ECS实例的CPU使用率数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'cpu_total')
@@ -46,8 +46,8 @@ def CMS_GetCpuUsageData(
 
 @tools.append
 def CMS_GetCpuLoadavgData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取CPU一分钟平均负载指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'load_1m')
@@ -55,8 +55,8 @@ def CMS_GetCpuLoadavgData(
 
 @tools.append
 def CMS_GetCpuloadavg5mData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取CPU五分钟平均负载指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'load_5m')
@@ -64,16 +64,16 @@ def CMS_GetCpuloadavg5mData(
 
 @tools.append
 def CMS_GetCpuloadavg15mData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取CPU十五分钟平均负载指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'load_15m')
 
 @tools.append
 def CMS_GetMemUsedData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取内存使用量指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'memory_usedspace')
@@ -81,8 +81,8 @@ def CMS_GetMemUsedData(
 
 @tools.append
 def CMS_GetMemUsageData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取内存利用率指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'memory_usedutilization')
@@ -90,8 +90,8 @@ def CMS_GetMemUsageData(
 
 @tools.append
 def CMS_GetDiskUsageData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取磁盘利用率指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'diskusage_utilization')
@@ -99,8 +99,8 @@ def CMS_GetDiskUsageData(
 
 @tools.append
 def CMS_GetDiskTotalData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取磁盘分区总容量指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'diskusage_total')
@@ -108,8 +108,8 @@ def CMS_GetDiskTotalData(
 
 @tools.append
 def CMS_GetDiskUsedData(
-    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou')
 ):
     """获取磁盘分区使用量指标数据"""
     return _get_cms_metric_data(RegionId, InstanceIds, 'diskusage_used')

alibaba_cloud_ops_mcp_server-0.9.1/src/alibaba_cloud_ops_mcp_server/tools/common_api_tools.py

@@ -0,0 +1,74 @@
+import logging
+import os
+from pydantic import Field
+from alibabacloud_tea_openapi import models as open_api_models
+from alibabacloud_tea_util import models as util_models
+from alibabacloud_tea_openapi.client import Client as OpenApiClient
+from alibabacloud_openapi_util.client import Client as OpenApiUtilClient
+from alibaba_cloud_ops_mcp_server.alibabacloud.api_meta_client import ApiMetaClient
+from alibaba_cloud_ops_mcp_server.alibabacloud.static import PROMPT_UNDERSTANDING
+from alibaba_cloud_ops_mcp_server.tools.api_tools import create_client, _tools_api_call
+
+END_STATUSES = ['Success', 'Failed', 'Cancelled']
+
+tools = []
+
+_CUSTOM_SERVICE_LIST = None
+
+logger = logging.getLogger(__name__)
+
+
+def set_custom_service_list(service_list):
+    global _CUSTOM_SERVICE_LIST
+    _CUSTOM_SERVICE_LIST = service_list
+
+
+@tools.append
+def PromptUnderstanding() -> str:
+    """
+    Always use this tool first to understand the user's query and convert it into suggestions from Alibaba Cloud experts.
+    """
+    global _CUSTOM_SERVICE_LIST
+
+    content = PROMPT_UNDERSTANDING
+    if _CUSTOM_SERVICE_LIST:
+        import re
+        pattern = r'Supported Services\s*:\s*\n(?:\s{3}- .+?\n)+'
+        replacement = f"Supported Services:\n   - " + "\n   - ".join([f"{k}: {v}" for k, v in _CUSTOM_SERVICE_LIST])
+        content = re.sub(pattern, replacement, content, flags=re.DOTALL)
+
+    return content
+
+
+@tools.append
+def ListAPIs(
+        service: str = Field(description='AlibabaCloud service code')
+):
+    """
+    Use PromptUnderstanding tool first to understand the user's query, Get the corresponding API list information through the service name to prepare for the subsequent selection of the appropriate API to call
+    """
+    return ApiMetaClient.get_apis_in_service(service)
+
+
+@tools.append
+def GetAPIInfo(
+        service: str = Field(description='AlibabaCloud service code'),
+        api: str = Field(description='AlibabaCloud api name'),
+):
+    """
+    Use PromptUnderstanding tool first to understand the user's query, After specifying the service name and API name, get the detailed API META of the corresponding API
+    """
+    data, version = ApiMetaClient.get_api_meta(service, api)
+    return data.get('parameters')
+
+
+@tools.append
+def CommonAPICaller(
+        service: str = Field(description='AlibabaCloud service code'),
+        api: str = Field(description='AlibabaCloud api name'),
+        parameters: dict = Field(description='AlibabaCloud ECS instance ID List', default={}),
+):
+    """
+    Use PromptUnderstanding tool first to understand the user's query, Perform the actual call by specifying the Service, API, and Parameters
+    """
+    return _tools_api_call(service, api, parameters, None)