Allows elasticsearch7' package to be used instead of elasticsearch'.

baidubce · Apr 15, 2024 · cfe4c77 · cfe4c77
1 parent 1a82f23
commit cfe4c77
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/appbuilder/core/components/retriever/bes/bes_retriever.py b/appbuilder/core/components/retriever/bes/bes_retriever.py
@@ -66,8 +66,13 @@ def _lazy_import_es(self):
                 self._es = Elasticsearch
                 self._helpers = helpers
             except ImportError:
-                raise ImportError("Elasticsearch module is not installed. "
-                                  "Please install it using 'pip install elasticsearch==7.11.0'.")
+                try:
+                    from elasticsearch7 import Elasticsearch, helpers
+                    self._es = Elasticsearch
+                    self._helpers = helpers
+                except ImportError:
+                    raise ImportError("Elasticsearch module is not installed. "
+                                      "Please install it using 'pip install elasticsearch==7.11.0'.")
 
     @staticmethod
     def generate_id(length=16):