Merge remote-tracking branch 'upstream/main'

isthaison · Dec 10, 2024 · 3794a81 · 3794a81
2 parents 46ac7ab + d5a322a
commit 3794a81
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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}")