diff --git a/llama_stack/providers/utils/kvstore/redis/redis.py b/llama_stack/providers/utils/kvstore/redis/redis.py index 8a7f3464b..ca34f0fad 100644 --- a/llama_stack/providers/utils/kvstore/redis/redis.py +++ b/llama_stack/providers/utils/kvstore/redis/redis.py @@ -48,5 +48,27 @@ class RedisKVStoreImpl(KVStore): async def range(self, start_key: str, end_key: str) -> List[str]: start_key = self._namespaced_key(start_key) end_key = self._namespaced_key(end_key) + cursor = 0 + pattern = start_key + "*" # Match all keys starting with start_key prefix + matching_keys = [] + while True: + cursor, keys = await self.redis.scan(cursor, match=pattern, count=1000) - return await self.redis.zrangebylex(start_key, end_key) + for key in keys: + key_str = key.decode("utf-8") if isinstance(key, bytes) else key + if start_key <= key_str <= end_key: + matching_keys.append(key) + + if cursor == 0: + break + + # Then fetch all values in a single MGET call + if matching_keys: + values = await self.redis.mget(matching_keys) + return [ + value.decode("utf-8") if isinstance(value, bytes) else value + for value in values + if value is not None + ] + + return []