---
title: "Websocket"
description: "WebSocket connection management for MCP implementations API Documentation"
icon: "code"
github: "https://github.com/mcp-use/mcp-use/blob/main/libraries/python/mcp_use/client/task_managers/websocket.py"
---

import {RandomGradientBackground} from "/snippets/gradient.jsx"

<Callout type="info" title="Source Code">
View the source code for this module on GitHub: <a href='https://github.com/mcp-use/mcp-use/blob/main/libraries/python/mcp_use/client/task_managers/websocket.py' target='_blank' rel='noopener noreferrer'>https://github.com/mcp-use/mcp-use/blob/main/libraries/python/mcp_use/client/task_managers/websocket.py</a>
</Callout>

WebSocket connection management for MCP implementations.

This module provides a connection manager for WebSocket-based MCP connections.

## WebSocketConnectionManager

<div>
<RandomGradientBackground className="rounded-lg p-4 w-full h-full rounded-full">
<div className="text-black">
<div className="text-black font-bold text-xl mb-2 mt-8"><code className="!text-black">class</code> WebSocketConnectionManager</div>

Connection manager for WebSocket-based MCP connections.

    This class handles the lifecycle of WebSocket connections, ensuring proper
    connection establishment and cleanup.

</div>
</RandomGradientBackground>
```python
from mcp_use.client.task_managers.websocket import WebSocketConnectionManager
```

<Card type="info">
### `method` __init__

Initialize a new WebSocket connection manager.


**Parameters**
><ParamField body="url" type="str" required="True" >   The WebSocket URL to connect to </ParamField>
><ParamField body="headers" type="dict[str, str] | None" default="None" >   Optional HTTP headers </ParamField>

**Signature**
```python wrap
def __init__(url: str, headers: dict[str, str] | None = None):
```

</Card>
</div>
