alibaba-cloud-ops-mcp-server 0.7.2__py3-none-any.whl

alibaba_cloud_ops_mcp_server/__init__.py

@@ -0,0 +1,11 @@
+from . import server
+import asyncio
+import logging
+
+logging.getLogger().setLevel(logging.INFO)
+
+def main():
+    """Main entry point for the package."""
+    asyncio.run(server.main())
+
+__all__ = ['main', 'server']

alibaba_cloud_ops_mcp_server/api_meta_client.py

@@ -0,0 +1,164 @@
+# -------------------------------------------------------------------------------
+# Copyright (c) 2019 Aliyun.com All right reserved. This software is the
+# confidential and proprietary information of Aliyun.com ("Confidential
+# Information"). You shall not disclose such Confidential Information and shall
+# use it only in accordance with the terms of the license agreement you entered
+# into with Aliyun.com .
+# -------------------------------------------------------------------------------
+import requests
+
+API_META_KEYS = (VERSION, RESPONSES, SCHEMA, PROPERTIES, HTTP_SUCCESS_CODE, DEFAULT_VERSION, CODE, REF, APIS,
+                 SERVICE_KEY, NAME, IN, PARAMETERS, STYLE, BODY) \
+    = ('version', 'responses', 'schema', 'properties', '200', 'defaultVersion', 'code', '$ref', 'apis', 'service',
+       'name', 'in', 'parameters', 'style', 'body')
+
+
+
+class ApiMetaClient:
+    PATH = 'path'
+    METHODS = 'methods'
+    BASE_URL = 'https://api.aliyun.com/meta/v1'
+    POP_API_NAME = (GET_PRODUCT_LIST, GET_API_OVERVIEW, GET_API_INFO, GET_APIDOCS) = \
+        ('GetProductList', 'GetApiOverview', 'GetApiInfo', 'GetAPIDocs')
+
+    config = {
+        'GetProductList': {'path': 'products.json'},
+        'GetApiOverview': {'path': 'products/{service}/versions/{version}/overview.json'},
+        'GetApiInfo': {'path': 'products/{service}/versions/{version}/apis/{api}/api.json'},
+        'GetAPIDocs': {'path': 'products/{service}/versions/{version}/api-docs.json'},
+    }
+
+    @classmethod
+    def get_response_from_pop_api(cls, pop_api_name, service=None, api=None, version=None):
+        try:
+            api_config = cls.config.get(pop_api_name)
+            try:
+                formatted_path = api_config[cls.PATH].format(service=service, api=api, version=version)
+            except KeyError as e:
+                raise Exception(f'Failed to format path, path: {api_config[cls.PATH]}, error: {e}')
+
+            url = f'{cls.BASE_URL}/{formatted_path}'
+            response = requests.get(url)
+            return response.json()
+        except Exception as e:
+            raise Exception(f'Failed to get response from pop api, url: {url}, error: {e}')
+
+    @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)
+        return version
+
+    @classmethod
+    def get_service_style(cls, service):
+        data = cls.get_response_from_pop_api(cls.GET_PRODUCT_LIST)
+        style = next((item.get(STYLE) for item in data if item.get(CODE).lower() == service), 'RPC')
+        return style
+
+    @classmethod
+    def get_standard_service_and_api(cls, service, api=None, version=None):
+        data = cls.get_response_from_pop_api(cls.GET_PRODUCT_LIST)
+        service_standard = (next((item.get(CODE) for item in data if item.get(CODE).lower() == service), None))
+        api_standard = None
+        if api:
+            apis = cls.get_response_from_pop_api(cls.GET_API_OVERVIEW, service=service_standard,
+                                                 version=version).get(APIS, {})
+            for api_name in apis:
+                if api_name.lower() == api.lower():
+                    api_standard = api_name
+        return service_standard, api_standard
+
+    @classmethod
+    def get_api_meta(cls, service, api):
+        service = service.lower()
+        # API_META不包含ROA类型的API，需要通过POP平台的API GetProductList获取Service对应的Version
+        # 获取POP平台API META参考文档：https://api.aliyun.com/openmeta/guide
+        version = cls.get_service_version(service)
+        service_standard, api_standard = cls.get_standard_service_and_api(service, api, version)
+        data = cls.get_response_from_pop_api(cls.GET_API_INFO, service_standard, api_standard, version)
+        return data, version
+
+    @classmethod
+    def get_response_from_api_meta(cls, service, api):
+        api_meta, version = cls.get_api_meta(service, api)
+        property_values = api_meta.get(RESPONSES, {}).get(HTTP_SUCCESS_CODE, {}).get(SCHEMA, {}).get(PROPERTIES, {})
+        return property_values, version
+
+    @classmethod
+    def get_ref_api_meta(cls, data, service, version):
+        service_standard, _ = cls.get_standard_service_and_api(service=service, version=version)
+        current_data = cls.get_response_from_pop_api(cls.GET_API_OVERVIEW, service=service_standard, version=version)
+        ref_path = data.get(REF)
+        path = ref_path.lstrip('#/').split('/')
+        for _key in path:
+            if _key in current_data:
+                current_data = current_data[_key]
+            else:
+                raise KeyError(f"Path {_key} not found in the JSON data.")
+
+        return current_data
+
+    @classmethod
+    def get_api_parameters(cls, service, api, params_in=''):
+        """
+        params_in: 过滤参数位置，取值：'host', 'query', 'body', 'header'，若为空，则返回所有参数
+        """
+        api_meta, _ = cls.get_api_meta(service, api)
+        parameters = api_meta.get(PARAMETERS)
+        param_names = []
+        additional_props = []
+        # 避免循环引用
+        visited_refs = set()
+
+        def get_ref(data, _):
+            props = []
+            if not isinstance(data, dict):
+                return props
+            if REF in data:
+                ref_path = data.get(REF)
+                if ref_path in visited_refs:
+                    return props
+                visited_refs.add(ref_path)
+                referenced_schema = cls.get_ref_api_meta(data, service, _)
+                props.extend(get_ref(referenced_schema, _))
+                return props
+            if PROPERTIES in data:
+                for prop_name, prop_details in data.get(PROPERTIES, {}).items():
+                    props.append(prop_name)
+                    if isinstance(prop_details, dict) and REF in prop_details:
+                        props.extend(get_ref(prop_details, _))
+            return props
+
+        for param in parameters:
+            if params_in and param.get(IN) != params_in:
+                continue
+            param_name = param.get(NAME)
+            if param_name:
+                param_names.append(param_name)
+            schema = param.get(SCHEMA, {})
+            extracted_props = get_ref(schema, _)
+            additional_props.extend(extracted_props)
+        combined_params = param_names + additional_props
+        return combined_params
+
+    @classmethod
+    def get_apis_in_service(cls, service, version):
+        data = cls.get_response_from_pop_api(cls.GET_API_OVERVIEW, service=service, version=version)
+        apis = list(data[APIS].keys())
+        return apis
+
+    @classmethod
+    def get_api_field(cls, field_type, service, api, default=None):
+        try:
+            data, _ = cls.get_api_meta(service, api)
+            return data.get(field_type, default)
+        except Exception as e:
+            return default
+
+    @classmethod
+    def get_api_body_style(cls, service, api):
+        parameters = cls.get_api_field(PARAMETERS, service, api)
+        body_style = None
+        if parameters:
+            body_style = next((param.get(STYLE) for param in parameters if param.get(IN) == BODY), None)
+        return body_style

alibaba_cloud_ops_mcp_server/cms_tools.py

@@ -0,0 +1,120 @@
+from pydantic import Field
+from typing import List
+import os
+import json
+
+from alibabacloud_cms20190101.client import Client as cms20190101Client
+from alibabacloud_tea_openapi import models as open_api_models
+from alibabacloud_cms20190101 import models as cms_20190101_models
+from alibabacloud_tea_util import models as util_models
+from alibabacloud_tea_util.client import Client as UtilClient
+
+
+END_STATUSES = ['Success', 'Failed', 'Cancelled']
+
+
+tools = []
+
+def create_client(region_id: str) -> cms20190101Client:
+    config = open_api_models.Config(
+        access_key_id=os.environ['ALIBABA_CLOUD_ACCESS_KEY_ID'],
+        access_key_secret=os.environ['ALIBABA_CLOUD_ACCESS_KEY_SECRET'],
+        user_agent='alibaba-cloud-ops-mcp-server',
+    )
+    config.endpoint = f'metrics.{region_id}.aliyuncs.com'
+    return cms20190101Client(config)
+
+
+def _get_cms_metric_data(region_id: str, instance_ids: List[str], metric_name: str):
+    client = create_client(region_id)
+    dimesion = []
+    for instance_id in instance_ids:
+        dimesion.append({
+            'instanceId': instance_id
+        })
+    describe_metric_last_request = cms_20190101_models.DescribeMetricLastRequest(
+        namespace='acs_ecs_dashboard',
+        metric_name=metric_name,
+        dimensions=json.dumps(dimesion),
+    )
+    describe_metric_last_resp = client.describe_metric_last(describe_metric_last_request)
+    return describe_metric_last_resp.body.datapoints
+
+@tools.append
+def GetCpuUsageData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取ECS实例的CPU使用率数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'cpu_total')
+
+
+@tools.append
+def GetCpuLoadavgData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取CPU一分钟平均负载指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'load_1m')
+
+
+@tools.append
+def GetCpuloadavg5mData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取CPU五分钟平均负载指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'load_5m')
+    
+
+@tools.append
+def GetCpuloadavg15mData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取CPU十五分钟平均负载指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'load_15m')
+
+@tools.append
+def GetMemUsedData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取内存使用量指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'memory_usedspace')
+
+
+@tools.append
+def GetMemUsageData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取内存利用率指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'memory_usedutilization')
+
+
+@tools.append
+def GetDiskUsageData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取磁盘利用率指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'diskusage_utilization')
+
+
+@tools.append
+def GetDiskTotalData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取磁盘分区总容量指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'diskusage_total')
+
+
+@tools.append
+def GetDiskUsedData(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """获取磁盘分区使用量指标数据"""
+    return _get_cms_metric_data(RegionId, InstanceIds, 'diskusage_used')

alibaba_cloud_ops_mcp_server/config.py

@@ -0,0 +1,36 @@
+"""
+========================================
+Tools config
+========================================
+"""
+
+config = {
+    'ecs': [
+        'DescribeInstances',
+        'DescribeRegions',
+        'DescribeZones',
+        'DescribeAccountAttributes',
+        'DescribeAvailableResource',
+        'DescribeImages',
+        'DescribeSecurityGroups',
+        'DeleteInstances'
+    ],
+    'Vpc': [
+        'DescribeVpcs',
+        'DescribeVSwitches'
+    ]
+}
+
+
+
+
+
+
+
+
+
+
+
+
+
+

alibaba_cloud_ops_mcp_server/oos_tools.py

@@ -0,0 +1,196 @@
+from pydantic import Field
+from typing import List
+import os
+import json
+import time
+
+from alibabacloud_oos20190601.client import Client as oos20190601Client
+from alibabacloud_tea_openapi import models as open_api_models
+from alibabacloud_oos20190601 import models as oos_20190601_models
+from alibabacloud_tea_util import models as util_models
+from alibabacloud_tea_util.client import Client as UtilClient
+
+
+END_STATUSES = ['Success', 'Failed', 'Cancelled']
+
+
+tools = []
+
+def create_client(region_id: str) -> oos20190601Client:
+    config = open_api_models.Config(
+        access_key_id=os.environ['ALIBABA_CLOUD_ACCESS_KEY_ID'],
+        access_key_secret=os.environ['ALIBABA_CLOUD_ACCESS_KEY_SECRET'],
+        user_agent='alibaba-cloud-ops-mcp-server',
+    )
+    config.endpoint = f'oos.{region_id}.aliyuncs.com'
+    return oos20190601Client(config)
+
+
+def _start_execution_sync(region_id: str, template_name: str, parameters: dict):
+    client = create_client(region_id=region_id)
+    start_execution_request = oos_20190601_models.StartExecutionRequest(
+        region_id=region_id,
+        template_name=template_name,
+        parameters=json.dumps(parameters)
+    )
+    start_execution_resp = client.start_execution(start_execution_request)
+    execution_id = start_execution_resp.body.execution.execution_id
+
+    while True:
+        list_executions_request = oos_20190601_models.ListExecutionsRequest(
+            region_id=region_id,
+            execution_id=execution_id
+        )
+        list_executions_resp = client.list_executions(list_executions_request)
+        status = list_executions_resp.body.executions[0].status
+        if status in END_STATUSES:
+            return list_executions_resp.body
+        time.sleep(1)
+@tools.append
+def RunCommand(
+     RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+     CommandType: str = Field(description='ECS实例上执行的命令类型，可选值：RunShellScript，RunPythonScript，RunPerlScript，RunBatScript，RunPowerShellScript', default='RunShellScript'),
+     Command: str = Field(description='ECS实例上执行的命令内容'),
+):
+    """批量在多台ECS实例上运行云助手命令，适用于需要同时管理多台ECS实例的场景，如应用程序管理和资源标记操作等。"""
+    
+    parameters = {
+        'regionId': RegionId,
+        'resourceType': 'ALIYUN::ECS::Instance',
+        'targets': {
+            'ResourceIds': InstanceIds,
+            'RegionId': RegionId,
+            'Type': 'ResourceIds',
+            'Parameters': {
+                'RegionId': RegionId,
+                'Status': 'Running'
+            }
+        },
+        "commandType": CommandType,
+        "commandContent": Command
+    }
+    return _start_execution_sync(region_id=RegionId, template_name='ACS-ECS-BulkyRunCommand', parameters=parameters)
+    
+
+@tools.append
+def StartInstances(
+     RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+):
+    """批量启动ECS实例，适用于需要同时管理和启动多台ECS实例的场景，例如应用部署和高可用性场景。"""
+    
+    parameters = {
+        'regionId': RegionId,
+        'resourceType': 'ALIYUN::ECS::Instance',
+        'targets': {
+            'ResourceIds': InstanceIds,
+            'RegionId': RegionId,
+            'Type': 'ResourceIds'
+        }
+    }
+    return _start_execution_sync(region_id=RegionId, template_name='ACS-ECS-BulkyStartInstances', parameters=parameters)
+
+
+@tools.append
+def StopInstances(
+     RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+     ForeceStop: bool = Field(description='是否强制关机', default=False),
+):
+    """批量停止ECS实例，适用于需要同时管理和停止多台ECS实例的场景。"""
+    
+    parameters = {
+        'regionId': RegionId,
+        'resourceType': 'ALIYUN::ECS::Instance',
+        'targets': {
+            'ResourceIds': InstanceIds,
+            'RegionId': RegionId,
+            'Type': 'ResourceIds'
+        },
+        'forceStop': ForeceStop
+    }
+    return _start_execution_sync(region_id=RegionId, template_name='ACS-ECS-BulkyStopInstances', parameters=parameters)
+
+
+@tools.append
+def RebootInstances(
+     RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+     InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+     ForeceStop: bool = Field(description='是否强制关机', default=False),
+):
+    """批量重启ECS实例，适用于需要同时管理和重启多台ECS实例的场景。"""
+    
+    parameters = {
+        'regionId': RegionId,
+        'resourceType': 'ALIYUN::ECS::Instance',
+        'targets': {
+            'ResourceIds': InstanceIds,
+            'RegionId': RegionId,
+            'Type': 'ResourceIds'
+        },
+        'forceStop': ForeceStop
+    }
+    return _start_execution_sync(region_id=RegionId, template_name='ACS-ECS-BulkyRebootInstances', parameters=parameters)
+
+
+@tools.append
+def RunInstances(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    ImageId: str = Field(description='镜像ID'),
+    InstanceType: str = Field(description='实例规格'),
+    SecurityGroupId: str = Field(description='安全组ID'),
+    VSwitchId: str = Field(description='交换机ID'),
+    Amount: int = Field(description='创建数量', default=1),
+    InstanceName: str = Field(description='实例名称', default=''),
+):
+    """批量创建ECS实例，适用于需要同时创建多台ECS实例的场景，例如应用部署和高可用性场景。"""
+
+    parameters = {
+        'imageId': ImageId,
+        'instanceType': InstanceType,
+        'securityGroupId': SecurityGroupId,
+        'vSwitchId': VSwitchId,
+        'amount': Amount,
+        'instanceName': InstanceName
+    }
+    return _start_execution_sync(region_id=RegionId, template_name='ACS-ECS-RunInstances', parameters=parameters)
+
+
+@tools.append
+def ResetPassword(
+    RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+    InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+    Password: str = Field(description='ECS实例的密码,8-30个字符且只能包含以下限制条件中的字符：小写字母，大写字母，数字，只可包含特殊字符（）~！@#$%^&*-_+=（40：<>，？/'),
+):
+    """批量修改ECS实例的密码，请注意，本操作将会重启ECS实例"""
+    parameters = {
+        'regionId': RegionId,
+        'resourceType': 'ALIYUN::ECS::Instance',
+        'targets': {
+            'ResourceIds': InstanceIds,
+            'RegionId': RegionId,
+            'Type': 'ResourceIds'
+        },
+        'password': Password
+    }
+    return _start_execution_sync(region_id=RegionId, template_name='ACS-ECS-BulkyResetPassword', parameters=parameters)
+
+@tools.append
+def ReplaceSystemDisk(
+        RegionId: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+        InstanceIds: List[str] = Field(description='AlibabaCloud ECS instance ID List'),
+        ImageId: str = Field(description='镜像ID')
+):
+    """批量替换ECS实例的系统盘，更换操作系统"""
+    parameters = {
+        'regionId': RegionId,
+        'resourceType': 'ALIYUN::ECS::Instance',
+        'targets': {
+            'ResourceIds': InstanceIds,
+            'RegionId': RegionId,
+            'Type': 'ResourceIds'
+        },
+        'imageId': ImageId
+    }
+    return _start_execution_sync(region_id=RegionId, template_name='ACS-ECS-BulkyReplaceSystemDisk', parameters=parameters)

alibaba_cloud_ops_mcp_server/server.py

@@ -0,0 +1,197 @@
+# server.py
+import os
+from mcp.server.fastmcp import FastMCP, Context
+from pydantic import Field
+import click
+import logging
+
+import inspect
+import types
+from dataclasses import make_dataclass, 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.api_meta_client import ApiMetaClient
+from alibaba_cloud_ops_mcp_server.config import config
+
+from alibaba_cloud_ops_mcp_server import oos_tools
+from alibaba_cloud_ops_mcp_server import cms_tools
+
+logger = logging.getLogger(__name__)
+
+
+type_map = {
+    'string': str,
+    'integer': int,
+    'boolean': bool,
+    'array': list,
+    'object': dict,
+    'number': float
+}
+
+
+def create_client(service: str, region_id: str) -> OpenApiClient:
+    config = open_api_models.Config(
+        access_key_id=os.environ['ALIBABA_CLOUD_ACCESS_KEY_ID'],
+        access_key_secret=os.environ['ALIBABA_CLOUD_ACCESS_KEY_SECRET'],
+        user_agent='alibaba-cloud-ops-mcp-server',
+    )
+    if isinstance(service, str):
+        service = service.lower()
+    config.endpoint = f'{service}.{region_id}.aliyuncs.com'
+    return OpenApiClient(config)
+
+
+def tools_api_call(service: str, api: str, parameters: dict, ctx: Context):
+    service = service.lower()
+    api_meta, _ = ApiMetaClient.get_api_meta(service, api)
+    version = ApiMetaClient.get_service_version(service)
+    method = 'POST' if api_meta.get('methods', [])[0] == 'post' else 'GET'
+    path = api_meta.get('path', '/')
+    style = ApiMetaClient.get_service_style(service)
+    req = open_api_models.OpenApiRequest(
+        query=OpenApiUtilClient.query(parameters)
+    )
+    params = open_api_models.Params(
+        action=api,
+        version=version,
+        protocol='HTTPS',
+        pathname=path,
+        method=method,
+        auth_type='AK',
+        style=style,
+        req_body_type='formData',
+        body_type='json'
+    )
+    client = create_client(service, parameters.get('RegionId', 'cn-hangzhou'))
+    runtime = util_models.RuntimeOptions()
+    return client.call_api(params, req, runtime)
+
+
+def create_parameter_schema(fields: dict):
+    return make_dataclass("ParameterSchema", [(name, type_, value) for name, (type_, value) in fields.items()])
+
+
+def create_function_schemas(service, api, api_meta):
+    schemas = {}
+    schemas[api] = {}
+    parameters = api_meta['parameters']
+    for parameter in parameters:
+        name = parameter.get('name')
+        # TODO 目前忽略了带'.'的参数
+        if '.' in name:
+            continue
+        schema = parameter.get('schema', '')
+        description = schema.get('description', '')
+        example = schema.get('example', '')
+        type_ = schema.get('type', '')
+        description = f'{description} 请注意，提供参数要严格按照参数的类型和参数示例的提示，如果提到参数为String，且为一个 JSON 数组字符串，应在数组内使用单引号包裹对应的参数以避免转义问题，并在最外侧用双引号包裹以确保其是字符串，否则可能会导致参数解析错误。参数类型: {type_},参数示例：{example}'
+        required = schema.get('required', False)
+        python_type = type_map.get(type_, str)
+        field_info = (
+            python_type,
+            field(
+                default=None,
+                metadata={'description': description, 'required': required}
+            )
+        )
+        schemas[api][name] = field_info
+    if 'RegionId' not in schemas[api]:
+        schemas[api]['RegionId'] = (
+            str,
+            field(
+                default=None,
+                metadata={'description': '地域ID', 'required': False}
+            )
+        )
+    return schemas
+
+
+def create_tool_function_with_signature(service: str, function_name: str, fields: dict, description: str):
+    """
+    Dynamically creates a lambda function with a custom signature based on the provided fields.
+    """
+    parameters = []
+    annotations = {}
+    defaults = {}
+
+    for name, (type_, field_info) in fields.items():
+        field_description = field_info.metadata.get('description', '')
+        is_required = field_info.metadata.get('required', False)
+        default_value = field_info.default if not is_required else ...
+
+        field_default = Field(default=default_value, description=field_description)
+        parameters.append(inspect.Parameter(
+            name=name,
+            kind=inspect.Parameter.POSITIONAL_OR_KEYWORD,
+            default=field_default,
+            annotation=type_
+        ))
+        annotations[name] = type_
+        defaults[name] = field_default
+
+    signature = inspect.Signature(parameters)
+
+    def func_code(*args, **kwargs):
+        bound_args = signature.bind(*args, **kwargs)
+        bound_args.apply_defaults()
+
+        return tools_api_call(
+            service=service,
+            api=function_name,
+            parameters=bound_args.arguments,
+            ctx=None
+        )
+
+    func = types.FunctionType(
+        func_code.__code__,
+        globals(),
+        function_name,
+        None,
+        func_code.__closure__
+    )
+    func.__signature__ = signature
+    func.__annotations__ = annotations
+    func.__defaults__ = tuple(defaults.values())
+    func.__doc__ = description
+
+    return func
+
+
+def create_and_decorate_tool(mcp: FastMCP, service: str, api: str):
+    """Create a tool function for a Lambda function."""
+    api_meta, _ = ApiMetaClient.get_api_meta(service, api)
+    fields = create_function_schemas(service, api, api_meta).get(api, {})
+    description = api_meta.get('summary', '')
+    dynamic_lambda = create_tool_function_with_signature(service, api, fields, description)
+    decorated_function = mcp.tool(name=api)(dynamic_lambda)
+
+    return decorated_function
+
+
+@click.command()
+@click.option(
+    "--transport",
+    type=click.Choice(["stdio", "sse"]),
+    default="stdio",
+    help="Transport type",
+)
+def main(transport: str):
+    # Create an MCP server
+    mcp = FastMCP("alibaba-cloud-ops-mcp-server")
+    for tool in oos_tools.tools:
+        mcp.add_tool(tool)
+    for tool in cms_tools.tools:
+        mcp.add_tool(tool)
+    for service_code, apis in config.items():
+        for api_name in apis:
+            create_and_decorate_tool(mcp, service_code, api_name)
+
+    # 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.7.2.dist-info/METADATA

@@ -0,0 +1,72 @@
+Metadata-Version: 2.4
+Name: alibaba-cloud-ops-mcp-server
+Version: 0.7.2
+Summary: A MCP server for Alibaba Cloud
+Author-email: Zheng Dayu <dayu.zdy@alibaba-inc.com>
+License-File: LICENSE
+Requires-Python: >=3.11
+Requires-Dist: alibabacloud-cms20190101>=3.1.4
+Requires-Dist: alibabacloud-ecs20140526>=6.1.0
+Requires-Dist: alibabacloud-oos20190601>=3.4.1
+Requires-Dist: click>=8.1.8
+Requires-Dist: mcp[cli]>=1.6.0
+Description-Content-Type: text/markdown
+
+# alibaba-cloud-ops-mcp-server
+
+## Prepare
+
+Install [uv](https://github.com/astral-sh/uv)
+
+```bash
+# On macOS and Linux.
+curl -LsSf https://astral.sh/uv/install.sh | sh
+```
+
+## Start
+
+Start from local development environment
+
+```bash
+ALIBABA_CLOUD_ACCESS_KEY_ID=<Your AccessKeyId> ALIBABA_CLOUD_ACCESS_KEY_SECRET=<Your AccessKeySecret> uv run src/alibaba_cloud_ops_mcp_server/server.py --transport sse
+```
+
+Start from package
+
+```bash
+ALIBABA_CLOUD_ACCESS_KEY_ID=<Your AccessKeyId> ALIBABA_CLOUD_ACCESS_KEY_SECRET=<Your AccessKeySecret> uvx alibaba-cloud-ops-mcp-server@latest --transport sse
+```
+
+## Debug
+
+Use [VS Code](https://code.visualstudio.com/) + [Cline](https://cline.bot/) to config MCP Server.
+
+## Tools
+
+| **Product** | **Tool** | **Function** | **Implematation** | **Status** |
+| --- | --- | --- | --- | --- |
+| ECS | RunCommand | Run Command | OOS | Done |
+| | StartInstances | Start Instances | OOS | Done |
+| | StopInstances | Stop Instances | OOS | Done |
+| | RebootInstances | Reboot Instances | OOS | Done |
+| | DescribeInstances | View Instances | API | Done |
+| | DescribeRegions | View Regions | API | Done |
+| | DescribeZones | View Zones | API | Done |
+| | DescribeAvailableResource | View Resource Inventory | API | Done |
+| | DescribeImages | View Images | API | Done |
+| | DescribeSecurityGroups | View Security Groups | API | Done |
+| | RunInstances | Create Instances | OOS | Done |
+| | DeleteInstances | Delete Instances | API | Done |
+| | ResetPassword | Modify Password | OOS | Done |
+| | ReplaceSystemDisk | Replace Operating System | OOS | Done |
+| VPC | DescribeVpcs | View VPCs | API | Done |
+| | DescribeVSwitches | View VSwitches | API | Done |
+| CloudMonitor | GetCpuUsageData | Get CPU Usage Data for ECS Instances | API | Done |
+| | GetCpuLoadavgData | Get CPU One-Minute Average Load Metric Data | API | Done |
+| | GetCpuloadavg5mData | Get CPU Five-Minute Average Load Metric Data | API | Done |
+| | GetCpuloadavg15mData | Get CPU Fifteen-Minute Average Load Metric Data | API | Done |
+| | GetMemUsedData | Get Memory Usage Metric Data | API | Done |
+| | GetMemUsageData | Get Memory Utilization Metric Data | API | Done |
+| | GetDiskUsageData | Get Disk Utilization Metric Data | API | Done |
+| | GetDiskTotalData | Get Total Disk Partition Capacity Metric Data | API | Done |
+| | GetDiskUsedData | Get Disk Partition Usage Metric Data | API | Done |

alibaba_cloud_ops_mcp_server-0.7.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+alibaba_cloud_ops_mcp_server/__init__.py,sha256=BaluUNyRz8Qw-X7Y0ywDezwbkqiSvWlSYn2452XeGcA,213
+alibaba_cloud_ops_mcp_server/api_meta_client.py,sha256=pxJztmkmcqqI3djECziAIMc3ZxsS8crqSc6AbneOX2I,7315
+alibaba_cloud_ops_mcp_server/cms_tools.py,sha256=5IhPtkHSwqLsy7D3lGznovQ94e-XBy6HBE7qLDMoYaQ,4464
+alibaba_cloud_ops_mcp_server/config.py,sha256=MJRa5MLJgftSBitrbzTX8fPAV3-7_7FVxNqPvQyJiac,463
+alibaba_cloud_ops_mcp_server/oos_tools.py,sha256=zXniXIPjTKYYMaP2oS9KtuSAWeLixeGhjqIMNgR4Hqg,7864
+alibaba_cloud_ops_mcp_server/server.py,sha256=MCXk-nVDEuSuTO9E3duRa_r77ALz0KoJzQI94eHCZVw,6612
+alibaba_cloud_ops_mcp_server-0.7.2.dist-info/METADATA,sha256=DlUj7F9mfEAuvtMF4S4n2XgEzjtpGSe28iPwXaNHaJw,2847
+alibaba_cloud_ops_mcp_server-0.7.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+alibaba_cloud_ops_mcp_server-0.7.2.dist-info/entry_points.txt,sha256=ESGAWXKEp184forhs7VzTD4P1AUdZz6vCW6hRUKITGw,83
+alibaba_cloud_ops_mcp_server-0.7.2.dist-info/licenses/LICENSE,sha256=gQgVkp2ttRCjodiPpXZZR-d7JnrYIYNiHk1YDUYgpa4,11331
+alibaba_cloud_ops_mcp_server-0.7.2.dist-info/RECORD,,

alibaba_cloud_ops_mcp_server-0.7.2.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

alibaba_cloud_ops_mcp_server-0.7.2.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+alibaba-cloud-ops-mcp-server = alibaba_cloud_ops_mcp_server:main

alibaba_cloud_ops_mcp_server-0.7.2.dist-info/licenses/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+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.