Fix handling of indexing error (#33)

Zipstack · Apr 9, 2024 · aed4fe6 · aed4fe6
1 parent 7ed943d
commit aed4fe6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/unstract/sdk/exceptions.py b/src/unstract/sdk/exceptions.py
@@ -12,5 +12,6 @@ def __str__(self) -> str:
 
 class IndexingError(SdkError):
     def __init__(self, message: str = ""):
-        prefix = "Error with indexing. "
-        super().__init__(prefix + message)
+        if "404" in message:
+            message = "Index not found. Please check vector db adapter settings."
+        super().__init__(message)
diff --git a/src/unstract/sdk/index.py b/src/unstract/sdk/index.py
@@ -293,7 +293,7 @@ def index_file(
                     f"Error adding nodes to vector db: {e}",
                     level=LogLevel.ERROR,
                 )
-                raise SdkError(f"Error adding nodes to vector db: {e}")
+                raise IndexingError(str(e)) from e
             self.tool.stream_log("Added nodes to vector db")
 
         self.tool.stream_log("File has been indexed successfully")