From 8dfb8638bed623a5e340ee34716a41abbdb0b537 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Mon, 8 Sep 2025 15:31:56 +0300 Subject: [PATCH] Fix SSE server setup --- src/haiku/rag/app.py | 2 +- tests/test_app.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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")