diff --git a/litellm/proxy/db/exception_handler.py b/litellm/proxy/db/exception_handler.py index 379e0012ea8..5e05d75cc13 100644 --- a/litellm/proxy/db/exception_handler.py +++ b/litellm/proxy/db/exception_handler.py @@ -203,6 +203,8 @@ class PrismaDBExceptionHandler: ), ): return True + if PrismaDBExceptionHandler.is_prisma_engine_internal_error(e): + return True if PrismaDBExceptionHandler.is_database_capacity_error(e): return False if isinstance(e, _exception_types(prisma.errors.PrismaError)): diff --git a/tests/unit/proxy/db/test_exception_handler.py b/tests/unit/proxy/db/test_exception_handler.py index 75346ca08a6..fc0ffb0b64d 100644 --- a/tests/unit/proxy/db/test_exception_handler.py +++ b/tests/unit/proxy/db/test_exception_handler.py @@ -195,6 +195,7 @@ def test_is_database_service_unavailable_error_prisma_engine_malformed_payload() assert "no attribute 'get'" in str(exc_info.value) assert PrismaDBExceptionHandler.is_database_service_unavailable_error(exc_info.value) is True + assert PrismaDBExceptionHandler.is_database_transport_error(exc_info.value) is True def test_is_prisma_engine_internal_error_excludes_application_attributeerror():