From d19d2b7f000d901b1c6ace3a9e38cd4185b9e248 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Fri, 31 Oct 2025 12:50:08 +0200 Subject: [PATCH] Fix tests --- tests/test_app.py | 4 ++-- tests/test_cli.py | 4 +++- tests/test_reranker.py | 4 +++- 3 files changed, 8 insertions(+), 4 deletions(-) diff --git a/tests/test_app.py b/tests/test_app.py index b6111ddc..73657202 100644 --- a/tests/test_app.py +++ b/tests/test_app.py @@ -159,7 +159,7 @@ async def test_search(app: HaikuRAGApp, monkeypatch): with patch("haiku.rag.app.HaikuRAG", return_value=mock_client): await app.search("query") - mock_client.search.assert_called_once_with("query", limit=5) + mock_client.search.assert_called_once_with("query", limit=5, filter=None) assert mock_rich_print_search.call_count == len(mock_results) @@ -176,7 +176,7 @@ async def test_search_no_results(app: HaikuRAGApp, monkeypatch): with patch("haiku.rag.app.HaikuRAG", return_value=mock_client): await app.search("query") - mock_client.search.assert_called_once_with("query", limit=5) + mock_client.search.assert_called_once_with("query", limit=5, filter=None) mock_print.assert_called_once_with("[yellow]No results found.[/yellow]") diff --git a/tests/test_cli.py b/tests/test_cli.py index de173f4a..f4e73fde 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -170,7 +170,9 @@ def test_search(): result = runner.invoke(cli, ["search", "query"]) assert result.exit_code == 0 - mock_app_instance.search.assert_called_once_with(query="query", limit=5) + mock_app_instance.search.assert_called_once_with( + query="query", limit=5, filter=None + ) def test_serve_no_flags(): diff --git a/tests/test_reranker.py b/tests/test_reranker.py index 2376dc47..399234b3 100644 --- a/tests/test_reranker.py +++ b/tests/test_reranker.py @@ -28,8 +28,10 @@ chunks = [ @pytest.mark.asyncio async def test_reranker_base(): + from haiku.rag.config import Config + reranker = RerankerBase() - assert reranker._model == "" + assert reranker._model == Config.reranking.model with pytest.raises(NotImplementedError): await reranker.rerank("query", [])