Fix MCP test
This commit is contained in:
parent
90105a861a
commit
03289bb644
1 changed files with 1 additions and 1 deletions
|
|
@ -202,7 +202,7 @@ async def test_serve(app: HaikuRAGApp, monkeypatch, transport):
|
||||||
elif transport == "sse":
|
elif transport == "sse":
|
||||||
mock_server.run_sse_async.assert_called_once()
|
mock_server.run_sse_async.assert_called_once()
|
||||||
else:
|
else:
|
||||||
mock_server.run_http_async.assert_called_once_with("streamable-http")
|
mock_server.run_http_async.assert_called_once_with(transport="streamable-http")
|
||||||
|
|
||||||
mock_task.cancel.assert_called_once()
|
mock_task.cancel.assert_called_once()
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue