Stabilize websocket connection typing

Co-authored-by: P4x-ng <P4X-ng@users.noreply.github.com>

Author: cursoragent

cdp/connection.py

@@ -11,22 +11,33 @@
 import json
 import logging
 import typing
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 
 try:
     import websockets
-    from websockets.client import WebSocketClientProtocol
     WEBSOCKETS_AVAILABLE = True
 except ImportError:
     WEBSOCKETS_AVAILABLE = False
-    WebSocketClientProtocol = typing.Any  # type: ignore
 
 from cdp.util import parse_json_event, T_JSON_DICT
 
 
 logger = logging.getLogger(__name__)
 
 
+class WebSocketConnection(typing.Protocol):
+    """Minimal websocket surface used by CDPConnection."""
+
+    async def send(self, message: str) -> None:
+        ...
+
+    async def recv(self) -> str:
+        ...
+
+    async def close(self) -> None:
+        ...
+
+
 class CDPError(Exception):
     """Base exception for CDP errors."""
     pass
@@ -92,7 +103,7 @@ def __init__(self, url: str, timeout: float = 30.0):
 
         self.url = url
         self.timeout = timeout
-        self._ws: typing.Optional[WebSocketClientProtocol] = None
+        self._ws: typing.Optional[WebSocketConnection] = None
         self._next_command_id = 1
         self._pending_commands: typing.Dict[int, PendingCommand] = {}
         self._event_queue: asyncio.Queue = asyncio.Queue()