Spaces:
Runtime error
Runtime error
File size: 1,178 Bytes
b3d3593 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import functools
def wrap_exceptions(exception_cls, message=None):
"""Wrap exceptions raised by a function with a custom exception class."""
def decorated(f):
@functools.wraps(f)
def wrapped(*args, **kwargs):
try:
return f(*args, **kwargs)
except Exception as e:
raise exception_cls(message) from e
return wrapped
return decorated
def async_wrap_exceptions(exception_cls, message=None):
"""Wrap exceptions raised by an async function with a custom exception class."""
def decorated(f):
@functools.wraps(f)
async def wrapped(*args, **kwargs):
try:
return await f(*args, **kwargs)
except Exception as e:
raise exception_cls(message) from e
return wrapped
return decorated
class TaskPlanningException(Exception):
pass
class TaskParsingException(Exception):
pass
class ModelScrapingException(Exception):
pass
class ModelSelectionException(Exception):
pass
class ModelInferenceException(Exception):
pass
class ResponseGenerationException(Exception):
pass
|