Merge pull request #49 from callmefisher/miku_0625

update miku mcp tools

Author: callmefisher

CHANGELOG.md

@@ -1,3 +1,6 @@
+# v1.3.1
+- 直播服务增加更多的tools支持
+
 # v1.3.0
 - 支持直播服务
 

README.md

@@ -21,8 +21,12 @@ Server 来访问七牛云存储、智能多媒体服务等。
   - 根据链接刷新文件
   - 根据链接预取文件
 - 直播
-    - 创建直播空间bucket和直播流
-    - 绑定推拉流域名,获取推拉流直播地址
+    - 创建直播空间bucket
+    - 创建直播流
+    - 获取直播空间列表
+    - 获取流列表
+    - 绑定推拉流域名
+    - 获取推拉流直播地址
     - 获取直播用量
 
 ## 环境要求
@@ -82,25 +86,6 @@ powershell -ExecutionPolicy ByPass -c "irm https://astral.sh/uv/install.ps1 | ie
        }
      }
    }
-   ```
-
-   ```
-   直播mcp配置
-   {
-     "mcpServers": {
-       "qiniu": {
-         "command": "uvx",
-         "args": [
-           "qiniu-mcp-server"
-         ],
-         "env": {
-           "QINIU_API_KEY": "YOUR_API_KEY",
-           "QINIU_ENDPOINT_URL": "YOUR_ENDPOINT_URL"
-        },
-         "disabled": false
-       }
-     }
-   }
    ```
     3. 点击 qiniu MCP Server 的链接开关进行连接
 4. 在 Cline 中创建一个聊天窗口，此时我们可以和 AI 进行交互来使用 qiniu-mcp-server ，下面给出几个示例：

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "qiniu-mcp-server"
-version = "1.3.0"
+version = "1.3.1"
 description = "A MCP server project of Qiniu."
 requires-python = ">=3.12"
 authors = [

src/mcp_server/README.md

@@ -103,6 +103,74 @@
    - Returns:
      - 操作状态信息
 
+
+### 直播 工具
+
+1. `CreateBucket`
+    - 创建新的直播空间
+    - Inputs:
+        - `bucket` (string):  bucket 名称
+    - Returns:
+        - `status_code` (integer): 创建结果code，200代表成功
+
+2. `CreateStream`
+    - 在直播空间下，创建新的直播流
+    - Inputs:
+        - `bucket` (string):  bucket 名称
+        - `stream` (string): 流名称
+    - Returns:
+        - `status_code` (integer): 创建结果code，200代表成功
+3. `ListBuckets`
+    - 查询当前用户配置的 Bucket
+    - Inputs:
+        - `prefix` (string, optional): Bucket 名称前缀，用于筛选特定前缀的 Bucket
+    - Returns:
+        - 满足条件的 Bucket 列表及其详细信息
+4. `ListStreams`
+    - 列举指定 Bucket 中的流列表
+    - Inputs:
+        - `bucket` (string): bucket 名称
+    - Returns:
+        - 满足条件的流列表及其详细信息
+5. `BindPushDomain`
+    - 为 Bucket 绑定推流域名
+    - Inputs:
+        - `bucket` (string): bucket 名称
+        - `domain` (string): domain 名称
+        - `type` (string): 推流 类型,可选项rtmp,whip
+    - Returns:
+        - `status_code` (integer): 创建结果code，200代表成功
+6. `BindPlayDomain`
+    - 为 Bucket 绑定播放域名
+    - Inputs:
+        - `bucket` (string): bucket 名称
+        - `domain` (string): domain 名称
+        - `type` (string): 播放协议类型,可选项srt,hls,flv,whep,live
+    - Returns:
+        - `status_code` (integer): 创建结果code，200代表成功
+7. `GetPushUrls`
+    - 获取推流地址
+    - Inputs:
+        - `bucket` (string): bucket 名称
+        - `stream` (string): 流名称
+    - Returns:
+        - `rtmp` (integer): rtmp推流地址
+        - `whip` (integer): webrtc低延迟直播推流地址
+8. `GetPlayUrls`
+    - 获取播放地址
+    - Inputs:
+        - `bucket` (string): bucket 名称
+        - `stream` (string): 流名称
+    - Returns:
+        - `flv` (integer): flv 拉流地址
+        - `hls` (integer): hls 拉流地址
+        - `whep` (integer): 低延迟直播拉流地址
+9. `QueryLiveTrafficStats`
+    - 查询直播流量
+    - Inputs:
+        - `bucket` (string): bucket 名称
+    - Returns:
+        - `data` (Array of integer): 每5分钟的直播流量
 ### 其他工具
 
 1. `Version`

src/mcp_server/config/config.py

@@ -8,7 +8,8 @@
 
 _CONFIG_ENV_KEY_ACCESS_KEY = "QINIU_ACCESS_KEY"
 _CONFIG_ENV_KEY_SECRET_KEY = "QINIU_SECRET_KEY"
-_CONFIG_ENV_KEY_API_KEY = "QINIU_API_KEY"
+_CONFIG_ENV_LIVE_API_KEY = "QINIU_LIVE_API_KEY"
+_CONFIG_ENV_LIVE_ENDPOINT = "QINIU_LIVE_ENDPOINT"
 _CONFIG_ENV_KEY_ENDPOINT_URL = "QINIU_ENDPOINT_URL"
 _CONFIG_ENV_KEY_REGION_NAME = "QINIU_REGION_NAME"
 _CONFIG_ENV_KEY_BUCKETS = "QINIU_BUCKETS"
@@ -23,7 +24,8 @@
 class Config:
     access_key: str
     secret_key: str
-    api_key: str
+    live_api_key: str
+    live_endpoint: str
     endpoint_url: str
     region_name: str
     buckets: List[str]
@@ -33,7 +35,8 @@ def load_config() -> Config:
     config = Config(
         access_key=os.getenv(_CONFIG_ENV_KEY_ACCESS_KEY),
         secret_key=os.getenv(_CONFIG_ENV_KEY_SECRET_KEY),
-        api_key=os.getenv(_CONFIG_ENV_KEY_API_KEY),
+        live_api_key=os.getenv(_CONFIG_ENV_LIVE_API_KEY),
+        live_endpoint=os.getenv(_CONFIG_ENV_LIVE_ENDPOINT),
         endpoint_url=os.getenv(_CONFIG_ENV_KEY_ENDPOINT_URL),
         region_name=os.getenv(_CONFIG_ENV_KEY_REGION_NAME),
         buckets=_get_configured_buckets_from_env(),
@@ -43,8 +46,10 @@ def load_config() -> Config:
         config.access_key = "YOUR_QINIU_ACCESS_KEY"
     if not config.secret_key or len(config.secret_key) == 0:
         config.secret_key = "YOUR_QINIU_SECRET_KEY"
-    if not config.api_key or len(config.api_key) == 0:
-        config.api_key = "YOUR_QINIU_API_KEY"
+    if not config.live_api_key or len(config.live_api_key) == 0:
+        config.live_api_key = "YOUR_QINIU_LIVE_API_KEY"
+    if not config.live_endpoint or len(config.live_endpoint) == 0:
+        config.live_endpoint = "mls.cn-east-1.qiniumiku.com"
     if not config.endpoint_url or len(config.endpoint_url) == 0:
         config.endpoint_url = "YOUR_QINIU_ENDPOINT_URL"
     if not config.region_name or len(config.region_name) == 0:

src/mcp_server/core/live_streaming/__init__.py

@@ -4,8 +4,8 @@
 
 
 def load(cfg: config.Config):
-    miku = LiveStreamingService(cfg)
-    register_tools(miku)
+    live = LiveStreamingService(cfg)
+    register_tools(live)
 
 
 __all__ = ["load"]