Update TrafficRoute MCP (#87)

Author: kingcean

README.md

@@ -75,7 +75,6 @@ MCP Server列表 与 **[火山引擎大模型生态广场](https://www.volcengin
 
 - **[CDN MCP](https://github.com/volcengine/mcp-server/tree/main/server/mcp_server_cdn)**：火山引擎 CDN 智能分析助手。  
 - **[边缘计算节点 MCP](https://github.com/volcengine/mcp-server/tree/main/server/mcp_server_veen)**：申请、配置、查阅在边缘计算节点，包括虚拟机、镜像、裸金属，及对应的网络配置。  
-- **[Traffic Route MCP](https://github.com/volcengine/mcp-server/tree/main/server/mcp_server_traffic_route)**：各种类型的 DNS 节点链路配置。  
 
 
 ### **大数据**  
@@ -101,6 +100,7 @@ MCP Server列表 与 **[火山引擎大模型生态广场](https://www.volcengin
 
 - **[证书中心 MCP](https://github.com/volcengine/mcp-server/tree/main/server/mcp_server_certificate_center)**：通过自然语言驱动管理证书服务。  
 - **[域名服务 MCP](https://github.com/volcengine/mcp-server/tree/main/server/mcp_server_domain_service)**：通过自然语言高效查询并注册域名。  
+- **[TrafficRoute MCP](https://github.com/volcengine/mcp-server/tree/main/server/mcp_server_traffic_route)**：各种类型的 DNS 节点链路配置。
 
 
 ### **管理与治理**  

server/mcp_server_traffic_route/README.md

@@ -8,11 +8,11 @@ v0.1
 
 ###  Short Description
 
-The DNS routing service that allows users to configure DNS rules to ensure that requests from clients reach the desired service nodes.
+Support for configuring DNS routing rules to ensure that requests from clients reach the desired service nodes.
 
 ### Long Description
 
-The DNS routing service that allows users to configure DNS rules to ensure that requests from clients reach the desired service nodes.
+The DNS routing service that allows users to configure DNS routing rules to ensure that requests from clients reach the desired service nodes.
 
 ## Category
 
@@ -26,20 +26,15 @@ DNS，Domain
 
 This MCP Server product provides the following Tools (capabilities):
 
-### Tool 1: list-zones
+### Tool 1: list_zones
 
 List DNS on TrafficRoute.
 
-### Tool 2: create-zone
+### Tool 2: create_zone
 
-Add a domain.
-After adding, you can use `create-record` to add DNS record.
+Add a domain configuration.
 
-### Tool 3: create-record
-
-Add a DNS record.
-
-### Tool 4: list-records
+### Tool 3: list_records
 
 Get all records of the specific DNS.
 
@@ -89,8 +84,8 @@ Use a client to interact with the server:
         "mcp-server-traffic-route"
       ],
       "env": {
-        "VOLC_ACCESSKEY": "<Your-Volcengine-AK>",
-        "VOLC_SECRETKEY": "<Your-Volcengine-SK>"
+        "VOLCENGINE_ACCESS_KEY": "Your Volcengine AK",
+        "VOLCENGINE_SECRET_KEY": "Your Volcengine SK"
       }
     }
   }

server/mcp_server_traffic_route/README_zh.md

@@ -8,11 +8,11 @@ v0.1
 
 ### 短描述
 
-TrafficRoute 用于配置 DNS 规则，以确保终端请求能够正确快捷到达对应的服务节点。
+支持对各种类型的 DNS 节点链路进行配置。
 
 ### 长描述
 
-TrafficRoute 用于配置 DNS 规则，以确保终端请求能够正确快捷到达对应的服务节点。
+TrafficRoute 用于配置 DNS 链路规则，以确保终端请求能够正确快捷到达对应的服务节点。
 
 ## 分类
 
@@ -26,20 +26,15 @@ DNS，Domain
 
 本 MCP Server 产品提供以下 Tools (工具/能力):
 
-### Tool 1: list-zones
+### Tool 1: list_zones
 
 获取在 TrafficRoute 上的解析域名列表。
 
-### Tool 2: create-zone
+### Tool 2: create_zone
 
-添加域名。
-添加完毕后，后续可以通过 `create-record` 工具来添加解析。
+添加域名配置。
 
-### Tool 3: create-record
-
-给指定域名增加解析记录。
-
-### Tool 4: list-records
+### Tool 3: list_records
 
 获取域名的全部解析记录列表。
 
@@ -89,8 +84,8 @@ uv run mcp-server-traffic-route -t sse
         "mcp-server-traffic-route"
       ],
       "env": {
-        "VOLC_ACCESSKEY": "<Your-Volcengine-AK>",
-        "VOLC_SECRETKEY": "<Your-Volcengine-SK>"
+        "VOLCENGINE_ACCESS_KEY": "Your Volcengine AK",
+        "VOLCENGINE_SECRET_KEY": "Your Volcengine SK"
       }
     }
   }

server/mcp_server_traffic_route/nodejs/src/tools/tr/all.ts

@@ -36,11 +36,11 @@ export const trafficRouteTools: IProxyToolModel[] = [
     name: 'create-record',
     description: '给指定域名增加解析记录。',
     args: {
-      ZID: z.number(),
-      Host: z.string(),
-      Type: z.string(),
-      Value: z.string(),
-      TTL: z.number().default(600).optional(),
+      ZID: z.number().describe('关联的域名 ID'),
+      Host: z.string().describe('主机名'),
+      Type: z.string().describe('解析记录类型'),
+      Value: z.string().describe('解析记录值'),
+      TTL: z.number().default(600).optional().describe('生存时间，单位为秒'),
       Weight: z.number().default(1).optional().describe('权重'),
     },
     action: 'CreateRecord',
@@ -54,8 +54,8 @@ export const trafficRouteTools: IProxyToolModel[] = [
       RecordID: z.string().describe('解析记录 ID'),
       Host: z.string().describe('域名'),
       Line: z.string(),
-      Type: z.string(),
-      Value: z.string(),
+      Type: z.string().describe('解析记录类型'),
+      Value: z.string().describe('解析记录值'),
       TTL: z.number().default(600).optional(),
       Weight: z.number().default(1).optional().describe('权重'),
     },
@@ -67,7 +67,7 @@ export const trafficRouteTools: IProxyToolModel[] = [
     name: 'list-records',
     description: '获取域名的全部解析记录列表。',
     args: {
-      ZID: z.number()
+      ZID: z.number().describe('关联的域名 ID')
     },
     action: 'ListRecords',
     service: trafficRouteService,

server/mcp_server_traffic_route/nodejs/src/utils/tools.ts

@@ -119,6 +119,16 @@ const store = {
   services: {} as Record<string, Service>
 }
 
+/**
+ * Gets the access key ID and secret key from environment variables.
+ */
+function aksk() {
+  return {
+    ak: process.env.VOLCENGINE_ACCESS_KEY ?? process.env.VOLC_ACCESS_KEY_ID,
+    sk: process.env.VOLCENGINE_SECRET_KEY ?? process.env.VOLC_ACCESS_KEY_SECRET,
+  }
+}
+
 /**
  * Generates the tool result by text.
  * @param value The input text to convert.
@@ -180,9 +190,10 @@ export function registerService(key: string | undefined, service: Service | Serv
     }
 
     const obj = new Service(service);
-    if (process.env.VOLC_ACCESS_KEY_ID && process.env.VOLC_ACCESS_KEY_SECRET) {
-      obj.setAccessKeyId(process.env.VOLC_ACCESS_KEY_ID);
-      obj.setSecretKey(process.env.VOLC_ACCESS_KEY_SECRET);
+    const { ak, sk } = aksk();
+    if (ak && sk) {
+      obj.setAccessKeyId(ak);
+      obj.setSecretKey(sk);
     }
 
     store.services[key] = obj;
@@ -194,13 +205,14 @@ export function registerService(key: string | undefined, service: Service | Serv
  * Update credential of each service.
  */
 export function batchUpdateAccessKey() {
-  if (!process.env.VOLC_ACCESS_KEY_ID || !process.env.VOLC_ACCESS_KEY_SECRET) return;
+  const { ak, sk } = aksk();
+  if (!ak || !sk) return;
   for (const key in store.services) {
     if (typeof key !== 'string') continue;
     const service = store.services[key];
     if (service instanceof Service) {
-      service.setAccessKeyId(process.env.VOLC_ACCESS_KEY_ID);
-      service.setSecretKey(process.env.VOLC_ACCESS_KEY_SECRET);
+      service.setAccessKeyId(ak);
+      service.setSecretKey(sk);
     }
   }
 }

server/mcp_server_traffic_route/python/vcloud/base/aksk.py

@@ -1,13 +1,13 @@
 import os
 
 def get_ak():
-    s = os.getenv("VOLC_ACCESSKEY")
+    s = os.getenv("VOLCENGINE_ACCESS_KEY")
     if (s is None):
         s = os.getenv("VOLC_ACCESS_KEY_ID")
     return s
 
 def get_sk():
-    s = os.getenv("VOLC_SECRETKEY")
+    s = os.getenv("VOLCENGINE_SECRET_KEY")
     if (s is None):
         s = os.getenv("VOLC_ACCESS_KEY_SECRET")
     return s

server/mcp_server_traffic_route/python/vcloud/base/base_service.py

@@ -1,5 +1,5 @@
 # coding:utf-8
-from src.base.base_trait import BaseTrait  # Modify it if necessary
+from vcloud.base.base_trait import BaseTrait  # Modify it if necessary
 
 
 class BaseService(BaseTrait):

server/mcp_server_traffic_route/python/vcloud/traffic_route/api/api.py

@@ -1,12 +1,19 @@
 from vcloud.base.base_service import BaseService
 from .config import *
-from ...base.aksk import get_ak, get_sk
+import os
+
+class TrafficRouteAPI(BaseService):
 
-class VeenedgeAPI(BaseService):
     def __init__(self):
+        if os.getenv("VOLCENGINE_REGION") is None:
+            region = "cn-north-1"
+        else:
+            region = os.getenv("VOLCENGINE_REGION")
+
         super().__init__(
-            ak=get_ak(),
-            sk=get_sk(),
+            ak=os.getenv("VOLCENGINE_ACCESS_KEY"),
+            sk=os.getenv("VOLCENGINE_SECRET_KEY"),
+            region=region,
             service_info_map=service_info_map,
         )
         self.set_api_info(api_info)