lightpdf-aipdf-mcp 0.1.148__py3-none-any.whl → 0.1.150__py3-none-any.whl

lightpdf_aipdf_mcp/core/__init__.py

@@ -0,0 +1 @@
+"""Core processing logic"""

lightpdf_aipdf_mcp/core/processor.py

@@ -0,0 +1,460 @@
+"""核心处理模块 - 完整的业务逻辑处理"""
+import json
+import asyncio
+from typing import List, Dict, Any, Optional, Callable, TypeVar
+import mcp.types as types
+
+from ..utils.common import Logger, FileHandler, BaseResult
+from ..services.converter import Converter, ConversionResult
+from ..services.editor import Editor, EditResult, EditType
+from ..services.summarizer import Summarizer
+from ..services.ocr import OcrClient
+from ..services.translator import Translator
+
+T = TypeVar('T')
+
+
+def generate_result_report(results: List[BaseResult]) -> str:
+    """生成通用结果报告
+    
+    Args:
+        results: 结果列表
+        
+    Returns:
+        str: JSON格式的报告文本
+    """
+    # 统计结果
+    success_count = sum(1 for r in results if r.success)
+    failed_count = len(results) - success_count
+    
+    # 构建结果JSON对象
+    report_obj = {
+        "total": len(results),
+        "success_count": success_count,
+        "failed_count": failed_count,
+        "success_files": [],
+        "failed_files": []
+    }
+    
+    for result in results:
+        if result.success:
+            # 添加成功的文件信息
+            file_info = {
+                "original_name": result.original_name,
+                "debug": {
+                    "task_id": result.task_id
+                }
+            }
+            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)
+        else:
+            # 添加失败的文件信息
+            file_info = {
+                "error_message": result.error_message,
+                "original_name": result.original_name,
+                "debug": {
+                    "task_id": result.task_id
+                }
+            }
+            report_obj["failed_files"].append(file_info)
+    
+    # 返回JSON字符串
+    return json.dumps(report_obj, ensure_ascii=False)
+
+
+async def process_batch_files(
+    file_objects: List[Dict[str, str]], 
+    logger: Logger, 
+    process_func: Callable[[str, Optional[str], Optional[str]], T],
+    operation_desc: Optional[str] = None
+) -> List[T]:
+    """通用批处理文件函数
+    
+    Args:
+        file_objects: 文件对象列表，每个对象包含path和可选的password及name
+        logger: 日志记录器
+        process_func: 处理单个文件的异步函数，接收file_path、password和original_name参数
+        operation_desc: 操作描述，用于日志记录
+    
+    Returns:
+        List[T]: 处理结果列表
+    """
+    if len(file_objects) > 1 and operation_desc:
+        await logger.log("info", f"开始批量{operation_desc}，共 {len(file_objects)} 个文件")
+        
+        # 并发处理文件，限制并发数为6
+        semaphore = asyncio.Semaphore(6)
+        
+        async def process_with_semaphore(file_obj: Dict[str, str]) -> T:
+            async with semaphore:
+                file_path = file_obj["path"]
+                password = file_obj.get("password")
+                original_name = file_obj.get("name")
+                return await process_func(file_path, password, original_name)
+        
+        # 创建任务列表
+        tasks = [process_with_semaphore(file_obj) for file_obj in file_objects]
+        return await asyncio.gather(*tasks)
+    else:
+        # 单文件处理
+        file_path = file_objects[0]["path"]
+        password = file_objects[0].get("password")
+        original_name = file_objects[0].get("name")
+        return [await process_func(file_path, password, original_name)]
+
+
+async def process_conversion_file(
+    file_path: str, 
+    format: str, 
+    converter: Converter, 
+    extra_params: Optional[Dict[str, Any]] = None, 
+    password: Optional[str] = None,
+    original_name: Optional[str] = None
+) -> ConversionResult:
+    """处理单个文件转换"""
+    is_page_numbering = format == "number-pdf"
+    
+    if is_page_numbering and extra_params:
+        # 对于添加页码，使用add_page_numbers方法
+        return await converter.add_page_numbers(
+            file_path, 
+            extra_params.get("start_num", 1),
+            extra_params.get("position", "5"),
+            extra_params.get("margin", 30),
+            password,
+            original_name
+        )
+    else:
+        # 处理extra_params
+        if extra_params is None:
+            extra_params = {}
+        
+        # 参数名称映射：将image_quality映射为image-quality
+        if "image_quality" in extra_params:
+            extra_params["image-quality"] = extra_params.get("image_quality")
+        
+        # 直接传递 merge_all 参数（如有）
+        # 其它逻辑交由 converter.convert_file 处理
+        return await converter.convert_file(file_path, format, extra_params, password, original_name)
+
+
+async def process_edit_file(
+    file_path: str, 
+    edit_type: str, 
+    editor: Editor, 
+    extra_params: Dict[str, Any] = None,
+    password: Optional[str] = None,
+    original_name: Optional[str] = None
+) -> EditResult:
+    """处理单个文件编辑"""
+    if edit_type == "decrypt":
+        return await editor.decrypt_pdf(file_path, password, original_name)
+    elif edit_type == "add_text_watermark":
+        return await editor.add_text_watermark(
+            file_path=file_path,
+            text=extra_params.get("text", "文本水印"),
+            position=extra_params.get("position", "center"),
+            opacity=extra_params.get("opacity", 1.0),
+            range=extra_params.get("range", ""),
+            layout=extra_params.get("layout", "on"),
+            font_family=extra_params.get("font_family"),
+            font_size=extra_params.get("font_size"),
+            font_color=extra_params.get("font_color"),
+            password=password,
+            original_name=original_name
+        )
+    elif edit_type == "add_image_watermark":
+        return await editor.add_image_watermark(
+            file_path=file_path,
+            image_url=extra_params.get("image_url"),
+            position=extra_params.get("position", "center"),
+            opacity=extra_params.get("opacity", 0.7),
+            range=extra_params.get("range", ""),
+            layout=extra_params.get("layout", "on"),
+            password=password,
+            original_name=original_name
+        )
+    elif edit_type == "encrypt":
+        return await editor.encrypt_pdf(
+            file_path=file_path,
+            password=extra_params.get("password", ""),
+            provider=extra_params.get("provider", ""),
+            original_password=password,
+            original_name=original_name
+        )
+    elif edit_type == "compress":
+        return await editor.compress_pdf(
+            file_path=file_path,
+            image_quantity=extra_params.get("image_quantity", 60),
+            password=password,
+            original_name=original_name
+        )
+    elif edit_type == "split":
+        return await editor.split_pdf(
+            file_path=file_path,
+            pages=extra_params.get("pages", ""),
+            password=password,
+            split_type=extra_params.get("split_type", "page"),
+            merge_all=extra_params.get("merge_all", 1),
+            original_name=original_name
+        )
+    elif edit_type == "merge":
+        # 对于合并操作，我们需要特殊处理，因为它需要处理多个文件
+        return EditResult(
+            success=False, 
+            file_path=file_path, 
+            error_message="合并操作需要使用特殊处理流程",
+            original_name=original_name
+        )
+    elif edit_type == "rotate":
+        # 从extra_params获取旋转参数列表
+        rotation_arguments = extra_params.get("rotates", [])
+        
+        # 验证旋转参数列表
+        if not rotation_arguments:
+            return EditResult(
+                success=False, 
+                file_path=file_path, 
+                error_message="旋转操作需要至少提供一个旋转参数",
+                original_name=original_name
+            )
+        
+        # 构建angle_params字典: {"90": "2-4,6-8", "180": "all"}
+        angle_params = {}
+        for arg in rotation_arguments:
+            angle = str(arg.get("angle", 90))
+            pages = arg.get("pages", "all") or "all"  # 确保空字符串转为"all"
+            angle_params[angle] = pages
+        
+        # 直接调用rotate_pdf方法，传入角度参数字典
+        return await editor.rotate_pdf(
+            file_path=file_path,
+            angle_params=angle_params,
+            password=password,
+            original_name=original_name
+        )
+    elif edit_type == "remove_margin":
+        # 直接调用remove_margin方法，不需要额外参数
+        return await editor.remove_margin(
+            file_path=file_path,
+            password=password,
+            original_name=original_name
+        )
+    elif edit_type == "extract_image":
+        # 调用extract_images方法提取图片
+        return await editor.extract_images(
+            file_path=file_path,
+            format=extra_params.get("format", "png"),
+            password=password,
+            original_name=original_name
+        )
+    else:
+        return EditResult(
+            success=False, 
+            file_path=file_path, 
+            error_message=f"不支持的编辑类型: {edit_type}",
+            original_name=original_name
+        )
+
+
+async def process_tool_call(
+    logger: Logger, 
+    file_objects: List[Dict[str, str]], 
+    operation_config: Dict[str, Any]
+) -> types.TextContent:
+    """通用工具调用处理函数
+    
+    Args:
+        logger: 日志记录器
+        file_objects: 文件对象列表，每个对象包含path和可选的password
+        operation_config: 操作配置，包括操作类型、格式、参数等
+        
+    Returns:
+        types.TextContent: 包含处理结果的文本内容
+    """
+    file_handler = FileHandler(logger)
+    editor = Editor(logger, file_handler)
+    extra_params = operation_config.get("extra_params", {})
+
+    # 新增：摘要操作分支
+    if operation_config.get("is_summarize_operation"):
+        summarizer = Summarizer(logger, file_handler)
+
+        results = await process_batch_files(
+            file_objects,
+            logger,
+            lambda file_path, password, original_name: summarizer.summarize_pdf(
+                file_path=file_path,
+                prompt=extra_params.get("prompt", "Give me a summary of the document."),
+                language=extra_params.get("language", "en"),
+                password=password,
+                original_name=original_name
+            ),
+            "PDF摘要"
+        )
+        report_msg = generate_result_report(results)
+
+    # 新增：OCR操作分支
+    elif operation_config.get("is_ocr_operation"):
+        ocr_client = OcrClient(logger, file_handler)
+
+        results = await process_batch_files(
+            file_objects,
+            logger,
+            lambda file_path, password, original_name: ocr_client.ocr_document(
+                file_path=file_path,
+                format=extra_params.get("format", "pdf"),
+                language=extra_params.get("language", "English,Digits,ChinesePRC"),
+                password=password,
+                original_name=original_name
+            ),
+            "文档OCR识别"
+        )
+        report_msg = generate_result_report(results)
+
+    # 新增：翻译操作分支
+    elif operation_config.get("is_translate_operation"):
+        translator = Translator(logger, file_handler)
+
+        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)
+
+    # 根据操作类型选择不同的处理逻辑
+    elif operation_config.get("is_edit_operation"):
+        # 编辑操作
+        edit_type = operation_config.get("edit_type", "")
+
+        # 获取操作描述
+        edit_map = {
+            "decrypt": "解密", 
+            "add_text_watermark": "添加文本水印", 
+            "add_image_watermark": "添加图片水印", 
+            "encrypt": "加密", 
+            "compress": "压缩", 
+            "split": "拆分", 
+            "merge": "合并", 
+            "rotate": "旋转",
+            "remove_margin": "去除白边"
+        }
+        operation_desc = f"PDF{edit_map.get(edit_type, edit_type)}"
+
+        # 处理文件
+        results = await process_batch_files(
+            file_objects, 
+            logger,
+            lambda file_path, password, original_name: process_edit_file(
+                file_path, edit_type, editor, extra_params, password, original_name
+            ),
+            operation_desc
+        )
+
+        # 生成报告
+        report_msg = generate_result_report(results)
+
+    else:
+        # 转换操作
+        converter = Converter(logger, file_handler)
+        format = operation_config.get("format", "")
+
+        # 新增：特殊处理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_url, "oss_file": ""}]},
+                    password=password,
+                    original_name=original_name
+                ),
+                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)
+                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={"pages": [{"url": oss_url, "oss_file": ""}]},
+                        password=password,
+                        original_name=original_name
+                    )
+                else:
+                    return await process_conversion_file(
+                        file_path, format, converter, extra_params, password, original_name
+                    )
+
+            results = await process_batch_files(
+                file_objects,
+                logger,
+                pdf_convert_dispatcher,
+                f"转换为 {format} 格式"
+            )
+
+            report_msg = generate_result_report(results)
+
+        else:
+            # 获取操作描述
+            if format == "doc-repair":
+                operation_desc = "去除水印"
+            elif format == "number-pdf":
+                operation_desc = "添加页码"
+            elif format == "flatten-pdf":
+                operation_desc = "展平PDF"
+            elif format == "pdf-replace-text":
+                operation_desc = "替换文本"
+            else:
+                operation_desc = f"转换为 {format} 格式"
+
+            # 处理文件
+            results = await process_batch_files(
+                file_objects,
+                logger,
+                lambda file_path, password, original_name: process_conversion_file(
+                    file_path, format, converter, extra_params, password, original_name
+                ),
+                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) 

lightpdf_aipdf_mcp/models/__init__.py

@@ -0,0 +1 @@
+"""Data models and schemas"""

lightpdf_aipdf_mcp/models/schemas.py

@@ -0,0 +1,9 @@
+"""FastMCP数据模型定义"""
+from typing import Optional
+from pydantic import BaseModel, Field
+
+class FileObject(BaseModel):
+    """文件对象模型"""
+    path: str = Field(description="文件URL，必须包含协议，支持http/https/oss")
+    password: Optional[str] = Field(None, description="文档密码，如果文档受密码保护则需要提供")
+    name: Optional[str] = Field(None, description="原始文件名") 

lightpdf_aipdf_mcp/services/__init__.py

@@ -0,0 +1 @@
+"""Business service modules"""

lightpdf_aipdf_mcp/{converter.py → services/converter.py}

@@ -6,7 +6,7 @@ from dataclasses import dataclass
 from enum import Enum
 from typing import List, Optional, Set
 
-from .common import BaseResult, Logger, FileHandler, BaseApiClient
+from ..utils.common import BaseResult, Logger, FileHandler, BaseApiClient, require_api_key
 
 class InputFormat(str, Enum):
     """支持的输入文件格式"""
@@ -21,6 +21,17 @@ class InputFormat(str, Enum):
     CAJ = "caj"
     OFD = "ofd"
     HTML = "html"
+    MARKDOWN = "md"
+    RTF = "rtf"
+    ODG = "odg"  # OpenDocument Graphics
+    ODS = "ods"  # OpenDocument Spreadsheet
+    ODP = "odp"  # OpenDocument Presentation
+    ODT = "odt"  # OpenDocument Text
+    TXT = "txt"  # Plain Text
+    HEIC = "heic"  # High Efficiency Image Container
+    SVG = "svg"  # Scalable Vector Graphics
+    TIFF = "tiff"  # Tagged Image File Format
+    WEBP = "webp"  # WebP Image Format
 
 class OutputFormat(str, Enum):
     """支持的输出文件格式"""
@@ -34,6 +45,9 @@ class OutputFormat(str, Enum):
     HTML = "html"
     TEXT = "txt"
     CSV = "csv"
+    MARKDOWN = "md"  # Markdown
+    RTF = "rtf"  # Rich Text Format
+    TEX = "tex"  # LaTeX
 
 # 文件扩展名到输入格式的映射
 INPUT_EXTENSIONS = {
@@ -48,6 +62,18 @@ INPUT_EXTENSIONS = {
     ".caj": InputFormat.CAJ,
     ".ofd": InputFormat.OFD,
     ".html": InputFormat.HTML,
+    ".md": InputFormat.MARKDOWN,
+    ".rtf": InputFormat.RTF,
+    ".odg": InputFormat.ODG,
+    ".ods": InputFormat.ODS,
+    ".odp": InputFormat.ODP,
+    ".odt": InputFormat.ODT,
+    ".txt": InputFormat.TXT,
+    ".heic": InputFormat.HEIC,
+    ".svg": InputFormat.SVG,
+    ".tiff": InputFormat.TIFF,
+    ".tif": InputFormat.TIFF,  # TIFF的另一种扩展名
+    ".webp": InputFormat.WEBP,
 }
 
 # 输入格式到可用输出格式的映射
@@ -62,6 +88,9 @@ FORMAT_CONVERSION_MAP = {
         OutputFormat.HTML,   # PDF转HTML
         OutputFormat.TEXT,   # PDF转文本
         OutputFormat.CSV,    # PDF转CSV
+        OutputFormat.MARKDOWN,  # PDF转Markdown
+        OutputFormat.RTF,    # PDF转RTF
+        OutputFormat.TEX,    # PDF转LaTeX
     },
     InputFormat.WORD: {OutputFormat.PDF},    # Word转PDF
     InputFormat.EXCEL: {OutputFormat.PDF},   # Excel转PDF
@@ -73,6 +102,18 @@ FORMAT_CONVERSION_MAP = {
     InputFormat.CAJ: {OutputFormat.PDF},     # CAJ转PDF
     InputFormat.OFD: {OutputFormat.PDF},     # OFD转PDF
     InputFormat.HTML: {OutputFormat.PDF},    # HTML转PDF
+    # 新增格式转换支持
+    InputFormat.MARKDOWN: {OutputFormat.PDF},  # Markdown转PDF
+    InputFormat.RTF: {OutputFormat.PDF},       # RTF转PDF
+    InputFormat.ODG: {OutputFormat.PDF},       # ODG转PDF
+    InputFormat.ODS: {OutputFormat.PDF},       # ODS转PDF
+    InputFormat.ODP: {OutputFormat.PDF},       # ODP转PDF
+    InputFormat.ODT: {OutputFormat.PDF},       # ODT转PDF
+    InputFormat.TXT: {OutputFormat.PDF},       # TXT转PDF
+    InputFormat.HEIC: {OutputFormat.PDF},      # HEIC转PDF
+    InputFormat.SVG: {OutputFormat.PDF},       # SVG转PDF
+    InputFormat.TIFF: {OutputFormat.PDF},      # TIFF转PDF
+    InputFormat.WEBP: {OutputFormat.PDF},      # WEBP转PDF
 }
 
 # 扩展FileHandler类的方法
@@ -156,6 +197,7 @@ class Converter(BaseApiClient):
         # 调用convert_file方法处理API请求
         return await self.convert_file(file_path, "number-pdf", extra_params, password, original_name)
         
+    @require_api_key
     async def convert_file(self, file_path: str, format: str, extra_params: dict = None, password: str = None, original_name: Optional[str] = None) -> ConversionResult:
         """转换单个文件
         
@@ -169,10 +211,6 @@ class Converter(BaseApiClient):
         Returns:
             ConversionResult: 转换结果
         """
-        if not self.api_key:
-            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-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
@@ -324,7 +362,7 @@ class Converter(BaseApiClient):
         self.api_base_url = f"https://{self.api_endpoint}/tasks/document/conversion"
         if format == "pdf":
             ext = self.file_handler.get_file_extension(file_path)
-            direct_pdf_exts = {".docx", ".xlsx", ".pptx", ".jpg", ".jpeg", ".png", ".dwg", ".caj", ".ofd", ".html", ".txt", ".tex", ".odt"}
+            direct_pdf_exts = {".docx", ".xlsx", ".pptx", ".jpg", ".jpeg", ".png", ".dwg", ".caj", ".ofd", ".html", ".txt", ".tex", ".odt", ".md", ".rtf", ".odg", ".ods", ".odp", ".heic", ".svg", ".tiff", ".tif", ".webp"}
             if ext == ".html" or (file_path and file_path.startswith(("http://", "https://")) and ext not in direct_pdf_exts):
                 self.api_base_url = self.api_wkhtmltopdf_url
 

lightpdf_aipdf_mcp/{create_pdf.py → services/create_pdf.py}

@@ -4,7 +4,7 @@ from typing import Optional
 import os
 import uuid
 import httpx
-from .common import Logger, FileHandler, BaseResult, BaseApiClient
+from ..utils.common import Logger, FileHandler, BaseResult, BaseApiClient, require_api_key
 from .editor import Editor, EditResult, EditType
 
 @dataclass
@@ -26,6 +26,7 @@ class PDFCreator(BaseApiClient):
             "no": "Norsk", "pl": "Polski", "sv": "Svenska", "tr": "Türkçe"
         }
 
+    @require_api_key
     async def create_pdf_from_prompt(
         self, 
         prompt: str, 

lightpdf_aipdf_mcp/{editor.py → services/editor.py}

@@ -7,7 +7,7 @@ from typing import List, Optional, Dict, Any
 
 import httpx
 
-from .common import BaseResult, Logger, FileHandler, BaseApiClient
+from ..utils.common import BaseResult, Logger, FileHandler, BaseApiClient, require_api_key
 from .converter import InputFormat
 
 class EditType(str, Enum):
@@ -468,6 +468,7 @@ class Editor(BaseApiClient):
         # 调用edit_pdf方法处理API请求
         return await self.edit_pdf(file_path, EditType.EXTRACT_IMAGE, extra_params, password, original_name)
 
+    @require_api_key
     async def edit_pdf(self, file_path: str, edit_type: EditType, extra_params: Dict[str, Any] = None, password: Optional[str] = None, original_name: Optional[str] = None) -> EditResult:
         """编辑PDF文件
         
@@ -486,10 +487,6 @@ class Editor(BaseApiClient):
         Returns:
             EditResult: 编辑结果
         """
-        if not self.api_key:
-            await self.logger.error("未找到API_KEY。请在客户端配置API_KEY环境变量。")
-            return EditResult(success=False, file_path=file_path, error_message="未找到API_KEY。请在客户端配置API_KEY环境变量。", original_name=original_name)
-
         # 验证文件
         exists = await self.file_handler.validate_file_exists(file_path)
         if not exists:

lightpdf_aipdf_mcp/{ocr.py → services/ocr.py}

@@ -2,7 +2,7 @@ from dataclasses import dataclass
 import os
 import httpx
 from typing import Optional, Dict, Any
-from .common import Logger, BaseResult, FileHandler, BaseApiClient
+from ..utils.common import Logger, BaseResult, FileHandler, BaseApiClient, require_api_key
 
 @dataclass
 class OcrResult(BaseResult):
@@ -15,11 +15,8 @@ class OcrClient(BaseApiClient):
         super().__init__(logger, file_handler)
         self.api_base_url = f"https://{self.api_endpoint}/tasks/document/ocr"
 
+    @require_api_key
     async def ocr_document(self, file_path: str, format: str = "pdf", language: Optional[str] = None, password: Optional[str] = None, original_name: Optional[str] = None) -> OcrResult:
-        if not self.api_key:
-            await self.logger.error("未找到API_KEY。请在客户端配置API_KEY环境变量。")
-            return OcrResult(success=False, file_path=file_path, error_message="未找到API_KEY", original_name=original_name)
-
         # 构建API参数
         extra_params = {
             "format": format or "pdf"

lightpdf_aipdf_mcp/{summarizer.py → services/summarizer.py}

@@ -2,7 +2,7 @@ from dataclasses import dataclass
 import os
 import httpx
 from typing import Optional
-from .common import Logger, BaseResult, FileHandler, BaseApiClient
+from ..utils.common import Logger, BaseResult, FileHandler, BaseApiClient, require_api_key
 
 @dataclass
 class SummarizeResult(BaseResult):
@@ -14,11 +14,8 @@ class Summarizer(BaseApiClient):
     def __init__(self, logger: Logger, file_handler: FileHandler):
         super().__init__(logger, file_handler)
 
+    @require_api_key
     async def summarize_pdf(self, file_path: str, prompt: str, language: Optional[str] = None, password: Optional[str] = None, original_name: Optional[str] = None) -> SummarizeResult:
-        if not self.api_key:
-            await self.logger.error("未找到API_KEY。请在客户端配置API_KEY环境变量。")
-            return SummarizeResult(success=False, file_path=file_path, error_message="未找到API_KEY", original_name=original_name)
-
         # 构建API参数
         extra_params = {
             "po": "lightpdf"