Skip to content

Commit 1171e0e

Browse files
ofekshenawaofekshenawa
committed
Fix resp 3 errors
1 parent 5044695 commit 1171e0e

File tree

2 files changed

+24
-24
lines changed

2 files changed

+24
-24
lines changed

‎tests/test_asyncio/test_search.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1874,7 +1874,7 @@ async def test_async_svs_vamana_distance_metrics(decoded_r: redis.Redis):
18741874
for i, vec in enumerate(vectors):
18751875
await decoded_r.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
18761876

1877-
query = Query("*=>[KNN 2 @v $vec as score]").sort_by("score")
1877+
query = Query("*=>[KNN 2 @v $vec as score]").sort_by("score").no_content()
18781878
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
18791879

18801880
res = await decoded_r.ft().search(query, query_params=query_params)
@@ -1907,7 +1907,7 @@ async def test_async_svs_vamana_vector_types(decoded_r: redis.Redis):
19071907
f"doc16_{i}", "v16", np.array(vec, dtype=np.float16).tobytes()
19081908
)
19091909

1910-
query = Query("*=>[KNN 2 @v16 $vec as score]")
1910+
query = Query("*=>[KNN 2 @v16 $vec as score]").no_content()
19111911
query_params = {"vec": np.array(vectors[0], dtype=np.float16).tobytes()}
19121912

19131913
res = await decoded_r.ft("idx16").search(query, query_params=query_params)
@@ -1944,7 +1944,7 @@ async def test_async_svs_vamana_compression(decoded_r: redis.Redis):
19441944
vectors.append(vec)
19451945
await decoded_r.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
19461946

1947-
query = Query("*=>[KNN 5 @v $vec as score]")
1947+
query = Query("*=>[KNN 5 @v $vec as score]").no_content()
19481948
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
19491949

19501950
res = await decoded_r.ft().search(query, query_params=query_params)
@@ -1983,7 +1983,7 @@ async def test_async_svs_vamana_build_parameters(decoded_r: redis.Redis):
19831983
vectors.append(vec)
19841984
await decoded_r.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
19851985

1986-
query = Query("*=>[KNN 3 @v $vec as score]")
1986+
query = Query("*=>[KNN 3 @v $vec as score]").no_content()
19871987
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
19881988

19891989
res = await decoded_r.ft().search(query, query_params=query_params)

‎tests/test_search.py‎

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -2883,7 +2883,7 @@ def test_svs_vamana_l2_distance_metric(client):
28832883
for i, vec in enumerate(vectors):
28842884
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
28852885

2886-
query = Query("*=>[KNN 3 @v $vec as score]").sort_by("score")
2886+
query = Query("*=>[KNN 3 @v $vec as score]").sort_by("score").no_content()
28872887
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
28882888

28892889
res = client.ft().search(query, query_params=query_params)
@@ -2914,7 +2914,7 @@ def test_svs_vamana_cosine_distance_metric(client):
29142914
for i, vec in enumerate(vectors):
29152915
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
29162916

2917-
query = Query("*=>[KNN 3 @v $vec as score]").sort_by("score")
2917+
query = Query("*=>[KNN 3 @v $vec as score]").sort_by("score").no_content()
29182918
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
29192919

29202920
res = client.ft().search(query, query_params=query_params)
@@ -2945,7 +2945,7 @@ def test_svs_vamana_ip_distance_metric(client):
29452945
for i, vec in enumerate(vectors):
29462946
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
29472947

2948-
query = Query("*=>[KNN 3 @v $vec as score]").sort_by("score")
2948+
query = Query("*=>[KNN 3 @v $vec as score]").sort_by("score").no_content()
29492949
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
29502950

29512951
res = client.ft().search(query, query_params=query_params)
@@ -2972,7 +2972,7 @@ def test_vector_search_with_int8_type(client):
29722972
client.hset("b", "v", np.array(b, dtype=np.int8).tobytes())
29732973
client.hset("c", "v", np.array(c, dtype=np.int8).tobytes())
29742974

2975-
query = Query("*=>[KNN 2 @v $vec as score]")
2975+
query = Query("*=>[KNN 2 @v $vec as score]").no_content()
29762976
query_params = {"vec": np.array(a, dtype=np.int8).tobytes()}
29772977

29782978
assert 2 in query.get_args()
@@ -3003,7 +3003,7 @@ def test_vector_search_with_uint8_type(client):
30033003
client.hset("b", "v", np.array(b, dtype=np.uint8).tobytes())
30043004
client.hset("c", "v", np.array(c, dtype=np.uint8).tobytes())
30053005

3006-
query = Query("*=>[KNN 2 @v $vec as score]")
3006+
query = Query("*=>[KNN 2 @v $vec as score]").no_content()
30073007
query_params = {"vec": np.array(a, dtype=np.uint8).tobytes()}
30083008

30093009
assert 2 in query.get_args()
@@ -3121,7 +3121,7 @@ def test_svs_vamana_float16_type(client):
31213121
for i, vec in enumerate(vectors):
31223122
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float16).tobytes())
31233123

3124-
query = Query("*=>[KNN 2 @v $vec as score]")
3124+
query = Query("*=>[KNN 2 @v $vec as score]").no_content()
31253125
query_params = {"vec": np.array(vectors[0], dtype=np.float16).tobytes()}
31263126

31273127
res = client.ft().search(query, query_params=query_params)
@@ -3152,7 +3152,7 @@ def test_svs_vamana_float32_type(client):
31523152
for i, vec in enumerate(vectors):
31533153
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
31543154

3155-
query = Query("*=>[KNN 2 @v $vec as score]")
3155+
query = Query("*=>[KNN 2 @v $vec as score]").no_content()
31563156
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
31573157

31583158
res = client.ft().search(query, query_params=query_params)
@@ -3239,7 +3239,7 @@ def test_svs_vamana_lvq8_compression(client):
32393239
vectors.append(vec)
32403240
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
32413241

3242-
query = Query("*=>[KNN 5 @v $vec as score]")
3242+
query = Query("*=>[KNN 5 @v $vec as score]").no_content()
32433243
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
32443244

32453245
res = client.ft().search(query, query_params=query_params)
@@ -3296,7 +3296,7 @@ def test_svs_vamana_compression_with_both_vector_types(client):
32963296
client.hset(f"doc32_{i}", "v32", np.array(vec, dtype=np.float32).tobytes())
32973297

32983298
# Test both indices
3299-
query = Query("*=>[KNN 3 @v16 $vec as score]")
3299+
query = Query("*=>[KNN 3 @v16 $vec as score]").no_content()
33003300
res16 = client.ft("idx16").search(
33013301
query,
33023302
query_params={
@@ -3306,7 +3306,7 @@ def test_svs_vamana_compression_with_both_vector_types(client):
33063306
},
33073307
)
33083308

3309-
query = Query("*=>[KNN 3 @v32 $vec as score]")
3309+
query = Query("*=>[KNN 3 @v32 $vec as score]").no_content()
33103310
res32 = client.ft("idx32").search(
33113311
query,
33123312
query_params={
@@ -3349,7 +3349,7 @@ def test_svs_vamana_construction_window_size(client):
33493349
vectors.append(vec)
33503350
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
33513351

3352-
query = Query("*=>[KNN 5 @v $vec as score]")
3352+
query = Query("*=>[KNN 5 @v $vec as score]").no_content()
33533353
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
33543354

33553355
res = client.ft().search(query, query_params=query_params)
@@ -3386,7 +3386,7 @@ def test_svs_vamana_graph_max_degree(client):
33863386
vectors.append(vec)
33873387
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
33883388

3389-
query = Query("*=>[KNN 6 @v $vec as score]")
3389+
query = Query("*=>[KNN 6 @v $vec as score]").no_content()
33903390
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
33913391

33923392
res = client.ft().search(query, query_params=query_params)
@@ -3423,7 +3423,7 @@ def test_svs_vamana_search_window_size(client):
34233423
vectors.append(vec)
34243424
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
34253425

3426-
query = Query("*=>[KNN 8 @v $vec as score]")
3426+
query = Query("*=>[KNN 8 @v $vec as score]").no_content()
34273427
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
34283428

34293429
res = client.ft().search(query, query_params=query_params)
@@ -3455,7 +3455,7 @@ def test_svs_vamana_epsilon_parameter(client):
34553455
vectors.append(vec)
34563456
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
34573457

3458-
query = Query("*=>[KNN 5 @v $vec as score]")
3458+
query = Query("*=>[KNN 5 @v $vec as score]").no_content()
34593459
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
34603460

34613461
res = client.ft().search(query, query_params=query_params)
@@ -3495,7 +3495,7 @@ def test_svs_vamana_all_build_parameters_combined(client):
34953495
vectors.append(vec)
34963496
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
34973497

3498-
query = Query("*=>[KNN 7 @v $vec as score]")
3498+
query = Query("*=>[KNN 7 @v $vec as score]").no_content()
34993499
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
35003500

35013501
res = client.ft().search(query, query_params=query_params)
@@ -3540,7 +3540,7 @@ def test_svs_vamana_comprehensive_configuration(client):
35403540
vectors.append(vec)
35413541
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float16).tobytes())
35423542

3543-
query = Query("*=>[KNN 10 @v $vec as score]")
3543+
query = Query("*=>[KNN 10 @v $vec as score]").no_content()
35443544
query_params = {"vec": np.array(vectors[0], dtype=np.float16).tobytes()}
35453545

35463546
res = client.ft().search(query, query_params=query_params)
@@ -3658,7 +3658,7 @@ def test_svs_vamana_large_dimension_vectors(client):
36583658
vectors.append(vec)
36593659
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
36603660

3661-
query = Query("*=>[KNN 5 @v $vec as score]")
3661+
query = Query("*=>[KNN 5 @v $vec as score]").no_content()
36623662
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
36633663

36643664
res = client.ft().search(query, query_params=query_params)
@@ -3697,7 +3697,7 @@ def test_svs_vamana_training_threshold_behavior(client):
36973697
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
36983698

36993699
if i >= 5:
3700-
query = Query("*=>[KNN 3 @v $vec as score]")
3700+
query = Query("*=>[KNN 3 @v $vec as score]").no_content()
37013701
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
37023702
res = client.ft().search(query, query_params=query_params)
37033703

@@ -3733,7 +3733,7 @@ def test_svs_vamana_different_k_values(client):
37333733
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
37343734

37353735
for k in [1, 3, 5, 10, 15]:
3736-
query = Query(f"*=>[KNN {k} @v $vec as score]")
3736+
query = Query(f"*=>[KNN {k} @v $vec as score]").no_content()
37373737
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
37383738
res = client.ft().search(query, query_params=query_params)
37393739

@@ -3792,7 +3792,7 @@ def test_svs_vamana_vector_search_with_parameters(client):
37923792
for i, vec in enumerate(vectors):
37933793
client.hset(f"doc{i}", "v", np.array(vec, dtype=np.float32).tobytes())
37943794

3795-
query = Query("*=>[KNN 3 @v $vec as score]")
3795+
query = Query("*=>[KNN 3 @v $vec as score]").no_content()
37963796
query_params = {"vec": np.array(vectors[0], dtype=np.float32).tobytes()}
37973797

37983798
res = client.ft().search(query, query_params=query_params)

0 commit comments

Comments
 (0)