Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 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
18 changes: 3 additions & 15 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,16 @@ 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:
"""
Wraps an exception in a HTTPException.
"""
if 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