浏览代码

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)
tags/v0.18.0
Zhichang Yu 7 个月前
父节点
当前提交
ca9c3e59fa
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 2 次插入3 次删除
  1. 2
    3
      rag/utils/redis_conn.py

+ 2
- 3
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

正在加载...
取消
保存