no spacy import at text_similarity file

Author: manuelporrasojeda

toolium/utils/ai_utils/spacy.py

@@ -39,6 +39,8 @@ def get_spacy_model(model_name):
     :param model_name: spaCy model name
     :return: spaCy model
     """
+    if spacy is None:
+        return None
     return spacy.load(model_name)
 
 

toolium/utils/ai_utils/text_similarity.py

@@ -19,11 +19,6 @@
 import json
 import logging
 
-# AI library imports must be optional to allow installing Toolium without `ai` extra dependency
-try:
-    import spacy
-except ImportError:
-    spacy = None
 try:
     from sentence_transformers import SentenceTransformer
 except ImportError:
@@ -52,11 +47,11 @@ def get_text_similarity_with_spacy(text, expected_text, model_name=None):
     # - Normalizing texts (lowercase, extra points, etc.)
     # - Use only models that include word vectors (e.g., 'en_core_news_md' or 'en_core_news_lg')
     # - Preprocessing texts. Now we only preprocess negations.
-    if spacy is None:
-        raise ImportError("spaCy is not installed. Please run 'pip install toolium[ai]' to use spaCy features")
     config = DriverWrappersPool.get_default_wrapper().config
     model_name = model_name or config.get_optional('AI', 'spacy_model', 'en_core_web_md')
     model = get_spacy_model(model_name)
+    if model is None:
+        raise ImportError("spaCy is not installed. Please run 'pip install toolium[ai]' to use spaCy features")
     text = model(preprocess_with_ud_negation(text, model))
     expected_text = model(preprocess_with_ud_negation(expected_text, model))
     similarity = model(text).similarity(model(expected_text))