media-agent-mcp 2.6.4__tar.gz → 2.6.6__tar.gz

{media_agent_mcp-2.6.4 → media_agent_mcp-2.6.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 2.6.4
+Version: 2.6.6
 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.4 → media_agent_mcp-2.6.6}/pyproject.toml

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

{media_agent_mcp-2.6.4 → media_agent_mcp-2.6.6}/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):
@@ -93,6 +97,119 @@ class IgnoreClosedResourceErrorMiddleware:
             logger.warning("SSE client disconnected (ClosedResourceError). Ignoring.")
             return
 
+# Enhanced middleware that supports reconnection for SSE clients
+class ReconnectableSSEMiddleware:
+    def __init__(self, app):
+        self.app = app
+        self.connections = {}  # 存储活跃连接
+
+    async def __call__(self, scope, receive, send):
+        # 为每个连接生成唯一ID
+        connection_id = scope.get('client', ('unknown', 0))[0] + ':' + str(scope.get('client', ('unknown', 0))[1])
+        
+        # 包装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}")
+            
+            # 调用原始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
+            
+            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("正在关闭当前服务器以释放端口...")
+            # 发送SIGTERM信号给当前进程，但不立即退出
+            # 这会触发uvicorn的关闭流程，释放端口
+            os.kill(os.getpid(), signal.SIGUSR1)
+            
+            # 等待端口释放
+            await asyncio.sleep(2)
+            
+            # 启动新进程
+            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
+            
+            # 重置超过窗口期的错误计数
+            if current_time - self.last_error_time > self.error_window:
+                self.error_count = 0
+            
+            self.last_error_time = current_time
+            self.error_count += 1
+            
+            logger.warning(f"SSE client disconnected (ClosedResourceError). Error count: {self.error_count}/{self.max_errors}")
+            
+            # 如果错误次数超过阈值，重启应用
+            if self.error_count >= self.max_errors:
+                logger.error(f"Too many connection errors ({self.error_count}). Restarting application...")
+                
+                # 启动一个新进程来重启应用
+                # 使用当前脚本的路径和参数
+                args = sys.argv.copy()
+                
+                # 确保我们不会立即重启（可能导致无限循环）
+                logger.info(f"Cooling down for {self.restart_cooldown} seconds before restart...")
+                await asyncio.sleep(self.restart_cooldown)
+                
+                # 使用subprocess启动新进程
+                subprocess.Popen(args, env=os.environ.copy())
+                
+                # 发送SIGTERM信号给当前进程
+                logger.info("Sending termination signal to current process...")
+                os.kill(os.getpid(), signal.SIGTERM)
+                
+                # 等待进程终止
+                await asyncio.sleep(1)
+                
+                # 如果进程还在运行，强制退出
+                sys.exit(0)
+            
+            return
+
 # Initialize FastMCP server (will be configured in main function)
 load_dotenv()
 mcp = FastMCP("Media-Agent-MCP-Async")
@@ -537,7 +654,7 @@ def main():
             mcp.settings.port = args.port
             # Use uvicorn to run SSE app with extended keep-alive timeout (5 minutes)
             uvicorn.run(
-                IgnoreClosedResourceErrorMiddleware(mcp.sse_app()),
+                RestartOnErrorMiddleware(mcp.sse_app()),
                 host=args.host,
                 port=args.port,
                 timeout_keep_alive=300

{media_agent_mcp-2.6.4 → media_agent_mcp-2.6.6}/src/media_agent_mcp/server.py

@@ -21,6 +21,12 @@ import json
 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
 
@@ -50,6 +56,83 @@ class IgnoreClosedResourceErrorMiddleware:
             await self.app(scope, receive, send)
         except anyio.ClosedResourceError:
             logger.warning("SSE client disconnected (ClosedResourceError). Ignoring.")
+            return
+
+# Enhanced middleware that supports reconnection for SSE clients
+class ReconnectableSSEMiddleware:
+    def __init__(self, app):
+        self.app = app
+        self.connections = {}  # 存储活跃连接
+
+    async def __call__(self, scope, receive, send):
+        # 为每个连接生成唯一ID
+        connection_id = scope.get('client', ('unknown', 0))[0] + ':' + str(scope.get('client', ('unknown', 0))[1])
+        
+        # 包装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}")
+            
+            # 调用原始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
+            
+            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)
@@ -590,7 +673,7 @@ def main():
     # Configure and run the server
     if args.transport == 'sse':
         # SSE transport
-        uvicorn.run(IgnoreClosedResourceErrorMiddleware(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.4 → media_agent_mcp-2.6.6}/src/media_agent_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 2.6.4
+Version: 2.6.6
 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