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]>
- 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 |
-
|
|
|
167 |
docStoreConn = rag.utils.es_conn.ESConnection()
|
168 |
-
elif
|
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}")
|