From 7d4f1c06451394f1a90fea781fc64dc836efae6b Mon Sep 17 00:00:00 2001 From: Jin Hai Date: Tue, 10 Dec 2024 11:26:10 +0800 Subject: [PATCH] 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 --- api/settings.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/api/settings.py b/api/settings.py index 94225bc44..cd9154864 100644 --- a/api/settings.py +++ b/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}")