Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions src/tenantq/search.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,13 @@ def search(
prefetch_limit: int = 50,
) -> List[SearchHit]:
"""Run a tenant-isolated search in the requested retrieval mode."""
if query is None or not str(query).strip():
raise ValueError(
"query is required: empty or whitespace-only query would produce "
"degenerate embeddings and arbitrary-looking hits"
)
query = str(query).strip()

qfilter = build_filter(tenant_id, category, created_after, created_before)
params = models.SearchParams(hnsw_ef=settings.hnsw.hnsw_ef)

Expand Down
28 changes: 28 additions & 0 deletions tests/test_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,10 @@

from __future__ import annotations

from unittest.mock import MagicMock

import pytest

from tenantq.search import search


Expand Down Expand Up @@ -38,3 +42,27 @@ def test_created_at_range_filter(ingested, settings, embedder, dataset):
by_id = {d.id: d for d in dataset.documents}
for h in hits:
assert lo <= by_id[h.id].created_at <= hi


@pytest.mark.parametrize("bad", ["", " ", "\t", "\n"])
def test_search_rejects_empty_or_whitespace_query(settings, bad):
client = MagicMock()
embedder = MagicMock()
with pytest.raises(ValueError, match="query is required"):
search(client, settings, embedder, bad, tenant_id="acme", mode="hybrid")
embedder.embed_dense.assert_not_called()
embedder.embed_sparse.assert_not_called()
client.query_points.assert_not_called()


def test_search_accepts_query_with_internal_spaces(ingested, settings, embedder):
hits = search(
ingested,
settings,
embedder,
" neural network ",
tenant_id="acme",
mode="dense",
limit=5,
)
assert len(hits) > 0
Loading