jinhai-2012 commited on
Commit
da54ddf
·
1 Parent(s): 86cdc50

Case insensitive when set doc engine (#3954)

Browse files

### 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 <[email protected]>

Files changed (1) hide show
  1. api/settings.py +3 -2
api/settings.py CHANGED
@@ -163,9 +163,10 @@ def init_settings():
163
 
164
  global DOC_ENGINE, docStoreConn, retrievaler, kg_retrievaler
165
  DOC_ENGINE = os.environ.get('DOC_ENGINE', "elasticsearch")
166
- if DOC_ENGINE == "elasticsearch":
 
167
  docStoreConn = rag.utils.es_conn.ESConnection()
168
- elif DOC_ENGINE == "infinity":
169
  docStoreConn = rag.utils.infinity_conn.InfinityConnection()
170
  else:
171
  raise Exception(f"Not supported doc engine: {DOC_ENGINE}")
 
163
 
164
  global DOC_ENGINE, docStoreConn, retrievaler, kg_retrievaler
165
  DOC_ENGINE = os.environ.get('DOC_ENGINE', "elasticsearch")
166
+ lower_case_doc_engine = DOC_ENGINE.lower()
167
+ if lower_case_doc_engine == "elasticsearch":
168
  docStoreConn = rag.utils.es_conn.ESConnection()
169
+ elif lower_case_doc_engine == "infinity":
170
  docStoreConn = rag.utils.infinity_conn.InfinityConnection()
171
  else:
172
  raise Exception(f"Not supported doc engine: {DOC_ENGINE}")