danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix SSE server setup

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-09-08 15:31:56 +03:00
parent fed914c6af
commit 8dfb8638be
No known key found for this signature in database
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/haiku/rag/app.py
View file

@ -170,7 +170,7 @@ class HaikuRAGApp:
if transport == "stdio": if transport == "stdio":
await server.run_stdio_async() await server.run_stdio_async()
elif transport == "sse": elif transport == "sse":
await server.run_sse_async("sse") await server.run_sse_async()
else: else:
await server.run_http_async("streamable-http") await server.run_http_async("streamable-http")
except KeyboardInterrupt: except KeyboardInterrupt:

2
tests/test_app.py
View file

@ -200,7 +200,7 @@ async def test_serve(app: HaikuRAGApp, monkeypatch, transport):
if transport == "stdio": if transport == "stdio":
mock_server.run_stdio_async.assert_called_once() mock_server.run_stdio_async.assert_called_once()
elif transport == "sse": elif transport == "sse":
mock_server.run_sse_async.assert_called_once_with("sse") 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("streamable-http")