AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 08:26:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Call register_scripts on connecting redis (#6361)

Browse Source 
### 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)
This commit is contained in:
Zhichang Yu 2025-03-20 23:20:37 +08:00 committed by GitHub
parent 674f94228b
commit ca9c3e59fa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
rag/utils/redis_conn.py
View File

@ -62,12 +62,10 @@ class RedisDB:
self.REDIS = None self.REDIS = None
self.config = settings.REDIS self.config = settings.REDIS
self.__open__() self.__open__()
self.register_scripts()
def register_scripts(self) -> None: def register_scripts(self) -> None:
cls = self.__class__ cls = self.__class__
client = self.REDIS 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): def __open__(self):
@ -79,6 +77,7 @@ class RedisDB:
password=self.config.get("password"), password=self.config.get("password"),
decode_responses=True, decode_responses=True,
) )
self.register_scripts()
except Exception: except Exception:
logging.warning("Redis can't be connected.") logging.warning("Redis can't be connected.")
return self.REDIS return self.REDIS