Merge branch 'master' into ps_fix_readthedocs_yml_format

Author: petyaslavova

redis/commands/core.py

@@ -35,6 +35,7 @@
     GroupT,
     KeysT,
     KeyT,
+    Number,
     PatternT,
     ResponseT,
     ScriptTextT,
@@ -2567,7 +2568,7 @@ class ListCommands(CommandsProtocol):
     """
 
     def blpop(
-        self, keys: List, timeout: Optional[int] = 0
+        self, keys: List, timeout: Optional[Number] = 0
     ) -> Union[Awaitable[list], list]:
         """
         LPOP a value off of the first non-empty list
@@ -2588,7 +2589,7 @@ def blpop(
         return self.execute_command("BLPOP", *keys)
 
     def brpop(
-        self, keys: List, timeout: Optional[int] = 0
+        self, keys: List, timeout: Optional[Number] = 0
     ) -> Union[Awaitable[list], list]:
         """
         RPOP a value off of the first non-empty list
@@ -2609,7 +2610,7 @@ def brpop(
         return self.execute_command("BRPOP", *keys)
 
     def brpoplpush(
-        self, src: str, dst: str, timeout: Optional[int] = 0
+        self, src: str, dst: str, timeout: Optional[Number] = 0
     ) -> Union[Awaitable[Optional[str]], Optional[str]]:
         """
         Pop a value off the tail of ``src``, push it on the head of ``dst``

redis/commands/search/commands.py

@@ -586,7 +586,7 @@ def _get_aggregate_result(
 
     def profile(
         self,
-        query: Union[str, Query, AggregateRequest],
+        query: Union[Query, AggregateRequest],
         limited: bool = False,
         query_params: Optional[Dict[str, Union[str, int, float]]] = None,
     ):
@@ -596,7 +596,7 @@ def profile(
 
         ### Parameters
 
-        **query**: This can be either an `AggregateRequest`, `Query` or string.
+        **query**: This can be either an `AggregateRequest` or `Query`.
         **limited**: If set to True, removes details of reader iterator.
         **query_params**: Define one or more value parameters.
         Each parameter has a name and a value.

tests/test_search.py

@@ -2856,6 +2856,64 @@ def test_vector_search_with_default_dialect(client):
         assert res["total_results"] == 2
 
 
+@pytest.mark.redismod
+@skip_if_server_version_lt("7.9.0")
+def test_vector_search_with_int8_type(client):
+    client.ft().create_index(
+        (VectorField("v", "FLAT", {"TYPE": "INT8", "DIM": 2, "DISTANCE_METRIC": "L2"}),)
+    )
+
+    a = [1.5, 10]
+    b = [123, 100]
+    c = [1, 1]
+
+    client.hset("a", "v", np.array(a, dtype=np.int8).tobytes())
+    client.hset("b", "v", np.array(b, dtype=np.int8).tobytes())
+    client.hset("c", "v", np.array(c, dtype=np.int8).tobytes())
+
+    query = Query("*=>[KNN 2 @v $vec as score]")
+    query_params = {"vec": np.array(a, dtype=np.int8).tobytes()}
+
+    assert 2 in query.get_args()
+
+    res = client.ft().search(query, query_params=query_params)
+    if is_resp2_connection(client):
+        assert res.total == 2
+    else:
+        assert res["total_results"] == 2
+
+
+@pytest.mark.redismod
+@skip_if_server_version_lt("7.9.0")
+def test_vector_search_with_uint8_type(client):
+    client.ft().create_index(
+        (
+            VectorField(
+                "v", "FLAT", {"TYPE": "UINT8", "DIM": 2, "DISTANCE_METRIC": "L2"}
+            ),
+        )
+    )
+
+    a = [1.5, 10]
+    b = [123, 100]
+    c = [1, 1]
+
+    client.hset("a", "v", np.array(a, dtype=np.uint8).tobytes())
+    client.hset("b", "v", np.array(b, dtype=np.uint8).tobytes())
+    client.hset("c", "v", np.array(c, dtype=np.uint8).tobytes())
+
+    query = Query("*=>[KNN 2 @v $vec as score]")
+    query_params = {"vec": np.array(a, dtype=np.uint8).tobytes()}
+
+    assert 2 in query.get_args()
+
+    res = client.ft().search(query, query_params=query_params)
+    if is_resp2_connection(client):
+        assert res.total == 2
+    else:
+        assert res["total_results"] == 2
+
+
 @pytest.mark.redismod
 @skip_ifmodversion_lt("2.4.3", "search")
 def test_search_query_with_different_dialects(client):