Skip to content

Commit 719b391

Browse files
Fix unit tests with updated top_k behavior
1 parent d2dd2f5 commit 719b391

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_semantic_search.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -827,11 +827,11 @@ def test_fetches_max_then_falls_back_per_connector(
827827
top_k=5,
828828
)
829829

830-
# First call: broad search without top_k (let backend decide, filter client-side)
830+
# First call: passes user's top_k to backend
831831
# Second call: per-connector fallback for "bamboohr" since first returned nothing
832832
assert mock_search.call_count == 2
833833
first_call = mock_search.call_args_list[0].kwargs
834-
assert first_call["top_k"] is None
834+
assert first_call["top_k"] == 5
835835
assert first_call["connector"] is None
836836
second_call = mock_search.call_args_list[1].kwargs
837837
assert second_call["connector"] == "bamboohr"

0 commit comments

Comments
 (0)