lightpdf-aipdf-mcp 0.1.82__py3-none-any.whl → 0.1.83__py3-none-any.whl

lightpdf_aipdf_mcp/server.py

@@ -139,15 +139,8 @@ async def process_conversion_file(
         # 处理extra_params
         if extra_params is None:
             extra_params = {}
-            
-        # 处理is_long_image参数，如果需要转换为长图，则添加merge_all=1参数
-        if extra_params.get("is_long_image") and format in ["jpg", "jpeg", "png"]:
-            extra_params["merge_all"] = 1
-            # 从extra_params中移除is_long_image，因为API不需要这个参数
-            if "is_long_image" in extra_params:
-                del extra_params["is_long_image"]
-                
-        # 对于其他操作，使用convert_file方法
+        # 直接传递 merge_all 参数（如有）
+        # 其它逻辑交由 converter.convert_file 处理
         return await converter.convert_file(file_path, format, extra_params, password, original_name)
 
 async def process_edit_file(
@@ -312,16 +305,17 @@ async def process_tool_call(
         converter = Converter(logger, file_handler)
         format = operation_config.get("format", "")
         extra_params = operation_config.get("extra_params")
-        is_watermark_removal = operation_config.get("is_watermark_removal", False)
-        is_page_numbering = operation_config.get("is_page_numbering", False)
         
         # 获取操作描述
-        if is_watermark_removal:
+        if format == "doc-repair":
             operation_desc = "去除水印"
             task_type = "水印去除"
-        elif is_page_numbering:
+        elif format == "number-pdf":
             operation_desc = "添加页码"
             task_type = "添加页码"
+        elif format == "flatten-pdf":
+            operation_desc = "展平PDF"
+            task_type = "展平PDF"
         else:
             operation_desc = f"转换为 {format} 格式"
             task_type = "转换"
@@ -390,10 +384,11 @@ async def handle_list_tools() -> list[types.Tool]:
                         "description": "Target format",
                         "enum": ["pdf", "docx", "xlsx", "pptx", "jpg", "jpeg", "png", "html", "txt", "csv"]
                     },
-                    "is_long_image": {
-                        "type": "boolean",
-                        "description": "Whether to convert to a long image. Only valid when format is jpg/jpeg/png",
-                        "default": False
+                    "merge_all": {
+                        "type": "integer",
+                        "enum": [0, 1],
+                        "default": 0,
+                        "description": "Whether to merge results: 1 = merge all, 0 = separate. Only valid for: PDF to Excel (1: all pages to one sheet, 0: each page to a sheet), PDF to Image (1: merge to long image, 0: each page to an image), Image to PDF (1: all images to one PDF, 0: each image to a PDF)"
                     }
                 },
                 "required": ["files", "format"]
@@ -862,6 +857,38 @@ async def handle_list_tools() -> list[types.Tool]:
                 },
                 "required": ["files"]
             }
+        ),
+        types.Tool(
+            name="flatten_pdf",
+            description="Flatten PDF files (convert editable elements such as text, form fields, annotations, and layers into non-editable static content or fixed content)",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "files": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "path": {
+                                    "type": "string",
+                                    "description": "PDF file URL to flatten, must include protocol, supports http/https/oss"
+                                },
+                                "password": {
+                                    "type": "string",
+                                    "description": "PDF document password, required if the document is password-protected"
+                                },
+                                "name": {
+                                    "type": "string",
+                                    "description": "Original filename of the document"
+                                }
+                            },
+                            "required": ["path"]
+                        },
+                        "description": "List of PDF files to flatten, each containing path and optional password"
+                    }
+                },
+                "required": ["files"]
+            }
         )
     ]
 
@@ -874,23 +901,21 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
     TOOL_CONFIG = {
         "convert_document": {
             "format_key": "format",  # 从arguments获取format
-            "is_watermark_removal": False,
-            "is_page_numbering": False,
             "is_edit_operation": False,
         },
         "remove_watermark": {
             "format": "doc-repair",  # 固定format
-            "is_watermark_removal": True,
-            "is_page_numbering": False,
             "is_edit_operation": False,
         },
         "add_page_numbers": {
             "format": "number-pdf",  # 固定format
-            "is_watermark_removal": False,
-            "is_page_numbering": True,
             "is_edit_operation": False,
             "param_keys": ["start_num", "position", "margin"]  # 需要从arguments获取的参数
         },
+        "flatten_pdf": {
+            "format": "flatten-pdf",  # 固定format
+            "is_edit_operation": False
+        },
         "unlock_pdf": {
             "edit_type": "decrypt",  # 编辑类型
             "is_edit_operation": True,  # 标记为编辑操作
@@ -995,12 +1020,6 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
             if name == "protect_pdf" and "password" in arguments:
                 operation_config["extra_params"]["password"] = arguments.get("password")
         
-        # 处理convert_document工具的is_long_image参数
-        if name == "convert_document" and "is_long_image" in arguments:
-            if operation_config.get("extra_params") is None:
-                operation_config["extra_params"] = {}
-            operation_config["extra_params"]["is_long_image"] = arguments.get("is_long_image", False)
-            
         # 特殊处理merge_pdfs工具
         if name == "merge_pdfs":
             # 创建编辑器

{lightpdf_aipdf_mcp-0.1.82.dist-info → lightpdf_aipdf_mcp-0.1.83.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lightpdf-aipdf-mcp
-Version: 0.1.82
+Version: 0.1.83
 Summary: MCP Server for LightPDF AI-PDF
 Author: LightPDF Team
 License: Proprietary

{lightpdf_aipdf_mcp-0.1.82.dist-info → lightpdf_aipdf_mcp-0.1.83.dist-info}/RECORD

@@ -2,8 +2,8 @@ lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6
 lightpdf_aipdf_mcp/common.py,sha256=_UO1f6S9Qr_3k6u5iBpdVDpvTK5U-tHEpu9KsDGqV8Y,6635
 lightpdf_aipdf_mcp/converter.py,sha256=f9YuDtOmXBGlMmS3O4Xn3rdWljY9XcNxu0CjftH4s0o,14726
 lightpdf_aipdf_mcp/editor.py,sha256=O7wF_HWs5l-IiXLbZYLNYjj1ygo2v4yGJEYMJtn7jpo,26916
-lightpdf_aipdf_mcp/server.py,sha256=_G85-okjOSfvzBkj_7vehs6caJNeMB1-ACGn5ji6aOI,47557
-lightpdf_aipdf_mcp-0.1.82.dist-info/METADATA,sha256=vkLJNOf6YwBJtswwg4t7eLqE5r1fOogBWK-gyG07Lis,8119
-lightpdf_aipdf_mcp-0.1.82.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.82.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.82.dist-info/RECORD,,
+lightpdf_aipdf_mcp/server.py,sha256=cWekqqhTfJNCyNA6UPgM-XSVhQl2QNireEhUH5uBBgQ,48454
+lightpdf_aipdf_mcp-0.1.83.dist-info/METADATA,sha256=UsoTcuqr1U4RfR1A-d5yJ62b1EQlSlqXRKDi8jkVQAo,8119
+lightpdf_aipdf_mcp-0.1.83.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.83.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.83.dist-info/RECORD,,