lightpdf-aipdf-mcp 0.1.41__py3-none-any.whl → 0.1.43__py3-none-any.whl

lightpdf_aipdf_mcp/common.py

@@ -57,12 +57,12 @@ class FileHandler:
     @staticmethod
     def is_url(path: str) -> bool:
         """检查路径是否为URL"""
-        return path.startswith(("http://", "https://"))
+        return path.startswith(("http://", "https://", "oss://"))
         
     @staticmethod
-    def is_oss_path(path: str) -> bool:
-        """检查路径是否为OSS路径"""
-        return path.startswith("oss://")
+    def is_oss_id(path: str) -> bool:
+        """检查路径是否为OSS ID"""
+        return path.startswith("oss_id://")
 
     @staticmethod
     def get_file_extension(file_path: str) -> str:
@@ -101,7 +101,7 @@ class FileHandler:
             bool: 文件是否存在
         """
         is_url = self.is_url(file_path)
-        is_oss = self.is_oss_path(file_path)
+        is_oss = self.is_oss_id(file_path)
         
         # 对于URL或OSS路径，假设它们是有效的
         if is_url or is_oss:

lightpdf_aipdf_mcp/converter.py

@@ -172,7 +172,7 @@ class Converter(BaseApiClient):
         actual_output_format = "pdf" if is_special_operation else format
 
         # 检查是否为URL或OSS路径，如果是则跳过文件格式检查
-        is_remote_path = self.file_handler.is_url(file_path) or self.file_handler.is_oss_path(file_path)
+        is_remote_path = self.file_handler.is_url(file_path) or self.file_handler.is_oss_id(file_path)
         
         if not is_remote_path:
             # 只对本地文件进行格式验证
@@ -295,9 +295,9 @@ class Converter(BaseApiClient):
             data.update(extra_params)
         
         # 检查是否为OSS路径
-        if self.file_handler.is_oss_path(file_path):
+        if self.file_handler.is_oss_id(file_path):
             # OSS路径处理方式，与URL类似，但提取resource_id
-            data["resource_id"] = file_path.split("oss://")[1]
+            data["resource_id"] = file_path.split("oss_id://")[1]
             # 使用JSON方式时添加Content-Type
             headers["Content-Type"] = "application/json"
             response = await client.post(

lightpdf_aipdf_mcp/editor.py

@@ -41,7 +41,7 @@ class Editor(BaseApiClient):
             bool: 如果是PDF文件则返回True，否则返回False
         """
         # 对于URL或OSS路径，跳过文件格式检查
-        if self.file_handler.is_url(file_path) or self.file_handler.is_oss_path(file_path):
+        if self.file_handler.is_url(file_path) or self.file_handler.is_oss_id(file_path):
             return True
             
         input_format = self.file_handler.get_input_format(file_path)
@@ -433,11 +433,11 @@ class Editor(BaseApiClient):
             data.update(extra_params)
         
         # 检查是否为OSS路径
-        if self.file_handler.is_oss_path(file_path):
-            # OSS路径处理方式，与URL类似，但提取resource_id
-            data["resource_id"] = file_path.split("oss://")[1]
+        if self.file_handler.is_oss_id(file_path):
             # 使用JSON方式时添加Content-Type
             headers["Content-Type"] = "application/json"
+            # OSS路径处理方式，与URL类似，但提取resource_id
+            data["resource_id"] = file_path.split("oss_id://")[1]
             response = await client.post(
                 self.api_base_url,
                 json=data,
@@ -445,9 +445,9 @@ class Editor(BaseApiClient):
             )
         # 检查是否为URL路径
         elif self.file_handler.is_url(file_path):
-            data["url"] = file_path
             # 使用JSON方式时添加Content-Type
             headers["Content-Type"] = "application/json"
+            data["url"] = file_path
             response = await client.post(
                 self.api_base_url,
                 json=data,
@@ -492,15 +492,15 @@ class Editor(BaseApiClient):
         
         for i, file_path in enumerate(file_paths):
             # 检查是否为URL或OSS路径
-            if self.file_handler.is_url(file_path):
-                # 对于URL或OSS路径，添加到inputs数组
-                input_item = {"url": file_path}
+            if self.file_handler.is_oss_id(file_path):
+                # 对于OSS路径，添加到inputs数组
+                input_item = {"resource_id": file_path.split("oss_id://")[1]}
                 if password:
                     input_item["password"] = password
                 url_inputs.append(input_item)
-            elif self.file_handler.is_oss_path(file_path):
-                # 对于OSS路径，添加到inputs数组
-                input_item = {"resource_id": file_path.split("oss://")[1]}
+            elif self.file_handler.is_url(file_path):
+                # 对于URL或OSS路径，添加到inputs数组
+                input_item = {"url": file_path}
                 if password:
                     input_item["password"] = password
                 url_inputs.append(input_item)

lightpdf_aipdf_mcp/server.py

@@ -3,6 +3,7 @@
 import asyncio
 import os
 import sys
+import argparse
 from typing import List, Dict, Any, Callable, TypeVar, Optional, Union
 
 # 第三方库导入
@@ -875,19 +876,71 @@ async def main():
     except Exception as e:
         print("LightPDF AI-PDF MCP Server (版本信息获取失败)", file=sys.stderr)
     
-    import mcp.server.stdio as stdio
+    # 解析命令行参数
+    parser = argparse.ArgumentParser(description="LightPDF AI-PDF MCP Server")
+    parser.add_argument("-p", "--port", type=int, default=0, help="指定SSE服务器的端口号，如果提供则使用SSE模式，否则使用stdio模式")
+    args = parser.parse_args()
     
-    async with stdio.stdio_server() as (read_stream, write_stream):
-        await app.run(
-            read_stream,
-            write_stream,
-            app.create_initialization_options(
-                notification_options=NotificationOptions()
+    initialization_options = app.create_initialization_options(
+        notification_options=NotificationOptions()
+    )
+    
+    if args.port:
+        from mcp.server.sse import SseServerTransport
+        from starlette.applications import Starlette
+        from starlette.routing import Mount, Route
+        import uvicorn
+            
+        # 使用SSE服务器
+        print(f"启动SSE服务器，端口号：{args.port}", file=sys.stderr)
+        
+        # 创建SSE传输
+        transport = SseServerTransport("/messages/")
+        
+        # 定义SSE连接处理函数
+        async def handle_sse(request):
+            async with transport.connect_sse(
+                request.scope, request.receive, request._send
+            ) as streams:
+                await app.run(
+                    streams[0], streams[1], initialization_options
+                )
+        
+        # 创建Starlette应用
+        sse_app = Starlette(routes=[
+            Route("/sse/", endpoint=handle_sse),
+            Mount("/messages/", app=transport.handle_post_message),
+        ])
+        
+        # 使用异步方式启动服务器
+        server = uvicorn.Server(uvicorn.Config(
+            app=sse_app,
+            host="0.0.0.0",
+            port=args.port,
+            log_level="warning"
+        ))
+        await server.serve()
+    else:
+        import mcp.server.stdio as stdio
+
+        # 使用stdio服务器
+        print("启动stdio服务器", file=sys.stderr)
+        async with stdio.stdio_server() as (read_stream, write_stream):
+            await app.run(
+                read_stream,
+                write_stream,
+                initialization_options
             )
-        )
 
 def cli_main():
-    asyncio.run(main())
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("服务器被用户中断", file=sys.stderr)
+        sys.exit(0)
+    except Exception as e:
+        print(f"服务器发生错误: {e}", file=sys.stderr)
+        sys.exit(1)
 
 if __name__ == "__main__":
     cli_main()

{lightpdf_aipdf_mcp-0.1.41.dist-info → lightpdf_aipdf_mcp-0.1.43.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lightpdf-aipdf-mcp
-Version: 0.1.41
+Version: 0.1.43
 Summary: MCP Server for LightPDF AI-PDF
 Author: LightPDF Team
 License: Proprietary
@@ -8,9 +8,11 @@ Requires-Python: >=3.8
 Requires-Dist: httpx
 Requires-Dist: httpx-sse
 Requires-Dist: mcp
+Requires-Dist: mcp[cli]
 Requires-Dist: pydantic
 Requires-Dist: pydantic-settings
 Requires-Dist: python-dotenv
+Requires-Dist: starlette
 Requires-Dist: uvicorn
 Description-Content-Type: text/markdown
 

lightpdf_aipdf_mcp-0.1.43.dist-info/RECORD

@@ -0,0 +1,9 @@
+lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
+lightpdf_aipdf_mcp/common.py,sha256=k0oNIKLzqo15UUJAkKJjYZBByLdBGpUPCV59FiqEgcY,6574
+lightpdf_aipdf_mcp/converter.py,sha256=MxjO1yqMqfOQ9B696IyjRbtB3oZpemtD7x8arz52zoc,13312
+lightpdf_aipdf_mcp/editor.py,sha256=CkL5mCM7CSUGD4ZAMoRdgBr3xBpJ-7bEC8InCip5q9A,22802
+lightpdf_aipdf_mcp/server.py,sha256=xErltv5YkV8j3Ja5cm_KgJsc9qqnyCC7RTYs1Id9tWI,37652
+lightpdf_aipdf_mcp-0.1.43.dist-info/METADATA,sha256=0SKzho1zsef_Ar44QCr2aREq7s9R49kfl1FPQRgYZTg,7928
+lightpdf_aipdf_mcp-0.1.43.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lightpdf_aipdf_mcp-0.1.43.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
+lightpdf_aipdf_mcp-0.1.43.dist-info/RECORD,,

lightpdf_aipdf_mcp-0.1.41.dist-info/RECORD

@@ -1,9 +0,0 @@
-lightpdf_aipdf_mcp/__init__.py,sha256=PPnAgpvJLYLVOTxnHDmJAulFnHJD6wuTwS6tRGjqq6s,141
-lightpdf_aipdf_mcp/common.py,sha256=zhd_XxO7cBiQ7A6fc1JJjcpJfRWCQBi3pfBSsrBRusg,6568
-lightpdf_aipdf_mcp/converter.py,sha256=SbP5CpDn1suoq9QSApw6kgqJXEc4gwF7Po3oLo5namg,13313
-lightpdf_aipdf_mcp/editor.py,sha256=6ELqub8uOW6kwkVz52YdFUyDmkMsN1BkTApSw6Bi-lg,22802
-lightpdf_aipdf_mcp/server.py,sha256=jXAcfLWhef5E3cle28bytDTTuMiwTMYAglbAFVF6Wlw,35737
-lightpdf_aipdf_mcp-0.1.41.dist-info/METADATA,sha256=EsB0LkJG393OYqZbfO-DzcOC0JGBujmuAx2C6sWbxGk,7879
-lightpdf_aipdf_mcp-0.1.41.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lightpdf_aipdf_mcp-0.1.41.dist-info/entry_points.txt,sha256=X7TGUe52N4sYH-tYt0YUGApeJgw-efQlZA6uAZmlmr4,63
-lightpdf_aipdf_mcp-0.1.41.dist-info/RECORD,,