|
|
|
|
|
|
|
|
|
|
|
|
|
|
global DOC_ENGINE, docStoreConn, retrievaler, kg_retrievaler |
|
|
global DOC_ENGINE, docStoreConn, retrievaler, kg_retrievaler |
|
|
DOC_ENGINE = os.environ.get('DOC_ENGINE', "elasticsearch") |
|
|
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() |
|
|
docStoreConn = rag.utils.es_conn.ESConnection() |
|
|
elif DOC_ENGINE == "infinity": |
|
|
|
|
|
|
|
|
elif lower_case_doc_engine == "infinity": |
|
|
docStoreConn = rag.utils.infinity_conn.InfinityConnection() |
|
|
docStoreConn = rag.utils.infinity_conn.InfinityConnection() |
|
|
else: |
|
|
else: |
|
|
raise Exception(f"Not supported doc engine: {DOC_ENGINE}") |
|
|
raise Exception(f"Not supported doc engine: {DOC_ENGINE}") |