alibaba-cloud-ops-mcp-server 0.9.13__py3-none-any.whl → 0.9.14__py3-none-any.whl

alibaba_cloud_ops_mcp_server/server.py

@@ -80,7 +80,13 @@ SUPPORTED_SERVICES_MAP = {
     default="domestic",
     help="Environment type: 'domestic' for domestic, 'international' for overseas (default: domestic)",
 )
-def main(transport: str, port: int, host: str, services: str, headers_credential_only: bool, env: str):
+@click.option(
+    "--code-deploy",
+    is_flag=True,
+    default=False,
+    help="Enable code deploy mode, only load 6 specific tools: OOS_CodeDeploy, OOS_GetDeployStatus, OOS_GetLastDeploymentInfo, LOCAL_ListDirectory, LOCAL_RunShellScript, LOCAL_AnalyzeDeployStack",
+)
+def main(transport: str, port: int, host: str, services: str, headers_credential_only: bool, env: str, code_deploy: bool):
     _setup_logging()
     # Create an MCP server
     mcp = FastMCP(
@@ -152,23 +158,46 @@ def main(transport: str, port: int, host: str, services: str, headers_credential
         settings.headers_credential_only = headers_credential_only
     if env:
         settings.env = env
-    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:
+    if code_deploy:
+        # Code deploy mode: only load 6 specific tools
+        code_deploy_tools = {
+            'OOS_CodeDeploy',
+            'OOS_GetDeployStatus',
+            'OOS_GetLastDeploymentInfo',
+            'ECS_DescribeInstances',
+            'LOCAL_ListDirectory',
+            'LOCAL_RunShellScript',
+            'LOCAL_AnalyzeDeployStack'
+        }
+        
+        # Load from application_management_tools
+        for tool in application_management_tools.tools:
+            if tool.__name__ in code_deploy_tools:
+                mcp.tool(tool)
+        
+        # Load from local_tools
+        for tool in local_tools.tools:
+            if tool.__name__ in code_deploy_tools:
+                mcp.tool(tool)
+    else:
+        # Normal mode: load all tools
+        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 application_management_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)
+        for tool in local_tools.tools:
             mcp.tool(tool)
-    for tool in oos_tools.tools:
-        mcp.tool(tool)
-    for tool in application_management_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)
-    for tool in local_tools.tools:
-        mcp.tool(tool)
 
     # Initialize and run the server
     logger.debug(f'mcp server is running on {transport} mode.')

alibaba_cloud_ops_mcp_server/tools/api_tools.py

@@ -1,4 +1,7 @@
 import os
+import time
+
+from Tea.exceptions import UnretryableException
 from mcp.server.fastmcp import FastMCP, Context
 from pydantic import Field
 import logging
@@ -90,9 +93,9 @@ def create_client(service: str, region_id: str) -> OpenApiClient:
 
 # JSON array parameter of type String
 ECS_LIST_PARAMETERS = {
-    'HpcClusterIds', 'DedicatedHostClusterIds', 'DedicatedHostIds', 
-    'InstanceIds', 'DeploymentSetIds', 'KeyPairNames', 'SecurityGroupIds', 
-    'diskIds', 'repeatWeekdays', 'timePoints', 'DiskIds', 'SnapshotLinkIds', 
+    'HpcClusterIds', 'DedicatedHostClusterIds', 'DedicatedHostIds',
+    'InstanceIds', 'DeploymentSetIds', 'KeyPairNames', 'SecurityGroupIds',
+    'diskIds', 'repeatWeekdays', 'timePoints', 'DiskIds', 'SnapshotLinkIds',
     'EipAddresses', 'PublicIpAddresses', 'PrivateIpAddresses'
 }
 
@@ -109,7 +112,7 @@ def _tools_api_call(service: str, api: str, parameters: dict, ctx: Context):
     method = 'POST' if api_meta.get('methods', ['post'])[0] == 'post' else 'GET'
     path = api_meta.get('path', '/')
     style = ApiMetaClient.get_service_style(service)
-    
+
     # Handling special parameter formats
     processed_parameters = parameters.copy()
     processed_parameters = {k: v for k, v in processed_parameters.items() if v is not None}
@@ -117,7 +120,7 @@ def _tools_api_call(service: str, api: str, parameters: dict, ctx: Context):
         for param_name, param_value in parameters.items():
             if param_name in ECS_LIST_PARAMETERS and isinstance(param_value, list):
                 processed_parameters[param_name] = json.dumps(param_value)
-    
+
     req = open_api_models.OpenApiRequest(
         query=OpenApiUtilClient.query(processed_parameters)
     )
@@ -135,9 +138,34 @@ def _tools_api_call(service: str, api: str, parameters: dict, ctx: Context):
     logger.info(f'Call API Request: Service: {service} API: {api} Method: {method} Parameters: {processed_parameters}')
     client = create_client(service, processed_parameters.get('RegionId', 'cn-hangzhou'))
     runtime = util_models.RuntimeOptions()
-    resp = client.call_api(params, req, runtime)
-    logger.info(f'Call API Response: {resp}')
-    return resp
+    
+    max_retries = 3
+    last_exception = None
+    
+    for attempt in range(max_retries):
+        try:
+            resp = client.call_api(params, req, runtime)
+            logger.info(f'Call API Response: {resp}')
+            return resp
+        except UnretryableException as e:
+            last_exception = e
+            error_msg = str(e)
+            # 检查错误消息是否包含 [Errno 9] Bad file descriptor
+            has_bad_fd = '[Errno 9] Bad file descriptor' in error_msg
+            
+            if has_bad_fd and attempt < max_retries - 1:
+                wait_time = (attempt + 1) * 1  # 递增等待时间：1秒、2秒
+                logger.warning(f'[_tools_api_call] UnretryableException with [Errno 9] Bad file descriptor (attempt {attempt + 1}/{max_retries}), retrying after {wait_time}s: {e}')
+                time.sleep(wait_time)
+            else:
+                # 如果不是可重试的错误，或者已经重试了3次，直接抛出异常
+                logger.error(f'Call API Error: {e}')
+                raise e
+    
+    # 如果所有重试都失败了，抛出最后一次的异常
+    if last_exception:
+        logger.error(f'[_tools_api_call] All retries failed, raising last exception: {last_exception}')
+        raise last_exception
 
 
 def _create_parameter_schema(fields: dict):

alibaba_cloud_ops_mcp_server/tools/application_management_tools.py

@@ -5,7 +5,7 @@ from alibaba_cloud_ops_mcp_server.tools.api_tools import _tools_api_call
 from pathlib import Path
 
 from pydantic import Field
-from typing import Optional, Tuple
+from typing import Optional, Tuple, List
 import json
 import time
 from alibabacloud_oos20190601.client import Client as oos20190601Client
@@ -78,7 +78,7 @@ def OOS_CodeDeploy(
 
 ):
     """
-    通过应用管理 API 部署应用到 ECS 实例。
+    部署应用到阿里云ECS实例。
 
     完整部署流程（在调用此工具之前）：
 
@@ -379,6 +379,30 @@ def OOS_GetDeployStatus(
     return response
 
 
+@_append_tool
+def ECS_DescribeInstances(
+        instance_ids: List[str] = Field(description='AlibabaCloud ECS instance ID List (required)'),
+        region_id: str = Field(description='AlibabaCloud region ID', default='cn-hangzhou'),
+):
+    """
+    查询指定ECS实例的详细信息。此工具要求必须提供实例ID列表，避免随意查询所有实例。
+    注意：此工具仅用于查询用户明确指定的实例信息，不允许用于扫描或枚举所有实例。
+    """
+    logger.info(f"[ECS_DescribeInstances] Input parameters: region_id={region_id}, instance_ids={instance_ids}")
+    
+    if not instance_ids:
+        raise ValueError("instance_ids is required and cannot be empty")
+    
+    describe_instances_request = ecs_20140526_models.DescribeInstancesRequest(
+        region_id=region_id,
+        instance_ids=json.dumps(instance_ids)
+    )
+    
+    response = _describe_instances_with_retry(region_id, describe_instances_request)
+    logger.info(f"[ECS_DescribeInstances] Response: {json.dumps(str(response), ensure_ascii=False)}")
+    return response
+
+
 def _handle_new_application_group(client, name, application_group_name, deploy_region_id,
                                   region_id_oss, bucket_name, object_name, version_id,
                                   is_internal_oss, port, instance_ids, application_start,

{alibaba_cloud_ops_mcp_server-0.9.13.dist-info → alibaba_cloud_ops_mcp_server-0.9.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alibaba-cloud-ops-mcp-server
-Version: 0.9.13
+Version: 0.9.14
 Summary: A MCP server for Alibaba Cloud
 Author-email: Zheng Dayu <dayu.zdy@alibaba-inc.com>, Zhao Shuaibo <zhaoshuaibo.zsb@alibaba-inc.com>
 License-File: LICENSE

{alibaba_cloud_ops_mcp_server-0.9.13.dist-info → alibaba_cloud_ops_mcp_server-0.9.14.dist-info}/RECORD

@@ -1,7 +1,7 @@
 alibaba_cloud_ops_mcp_server/__init__.py,sha256=BaluUNyRz8Qw-X7Y0ywDezwbkqiSvWlSYn2452XeGcA,213
 alibaba_cloud_ops_mcp_server/__main__.py,sha256=Q40p2HtWGvxj1JLvS7dn95NLzDhJNQ6JAgLLyCb4Y50,63
 alibaba_cloud_ops_mcp_server/config.py,sha256=PizctjXsQUWoMWBY1dFjNffVlZr9K6hNvqA4DpayR_o,513
-alibaba_cloud_ops_mcp_server/server.py,sha256=C3vb1VlVo2P_yJ1LjWiRMI8FAMkujUdD6j2NmB36kWs,8005
+alibaba_cloud_ops_mcp_server/server.py,sha256=l-a53wmZPSilVq6TL9UHLw4LboYSSFJhNtEfDW2y-e4,9138
 alibaba_cloud_ops_mcp_server/settings.py,sha256=R1jvMtgErWn_1MZ2Gq3xzBZeMtkGpSHvVutInh0Ix4s,167
 alibaba_cloud_ops_mcp_server/alibabacloud/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alibaba_cloud_ops_mcp_server/alibabacloud/api_meta_client.py,sha256=t2TSc0Gzcy_uEcaCgiHHuLoMiEGu3-NCtYmwYjyPWsY,7973
@@ -10,15 +10,15 @@ alibaba_cloud_ops_mcp_server/alibabacloud/utils.py,sha256=J3fywpnqMvIX19Fw87sKlK
 alibaba_cloud_ops_mcp_server/alibabacloud/static/PROMPT_UNDERSTANDING.md,sha256=QPubudP1bwDbWu0Js6MYb4cJd1B2zM_JGp53twYv5yc,3611
 alibaba_cloud_ops_mcp_server/alibabacloud/static/__init__.py,sha256=wJrYamaIb7e_kA4ILZpdP1f1TUUTXMGqEhA4IbSZ2Ts,230
 alibaba_cloud_ops_mcp_server/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-alibaba_cloud_ops_mcp_server/tools/api_tools.py,sha256=nEtrjp-hFYJs47rFreXJoQ25OSdAUDODSv8GqBtP54Q,9687
-alibaba_cloud_ops_mcp_server/tools/application_management_tools.py,sha256=Y-YidjZ5wwNRE_FMb6_bD7WDav4XjsMa3fbAIusrPRc,34284
+alibaba_cloud_ops_mcp_server/tools/api_tools.py,sha256=GFjummXzLDiPr8QQ_2LYmblhUXRxKw1E6Z1ZazmISSA,10910
+alibaba_cloud_ops_mcp_server/tools/application_management_tools.py,sha256=Bivn1UXBzWcdJj3EWzO4EpXiGAdVn18dZJVxVe1RpvI,35363
 alibaba_cloud_ops_mcp_server/tools/cms_tools.py,sha256=BmPTiP8wu9DsEHBQsvR7JH9nFkcKMTBuNuafFqSfVxU,4308
 alibaba_cloud_ops_mcp_server/tools/common_api_tools.py,sha256=ccQAWqS1I9F-fdOdjLcXN-dIhNqSbZV8T5ODuGXlfXM,2711
 alibaba_cloud_ops_mcp_server/tools/local_tools.py,sha256=Sl7Vl2mVQMLyMIZQDLoE-xVF0vNz_haMEspyYPDe6Sg,13133
 alibaba_cloud_ops_mcp_server/tools/oos_tools.py,sha256=cPEl05Y0rlNvCeO2SF46Y7Ewky4LXZ8__sD0_JCi8ek,13569
 alibaba_cloud_ops_mcp_server/tools/oss_tools.py,sha256=6yKo1FqQN3n9I-eDUW8MrnIZTHthy-worKc8XIsn-Nw,7427
-alibaba_cloud_ops_mcp_server-0.9.13.dist-info/METADATA,sha256=tB3_nzxnrar3_uL0HKcbQsTLLwMXCg86rQJAO6tGJRA,8794
-alibaba_cloud_ops_mcp_server-0.9.13.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-alibaba_cloud_ops_mcp_server-0.9.13.dist-info/entry_points.txt,sha256=ESGAWXKEp184forhs7VzTD4P1AUdZz6vCW6hRUKITGw,83
-alibaba_cloud_ops_mcp_server-0.9.13.dist-info/licenses/LICENSE,sha256=gQgVkp2ttRCjodiPpXZZR-d7JnrYIYNiHk1YDUYgpa4,11331
-alibaba_cloud_ops_mcp_server-0.9.13.dist-info/RECORD,,
+alibaba_cloud_ops_mcp_server-0.9.14.dist-info/METADATA,sha256=EsGiCnz8252l848ovAPAsNXNjCTOJGbRLTg1cO0wjLI,8794
+alibaba_cloud_ops_mcp_server-0.9.14.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+alibaba_cloud_ops_mcp_server-0.9.14.dist-info/entry_points.txt,sha256=ESGAWXKEp184forhs7VzTD4P1AUdZz6vCW6hRUKITGw,83
+alibaba_cloud_ops_mcp_server-0.9.14.dist-info/licenses/LICENSE,sha256=gQgVkp2ttRCjodiPpXZZR-d7JnrYIYNiHk1YDUYgpa4,11331
+alibaba_cloud_ops_mcp_server-0.9.14.dist-info/RECORD,,