media-agent-mcp 2.6.12__tar.gz → 2.6.14__tar.gz

{media_agent_mcp-2.6.12 → media_agent_mcp-2.6.14}/PKG-INFO

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

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

{media_agent_mcp-2.6.12 → media_agent_mcp-2.6.14}/src/media_agent_mcp/async_server.py

@@ -163,15 +163,15 @@ class SessionManager:
         """Get session object by ID."""
         return self._sessions.get(session_id)
     
-    def remove_session(self, session_id: str):
+    def remove_session(self, session_id: str, remove_route_mapping: bool = True):
         """Remove a session."""
         if session_id in self._sessions:
             del self._sessions[session_id]
-        if session_id in self._session_routes:
+        if remove_route_mapping and session_id in self._session_routes:
             del self._session_routes[session_id]
         if session_id in self._session_timestamps:
             del self._session_timestamps[session_id]
-        logger.info(f"Removed session: {session_id}")
+        logger.info(f"Removed session: {session_id} (route_mapping_removed: {remove_route_mapping})")
     
     def cleanup_expired_sessions(self, max_age_seconds: int = 3600):
         """Clean up sessions older than max_age_seconds."""
@@ -209,61 +209,25 @@ class SessionManager:
 # Global session manager
 session_manager = SessionManager()
 
-class SessionErrorHandlingMiddleware(BaseHTTPMiddleware):
-    """Middleware to handle ClosedResourceError and auto-regenerate sessions."""
+def _extract_session_id_from_url(url: str) -> Optional[str]:
+    """Extract session_id from URL."""
+    import re
+    from urllib.parse import urlparse, parse_qs
     
-    async def dispatch(self, request: Request, call_next):
-        try:
-            response = await call_next(request)
-            return response
-        except anyio.ClosedResourceError as e:
-            logger.warning(f"ClosedResourceError detected: {e}")
-            
-            # Extract session_id from request
-            session_id = self._extract_session_id(request)
-            if session_id:
-                # Generate new session ID
-                new_session_id = session_manager.generate_session_id()
-                
-                # Add route mapping
-                session_manager.add_route_mapping(session_id, new_session_id)
-                
-                # Remove old session
-                session_manager.remove_session(session_id)
-                
-                logger.info(f"Auto-generated new session {new_session_id} to replace expired session {session_id}")
-                
-                # Create a redirect response with new session ID
-                new_url = str(request.url).replace(f"session_id={session_id}", f"session_id={new_session_id}")
-                
-                from starlette.responses import RedirectResponse
-                return RedirectResponse(url=new_url, status_code=307)  # Temporary redirect preserving method
-            
-            # If no session_id found, re-raise the error
-            raise e
-        except Exception as e:
-            logger.error(f"Unexpected error in middleware: {e}")
-            raise e
-    
-    def _extract_session_id(self, request: Request) -> Optional[str]:
-        """Extract session_id from request URL or headers."""
-        # Try to get from query parameters
-        session_id = request.query_params.get('session_id')
-        if session_id:
-            return session_id
-        
-        # Try to get from path parameters
-        if hasattr(request, 'path_params') and 'session_id' in request.path_params:
-            return request.path_params['session_id']
-        
-        # Try to extract from URL path
-        import re
-        path = str(request.url.path)
-        match = re.search(r'session_id=([a-f0-9]+)', str(request.url))
-        if match:
-            return match.group(1)
-        
-        return None
+    # Parse the URL
+    parsed = urlparse(url)
+    
+    # Try to get from query parameters
+    query_params = parse_qs(parsed.query)
+    if 'session_id' in query_params:
+        return query_params['session_id'][0]
+    
+    # Try to extract from URL path using regex
+    match = re.search(r'session_id=([a-f0-9]+)', url)
+    if match:
+        return match.group(1)
+    
+    return None
 
 # Global exception handler for ClosedResourceError
 async def handle_closed_resource_error(request, exc):
@@ -279,7 +243,7 @@ async def handle_closed_resource_error(request, exc):
         # Generate new session ID
         new_session_id = session_manager.generate_session_id()
         session_manager.add_route_mapping(session_id, new_session_id)
-        session_manager.remove_session(session_id)
+        session_manager.remove_session(session_id, remove_route_mapping=False)
         
         logger.info(f"Global handler: Generated new session {new_session_id} to replace {session_id}")
         
@@ -825,16 +789,13 @@ def main():
             mcp.settings.host = args.host
             mcp.settings.port = args.port
             
-            # Get the SSE app and add session error handling middleware
+            # Get the SSE app and add global exception handler
             sse_app = mcp.sse_app()
             
-            # Add session error handling middleware
-            sse_app.add_middleware(SessionErrorHandlingMiddleware)
-            
             # Add global exception handler for ClosedResourceError
             sse_app.add_exception_handler(anyio.ClosedResourceError, handle_closed_resource_error)
             
-            logger.info("Added SessionErrorHandlingMiddleware and global ClosedResourceError handler for automatic session recovery")
+            logger.info("Added global ClosedResourceError handler for automatic session recovery")
             
             # Use uvicorn to run SSE app with extended keep-alive timeout (5 minutes)
             uvicorn.run(

{media_agent_mcp-2.6.12 → media_agent_mcp-2.6.14}/src/media_agent_mcp.egg-info/PKG-INFO

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