diff --git a/tests/entrypoints/serve/exception_handling/test_error_sanitization.py b/tests/entrypoints/serve/exception_handling/test_error_sanitization.py index 426d1ede74..98bf808965 100644 --- a/tests/entrypoints/serve/exception_handling/test_error_sanitization.py +++ b/tests/entrypoints/serve/exception_handling/test_error_sanitization.py @@ -18,6 +18,10 @@ def test_sanitize_message(): sanitize_message("<_io.BytesIO object at 0x7a95e299e750>") == "<_io.BytesIO object>" ) + assert ( + sanitize_message("") + == "" + ) class TestSanitizeMessageFilePaths: diff --git a/vllm/entrypoints/serve/exception_handling/utils.py b/vllm/entrypoints/serve/exception_handling/utils.py index 73d9992237..6c1dd20516 100644 --- a/vllm/entrypoints/serve/exception_handling/utils.py +++ b/vllm/entrypoints/serve/exception_handling/utils.py @@ -6,7 +6,7 @@ import regex as re def sanitize_message(message: str) -> str: """Strip memory addresses, tracebacks, and file paths from error messages.""" - message = re.sub(r" at 0x[0-9a-f]+>", ">", message) + message = re.sub(r" at 0x[0-9a-fA-F]+>", ">", message) message = re.sub(r'\n?\s*File "[^"]+", line \d+, in \S+(\n\s+.*)?', "", message) message = re.sub( r"/(?:home|usr|opt|var|tmp|root|lib|mnt|srv)(?:/[\w.\-]+)+", "", message