diff --git a/redis/commands/core.py b/redis/commands/core.py index 1d19e33f2c..6ab0602698 100644 --- a/redis/commands/core.py +++ b/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`` diff --git a/redis/commands/search/commands.py b/redis/commands/search/commands.py index e5e78578be..1db57c23a5 100644 --- a/redis/commands/search/commands.py +++ b/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.