lightpdf-aipdf-mcp 0.1.139__py3-none-any.whl → 0.1.141__py3-none-any.whl

lightpdf_aipdf_mcp/common.py

@@ -25,6 +25,7 @@ class Logger:
         self.context = context
         self.collect_info = collect_info
         self._info_log = []
+        self._debug = os.getenv("DEBUG")
 
     async def log(self, level: str, message: str, add_to_result: bool = True):
         """记录日志消息"""
@@ -40,7 +41,8 @@ class Logger:
         
         mcp_level = level_map.get(level.lower(), "info")
         
-        # print(f"mcp_level: {mcp_level}, message: {message}", file=sys.stderr)
+        if self._debug:
+            print(f"mcp_level: {mcp_level}, message: {message}", file=sys.stderr)
         # 直接调用session的send_log_message方法
         await self.context.session.send_log_message(mcp_level, message)
     

lightpdf_aipdf_mcp/converter.py

@@ -173,8 +173,8 @@ class Converter(BaseApiClient):
             await self.logger.error("未找到API_KEY。请在客户端配置API_KEY环境变量。")
             return ConversionResult(success=False, file_path=file_path, error_message="未找到API_KEY", original_name=original_name)
 
-        # 特殊格式：doc-repair用于去除水印，number-pdf用于添加页码，输出均为PDF
-        is_special_operation = format in ["doc-repair", "number-pdf", "flatten-pdf", "resize-pdf"]
+        # 特殊格式：doc-repair用于去除水印，number-pdf用于添加页码，pdf-replace-text用于替换文本，输出均为PDF
+        is_special_operation = format in ["doc-repair", "number-pdf", "flatten-pdf", "resize-pdf", "pdf-replace-text"]
         actual_output_format = "pdf" if is_special_operation else format
 
         # 检查是否为URL或OSS路径，如果是则跳过文件格式检查
@@ -214,6 +214,12 @@ class Converter(BaseApiClient):
                 await self.logger.error(error_msg)
                 return ConversionResult(success=False, file_path=file_path, error_message=error_msg, original_name=original_name)
 
+            # 如果是替换文本操作，检查是否PDF文件
+            if format == "pdf-replace-text" and input_format != InputFormat.PDF:
+                error_msg = "替换文本功能仅支持PDF文件"
+                await self.logger.error(error_msg)
+                return ConversionResult(success=False, file_path=file_path, error_message=error_msg, original_name=original_name)
+
             # 验证输出格式（除去特殊操作外）
             if not is_special_operation:
                 try:
@@ -252,6 +258,8 @@ class Converter(BaseApiClient):
             operation_desc = "展平PDF"
         elif format == "resize-pdf":
             operation_desc = "调整PDF大小"
+        elif format == "pdf-replace-text":
+            operation_desc = "替换文本"
         else:
             if is_remote_path:
                 operation_desc = f"转换为 {output_format.value.upper()} 格式"

lightpdf_aipdf_mcp/server.py

@@ -11,6 +11,9 @@ from urllib.request import url2pathname
 # 第三方库导入
 from dotenv import load_dotenv
 
+# 加载环境变量
+load_dotenv()
+
 # MCP相关导入
 from mcp.server.lowlevel import Server, NotificationOptions
 from mcp.server.models import InitializationOptions
@@ -24,9 +27,6 @@ from .translator import Translator, TranslateResult
 from .summarizer import Summarizer
 from .ocr import OcrClient
 
-# 加载环境变量
-load_dotenv()
-
 # 类型定义
 T = TypeVar('T', bound=BaseResult)
 ProcessFunc = Callable[[str], Any]
@@ -66,6 +66,7 @@ def generate_result_report(
             }
             if hasattr(result, "summary") and result.summary is not None:
                 file_info["summary"] = result.summary
+                file_info["instruction"] = "Return the 'summary' field content directly without any modification or additional processing."
             else:
                 file_info["download_url"] = result.download_url
             report_obj["success_files"].append(file_info)
@@ -444,6 +445,8 @@ async def process_tool_call(
                 operation_desc = "添加页码"
             elif format == "flatten-pdf":
                 operation_desc = "展平PDF"
+            elif format == "pdf-replace-text":
+                operation_desc = "替换文本"
             else:
                 operation_desc = f"转换为 {format} 格式"
 
@@ -1159,6 +1162,46 @@ async def handle_list_tools() -> list[types.Tool]:
                 "required": ["files"]
             }
         ),
+        types.Tool(
+            name="replace_text",
+            description="Replace or delete text in PDF files. When new_text is empty, the old_text will be deleted from the PDF.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "files": {
+                        "type": "array",
+                        "items": {
+                            "type": "object",
+                            "properties": {
+                                "path": {
+                                    "type": "string",
+                                    "description": "PDF file URL to replace text in, 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 replace text in, each containing path and optional password"
+                    },
+                    "old_text": {
+                        "type": "string",
+                        "description": "The text to be replaced or deleted"
+                    },
+                    "new_text": {
+                        "type": "string",
+                        "description": "The replacement text. If empty, the old_text will be deleted"
+                    }
+                },
+                "required": ["files", "old_text", "new_text"]
+            }
+        ),
         types.Tool(
             name="create_pdf",
             description="Create a PDF file from LaTeX source code string only. File upload is NOT supported. If you want to convert a TEX file to PDF, please use the convert_document tool instead. This tool only accepts pure LaTeX code as input.",
@@ -1344,6 +1387,11 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
             "is_edit_operation": False,
             "param_keys": ["page_size", "resolution"]
         },
+        "replace_text": {
+            "format": "pdf-replace-text",
+            "is_edit_operation": False,
+            "param_keys": ["old_text", "new_text"]
+        },
         "unlock_pdf": {
             "edit_type": "decrypt",  # 编辑类型
             "is_edit_operation": True,  # 标记为编辑操作

{lightpdf_aipdf_mcp-0.1.139.dist-info → lightpdf_aipdf_mcp-0.1.141.dist-info}/METADATA

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

{lightpdf_aipdf_mcp-0.1.139.dist-info → lightpdf_aipdf_mcp-0.1.141.dist-info}/RECORD

@@ -1,13 +1,13 @@
 lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
-lightpdf_aipdf_mcp/common.py,sha256=llEFz4XM7TAdmXe1oThEQOi_kyWcCCku5wscLOnMUAk,9147
-lightpdf_aipdf_mcp/converter.py,sha256=6e-p5zh6d5ijXtTgXuBtePp4xEQVMYt6F4j29cj4Kr4,14796
+lightpdf_aipdf_mcp/common.py,sha256=T4WKhtoWvDEosoU36ryZ-7IS62iNm_TL8H_jo6nFf7c,9214
+lightpdf_aipdf_mcp/converter.py,sha256=r8iO5R5vLNNKWdb6WSnwzTwwmp2TvgLXSIvvA4y___o,15336
 lightpdf_aipdf_mcp/create_pdf.py,sha256=oALIhOBo60D3Gu_li7d7FF0COhFfSTM-BJpc63r9iAs,2465
 lightpdf_aipdf_mcp/editor.py,sha256=BR-sWW9L7tybEPOhdc8W-uwdBoom19EPTmGDvy_2gMc,27941
 lightpdf_aipdf_mcp/ocr.py,sha256=IyzxisA6qtXcGTHZofpUYXYDdcIjUaaHcVUKpM7DH9A,2832
-lightpdf_aipdf_mcp/server.py,sha256=zuWZZcTWVIe6jpbDyqihWmyrjOjOSdydtU_EJWw8JFk,75272
+lightpdf_aipdf_mcp/server.py,sha256=gLzV_td5TVkt2bzTvJAFyDiEgJSiAWh4RtJJ-2G4m5c,77622
 lightpdf_aipdf_mcp/summarizer.py,sha256=2QMMgo_xxlEDSd_STPh7-1lBc4VRsL4SPSTijJPyb3I,5456
 lightpdf_aipdf_mcp/translator.py,sha256=nuZa4FpsA0xeRWAEGqSPIM55aJuazJX1m32uajowo7I,2778
-lightpdf_aipdf_mcp-0.1.139.dist-info/METADATA,sha256=UHO-YJeEKU2azol5BSXaLfynR1GOlcuyImnj1_mCeVY,8120
-lightpdf_aipdf_mcp-0.1.139.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.139.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.139.dist-info/RECORD,,
+lightpdf_aipdf_mcp-0.1.141.dist-info/METADATA,sha256=kYlwLqSPgIdI5O4-IE3y14MOdUlbGcHSUAJBXjpgxFE,8120
+lightpdf_aipdf_mcp-0.1.141.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.141.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.141.dist-info/RECORD,,