lightpdf-aipdf-mcp 0.1.114__py3-none-any.whl → 0.1.116__py3-none-any.whl

lightpdf_aipdf_mcp/converter.py

@@ -20,6 +20,7 @@ class InputFormat(str, Enum):
     CAD = "dwg"
     CAJ = "caj"
     OFD = "ofd"
+    HTML = "html"
 
 class OutputFormat(str, Enum):
     """支持的输出文件格式"""
@@ -46,6 +47,7 @@ INPUT_EXTENSIONS = {
     ".dwg": InputFormat.CAD,
     ".caj": InputFormat.CAJ,
     ".ofd": InputFormat.OFD,
+    ".html": InputFormat.HTML,
 }
 
 # 输入格式到可用输出格式的映射
@@ -70,6 +72,7 @@ FORMAT_CONVERSION_MAP = {
     InputFormat.CAD: {OutputFormat.PDF},     # CAD转PDF
     InputFormat.CAJ: {OutputFormat.PDF},     # CAJ转PDF
     InputFormat.OFD: {OutputFormat.PDF},     # OFD转PDF
+    InputFormat.HTML: {OutputFormat.PDF},    # HTML转PDF
 }
 
 # 扩展FileHandler类的方法
@@ -97,6 +100,7 @@ class Converter(BaseApiClient):
         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/conversion"
+        self.api_wkhtmltopdf_url = f"https://{api_endpoint}/tasks/document/wkhtmltopdf"
 
     async def add_page_numbers(self, file_path: str, start_num: int = 1, position: str = "5", margin: int = 30, password: str = None, original_name: Optional[str] = None) -> ConversionResult:
         """为PDF文档添加页码
@@ -327,6 +331,14 @@ class Converter(BaseApiClient):
         if extra_params:
             data.update(extra_params)
         
+        api_url = self.api_base_url
+        # html转pdf特殊接口处理
+        if format == "pdf":
+            # 检查是否html输入
+            ext = self.file_handler.get_file_extension(file_path)
+            if ext == ".html":
+                api_url = self.api_wkhtmltopdf_url
+        
         # 检查是否为OSS路径
         if self.file_handler.is_oss_id(file_path):
             # OSS路径处理方式，与URL类似，但提取resource_id
@@ -334,7 +346,7 @@ class Converter(BaseApiClient):
             # 使用JSON方式时添加Content-Type
             headers["Content-Type"] = "application/json"
             response = await client.post(
-                self.api_base_url,
+                api_url,
                 json=data,
                 headers=headers
             )
@@ -351,7 +363,7 @@ class Converter(BaseApiClient):
             # 使用JSON方式时添加Content-Type
             headers["Content-Type"] = "application/json"
             response = await client.post(
-                self.api_base_url,
+                api_url,
                 json=data,
                 headers=headers
             )
@@ -360,7 +372,7 @@ class Converter(BaseApiClient):
             with open(file_path, "rb") as f:
                 files = {"file": f}
                 response = await client.post(
-                    self.api_base_url,
+                    api_url,
                     files=files,
                     data=data,
                     headers=headers

lightpdf_aipdf_mcp/server.py

@@ -419,7 +419,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/HTML.\n\nDoes not support creating files from content.",
             inputSchema={
                 "type": "object",
                 "properties": {

{lightpdf_aipdf_mcp-0.1.114.dist-info → lightpdf_aipdf_mcp-0.1.116.dist-info}/METADATA

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

{lightpdf_aipdf_mcp-0.1.114.dist-info → lightpdf_aipdf_mcp-0.1.116.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=1SQcPTKXbf6UKqC1zJ4Vi-k7GxGTtPiVKyhGYAvK7DY,16299
+lightpdf_aipdf_mcp/converter.py,sha256=vAFB6XtDFt9NPS9yygKBuL4t43nZU--RqqKPC06Mvkg,16761
 lightpdf_aipdf_mcp/create_pdf.py,sha256=oALIhOBo60D3Gu_li7d7FF0COhFfSTM-BJpc63r9iAs,2465
 lightpdf_aipdf_mcp/editor.py,sha256=Ph_ZZNkDrsSMGarS8I625lgt0mEDOxr4AfDoSiG9Rvc,30000
-lightpdf_aipdf_mcp/server.py,sha256=CS9oEY1fd8ZXtsi1nNraNjMxj_fe6316huL8SjqrW4s,64997
+lightpdf_aipdf_mcp/server.py,sha256=Lwl9kQHLKCQ1szQkkBhw1xcbC3MsWy6fm2_1vBNQGuE,65002
 lightpdf_aipdf_mcp/translator.py,sha256=NbFDz-mZSD4qCNQVyV0W_0x6xXwbqs_7FiBU13JAxZs,4243
-lightpdf_aipdf_mcp-0.1.114.dist-info/METADATA,sha256=no8vn4LbutHZdWoV3eFKdWaBpdcl4lCg_ePTnqmmPo0,8120
-lightpdf_aipdf_mcp-0.1.114.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.114.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.114.dist-info/RECORD,,
+lightpdf_aipdf_mcp-0.1.116.dist-info/METADATA,sha256=YzPPjo3tOVo1pfA_4EI2FxtY3NtoPaPatJ7stiKpxBE,8120
+lightpdf_aipdf_mcp-0.1.116.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.116.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.116.dist-info/RECORD,,