Raise APIError

Zipstack · Sep 4, 2024 · 33695d0 · 33695d0
1 parent fbb8a21
commit 33695d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prompt-service/src/unstract/prompt_service/main.py b/prompt-service/src/unstract/prompt_service/main.py
@@ -245,7 +245,7 @@ def prompt_processor() -> Any:
                 RunLevel.RUN,
                 "Unable to obtain LLM / embedding / vectorDB",
             )
-            return APIError(message=msg)
+            raise APIError(message=msg)
 
         if output[PSKeys.TYPE] == PSKeys.TABLE:
             try:
@@ -323,7 +323,7 @@ def prompt_processor() -> Any:
                         RunLevel.RUN,
                         "Extracted file not present.",
                     )
-                    return APIError(message=msg)
+                    raise APIError(message=msg)
 
                 # TODO: Use vectorDB name when available
                 _publish_log(