JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: update valkey cluster search test to use top_k parameter (#4815)

Browse Source
This commit is contained in:
Kartik
2026-04-13 20:44:18 +05:30
committed by GitHub
parent ac9cdd4840
commit 1a53852d93
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/vector_stores/test_valkey.py
+1 -1
View File
@@ -983,7 +983,7 @@ def test_cluster_mode_search(valkey_db_cluster, mock_valkey_cluster_client):
mock_results.docs = [mock_doc] mock_results.docs = [mock_doc]
mock_valkey_cluster_client.ft.return_value.search.return_value = mock_results mock_valkey_cluster_client.ft.return_value.search.return_value = mock_results
results = valkey_db_cluster.search("test", np.random.rand(1536).tolist(), limit=5) results = valkey_db_cluster.search("test", np.random.rand(1536).tolist(), top_k=5)
assert len(results) == 1 assert len(results) == 1
assert results[0].id == "id1" assert results[0].id == "id1"