lightpdf-aipdf-mcp 0.1.87__py3-none-any.whl → 0.1.88__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,56 @@ 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"]
+            }
+        ),
     ]
 
 @app.call_tool()
@@ -1105,6 +1175,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 = {

lightpdf_aipdf_mcp/translator.py

@@ -0,0 +1,94 @@
+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):
+            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.88.dist-info}/METADATA

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

lightpdf_aipdf_mcp-0.1.88.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=khv6gJNWx8nhH6gpOu4jdeeL8p7xMPONEv7mAf8oWXQ,59952
+lightpdf_aipdf_mcp/translator.py,sha256=FACnFcnz1zNDdndR3tAgTfDDkfk1rJRRgWorFbiiEUk,3834
+lightpdf_aipdf_mcp-0.1.88.dist-info/METADATA,sha256=AOlJAa2HvSrrcN2I264tvQ5FR3-Gl4Nz7jqo1Ai4Wjs,8119
+lightpdf_aipdf_mcp-0.1.88.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.88.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.88.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,,