浏览代码

Case insensitive when set doc engine (#3954)

### What problem does this PR solve?

DOC_ENGINE="INFINITY" or "Infinity" or "Elasticsearch" also works

### Type of change

- [x] Refactoring

Signed-off-by: Jin Hai <haijin.chn@gmail.com>
tags/v0.15.0
Jin Hai 10 个月前
父节点
当前提交
7d4f1c0645
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3
    2
      api/settings.py

+ 3
- 2
api/settings.py 查看文件

@@ -163,9 +163,10 @@ def init_settings():

global DOC_ENGINE, docStoreConn, retrievaler, kg_retrievaler
DOC_ENGINE = os.environ.get('DOC_ENGINE', "elasticsearch")
if DOC_ENGINE == "elasticsearch":
lower_case_doc_engine = DOC_ENGINE.lower()
if lower_case_doc_engine == "elasticsearch":
docStoreConn = rag.utils.es_conn.ESConnection()
elif DOC_ENGINE == "infinity":
elif lower_case_doc_engine == "infinity":
docStoreConn = rag.utils.infinity_conn.InfinityConnection()
else:
raise Exception(f"Not supported doc engine: {DOC_ENGINE}")

正在加载...
取消
保存