From ca9c3e59faf11c481cdf2d05cac05163c118cc56 Mon Sep 17 00:00:00 2001 From: Zhichang Yu Date: Thu, 20 Mar 2025 23:20:37 +0800 Subject: [PATCH] Call register_scripts on connecting redis (#6361) ### What problem does this PR solve? Call register_scripts on connecting redis ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- rag/utils/redis_conn.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/rag/utils/redis_conn.py b/rag/utils/redis_conn.py index f505b5064..790df690a 100644 --- a/rag/utils/redis_conn.py +++ b/rag/utils/redis_conn.py @@ -62,13 +62,11 @@ class RedisDB: self.REDIS = None self.config = settings.REDIS self.__open__() - self.register_scripts() def register_scripts(self) -> None: cls = self.__class__ client = self.REDIS - if cls.lua_delete_if_equal is None: - cls.lua_delete_if_equal = client.register_script(cls.LUA_DELETE_IF_EQUAL_SCRIPT) + cls.lua_delete_if_equal = client.register_script(cls.LUA_DELETE_IF_EQUAL_SCRIPT) def __open__(self): try: @@ -79,6 +77,7 @@ class RedisDB: password=self.config.get("password"), decode_responses=True, ) + self.register_scripts() except Exception: logging.warning("Redis can't be connected.") return self.REDIS