lightpdf-aipdf-mcp 0.1.97__py3-none-any.whl → 0.1.99__py3-none-any.whl

lightpdf_aipdf_mcp/common.py

@@ -11,8 +11,8 @@ import httpx
 @dataclass
 class BaseResult:
     """基础结果数据类"""
-    success: bool
-    file_path: str
+    success: bool = False
+    file_path: Optional[str] = None
     error_message: Optional[str] = None
     download_url: Optional[str] = None
     original_name: Optional[str] = None

lightpdf_aipdf_mcp/create_pdf.py

@@ -0,0 +1,56 @@
+"""根据LaTeX内容创建PDF文件的接口（未实现具体逻辑）"""
+from typing import Optional
+import os
+import uuid
+from .common import Logger, FileHandler
+from .editor import Editor, EditResult, EditType
+
+async def create_pdf_from_latex(latex_content: str, output_path: Optional[str] = None, logger: Optional[Logger] = None):
+    """
+    根据LaTeX内容创建PDF文件。
+
+    参数：
+        latex_content (str): LaTeX源内容。
+        output_path (Optional[str]): 可选，输出PDF文件路径。
+        logger (Optional[Logger]): 日志对象。
+    返回：
+        dict: 包含生成结果的信息（如成功与否、PDF路径、错误信息等）。
+    """
+    tex_path = None
+    try:
+        # 1. 保存latex_content为本地.tex文件
+        try:
+            temp_dir = "./tmp"
+            os.makedirs(temp_dir, exist_ok=True)
+            tex_filename = f"latex_{uuid.uuid4().hex}.tex"
+            tex_path = os.path.join(temp_dir, tex_filename)
+            with open(tex_path, "w", encoding="utf-8") as f:
+                f.write(latex_content)
+        except Exception as e:
+            return EditResult(
+                error_message=f"保存LaTeX内容失败: {e}"
+            )
+
+        # 2. 调用Editor.edit_pdf并传递oss://tex2pdf参数生成PDF
+        try:
+            # 构造Logger和FileHandler
+            file_handler = FileHandler(logger)
+            editor = Editor(logger, file_handler)
+
+            # extra_params按txt转pdf方式
+            extra_params = {"pages": '[{"url": "oss://tex2pdf", "oss_file": ""}]'}
+
+            # 这里original_name可用tex_filename
+            result: EditResult = await editor.edit_pdf(tex_path, edit_type=EditType.EDIT, extra_params=extra_params, original_name=tex_filename)
+            # 3. 返回结果
+            return result
+        except Exception as e:
+            return EditResult(
+                error_message=f"PDF生成失败: {e}"
+            )
+    finally:
+        if tex_path and os.path.exists(tex_path):
+            try:
+                os.remove(tex_path)
+            except Exception:
+                pass

lightpdf_aipdf_mcp/editor.py

@@ -19,8 +19,8 @@ class EditType(str, Enum):
     ENCRYPT = "protect"      # 加密PDF
     DECRYPT = "unlock"       # 解密PDF
     ADD_WATERMARK = "watermark"  # 添加水印
-    REMOVE_MARGIN = "edit"   # 去除白边
     EXTRACT_IMAGE = "extract"  # 提取图片
+    EDIT = "edit"            # 编辑操作
 
 @dataclass
 class EditResult(BaseResult):
@@ -430,7 +430,7 @@ class Editor(BaseApiClient):
         }
         
         # 调用edit_pdf方法处理API请求
-        return await self.edit_pdf(file_path, EditType.REMOVE_MARGIN, extra_params, password, original_name)
+        return await self.edit_pdf(file_path, EditType.EDIT, extra_params, password, original_name)
 
     async def extract_images(self, file_path: str, format: str = "png", password: Optional[str] = None, original_name: Optional[str] = None) -> EditResult:
         """从PDF文件中提取图片

lightpdf_aipdf_mcp/server.py

@@ -18,7 +18,7 @@ import mcp.types as types
 # 本地导入
 from .common import BaseResult, Logger, FileHandler
 from .converter import Converter, ConversionResult
-from .editor import Editor, EditResult
+from .editor import Editor, EditResult, EditType
 from .translator import Translator, TranslateResult
 
 # 加载环境变量
@@ -347,7 +347,7 @@ async def process_tool_call(
                 logger,
                 lambda file_path, password, original_name: editor.edit_pdf(
                     file_path,
-                    edit_type="edit",
+                    edit_type=EditType.EDIT,
                     extra_params={"pages": [{"url": "oss://pdf2md", "oss_file": ""}]},
                     password=password,
                     original_name=original_name
@@ -363,7 +363,7 @@ async def process_tool_call(
                 lambda file_path, password, original_name: (
                     editor.edit_pdf(
                         file_path,
-                        edit_type="edit",
+                        edit_type=EditType.EDIT,
                         extra_params={"pages": [{"url": "oss://txt2pdf", "oss_file": ""}]},
                         password=password,
                         original_name=original_name
@@ -1140,6 +1140,20 @@ async def handle_list_tools() -> list[types.Tool]:
                 "required": ["files"]
             }
         ),
+        types.Tool(
+            name="create_pdf",
+            description="Create a PDF file from LaTeX content. Provide LaTeX source code and get a compiled PDF file as output.",
+            inputSchema={
+                "type": "object",
+                "properties": {
+                    "latex_content": {
+                        "type": "string",
+                        "description": "The LaTeX source content to be compiled into a PDF file."
+                    }
+                },
+                "required": ["latex_content"]
+            }
+        ),
     ]
 
 @app.call_tool()
@@ -1341,6 +1355,27 @@ async def handle_call_tool(name: str, arguments: dict | None) -> list[types.Text
         # 调用通用处理函数
         result = await process_tool_call(logger, file_objects, operation_config)
         return [result]
+
+    elif name == "create_pdf":
+        from .create_pdf import create_pdf_from_latex
+        latex_content = arguments.get("latex_content")
+        if not latex_content:
+            error_msg = "latex_content参数不能为空"
+            await logger.error(error_msg)
+            return [types.TextContent(type="text", text=error_msg)]
+
+        # 调用create_pdf_from_latex接口（未实现具体逻辑）
+        result = await create_pdf_from_latex(latex_content, logger=logger)
+        # 构建结果报告
+        report_msg = generate_result_report(
+            [result]
+        )
+        # 如果失败，记录错误
+        if not result.success:
+            await logger.error(report_msg)
+        
+        return [types.TextContent(type="text", text=report_msg)]
+
     else:
         error_msg = f"未知工具: {name}"
         await logger.error(error_msg, ValueError)

{lightpdf_aipdf_mcp-0.1.97.dist-info → lightpdf_aipdf_mcp-0.1.99.dist-info}/METADATA

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

lightpdf_aipdf_mcp-0.1.99.dist-info/RECORD

@@ -0,0 +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=Q_4vXcysqcJ7w6bUKY9JhoBDlCRcAJSpW7p0jh5qekU,16120
+lightpdf_aipdf_mcp/create_pdf.py,sha256=SULJRKFghbQy7VP9GPxS6auy5nMxBN61SxgDr-Wr_lA,2120
+lightpdf_aipdf_mcp/editor.py,sha256=vPaahaAV6pDZi_eheRGSgHzfZxc6ZqQyxHDgsNfdgjQ,29899
+lightpdf_aipdf_mcp/server.py,sha256=LVzQoDvliyvAQVISIoypc8PYyK4oQZU6IhaSRbk7jA8,64112
+lightpdf_aipdf_mcp/translator.py,sha256=NbFDz-mZSD4qCNQVyV0W_0x6xXwbqs_7FiBU13JAxZs,4243
+lightpdf_aipdf_mcp-0.1.99.dist-info/METADATA,sha256=OKX3x6wiU3S5IkRd03x2z-r5sZFkBZIV2GbYRYbHz8A,8119
+lightpdf_aipdf_mcp-0.1.99.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.99.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.99.dist-info/RECORD,,

lightpdf_aipdf_mcp-0.1.97.dist-info/RECORD

@@ -1,10 +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=Q_4vXcysqcJ7w6bUKY9JhoBDlCRcAJSpW7p0jh5qekU,16120
-lightpdf_aipdf_mcp/editor.py,sha256=CjxcZh-zDmb9UjHYj5spfx9HK8j-_voZ3_nmdwWGM8c,29908
-lightpdf_aipdf_mcp/server.py,sha256=40rntJFuBdTj1KMzTYzBL-FuZabmTOpb1_RC-ccJ_k0,62746
-lightpdf_aipdf_mcp/translator.py,sha256=NbFDz-mZSD4qCNQVyV0W_0x6xXwbqs_7FiBU13JAxZs,4243
-lightpdf_aipdf_mcp-0.1.97.dist-info/METADATA,sha256=QN959AXqlGsLIzKv_m60Q6-msz9Twzpgcf7daTs3cHU,8119
-lightpdf_aipdf_mcp-0.1.97.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.97.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.97.dist-info/RECORD,,