lightpdf-aipdf-mcp 0.1.87__py3-none-any.whl → 0.1.89__py3-none-any.whl

lightpdf_aipdf_mcp/server.py

@@ -19,6 +19,7 @@ import mcp.types as types
 from .common import BaseResult, Logger, FileHandler
 from .converter import Converter, ConversionResult
 from .editor import Editor, EditResult
+from .translator import Translator, TranslateResult
 
 # 加载环境变量
 load_dotenv()
@@ -277,9 +278,29 @@ async def process_tool_call(
     """
     file_handler = FileHandler(logger)
     editor = Editor(logger, file_handler)
-    
+    # 新增：翻译操作分支
+    if operation_config.get("is_translate_operation"):
+        translator = Translator(logger, file_handler)
+        extra_params = operation_config.get("extra_params", {})
+
+        results = await process_batch_files(
+            file_objects,
+            logger,
+            lambda file_path, password, original_name: translator.translate_pdf(
+                file_path=file_path,
+                source=extra_params.get("source", "auto"),
+                target=extra_params.get("target"),
+                output_type=extra_params.get("output_type", "mono"),
+                password=password,
+                original_name=original_name
+            ),
+            "PDF翻译"
+        )
+
+        report_msg = generate_result_report(results)
+
     # 根据操作类型选择不同的处理逻辑
-    if operation_config.get("is_edit_operation"):
+    elif operation_config.get("is_edit_operation"):
         # 编辑操作
         edit_type = operation_config.get("edit_type", "")
         extra_params = operation_config.get("extra_params")
@@ -397,7 +418,7 @@ async def handle_list_tools() -> list[types.Tool]:
     return [
         types.Tool(
             name="convert_document",
-            description="Document format conversion tool.\n\nPDF can be converted to: DOCX/XLSX/PPTX/Images (including long images)/HTML/TXT (for text extraction)/CSV;\nOther formats can be converted to PDF: DOCX/XLSX/PPTX/Images/CAD/CAJ/OFD.\n\nDoes not support creating files from content",
+            description="Document format conversion tool.\n\nPDF can be converted to: DOCX/XLSX/PPTX/Images (including long images)/HTML/TXT (for text extraction)/CSV;\nOther formats can be converted to PDF: DOCX/XLSX/PPTX/Images/CAD/CAJ/OFD.\n\nDoes not support creating files from content.",
             inputSchema={
                 "type": "object",
                 "properties": {
@@ -408,20 +429,20 @@ async def handle_list_tools() -> list[types.Tool]:
                             "properties": {
                                 "path": {
                                     "type": "string",
-                                    "description": "File URL, must include protocol, supports http/https/oss"
+                                    "description": "File URL, must include protocol, supports http/https/oss."
                                 },
                                 "password": {
                                     "type": "string",
-                                    "description": "Document password, required if the document is password-protected"
+                                    "description": "Document password, required if the document is password-protected."
                                 },
                                 "name": {
                                     "type": "string",
-                                    "description": "Original filename of the document"
+                                    "description": "Original filename of the document."
                                 }
                             },
                             "required": ["path"]
                         },
-                        "description": "List of files to convert, each containing path and optional password"
+                        "description": "List of files to convert, each containing path and optional password."
                     },
                     "format": {
                         "type": "string",
@@ -432,7 +453,7 @@ async def handle_list_tools() -> list[types.Tool]:
                         "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)"
+                        "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"]
@@ -964,7 +985,7 @@ async def handle_list_tools() -> list[types.Tool]:
         ),
         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)",
+            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": {
@@ -1025,7 +1046,96 @@ async def handle_list_tools() -> list[types.Tool]:
                 },
                 "required": ["files"]
             }
-        )
+        ),
+        types.Tool(
+            name="translate_pdf",
+            description="Translate PDF documents from a source language to a target language. Supports mono (target only) or dual (source/target bilingual) output.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "files": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "path": {
+                                    "type": "string",
+                                    "description": "PDF file URL, 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 translate, each containing path and optional password."
+                    },
+                    "source": {
+                        "type": "string",
+                        "description": "Source language. Supports 'auto' for automatic detection.",
+                        "enum": ["auto", "ar", "bg", "cz", "da", "de", "el", "en", "es", "fi", "fr", "hbs", "hi", "hu", "id", "it", "ja", "ko", "ms", "nl", "no", "pl", "pt", "ru", "sl", "sv", "th", "tr", "vi", "zh", "zh-tw"],
+                        "default": "auto"
+                    },
+                    "target": {
+                        "type": "string",
+                        "description": "Target language. Must be specified.",
+                        "enum": ["ar", "bg", "cz", "da", "de", "el", "en", "es", "fi", "fr", "hbs", "hi", "hu", "id", "it", "ja", "ko", "ms", "nl", "no", "pl", "pt", "ru", "sl", "sv", "th", "tr", "vi", "zh", "zh-tw"]
+                    },
+                    "output_type": {
+                        "type": "string",
+                        "description": "Output type: 'mono' for target language only, 'dual' for source/target bilingual output.",
+                        "enum": ["mono", "dual"],
+                        "default": "mono"
+                    }
+                },
+                "required": ["files", "target"]
+            }
+        ),
+        types.Tool(
+            name="resize-pdf",
+            description="Resize PDF pages. You can specify the target page size (a0/a1/a2/a3/a4/a5/a6/letter) and/or the image resolution (dpi, e.g., 72). If not set, the corresponding property will not be changed.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "files": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "path": {
+                                    "type": "string",
+                                    "description": "PDF file URL to resize, 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 resize, each containing path and optional password"
+                    },
+                    "page_size": {
+                        "type": "string",
+                        "description": "Target page size. Any valid page size name is supported (e.g., a4, letter, legal, etc.), or use width,height in points (pt, e.g., 595,842). If not set, page size will not be changed."
+                    },
+                    "resolution": {
+                        "type": "integer",
+                        "description": "Image resolution (dpi), e.g., 72. If not set, resolution will not be changed."
+                    }
+                },
+                "required": ["files"]
+            }
+        ),
     ]
 
 @app.call_tool()
@@ -1052,6 +1162,11 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
             "format": "flatten-pdf",  # 固定format
             "is_edit_operation": False
         },
+        "resize-pdf": {
+            "format": "resize-pdf",
+            "is_edit_operation": False,
+            "param_keys": ["page_size", "resolution"]
+        },
         "unlock_pdf": {
             "edit_type": "decrypt",  # 编辑类型
             "is_edit_operation": True,  # 标记为编辑操作
@@ -1105,6 +1220,10 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
             "is_edit_operation": True,
             "param_keys": []  # 不暴露provider
         },
+        "translate_pdf": {
+            "is_translate_operation": True,
+            "param_keys": ["source", "target", "output_type"]
+        },
     }
     
     DEFAULTS = {
@@ -1120,7 +1239,9 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
         "merge_all": 1,
         "angle": 90,
         "pages": "all",  # 更新默认值为"all"
-        "format": "png"  # 提取图片的默认格式
+        "format": "png",  # 提取图片的默认格式
+        "page_size": "",
+        "resolution": 0,
     }
     
     if name in TOOL_CONFIG:

lightpdf_aipdf_mcp/translator.py

@@ -0,0 +1,101 @@
+from dataclasses import dataclass
+import os
+import httpx
+from typing import Optional, Dict, Any
+from .common import Logger, BaseResult, FileHandler, BaseApiClient
+
+@dataclass
+class TranslateResult(BaseResult):
+    """翻译结果数据类"""
+    pass
+
+class Translator(BaseApiClient):
+    """PDF文档翻译器"""
+    def __init__(self, logger: Logger, file_handler: FileHandler):
+        super().__init__(logger, file_handler)
+        api_endpoint = os.getenv("API_ENDPOINT", "techsz.aoscdn.com/api")
+        self.api_base_url = f"https://{api_endpoint}/tasks/document/transdocument-local"
+
+    async def translate_pdf(self, file_path: str, source: str, target: str, output_type: str = "mono", password: Optional[str] = None, original_name: Optional[str] = None) -> TranslateResult:
+        if not self.api_key:
+            await self.logger.error("未找到API_KEY。请在客户端配置API_KEY环境变量。")
+            return TranslateResult(success=False, file_path=file_path, error_message="未找到API_KEY", original_name=original_name)
+
+        # 构建API参数
+        extra_params = {
+            "source": source or "auto",
+            "target": target,
+            "output_type": output_type or "mono"
+        }
+        if password:
+            extra_params["password"] = password
+        if original_name:
+            extra_params["filename"] = os.path.splitext(original_name)[0]
+
+        async with httpx.AsyncClient(timeout=3600.0) as client:
+            task_id = None
+            try:
+                # 创建翻译任务
+                task_id = await self._create_task(client, file_path, extra_params)
+                # 等待任务完成
+                download_url = await self._wait_for_task(client, task_id, "翻译")
+
+                await self.logger.log("info", "翻译完成。可通过下载链接获取结果文件。")
+                return TranslateResult(
+                    success=True,
+                    file_path=file_path,
+                    error_message=None,
+                    download_url=download_url,
+                    original_name=original_name,
+                    task_id=task_id
+                )
+            except Exception as e:
+                return TranslateResult(
+                    success=False,
+                    file_path=file_path,
+                    error_message=str(e),
+                    download_url=None,
+                    original_name=original_name,
+                    task_id=task_id
+                )
+
+    async def _create_task(self, client: httpx.AsyncClient, file_path: str, extra_params: dict = None) -> str:
+        await self.logger.log("info", "正在提交翻译任务...")
+        headers = {"X-API-KEY": self.api_key}
+        data = {}
+        if extra_params:
+            data.update(extra_params)
+        # 检查是否为OSS路径
+        if self.file_handler.is_oss_id(file_path):
+            data["resource_id"] = file_path.split("oss_id://")[1]
+            headers["Content-Type"] = "application/json"
+            response = await client.post(
+                self.api_base_url,
+                json=data,
+                headers=headers
+            )
+        elif self.file_handler.is_url(file_path):
+            # arxiv.org/pdf/特殊处理
+            if isinstance(file_path, str) and "arxiv.org/pdf/" in file_path:
+                from urllib.parse import urlparse, urlunparse
+                url_obj = urlparse(file_path)
+                if not url_obj.path.endswith(".pdf"):
+                    new_path = url_obj.path + ".pdf"
+                    file_path = urlunparse(url_obj._replace(path=new_path))
+            data["url"] = file_path
+            headers["Content-Type"] = "application/json"
+            response = await client.post(
+                self.api_base_url,
+                json=data,
+                headers=headers
+            )
+        else:
+            with open(file_path, "rb") as f:
+                files = {"file": f}
+                response = await client.post(
+                    self.api_base_url,
+                    files=files,
+                    data=data,
+                    headers=headers
+                )
+        return await self._handle_api_response(response, "创建翻译任务") 

{lightpdf_aipdf_mcp-0.1.87.dist-info → lightpdf_aipdf_mcp-0.1.89.dist-info}/METADATA

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

lightpdf_aipdf_mcp-0.1.89.dist-info/RECORD

@@ -0,0 +1,10 @@
+lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
+lightpdf_aipdf_mcp/common.py,sha256=_UO1f6S9Qr_3k6u5iBpdVDpvTK5U-tHEpu9KsDGqV8Y,6635
+lightpdf_aipdf_mcp/converter.py,sha256=f0gS8tAQlJ8uwJUVUmd9nAA4O9m558e9lAT2B_MxmIo,15135
+lightpdf_aipdf_mcp/editor.py,sha256=9teOqi2y2JbjcCI-kUhYpSXL-F75i7Mfr9E20KKyZP0,29909
+lightpdf_aipdf_mcp/server.py,sha256=uLb_gOB1uP6eN53BDMp2qLXQk_9GjKJjYJDSjb75iyo,62324
+lightpdf_aipdf_mcp/translator.py,sha256=NbFDz-mZSD4qCNQVyV0W_0x6xXwbqs_7FiBU13JAxZs,4243
+lightpdf_aipdf_mcp-0.1.89.dist-info/METADATA,sha256=O87D9f3cGiHt0FbLwdDLb_s5jD-v6c4MBK7X7HbAkkI,8119
+lightpdf_aipdf_mcp-0.1.89.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.89.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.89.dist-info/RECORD,,

lightpdf_aipdf_mcp-0.1.87.dist-info/RECORD

@@ -1,9 +0,0 @@
-lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
-lightpdf_aipdf_mcp/common.py,sha256=_UO1f6S9Qr_3k6u5iBpdVDpvTK5U-tHEpu9KsDGqV8Y,6635
-lightpdf_aipdf_mcp/converter.py,sha256=f0gS8tAQlJ8uwJUVUmd9nAA4O9m558e9lAT2B_MxmIo,15135
-lightpdf_aipdf_mcp/editor.py,sha256=9teOqi2y2JbjcCI-kUhYpSXL-F75i7Mfr9E20KKyZP0,29909
-lightpdf_aipdf_mcp/server.py,sha256=rn9QvHr5E-IaCx2V1pXuKtiXKf1-b3zns-VoxH6x2UE,56220
-lightpdf_aipdf_mcp-0.1.87.dist-info/METADATA,sha256=4WUxVWAPL13KBXsEKfI3iHn4qBxof-GMBNVcXZtLmkQ,8119
-lightpdf_aipdf_mcp-0.1.87.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.87.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.87.dist-info/RECORD,,