Skip to content

Commit 7c9d602

Browse files
stop shadowing werkzeug types
1 parent 7484fd8 commit 7c9d602

1 file changed

Lines changed: 13 additions & 5 deletions

File tree

tests/conftest.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,13 @@
6464
JSONRPCRequest,
6565
)
6666
from mcp.shared.message import SessionMessage
67-
from httpx import ASGITransport, Request, Response, AsyncByteStream, AsyncClient
67+
from httpx import (
68+
ASGITransport,
69+
Request as HttpxRequest,
70+
Response as HttpxResponse,
71+
AsyncByteStream,
72+
AsyncClient,
73+
)
6874
except ImportError:
6975
create_memory_object_stream = None
7076
create_task_group = None
@@ -76,8 +82,8 @@
7682
SessionMessage = None
7783

7884
ASGITransport = None
79-
Request = None
80-
Response = None
85+
HttpxRequest = None
86+
HttpxResponse = None
8187
AsyncByteStream = None
8288
AsyncClient = None
8389

@@ -814,7 +820,9 @@ def __init__(
814820
self.keep_sse_alive = keep_sse_alive
815821
super().__init__(app)
816822

817-
async def handle_async_request(self, request: "Request") -> "Response":
823+
async def handle_async_request(
824+
self, request: "HttpxRequest"
825+
) -> "HttpxResponse":
818826
scope = {
819827
"type": "http",
820828
"method": request.method,
@@ -873,7 +881,7 @@ async def __aiter__(self):
873881
pass
874882

875883
stream = StreamingBodyStream(body_receiver)
876-
response = Response(status_code=200, headers=[], stream=stream) # type: ignore
884+
response = HttpxResponse(status_code=200, headers=[], stream=stream) # type: ignore
877885

878886
asyncio.create_task(run_app())
879887
return response

0 commit comments

Comments
 (0)