http-mcp-client-server 0.1.0__py3-none-any.whl

http_mcp_client/__init__.py

@@ -0,0 +1,4 @@
+"""HTTP MCP Client-Server for video enhancement API."""
+
+__version__ = "0.1.0"
+__all__ = ["__version__"]

http_mcp_client/server.py

@@ -0,0 +1,356 @@
+#!/usr/bin/env python
+"""
+HTTP MCP Server - 提供视频增强功能的 MCP 服务
+
+通过 MCP 协议暴露 tools，内部调用 FastAPI HTTP Server。
+支持 URL 上传和本地文件上传（base64）。
+"""
+
+import argparse
+import asyncio
+import json
+import os
+from pathlib import Path
+from typing import Any
+
+import httpx
+from mcp.server import Server
+from mcp.server.stdio import stdio_server
+from mcp.types import TextContent, Tool
+
+
+class VideoEnhancementMCPServer:
+    """MCP Server for video enhancement"""
+
+    def __init__(self, base_url: str, api_key: str):
+        self.base_url = base_url.rstrip("/")
+        self.api_key = api_key
+        self.server = Server("video-enhancement")
+
+        # 注册 tools
+        self.server.list_tools()(self.list_tools)
+        self.server.call_tool()(self.call_tool)
+
+        # HTTP 客户端
+        self._client = httpx.AsyncClient(
+            base_url=self.base_url,
+            headers={
+                "Authorization": f"Bearer {api_key}",
+                "Content-Type": "application/json",
+            },
+            timeout=60.0,
+        )
+
+    async def list_tools(self) -> list[Tool]:
+        """列出所有可用的 tools"""
+        return [
+            Tool(
+                name="create_task",
+                description="""创建视频增强任务（异步）
+
+支持两种上传方式：
+1. URL 上传：提供视频 URL
+2. 本地上传：提供本地文件路径，MCP Server 自动读取并转为 base64
+
+参数说明：
+- video_source: 视频 URL 或本地文件路径
+- type: "url" 或 "local"
+- resolution: 目标分辨率
+""",
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "video_source": {
+                            "type": "string",
+                            "description": "视频URL地址或本地文件路径",
+                        },
+                        "type": {
+                            "type": "string",
+                            "enum": ["url", "local"],
+                            "default": "url",
+                            "description": "上传类型：url=网络视频，local=本地文件",
+                        },
+                        "resolution": {
+                            "type": "string",
+                            "enum": ["480p", "540p", "720p", "1080p", "2k"],
+                            "default": "720p",
+                            "description": "目标分辨率，默认720p",
+                        },
+                    },
+                    "required": ["video_source"],
+                },
+            ),
+            Tool(
+                name="get_task_status",
+                description="查询视频增强任务状态",
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "task_id": {
+                            "type": "string",
+                            "description": "任务ID",
+                        },
+                    },
+                    "required": ["task_id"],
+                },
+            ),
+            Tool(
+                name="enhance_video_sync",
+                description="""同步增强视频（阻塞等待完成）
+
+支持两种上传方式：
+1. URL 上传：提供视频 URL
+2. 本地上传：提供本地文件路径，MCP Server 自动读取并转为 base64
+
+参数说明：
+- video_source: 视频 URL 或本地文件路径
+- type: "url" 或 "local"
+- resolution: 目标分辨率
+- poll_interval: 轮询间隔（秒）
+- timeout: 超时时间（秒）
+""",
+                inputSchema={
+                    "type": "object",
+                    "properties": {
+                        "video_source": {
+                            "type": "string",
+                            "description": "视频URL地址或本地文件路径",
+                        },
+                        "type": {
+                            "type": "string",
+                            "enum": ["url", "local"],
+                            "default": "url",
+                            "description": "上传类型：url=网络视频，local=本地文件",
+                        },
+                        "resolution": {
+                            "type": "string",
+                            "enum": ["480p", "540p", "720p", "1080p", "2k"],
+                            "default": "720p",
+                            "description": "目标分辨率，默认720p",
+                        },
+                        "poll_interval": {
+                            "type": "number",
+                            "default": 5,
+                            "description": "轮询间隔（秒），默认5",
+                        },
+                        "timeout": {
+                            "type": "number",
+                            "default": 600,
+                            "description": "超时时间（秒），默认600",
+                        },
+                    },
+                    "required": ["video_source"],
+                },
+            ),
+        ]
+
+    async def call_tool(self, name: str, arguments: dict[str, Any]) -> list[TextContent]:
+        """调用 tool"""
+        try:
+            if name == "create_task":
+                result = await self._create_task(
+                    video_source=arguments["video_source"],
+                    source_type=arguments.get("type", "url"),
+                    resolution=arguments.get("resolution", "720p"),
+                )
+            elif name == "get_task_status":
+                result = await self._get_task_status(
+                    task_id=arguments["task_id"],
+                )
+            elif name == "enhance_video_sync":
+                result = await self._enhance_video_sync(
+                    video_source=arguments["video_source"],
+                    source_type=arguments.get("type", "url"),
+                    resolution=arguments.get("resolution", "720p"),
+                    poll_interval=arguments.get("poll_interval", 5),
+                    timeout=arguments.get("timeout", 600),
+                )
+            else:
+                result = {"success": False, "error": f"未知工具: {name}"}
+
+            return [TextContent(type="text", text=json.dumps(result, ensure_ascii=False, indent=2))]
+
+        except Exception as e:
+            return [
+                TextContent(
+                    type="text", text=json.dumps({"success": False, "error": str(e)}, ensure_ascii=False, indent=2)
+                )
+            ]
+
+    def _read_local_file(self, file_path: str) -> tuple[str, str]:
+        """
+        读取本地文件并转为 base64
+
+        Args:
+            file_path: 本地文件路径
+
+        Returns:
+            (base64_data, file_name)
+        """
+        import base64
+
+        path = Path(file_path)
+        if not path.exists():
+            raise FileNotFoundError(f"文件不存在: {file_path}")
+
+        # 检查文件大小（最大 100MB）
+        max_size = 100 * 1024 * 1024
+        file_size = path.stat().st_size
+        if file_size > max_size:
+            raise ValueError("文件大小超过 100MB 限制")
+
+        # 读取并编码
+        with open(path, "rb") as f:
+            file_data = f.read()
+
+        base64_data = base64.b64encode(file_data).decode("utf-8")
+        return base64_data, path.name
+
+    async def _create_task(self, video_source: str, source_type: str, resolution: str) -> dict:
+        """创建任务"""
+        # 根据类型构建请求
+        if source_type == "local":
+            # 本地上传：读取文件转 base64
+            file_data, file_name = self._read_local_file(video_source)
+            content_item = {
+                "type": "video_file",
+                "file_data": file_data,
+                "file_name": file_name,
+            }
+        else:
+            # URL 上传
+            content_item = {
+                "type": "video_url",
+                "video_url": {"url": video_source},
+            }
+
+        payload = {
+            "model": "avc-enhance",
+            "content": [content_item],
+            "resolution": resolution,
+        }
+
+        response = await self._client.post(
+            "/api/v3/contents/generations/tasks",
+            json=payload,
+        )
+        response.raise_for_status()
+
+        data = response.json()
+        if data.get("code", 0) != 0:
+            return {"success": False, "error": data.get("message", "Unknown error")}
+
+        return {
+            "success": True,
+            "task_id": data["data"]["task_id"],
+            "status": data["data"]["status"],
+        }
+
+    async def _get_task_status(self, task_id: str) -> dict:
+        """查询任务状态"""
+        response = await self._client.get(
+            f"/api/v3/contents/generations/tasks/{task_id}",
+        )
+        response.raise_for_status()
+
+        data = response.json()
+        if data.get("code", 0) != 0:
+            return {"success": False, "error": data.get("message", "Unknown error")}
+
+        result = data["data"]
+        return {
+            "success": True,
+            "task_id": result["task_id"],
+            "status": result["status"],
+            "progress": result.get("progress", 0),
+            "video_url": result.get("video_url"),
+            "error_message": result.get("error_message"),
+            "created_at": result.get("created_at"),
+            "updated_at": result.get("updated_at"),
+        }
+
+    async def _enhance_video_sync(
+        self,
+        video_source: str,
+        source_type: str,
+        resolution: str,
+        poll_interval: int,
+        timeout: int,
+    ) -> dict:
+        """同步增强视频"""
+        # 创建任务
+        create_result = await self._create_task(video_source, source_type, resolution)
+        if not create_result["success"]:
+            return create_result
+
+        task_id = create_result["task_id"]
+
+        # 轮询等待完成
+        start_time = asyncio.get_event_loop().time()
+        while True:
+            status = await self._get_task_status(task_id)
+            if not status["success"]:
+                return status
+
+            if status["status"] in ("completed", "failed"):
+                return status
+
+            elapsed = asyncio.get_event_loop().time() - start_time
+            if elapsed >= timeout:
+                return {
+                    "success": False,
+                    "error": f"任务超时: {task_id}",
+                    "task_id": task_id,
+                }
+
+            await asyncio.sleep(poll_interval)
+
+    async def run(self):
+        """运行 MCP Server (stdio mode)"""
+        async with stdio_server() as (read_stream, write_stream):
+            await self.server.run(
+                read_stream,
+                write_stream,
+                self.server.create_initialization_options(),
+            )
+
+    async def close(self):
+        """关闭客户端"""
+        await self._client.aclose()
+
+
+def main():
+    """主入口"""
+    parser = argparse.ArgumentParser(description="HTTP MCP Server for video enhancement")
+    parser.add_argument(
+        "--base-url",
+        default=os.getenv("HTTP_API_BASE_URL", "http://localhost:8000"),
+        help="FastAPI HTTP Server base URL",
+    )
+    parser.add_argument(
+        "--api-key",
+        default=os.getenv("HTTP_API_KEY", ""),
+        help="API key for authentication",
+    )
+
+    args = parser.parse_args()
+
+    if not args.api_key:
+        print("错误: 需要提供 --api-key 或设置 HTTP_API_KEY 环境变量", flush=True)
+        exit(1)
+
+    server = VideoEnhancementMCPServer(
+        base_url=args.base_url,
+        api_key=args.api_key,
+    )
+
+    try:
+        asyncio.run(server.run())
+    except KeyboardInterrupt:
+        print("\n服务已停止", flush=True)
+    finally:
+        asyncio.run(server.close())
+
+
+if __name__ == "__main__":
+    main()

http_mcp_client_server-0.1.0.dist-info/METADATA

@@ -0,0 +1,272 @@
+Metadata-Version: 2.4
+Name: http-mcp-client-server
+Version: 0.1.0
+Summary: HTTP MCP Client-Server for video enhancement API
+Project-URL: Homepage, https://github.com/yourusername/http-mcp-client-server
+Project-URL: Repository, https://github.com/yourusername/http-mcp-client-server
+Project-URL: Issues, https://github.com/yourusername/http-mcp-client-server/issues
+Author-email: Your Name <your.email@example.com>
+License: MIT
+License-File: LICENSE
+Keywords: client,enhancement,fastapi,http,mcp,server,video
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Multimedia :: Video
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: mcp>=1.0.0
+Requires-Dist: pydantic>=2.0.0
+Provides-Extra: dev
+Requires-Dist: build>=1.0.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: ruff>=0.6.0; extra == 'dev'
+Requires-Dist: twine>=5.0.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# http-mcp-client-server (Python)
+
+[![PyPI version](https://badge.fury.io/py/http-mcp-client-server.svg)](https://pypi.org/project/http-mcp-client-server/)
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+基于 MCP 协议的视频增强服务，作为 MCP Client-Server 与 FastAPI HTTP Server 交互。
+
+## 功能
+
+提供以下 MCP Tools：
+- `create_task` - 创建视频增强任务（支持 URL 或本地文件上传）
+- `get_task_status` - 查询任务状态
+- `enhance_video_sync` - 同步增强视频（阻塞等待）
+
+## 安装
+
+### 从 PyPI 安装（推荐）
+
+```bash
+# 使用 pip 安装
+pip install http-mcp-client-server
+
+# 或使用 uv 安装
+uv pip install http-mcp-client-server
+```
+
+### 从源码安装
+
+```bash
+git clone https://github.com/yourusername/http-mcp-client-server.git
+cd python_client
+
+# 使用 uv 安装（推荐）
+uv pip install -e ".[dev]"
+
+# 或使用 pip 安装
+pip install -e ".[dev]"
+```
+
+## 使用方法
+
+### 1. 命令行启动
+
+```bash
+# 直接运行（安装后）
+http-mcp-client-server --base-url http://localhost:8000 --api-key your-api-key
+
+# 或使用环境变量
+export HTTP_API_BASE_URL=http://localhost:8000
+export HTTP_API_KEY=your-api-key
+http-mcp-client-server
+```
+
+### 2. 在 Claude Desktop 中配置
+
+编辑 Claude Desktop 配置文件：
+
+**macOS**: `~/Library/Application Support/Claude/claude_desktop_config.json`
+
+**Windows**: `%APPDATA%/Claude/claude_desktop_config.json`
+
+```json
+{
+  "mcpServers": {
+    "video-enhancement": {
+      "command": "http-mcp-client-server",
+      "args": [
+        "--base-url",
+        "http://localhost:8000",
+        "--api-key",
+        "your-api-key"
+      ]
+    }
+  }
+}
+```
+
+### 3. 使用 uv run 运行（开发模式）
+
+```bash
+uv run http-mcp-client-server --base-url http://localhost:8000 --api-key your-api-key
+```
+
+## 提供的 Tools
+
+### create_task
+
+创建视频增强任务（异步）。
+
+**参数：**
+- `video_source` (string, required): 视频 URL 或本地文件路径
+- `type` (string, optional): 上传类型，默认 "url"
+  - 可选值: `"url"` - 网络视频URL, `"local"` - 本地文件路径
+- `resolution` (string, optional): 目标分辨率，默认 720p
+  - 可选值: 480p, 540p, 720p, 1080p, 2k
+
+**使用示例：**
+
+```python
+# URL 方式
+{
+  "video_source": "https://example.com/video.mp4",
+  "type": "url",
+  "resolution": "1080p"
+}
+
+# 本地文件方式
+{
+  "video_source": "/path/to/local/video.mp4",
+  "type": "local",
+  "resolution": "1080p"
+}
+```
+
+**返回值：**
+```json
+{
+  "success": true,
+  "task_id": "xxx",
+  "status": "wait"
+}
+```
+
+### get_task_status
+
+查询任务状态。
+
+**参数：**
+- `task_id` (string, required): 任务ID
+
+**使用示例：**
+```python
+{
+  "task_id": "task-123-abc"
+}
+```
+
+**返回值：**
+```json
+{
+  "success": true,
+  "task_id": "xxx",
+  "status": "completed",
+  "progress": 100,
+  "video_url": "https://...",
+  "error_message": null,
+  "created_at": "2024-01-01T00:00:00Z",
+  "updated_at": "2024-01-01T00:01:00Z"
+}
+```
+
+### enhance_video_sync
+
+同步增强视频（阻塞等待完成）。
+
+**参数：**
+- `video_source` (string, required): 视频 URL 或本地文件路径
+- `type` (string, optional): 上传类型，默认 "url"
+  - 可选值: `"url"` - 网络视频URL, `"local"` - 本地文件路径
+- `resolution` (string, optional): 目标分辨率，默认 720p
+- `poll_interval` (number, optional): 轮询间隔（秒），默认 5
+- `timeout` (number, optional): 超时时间（秒），默认 600
+
+**使用示例：**
+```python
+{
+  "video_source": "https://example.com/video.mp4",
+  "type": "url",
+  "resolution": "1080p",
+  "poll_interval": 5,
+  "timeout": 600
+}
+```
+
+**返回值：**
+```json
+{
+  "success": true,
+  "task_id": "xxx",
+  "status": "completed",
+  "progress": 100,
+  "video_url": "https://..."
+}
+```
+
+## 文件上传说明
+
+当 `type` 设置为 `"local"` 时，MCP Server 会：
+1. 读取本地文件
+2. 将文件转为 base64 编码
+3. 上传到视频增强服务
+
+**限制：**
+- 最大文件大小：100MB
+
+## 环境变量
+
+| 变量名 | 说明 | 默认值 |
+|--------|------|--------|
+| `HTTP_API_BASE_URL` | FastAPI HTTP Server 地址 | `http://localhost:8000` |
+| `HTTP_API_KEY` | API 认证密钥 | 无 |
+
+## 开发
+
+```bash
+# 克隆仓库
+git clone https://github.com/yourusername/http-mcp-client-server.git
+cd python_client
+
+# 安装开发依赖
+uv pip install -e ".[dev]"
+
+# 运行测试
+pytest
+
+# 代码格式化
+ruff format .
+ruff check --fix .
+```
+
+## 发布到 PyPI
+
+```bash
+# 安装构建工具
+uv pip install build twine
+
+# 构建分发包
+python -m build
+
+# 上传到 PyPI（测试）
+python -m twine upload --repository testpypi dist/*
+
+# 上传到 PyPI（正式）
+python -m twine upload dist/*
+```
+
+## License
+
+MIT License - 详见 [LICENSE](LICENSE) 文件

http_mcp_client_server-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+http_mcp_client/__init__.py,sha256=K7O3bmQwEKwkYvMsIwxLexqH4Kz2TzpsWTrGeSj8dkE,105
+http_mcp_client/server.py,sha256=J8f98ldu_EcfYcxadOcnwm_1oRloQB27Jyy7qDBB4vU,11916
+http_mcp_client_server-0.1.0.dist-info/METADATA,sha256=X6Y0BeiHXZrSNwdms0k2rUBXK6rNttCWf910GCsG3P0,6298
+http_mcp_client_server-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+http_mcp_client_server-0.1.0.dist-info/entry_points.txt,sha256=gnHpFJTKA7FGjMB1MqRALn6iD87wA7LO3Cv9dS0MIhw,71
+http_mcp_client_server-0.1.0.dist-info/licenses/LICENSE,sha256=QvyDolqEVJoMgCxrNb823xH5hJDH0ifntw7vzN-VGD4,1079
+http_mcp_client_server-0.1.0.dist-info/RECORD,,

http_mcp_client_server-0.1.0.dist-info/WHEEL

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

http_mcp_client_server-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+http-mcp-client-server = http_mcp_client.server:main

http_mcp_client_server-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 http-mcp-client-server
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.