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

Remove SSE transport, fix cli tests

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-09-20 10:57:48 +03:00
parent f582fb1be0
commit dce60fa461
No known key found for this signature in database
7 changed files with 12 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
docs/cli.md
View file

@ -111,9 +111,6 @@ haiku-rag serve
# stdio transport
haiku-rag serve --stdio
# SSE transport
haiku-rag serve --sse
```
## Settings

5
docs/mcp.md
View file

@ -19,7 +19,7 @@ The MCP server exposes `haiku.rag` as MCP tools for compatible MCP clients.
## Starting MCP Server
The MCP server starts automatically with the serve command and supports Streamable HTTP, stdio and SSE transports:
The MCP server starts automatically with the serve command and supports Streamable HTTP and stdio transports:
```bash
# Default streamable HTTP transport
@ -27,7 +27,4 @@ haiku-rag serve
# stdio transport (for Claude Desktop)
haiku-rag serve --stdio
# SSE transport
haiku-rag serve --sse
```

1
docs/server.md
View file

@ -11,7 +11,6 @@ haiku-rag serve
Transport options:
- Default - Streamable HTTP transport
- `--stdio` - Standard input/output transport
- `--sse` - Server-sent events transport
## File Monitoring

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

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

7
src/haiku/rag/cli.py
View file

@ -356,11 +356,6 @@ def serve(
"--stdio",
help="Run MCP server on stdio Transport",
),
sse: bool = typer.Option(
False,
"--sse",
help="Run MCP server on SSE transport",
),
) -> None:
"""Start the MCP server."""
from haiku.rag.app import HaikuRAGApp
@ -370,8 +365,6 @@ def serve(
transport = None
if stdio:
transport = "stdio"
elif sse:
transport = "sse"
asyncio.run(app.serve(transport=transport))

4
tests/test_app.py
View file

@ -172,7 +172,7 @@ async def test_search_no_results(app: HaikuRAGApp, monkeypatch):
@pytest.mark.asyncio
@pytest.mark.parametrize("transport", ["stdio", "sse", "http", None])
@pytest.mark.parametrize("transport", ["stdio", "http", None])
async def test_serve(app: HaikuRAGApp, monkeypatch, transport):
"""Test the serve method with different transports."""
mock_server = AsyncMock()
@ -199,8 +199,6 @@ 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()
else:
mock_server.run_http_async.assert_called_once_with(transport="streamable-http")

44
tests/test_cli.py
View file

@ -8,7 +8,7 @@ runner = CliRunner()
def test_list_documents():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.list_documents = AsyncMock()
mock_app.return_value = mock_app_instance
@ -20,7 +20,7 @@ def test_list_documents():
def test_add_document_text():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.add_document_from_text = AsyncMock()
mock_app.return_value = mock_app_instance
@ -34,7 +34,7 @@ def test_add_document_text():
def test_add_document_src():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.add_document_from_source = AsyncMock()
mock_app.return_value = mock_app_instance
@ -46,7 +46,7 @@ def test_add_document_src():
def test_get_document():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.get_document = AsyncMock()
mock_app.return_value = mock_app_instance
@ -58,7 +58,7 @@ def test_get_document():
def test_delete_document():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.delete_document = AsyncMock()
mock_app.return_value = mock_app_instance
@ -70,7 +70,7 @@ def test_delete_document():
def test_search():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.search = AsyncMock()
mock_app.return_value = mock_app_instance
@ -82,7 +82,7 @@ def test_search():
def test_serve():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.serve = AsyncMock()
mock_app.return_value = mock_app_instance
@ -94,7 +94,7 @@ def test_serve():
def test_serve_stdio():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.serve = AsyncMock()
mock_app.return_value = mock_app_instance
@ -105,32 +105,8 @@ def test_serve_stdio():
mock_app_instance.serve.assert_called_once_with(transport="stdio")
def test_serve_sse():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.serve = AsyncMock()
mock_app.return_value = mock_app_instance
result = runner.invoke(cli, ["serve", "--sse"])
assert result.exit_code == 0
mock_app_instance.serve.assert_called_once_with(transport="sse")
def test_serve_stdio_and_sse():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.serve = AsyncMock()
mock_app.return_value = mock_app_instance
result = runner.invoke(cli, ["serve", "--stdio", "--sse"])
assert result.exit_code == 1
assert "Error: Cannot use both --stdio and --http options" in result.stdout
def test_ask():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.ask = AsyncMock()
mock_app.return_value = mock_app_instance
@ -144,7 +120,7 @@ def test_ask():
def test_ask_with_cite():
with patch("haiku.rag.cli.HaikuRAGApp") as mock_app:
with patch("haiku.rag.app.HaikuRAGApp") as mock_app:
mock_app_instance = MagicMock()
mock_app_instance.ask = AsyncMock()
mock_app.return_value = mock_app_instance