diff --git a/src/haiku/rag/app.py b/src/haiku/rag/app.py index 74d52d0a..ad591fe1 100644 --- a/src/haiku/rag/app.py +++ b/src/haiku/rag/app.py @@ -170,7 +170,7 @@ class HaikuRAGApp: if transport == "stdio": await server.run_stdio_async() elif transport == "sse": - await server.run_sse_async("sse") + await server.run_sse_async() else: await server.run_http_async("streamable-http") except KeyboardInterrupt: diff --git a/tests/test_app.py b/tests/test_app.py index 037d0f9c..9da31ef1 100644 --- a/tests/test_app.py +++ b/tests/test_app.py @@ -200,7 +200,7 @@ async def test_serve(app: HaikuRAGApp, monkeypatch, transport): if transport == "stdio": mock_server.run_stdio_async.assert_called_once() elif transport == "sse": - mock_server.run_sse_async.assert_called_once_with("sse") + mock_server.run_sse_async.assert_called_once() else: mock_server.run_http_async.assert_called_once_with("streamable-http")