media-agent-mcp 2.6.5__tar.gz → 2.6.7__tar.gz

{media_agent_mcp-2.6.5 → media_agent_mcp-2.6.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 2.6.5
+Version: 2.6.7
 Summary: A Model Context Protocol server for media processing with AI tools
 Author-email: Media Agent Team <team@mediaagent.com>
 Keywords: mcp,ai,media,video,image,processing

{media_agent_mcp-2.6.5 → media_agent_mcp-2.6.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "media-agent-mcp"
-version = "2.6.5"
+version = "2.6.7"
 description = "A Model Context Protocol server for media processing with AI tools"
 readme = "README.md"
 requires-python = ">=3.12"

{media_agent_mcp-2.6.5 → media_agent_mcp-2.6.7}/src/media_agent_mcp/async_server.py

@@ -26,6 +26,10 @@ from dotenv import load_dotenv
 import uvicorn
 import anyio
 from functools import wraps
+import os
+import sys
+import signal
+import subprocess
 
 def async_retry(max_retries=3, delay=2):
     def decorator(func):
@@ -132,6 +136,72 @@ class ReconnectableSSEMiddleware:
             # 不抛出异常，允许客户端重连
             return
 
+# 在出现错误时重启应用的中间件
+class RestartOnErrorMiddleware:
+    def __init__(self, app):
+        self.app = app
+        self.restart_cooldown = 5  # 重启冷却时间（秒）
+        self.script_path = os.path.abspath(sys.argv[0])
+        self.args = sys.argv[1:]
+
+    async def __call__(self, scope, receive, send):
+        try:
+            await self.app(scope, receive, send)
+        except anyio.ClosedResourceError:
+            logger.warning("检测到 ClosedResourceError，准备重启应用...")
+            # 等待一段时间，确保日志输出
+            await asyncio.sleep(1)
+            
+            # 启动新进程，使用不同的端口避免冲突
+            # 解析当前命令行参数以获取当前端口
+            temp_parser = argparse.ArgumentParser()
+            temp_parser.add_argument('--port', type=int, default=8000)
+            temp_parser.add_argument('--host', type=str, default='127.0.0.1')
+            temp_parser.add_argument('--transport', type=str, default='stdio')
+            # 添加其他可能的参数，避免解析错误
+            temp_parser.add_argument('--run-be', action='store_true')
+            temp_parser.add_argument('--be-host', type=str, default='0.0.0.0')
+            temp_parser.add_argument('--be-port', type=int, default=5000)
+            temp_parser.add_argument('--version', action='store_true')
+            
+            # 忽略未知参数，避免解析错误
+            known_args, _ = temp_parser.parse_known_args(self.args)
+            
+            # 计算新端口，当前端口+1
+            new_port = known_args.port + 1
+            logger.info(f"将使用新端口 {new_port} 启动新进程")
+            
+            # 准备新的命令行参数
+            new_args = self.args.copy()
+            
+            # 如果命令行参数中有端口参数，则替换为新端口
+            if '--port' in new_args:
+                port_index = new_args.index('--port') + 1
+                if port_index < len(new_args):
+                    new_args[port_index] = str(new_port)
+            else:
+                new_args.extend(['--port', str(new_port)])
+                
+            logger.info(f"启动新进程: {self.script_path} {' '.join(new_args)}")
+            new_process = subprocess.Popen([sys.executable, self.script_path] + new_args)
+            
+            # 等待确认新进程已启动
+            logger.info(f"等待确认新进程已启动...")
+            await asyncio.sleep(3)
+            
+            # 检查新进程是否成功启动
+            if new_process.poll() is None:  # None表示进程仍在运行
+                logger.info(f"新进程成功启动，将在 {self.restart_cooldown} 秒后终止当前进程...")
+                await asyncio.sleep(self.restart_cooldown)
+                
+                # 终止当前进程
+                logger.info("终止当前进程")
+                sys.exit(0)
+            else:
+                logger.error(f"新进程启动失败，退出码: {new_process.returncode}，当前进程将继续运行")
+                return
+            return
+
 # Initialize FastMCP server (will be configured in main function)
 load_dotenv()
 mcp = FastMCP("Media-Agent-MCP-Async")
@@ -576,7 +646,7 @@ def main():
             mcp.settings.port = args.port
             # Use uvicorn to run SSE app with extended keep-alive timeout (5 minutes)
             uvicorn.run(
-                ReconnectableSSEMiddleware(mcp.sse_app()),
+                RestartOnErrorMiddleware(mcp.sse_app()),
                 host=args.host,
                 port=args.port,
                 timeout_keep_alive=300

{media_agent_mcp-2.6.5 → media_agent_mcp-2.6.7}/src/media_agent_mcp/server.py

@@ -22,6 +22,11 @@ from dotenv import load_dotenv
 import uvicorn
 import anyio
 import uuid
+import os
+import sys
+import signal
+import subprocess
+import asyncio
 
 from mcp.server.fastmcp import FastMCP
 
@@ -57,46 +62,78 @@ class IgnoreClosedResourceErrorMiddleware:
 class ReconnectableSSEMiddleware:
     def __init__(self, app):
         self.app = app
-        self.connections = {}
+        self.connections = {}  # 存储活跃连接
 
     async def __call__(self, scope, receive, send):
-        # Generate a unique connection ID
-        connection_id = str(uuid.uuid4())
+        # 为每个连接生成唯一ID
+        connection_id = scope.get('client', ('unknown', 0))[0] + ':' + str(scope.get('client', ('unknown', 0))[1])
         
-        # Store connection info
-        if scope["type"] == "http":
-            self.connections[connection_id] = {
-                "status": "active",
-                "path": scope.get("path", ""),
-                "created_at": anyio.current_time()
-            }
-            
-            # Wrap the send function to track connection state
-            original_send = send
+        # 包装send函数以跟踪连接状态
+        original_send = send
+        
+        async def wrapped_send(message):
+            if message.get('type') == 'http.response.start':
+                # 记录新连接
+                self.connections[connection_id] = {'active': True}
+                logger.info(f"New SSE connection established: {connection_id}")
+            elif message.get('type') == 'http.response.body' and message.get('more_body', False) is False:
+                # 连接结束
+                if connection_id in self.connections:
+                    self.connections[connection_id]['active'] = False
+                    logger.info(f"SSE connection closed normally: {connection_id}")
             
-            async def wrapped_send(message):
-                if message.get("type") == "http.response.body" and message.get("more_body", False) is False:
-                    # Connection is closing normally
-                    if connection_id in self.connections:
-                        self.connections[connection_id]["status"] = "closed"
-                return await original_send(message)
+            # 调用原始send
+            await original_send(message)
+        
+        try:
+            # 使用包装后的send函数
+            await self.app(scope, receive, wrapped_send)
+        except anyio.ClosedResourceError:
+            # 客户端断开连接
+            if connection_id in self.connections:
+                self.connections[connection_id]['active'] = False
             
-            try:
-                # Use the wrapped send function
-                await self.app(scope, receive, wrapped_send)
-            except anyio.ClosedResourceError:
-                # Client disconnected, but we'll allow reconnection
-                if connection_id in self.connections:
-                    self.connections[connection_id]["status"] = "disconnected"
-                logger.warning(f"SSE client disconnected (ClosedResourceError). Connection {connection_id[:8]} marked for reconnection.")
-                return
-            finally:
-                # Clean up connection tracking after some time
-                if connection_id in self.connections:
-                    self.connections.pop(connection_id, None)
-        else:
-            # For non-HTTP connections, just pass through
+            logger.warning(f"SSE client disconnected (ClosedResourceError): {connection_id}. Client can reconnect.")
+            # 不抛出异常，允许客户端重连
+            return
+
+# 在出现错误时重启应用的中间件
+class RestartOnErrorMiddleware:
+    def __init__(self, app):
+        self.app = app
+        self.restart_cooldown = 5  # 重启冷却时间（秒）
+        self.script_path = os.path.abspath(sys.argv[0])
+        self.args = sys.argv[1:]
+
+    async def __call__(self, scope, receive, send):
+        try:
             await self.app(scope, receive, send)
+        except anyio.ClosedResourceError:
+            logger.warning("检测到 ClosedResourceError，准备重启应用...")
+            # 等待一段时间，确保日志输出
+            await asyncio.sleep(1)
+            
+            # 启动新进程
+            logger.info(f"启动新进程: {self.script_path} {' '.join(self.args)}")
+            subprocess.Popen([sys.executable, self.script_path] + self.args)
+            
+            # 等待冷却时间
+            logger.info(f"等待 {self.restart_cooldown} 秒后终止当前进程...")
+            await asyncio.sleep(self.restart_cooldown)
+            
+            # 终止当前进程
+            logger.info("终止当前进程")
+            os.kill(os.getpid(), signal.SIGTERM)
+            return
+                
+                
+                
+                
+                
+                
+                
+            
+            return
 
 # Initialize FastMCP server (will be configured in main function)
 load_dotenv()
@@ -636,7 +673,7 @@ def main():
     # Configure and run the server
     if args.transport == 'sse':
         # SSE transport
-        uvicorn.run(ReconnectableSSEMiddleware(mcp.create_sse_app()), host=args.host, port=args.port)
+        uvicorn.run(RestartOnErrorMiddleware(mcp.create_sse_app()), host=args.host, port=args.port)
     else:
         # STDIO transport (default)
         mcp.run()

{media_agent_mcp-2.6.5 → media_agent_mcp-2.6.7}/src/media_agent_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 2.6.5
+Version: 2.6.7
 Summary: A Model Context Protocol server for media processing with AI tools
 Author-email: Media Agent Team <team@mediaagent.com>
 Keywords: mcp,ai,media,video,image,processing