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}")