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

baidubce · Apr 25, 2024 · 574bd56 · 574bd56
1 parent 1a82f23
commit 574bd56
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/appbuilder/core/components/retriever/bes/bes_retriever.py b/appbuilder/core/components/retriever/bes/bes_retriever.py
@@ -62,12 +62,17 @@ def helpers(self):
     def _lazy_import_es(self):
         if self._es is None or self._helpers is None:
             try:
-                from elasticsearch import Elasticsearch, helpers
+                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'.")
+                try:
+                    from elasticsearch import Elasticsearch, helpers
+                    self._es = Elasticsearch
+                    self._helpers = helpers
+                except ImportError:
+                    raise ImportError("Elasticsearch module is not installed. "
+                                      "Please install it using 'pip install elasticsearch7==7.11.0'.")
 
     @staticmethod
     def generate_id(length=16):