From 224574831c6aaabc1cb3361adeeec102a0651c5f Mon Sep 17 00:00:00 2001 From: Wang Qi Date: Thu, 23 Apr 2026 12:51:55 +0800 Subject: [PATCH] Add REDIS zcard (#14316) ### What problem does this PR solve? As description. ### Type of change - [x] Refactoring --- rag/utils/redis_conn.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/rag/utils/redis_conn.py b/rag/utils/redis_conn.py index 960e98af81..e3d5e4b3ea 100644 --- a/rag/utils/redis_conn.py +++ b/rag/utils/redis_conn.py @@ -284,6 +284,17 @@ class RedisDB: self.__open__() return 0 + def zcard(self, key: str): + try: + res = self.REDIS.zcard(key) + return res + except Exception as e: + logging.warning( + f"RedisDB.zcard {key} got exception: {e}" + ) + self.__open__() + return 0 + def incrby(self, key: str, increment: int): return self.REDIS.incrby(key, increment)