lightpdf-aipdf-mcp 0.1.128__py3-none-any.whl → 0.1.130__py3-none-any.whl

lightpdf_aipdf_mcp/converter.py

@@ -334,9 +334,11 @@ class Converter(BaseApiClient):
         api_url = self.api_base_url
         # html转pdf特殊接口处理
         if format == "pdf":
-            # 检查是否html输入
             ext = self.file_handler.get_file_extension(file_path)
-            if ext == ".html":
+            # 支持直接转PDF的文件后缀
+            direct_pdf_exts = {".docx", ".xlsx", ".pptx", ".jpg", ".jpeg", ".png", ".dwg", ".caj", ".ofd", ".html", ".txt", ".tex"}
+            # 本地.html 或 URL 且后缀不在direct_pdf_exts都走wkhtmltopdf
+            if ext == ".html" or (self.file_handler.is_url(file_path) and ext not in direct_pdf_exts):
                 api_url = self.api_wkhtmltopdf_url
         
         # 检查是否为OSS路径

lightpdf_aipdf_mcp/server.py

@@ -279,10 +279,11 @@ async def process_tool_call(
     """
     file_handler = FileHandler(logger)
     editor = Editor(logger, file_handler)
+    extra_params = operation_config.get("extra_params", {})
+
     # 新增：翻译操作分支
     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,
@@ -304,8 +305,7 @@ async def process_tool_call(
     elif operation_config.get("is_edit_operation"):
         # 编辑操作
         edit_type = operation_config.get("edit_type", "")
-        extra_params = operation_config.get("extra_params")
-        
+
         # 获取操作描述
         edit_map = {
             "decrypt": "解密", 
@@ -319,7 +319,7 @@ async def process_tool_call(
             "remove_margin": "去除白边"
         }
         operation_desc = f"PDF{edit_map.get(edit_type, edit_type)}"
-        
+
         # 处理文件
         results = await process_batch_files(
             file_objects, 
@@ -329,55 +329,60 @@ async def process_tool_call(
             ),
             operation_desc
         )
-        
+
         # 生成报告
-        report_msg = generate_result_report(
-            results
-        )
+        report_msg = generate_result_report(results)
+
     else:
         # 转换操作
         converter = Converter(logger, file_handler)
         format = operation_config.get("format", "")
-        extra_params = operation_config.get("extra_params")
 
-        # 新增：特殊处理PDF转Markdown和TXT转PDF
-        if format == "md":
-            # PDF转Markdown，调用Editor.edit_pdf，传递oss://pdf2md
+        # 新增：特殊处理PDF转Markdown和TEX（LaTeX）
+        if format in ("md", "tex"):
+            oss_map = {
+                "md": ("oss://pdf2md", "PDF转Markdown"),
+                "tex": ("oss://pdf2tex", "PDF转LaTeX")
+            }
+            oss_url, operation_desc = oss_map[format]
+
             results = await process_batch_files(
                 file_objects,
                 logger,
                 lambda file_path, password, original_name: editor.edit_pdf(
                     file_path,
                     edit_type=EditType.EDIT,
-                    extra_params={"pages": [{"url": "oss://pdf2md", "oss_file": ""}]},
+                    extra_params={"pages": [{"url": oss_url, "oss_file": ""}]},
                     password=password,
                     original_name=original_name
                 ),
-                "PDF转Markdown"
+                operation_desc
             )
 
             report_msg = generate_result_report(results)
+
         elif format == "pdf":
             # 只调用一次process_batch_files，在lambda里分流
             async def pdf_convert_dispatcher(file_path, password, original_name):
                 ext = file_handler.get_file_extension(file_path)
-                if ext in [".txt", ".tex"]:
-                    if ext == ".txt":
-                        extra_params = {"pages": [{"url": "oss://txt2pdf", "oss_file": ""}]}
-                    else:
-                        extra_params = {"pages": [{"url": "oss://tex2pdf", "oss_file": ""}]}
+                ext_map = {
+                    ".txt": ("oss://txt2pdf", "TXT转PDF"),
+                    ".tex": ("oss://tex2pdf", "LaTeX转PDF")
+                }
+                if ext in ext_map:
+                    oss_url, operation_desc = ext_map[ext]
                     return await editor.edit_pdf(
                         file_path,
                         edit_type=EditType.EDIT,
-                        extra_params=extra_params,
+                        extra_params={"pages": [{"url": oss_url, "oss_file": ""}]},
                         password=password,
                         original_name=original_name
                     )
                 else:
-                    extra_params = operation_config.get("extra_params")
                     return await process_conversion_file(
                         file_path, format, converter, extra_params, password, original_name
                     )
+
             results = await process_batch_files(
                 file_objects,
                 logger,
@@ -386,6 +391,7 @@ async def process_tool_call(
             )
 
             report_msg = generate_result_report(results)
+
         else:
             # 获取操作描述
             if format == "doc-repair":
@@ -396,7 +402,7 @@ async def process_tool_call(
                 operation_desc = "展平PDF"
             else:
                 operation_desc = f"转换为 {format} 格式"
-            
+
             # 处理文件
             results = await process_batch_files(
                 file_objects,
@@ -406,14 +412,14 @@ async def process_tool_call(
                 ),
                 operation_desc
             )
-            
+
             # 生成报告
             report_msg = generate_result_report(results)
-    
+
     # 如果全部失败，记录错误
     if not any(r.success for r in results):
         await logger.error(report_msg)
-    
+
     return types.TextContent(type="text", text=report_msg)
 
 # 创建Server实例
@@ -428,7 +434,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 (text extraction), CSV, or Markdown (md).\nOther formats (DOCX, XLSX, PPTX, images, CAD, CAJ, OFD, HTML, TEX (LaTeX), TXT) can be converted to PDF. For HTML to PDF, both local HTML files and any web page URL are supported.\n\nPDF to PDF conversion is not supported.\nOnly entire files can be converted.\nContent-based file creation is not supported.",
+            description="Document format conversion tool.\n\nPDF can be converted to: DOCX, XLSX, PPTX, images (including long images), HTML, TXT (text extraction), CSV, MD (Markdown), or TEX (LaTeX).\nOther formats (DOCX, XLSX, PPTX, images, CAD, CAJ, OFD, HTML, TEX (LaTeX), TXT) can be converted to PDF. For HTML to PDF, both local HTML files and any web page URL are supported.\n\nPDF to PDF conversion is not supported.\nOnly entire files can be converted.\nContent-based file creation is not supported.",
             inputSchema={
                 "type": "object",
                 "properties": {
@@ -456,8 +462,8 @@ async def handle_list_tools() -> list[types.Tool]:
                     },
                     "format": {
                         "type": "string",
-                        "description": "Target format. PDF can be converted to: DOCX, XLSX, PPTX, images (including long images), HTML, TXT (text extraction), CSV, or Markdown (md). Other formats (DOCX, XLSX, PPTX, images, CAD, CAJ, OFD, HTML, TEX (LaTeX), TXT) can be converted to PDF. For HTML to PDF, both local HTML files and any web page URL are supported. PDF to PDF conversion is not supported.",
-                        "enum": ["pdf", "docx", "xlsx", "pptx", "jpg", "jpeg", "png", "html", "txt", "csv", "md"]
+                        "description": "Target format. PDF can be converted to: DOCX, XLSX, PPTX, images (including long images), HTML, TXT (text extraction), CSV, MD (Markdown), or TEX (LaTeX). Other formats (DOCX, XLSX, PPTX, images, CAD, CAJ, OFD, HTML, TEX (LaTeX), TXT) can be converted to PDF. For HTML to PDF, both local HTML files and any web page URL are supported. PDF to PDF conversion is not supported.",
+                        "enum": ["pdf", "docx", "xlsx", "pptx", "jpg", "jpeg", "png", "html", "txt", "csv", "md", "tex"]
                     },
                     "merge_all": {
                         "type": "integer",

{lightpdf_aipdf_mcp-0.1.128.dist-info → lightpdf_aipdf_mcp-0.1.130.dist-info}/METADATA

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

{lightpdf_aipdf_mcp-0.1.128.dist-info → lightpdf_aipdf_mcp-0.1.130.dist-info}/RECORD

@@ -1,11 +1,11 @@
 lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
 lightpdf_aipdf_mcp/common.py,sha256=PhTf7Zg6mEgn1rTmJDHotXp-4xb2gWFf-Dy_t25qNdY,6660
-lightpdf_aipdf_mcp/converter.py,sha256=vAFB6XtDFt9NPS9yygKBuL4t43nZU--RqqKPC06Mvkg,16761
+lightpdf_aipdf_mcp/converter.py,sha256=_tXzNZZEaobPRjxXy3v__qHUzbDHIZRTIanmmpCrQB0,17058
 lightpdf_aipdf_mcp/create_pdf.py,sha256=oALIhOBo60D3Gu_li7d7FF0COhFfSTM-BJpc63r9iAs,2465
 lightpdf_aipdf_mcp/editor.py,sha256=cYJ6NlS9q_HJwL-Aw7mVwCT5CECMLWYlmR_ePhw_Ja4,30081
-lightpdf_aipdf_mcp/server.py,sha256=hiqN2U1CUUQxiCoLHdDY7jlPqPil-7zyzK15CpxXzQc,66352
+lightpdf_aipdf_mcp/server.py,sha256=LOjlD8AMJruM0uTB8V4ID3oRAspK65RnRft0OPXfr7w,66254
 lightpdf_aipdf_mcp/translator.py,sha256=NbFDz-mZSD4qCNQVyV0W_0x6xXwbqs_7FiBU13JAxZs,4243
-lightpdf_aipdf_mcp-0.1.128.dist-info/METADATA,sha256=YnRHN_r0Nsq3oMqjBNIkxe2HKRlrnYiSThcypiHdIC0,8120
-lightpdf_aipdf_mcp-0.1.128.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.128.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.128.dist-info/RECORD,,
+lightpdf_aipdf_mcp-0.1.130.dist-info/METADATA,sha256=00bQe_YGPdyVCfCElf8pai1__GpR-jFhYdR7FwPemL4,8120
+lightpdf_aipdf_mcp-0.1.130.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.130.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.130.dist-info/RECORD,,