浏览代码

refine redis connection (#599)

### What problem does this PR solve?

#591 

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.5.0
KevinHuSh 1年前
父节点
当前提交
8acc01a227
没有帐户链接到提交者的电子邮件
共有 2 个文件被更改,包括 3 次插入2 次删除
  1. 1
    0
      rag/app/table.py
  2. 2
    2
      rag/utils/redis_conn.py

+ 1
- 0
rag/app/table.py 查看文件

@@ -47,6 +47,7 @@ class Excel(ExcelParser):
cell.value for i,
cell in enumerate(
rows[0]) if i not in missed]
if not headers:continue
data = []
for i, r in enumerate(rows[1:]):
rn += 1

+ 2
- 2
rag/utils/redis_conn.py 查看文件

@@ -14,10 +14,10 @@ class RedisDB:
def __open__(self):
try:
self.REDIS = redis.Redis(host=self.config.get("host", "redis").split(":")[0],
self.REDIS = redis.StrictRedis(host=self.config["host"].split(":")[0],
port=int(self.config.get("host", ":6379").split(":")[1]),
db=int(self.config.get("db", 1)),
password=self.config.get("password"))
password=self.config["password"])
except Exception as e:
logging.warning("Redis can't be connected.")
return self.REDIS

正在加载...
取消
保存