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

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

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

@@ -1,6 +1,6 @@
 [project]
 name = "media-agent-mcp"
-version = "2.6.4"
+version = "2.6.5"
 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.5}/src/media_agent_mcp/async_server.py

@@ -93,6 +93,45 @@ 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
+
 # Initialize FastMCP server (will be configured in main function)
 load_dotenv()
 mcp = FastMCP("Media-Agent-MCP-Async")
@@ -537,7 +576,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()),
+                ReconnectableSSEMiddleware(mcp.sse_app()),
                 host=args.host,
                 port=args.port,
                 timeout_keep_alive=300

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

@@ -21,6 +21,7 @@ import json
 from dotenv import load_dotenv
 import uvicorn
 import anyio
+import uuid
 
 from mcp.server.fastmcp import FastMCP
 
@@ -52,6 +53,51 @@ 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):
+        # Generate a unique connection ID
+        connection_id = str(uuid.uuid4())
+        
+        # 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
+            
+            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)
+            
+            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
+            await self.app(scope, receive, send)
+
 # Initialize FastMCP server (will be configured in main function)
 load_dotenv()
 mcp = FastMCP("Media-Agent-MCP")
@@ -590,7 +636,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(ReconnectableSSEMiddleware(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.5}/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.5
 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