main
py 15 lines 611 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4 import regex as re
5
6
7 def sanitize_message(message: str) -> str:
8 """Strip memory addresses, tracebacks, and file paths from error messages."""
9 message = re.sub(r" at 0x[0-9a-fA-F]+>", ">", message)
10 message = re.sub(r'\n?\s*File "[^"]+", line \d+, in \S+(\n\s+.*)?', "", message)
11 message = re.sub(
12 r"/(?:home|usr|opt|var|tmp|root|lib|mnt|srv)(?:/[\w.\-]+)+", "<path>", message
13 )
14 message = re.sub(r"(?:/[\w\-]+)+/[\w\-]+\.\w+", "<path>", message)
15 return message.strip()