updated import error handling

cwpdntm · Jul 12, 2023 · fe3a07a · fe3a07a
1 parent 5810ef6
commit fe3a07a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/dsp/modules/azurecognitivesearch.py b/dsp/modules/azurecognitivesearch.py
@@ -1,11 +1,15 @@
 from typing import Optional, Union, Any
 
 from dsp.utils import dotdict
-
-from azure.core.credentials import AzureKeyCredential
-from azure.search.documents import SearchClient
-from azure.search.documents._paging import SearchItemPaged
-
+try:
+    from azure.core.credentials import AzureKeyCredential
+    from azure.search.documents import SearchClient
+    from azure.search.documents._paging import SearchItemPaged
+except ImportError as e:
+    raise ImportError(
+        "You need to install azure-search-documents library"
+        "Please use the command: pip install azure-search-documents"
+    )
 
 class AzureCognitiveSearch:
     """Wrapper for the Azure Congitive Search Retrieval."""

diff --git a/setup.py b/setup.py
@@ -20,8 +20,7 @@
         "regex",
         "ujson",
         "tqdm",
-        "datasets",
-        'azure-search-documents'        
+        "datasets",        
     ],
     classifiers=[
 		"Development Status :: 3 - Alpha",