vendor the whole shebang

tests/unit/conftest.py

@@ -30,7 +30,6 @@
 from asgi_lifespan import LifespanManager
 from faker import Faker
 from httpx import AsyncByteStream, Request, Response
-from httpx_ws import AsyncWebSocketSession, aconnect_ws
 from psycopg import Connection
 from pytest_postgresql import factories
 from sqlalchemy import URL, make_url
@@ -52,7 +51,8 @@
 from phoenix.server.types import BatchedCaller, DbSessionFactory
 from phoenix.session.client import Client
 from phoenix.trace.schemas import Span
-from tests.unit.ws_transport import ASGIWebSocketTransport
+from tests.unit.httpx_ws import AsyncWebSocketSession, aconnect_ws
+from tests.unit.httpx_ws.transport import ASGIWebSocketTransport
 
 
 def pytest_terminal_summary(

tests/unit/httpx_ws/__init__.py

@@ -0,0 +1,29 @@
+__version__ = "0.6.2"
+
+from ._api import (
+    AsyncWebSocketSession,
+    JSONMode,
+    WebSocketSession,
+    aconnect_ws,
+    connect_ws,
+)
+from ._exceptions import (
+    HTTPXWSException,
+    WebSocketDisconnect,
+    WebSocketInvalidTypeReceived,
+    WebSocketNetworkError,
+    WebSocketUpgradeError,
+)
+
+__all__ = [
+    "AsyncWebSocketSession",
+    "HTTPXWSException",
+    "JSONMode",
+    "WebSocketDisconnect",
+    "WebSocketInvalidTypeReceived",
+    "WebSocketNetworkError",
+    "WebSocketSession",
+    "WebSocketUpgradeError",
+    "aconnect_ws",
+    "connect_ws",
+]