From 1a53852d93b426e0606b3be31c35357cfb5b8918 Mon Sep 17 00:00:00 2001 From: Kartik Date: Mon, 13 Apr 2026 20:44:18 +0530 Subject: [PATCH] test: update valkey cluster search test to use top_k parameter (#4815) --- tests/vector_stores/test_valkey.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/vector_stores/test_valkey.py b/tests/vector_stores/test_valkey.py index bd30214d5..78d66615e 100644 --- a/tests/vector_stores/test_valkey.py +++ b/tests/vector_stores/test_valkey.py @@ -983,7 +983,7 @@ def test_cluster_mode_search(valkey_db_cluster, mock_valkey_cluster_client): mock_results.docs = [mock_doc] 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 results[0].id == "id1"