refac

backend/open_webui/main.py

@@ -1923,8 +1923,6 @@ async def chat_completion(
             # uses anyio task groups whose cancel scopes enforce
             # same-task exit.  Do NOT wrap in asyncio.shield() or
             # asyncio.wait_for() — both create a new task.
-            # MCPClient.disconnect() self-shields via
-            # anyio.CancelScope(shield=True).
             try:
                 if mcp_clients := metadata.get('mcp_clients'):
                     for client in reversed(list(mcp_clients.values())):
@@ -1932,14 +1930,29 @@ async def chat_completion(
                             await client.disconnect()
                         except Exception as e:
                             log.debug(f'Error disconnecting MCP client: {e}')
+                        except asyncio.CancelledError:
+                            # Let the client close asynchronously by GC
+                            pass
             except Exception as e:
                 log.debug(f'Error cleaning up MCP clients: {e}')
+            except asyncio.CancelledError:
+                pass
 
             try:
                 if metadata.get('chat_id'):
-                    event_emitter = await get_event_emitter(metadata, update_db=False)
-                    if event_emitter:
-                        await event_emitter({'type': 'chat:active', 'data': {'active': False}})
+                    async def emit_inactive_event():
+                        try:
+                            event_emitter = await get_event_emitter(metadata, update_db=False)
+                            if event_emitter:
+                                await event_emitter({'type': 'chat:active', 'data': {'active': False}})
+                        except Exception:
+                            pass
+                    
+                    try:
+                        # Shield the event emission so it finishes even if the main task is cancelled
+                        await asyncio.shield(emit_inactive_event())
+                    except asyncio.CancelledError:
+                        pass
             except Exception:
                 pass