Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion unstructured_platform_plugins/__version__.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__ = "0.0.33" # pragma: no cover
__version__ = "0.0.34" # pragma: no cover
4 changes: 2 additions & 2 deletions unstructured_platform_plugins/etl_uvicorn/api_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,8 +216,8 @@ async def _stream_response():
usage=usage,
message_channels=message_channels,
filedata_meta=filedata_meta_model.model_validate(filedata_meta.model_dump()),
status_code=wrap_error(exc).status_code,
status_code_text=f"[{exc.__class__.__name__}] {exc}",
status_code=exc.status_code,
status_code_text=exc.detail,
file_data=request_dict.get("file_data", None),
)
except UnrecoverableException as ex:
Expand Down
22 changes: 6 additions & 16 deletions unstructured_platform_plugins/etl_uvicorn/errors.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,10 +20,6 @@ class UserAuthError(UserError):
status_code: int = 403


class UnprocessableEntityError(UserError):
status_code: int = 422


class RateLimitError(UserError):
status_code: int = 429

Expand All @@ -36,24 +32,18 @@ class ProviderError(BaseError):
status_code: int = 500


class GatewayTimeoutError(BaseError):
status_code: int = 504


class CatchAllError(BaseError):
status_code: int = 512


def wrap_error(e: Exception) -> HTTPException:
if isinstance(e, ingest_errors.UserAuthError):
"""
Wraps an exception in a HTTPException.
"""
if isinstance(e, HTTPException):
return e
elif isinstance(e, ingest_errors.UserAuthError):
return UserAuthError(e)
elif isinstance(e, HTTPException):
if e.status_code == 400:
return UserError(e)
if e.status_code == 422:
return UnprocessableEntityError(e)
if e.status_code == 504:
return GatewayTimeoutError(e)
elif isinstance(e, ingest_errors.RateLimitError):
return RateLimitError(e)
elif isinstance(e, ingest_errors.QuotaError):
Expand Down