From 0402fffff78dfe37dddf9242b46da0d58de5af64 Mon Sep 17 00:00:00 2001 From: Yan Niznik Date: Fri, 9 Oct 2026 08:55:47 +0000 Subject: [PATCH] fix(proxy/db): classify prisma engine internal error as transport error to trigger reconnect --- litellm/proxy/db/exception_handler.py | 2 ++ tests/unit/proxy/db/test_exception_handler.py | 1 + 2 files changed, 3 insertions(+) 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():