degrade gracefully if no text2bednn

databio · Nov 9, 2023 · abd9077 · abd9077
1 parent 39de45b
commit abd9077
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/bbconf/bbconf.py b/bbconf/bbconf.py
@@ -182,7 +182,7 @@ def search_bed_by_text(
         """
         if self._t2bsi is None:
             raise BedBaseConfError(
-                "Can't perform search, qdrant_db credentials are not provided in config file"
+                "Can't perform search, ensure qdrant_db credentials in config file"
             )
         return self._t2bsi.nl_vec_search(query)
 
@@ -506,11 +506,15 @@ def _create_t2bsi_object(self) -> text2bednn.Text2BEDSearchInterface:
         :return: Text2BEDSearchInterface object
         """
 
-        return text2bednn.Text2BEDSearchInterface(
-            nl2vec_model=SentenceTransformer(os.getenv("HF_MODEL", DEFAULT_HF_MODEL)),
-            vec2vec_model=self._config[CFG_PATH_KEY][CFG_PATH_VEC2VEC_KEY],
-            search_backend=self.qdrant_backend,
-        )
+        try:
+            return text2bednn.Text2BEDSearchInterface(
+                nl2vec_model=SentenceTransformer(os.getenv("HF_MODEL", DEFAULT_HF_MODEL)),
+                vec2vec_model=self._config[CFG_PATH_KEY][CFG_PATH_VEC2VEC_KEY],
+                search_backend=self.qdrant_backend,
+            )
+        except Exception as e:
+            _LOGGER.error("Error in creating Text2BEDSearchInterface object: " + str(e))
+            return None
 
     def add_bed_to_qdrant(
         self,