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