feat: support upload

Author: YangSen-qn

README.md

@@ -7,6 +7,20 @@ Server 来访问七牛云存储、智能多媒体服务等。
 
 关于访问七牛云存储详细情况请参考 [基于 MCP 使用大模型访问七牛云存储](https://developer.qiniu.com/kodo/12914/mcp-aimodel-kodo)。
 
+能力集：
+- 存储
+  - 获取 Bucket 列表
+  - 获取 Bucket 中的文件列表
+  - 上传本地文件，以及给出文件内容进行上传
+  - 读取文件内容
+  - 获取文件下载链接
+- 智能多媒体
+  - 图片缩放
+  - 图片切圆角
+- CDN
+  - 根据链接刷新文件
+  - 根据链接预取文件
+
 ## 环境要求
 
 - Python 3.12 或更高版本

src/mcp_server/core/cdn/tools.py

@@ -29,7 +29,7 @@ def __init__(self, cdn: CDNService):
 
     @tools.tool_meta(
         types.Tool(
-            name="CDNPrefetchUrls",
+            name="cdn_prefetch_urls",
             description="Newly added resources are proactively retrieved by the CDN and stored on its cache nodes in advance. Users simply submit the resource URLs, and the CDN automatically triggers the prefetch process.",
             inputSchema={
                 "type": "object",
@@ -76,7 +76,7 @@ def prefetch_urls(self, **kwargs) -> list[types.TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="CDNRefresh",
+            name="cdn_refresh",
             description="This function marks resources cached on CDN nodes as expired. When users access these resources again, the CDN nodes will fetch the latest version from the origin server and store them anew.",
             inputSchema={
                 "type": "object",

src/mcp_server/core/media_processing/tools.py

@@ -17,7 +17,7 @@ def __init__(self, cli: MediaProcessingService):
 
     @tools.tool_meta(
         types.Tool(
-            name="ImageScaleByPercent",
+            name="image_scale_by_percent",
             description="""Image scaling tool that resizes images based on a percentage and returns information about the scaled image.
             The information includes the object_url of the scaled image, which users can directly use for HTTP GET requests to retrieve the image content or open in a browser to view the file.
             The image must be stored in a Qiniu Cloud Bucket.
@@ -70,7 +70,7 @@ def image_scale_by_percent(
 
     @tools.tool_meta(
         types.Tool(
-            name="ImageScaleBySize",
+            name="image_scale_by_size",
             description="""Image scaling tool that resizes images based on a specified width or height and returns information about the scaled image.
             The information includes the object_url of the scaled image, which users can directly use for HTTP GET requests to retrieve the image content or open in a browser to view the file.
             The image must be stored in a Qiniu Cloud Bucket.
@@ -134,7 +134,7 @@ def image_scale_by_size(
 
     @tools.tool_meta(
         types.Tool(
-            name="ImageRoundCorner",
+            name="image_round_corner",
             description="""Image rounded corner tool that processes images based on width, height, and corner radius, returning information about the processed image.
             If only radius_x or radius_y is set, the other parameter will be assigned the same value, meaning horizontal and vertical parameters will be identical.
             The information includes the object_url of the processed image, which users can directly use for HTTP GET requests to retrieve the image content or open in a browser to view the file.
@@ -203,7 +203,7 @@ def image_round_corner(self, **kwargs) -> list[types.TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="ImageInfo",
+            name="image_info",
             description="Retrieves basic image information, including image format, size, and color model.",
             inputSchema={
                 "type": "object",
@@ -240,7 +240,7 @@ def image_info(self, **kwargs) -> list[types.TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="GetFopStatus",
+            name="get_fop_status",
             description="Retrieves the execution status of a Fop operation.",
             inputSchema={
                 "type": "object",

src/mcp_server/core/storage/storage.py

@@ -172,11 +172,11 @@ def upload_text_data(self, bucket: str, key: str, data: str, overwrite: bool = F
         token = self.auth.upload_token(bucket=bucket, key=key, policy=policy)
         ret, info = qiniu.put_data(up_token=token, key=key, data=bytes(data, encoding="utf-8"))
         if info.status_code != 200:
-            raise Exception(f"Failed to upload object: {info.text}")
+            raise Exception(f"Failed to upload object: {info}")
 
         return self.get_object_url(bucket, key)
 
-    def upload_file(self, bucket: str, key: str, file_path: str, overwrite: bool = False) -> list[dict[str:Any]]:
+    def upload_local_file(self, bucket: str, key: str, file_path: str, overwrite: bool = False) -> list[dict[str:Any]]:
         policy = {
             "insertOnly": 1,
         }
@@ -188,14 +188,14 @@ def upload_file(self, bucket: str, key: str, file_path: str, overwrite: bool = F
         token = self.auth.upload_token(bucket=bucket, key=key, policy=policy)
         ret, info = qiniu.put_file(up_token=token, key=key, file_path=file_path)
         if info.status_code != 200:
-            raise Exception(f"Failed to upload object: {info.text}")
+            raise Exception(f"Failed to upload object: {info}")
 
         return self.get_object_url(bucket, key)
 
     def fetch_object(self, bucket: str, key: str, url: str):
         ret, info = self.bucket_manager.fetch(url, bucket, key=key)
         if info.status_code != 200:
-            raise Exception(f"Failed to fetch object: {info.text}")
+            raise Exception(f"Failed to fetch object: {info}")
 
         return self.get_object_url(bucket, key)
 

src/mcp_server/core/storage/tools.py

@@ -18,7 +18,7 @@ def __init__(self, storage: StorageService):
 
     @tools.tool_meta(
         types.Tool(
-            name="ListBuckets",
+            name="list_buckets",
             description="Return the Bucket you configured based on the conditions.",
             inputSchema={
                 "type": "object",
@@ -38,7 +38,7 @@ async def list_buckets(self, **kwargs) -> list[types.TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="ListObjects",
+            name="list_objects",
             description="List objects in Qiniu Cloud, list a part each time, you can set start_after to continue listing, when the number of listed objects is less than max_keys, it means that all files are listed. start_after can be the key of the last file in the previous listing.",
             inputSchema={
                 "type": "object",
@@ -70,7 +70,7 @@ async def list_objects(self, **kwargs) -> list[types.TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="GetObject",
+            name="get_object",
             description="Get an object contents from Qiniu Cloud bucket. In the GetObject request, specify the full key name for the object.",
             inputSchema={
                 "type": "object",
@@ -110,7 +110,7 @@ async def get_object(self, **kwargs) -> list[ImageContent] | list[TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="UploadTextData",
+            name="upload_text_data",
             description="Upload text data to Qiniu bucket.",
             inputSchema={
                 "type": "object",
@@ -121,7 +121,7 @@ async def get_object(self, **kwargs) -> list[ImageContent] | list[TextContent]:
                     },
                     "key": {
                         "type": "string",
-                        "description": "Key of the object to upload. Length Constraints: Minimum length of 1.",
+                        "description": "The key under which a file is saved in Qiniu Cloud Storage serves as the unique identifier for the file within that space, typically using the filename.",
                     },
                     "data": {
                         "type": "string",
@@ -142,7 +142,7 @@ def upload_text_data(self, **kwargs) -> list[types.TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="UploadFile",
+            name="upload_local_file",
             description="Upload a local file to Qiniu bucket.",
             inputSchema={
                 "type": "object",
@@ -153,7 +153,7 @@ def upload_text_data(self, **kwargs) -> list[types.TextContent]:
                     },
                     "key": {
                         "type": "string",
-                        "description": "Key of the object to upload. Length Constraints: Minimum length of 1.",
+                        "description": "The key under which a file is saved in Qiniu Cloud Storage serves as the unique identifier for the file within that space, typically using the filename.",
                     },
                     "file_path": {
                         "type": "string",
@@ -168,13 +168,13 @@ def upload_text_data(self, **kwargs) -> list[types.TextContent]:
             }
         )
     )
-    def upload_file(self, **kwargs) -> list[types.TextContent]:
-        urls = self.storage.upload_file(**kwargs)
+    def upload_local_file(self, **kwargs) -> list[types.TextContent]:
+        urls = self.storage.upload_local_file(**kwargs)
         return [types.TextContent(type="text", text=str(urls))]
 
     @tools.tool_meta(
         types.Tool(
-            name="FetchObject",
+            name="fetch_object",
             description="Fetch a http object to Qiniu bucket.",
             inputSchema={
                 "type": "object",
@@ -185,8 +185,14 @@ def upload_file(self, **kwargs) -> list[types.TextContent]:
                     },
                     "key": {
                         "type": "string",
-                    }
-                }
+                        "description": "The key under which a file is saved in Qiniu Cloud Storage serves as the unique identifier for the file within that space, typically using the filename.",
+                    },
+                    "url": {
+                        "type": "string",
+                        "description": "The URL of the object to fetch.",
+                    },
+                },
+                "required": ["bucket", "key", "url"],
             }
         )
     )
@@ -196,7 +202,7 @@ def fetch_object(self, **kwargs) -> list[types.TextContent]:
 
     @tools.tool_meta(
         types.Tool(
-            name="GetObjectURL",
+            name="get_object_url",
             description="Get the file download URL, and note that the Bucket where the file is located must be bound to a domain name. If using Qiniu Cloud test domain, HTTPS access will not be available, and users need to make adjustments for this themselves.",
             inputSchema={
                 "type": "object",
@@ -207,7 +213,7 @@ def fetch_object(self, **kwargs) -> list[types.TextContent]:
                     },
                     "key": {
                         "type": "string",
-                        "description": "Key of the object to get. Length Constraints: Minimum length of 1.",
+                        "description": "Key of the object to get.",
                     },
                     "disable_ssl": {
                         "type": "boolean",
@@ -235,8 +241,7 @@ def register_tools(storage: StorageService):
             tool_impl.list_objects,
             tool_impl.get_object,
             tool_impl.upload_text_data,
-            tool_impl.upload_file,
-            tool_impl.fetch_object,
+            tool_impl.upload_local_file,
             tool_impl.get_object_url,
         ]
     )

src/mcp_server/core/version/tools.py

@@ -11,7 +11,7 @@ def __init__(self):
 
     @tools.tool_meta(
         types.Tool(
-            name="Version",
+            name="version",
             description="qiniu mcp server version info.",
             inputSchema={
                 "type": "object",