init
This commit is contained in:
5
vllm/logging/__init__.py
Normal file
5
vllm/logging/__init__.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from vllm.logging.formatter import NewLineFormatter
|
||||
|
||||
__all__ = [
|
||||
"NewLineFormatter",
|
||||
]
|
||||
15
vllm/logging/formatter.py
Normal file
15
vllm/logging/formatter.py
Normal file
@@ -0,0 +1,15 @@
|
||||
import logging
|
||||
|
||||
|
||||
class NewLineFormatter(logging.Formatter):
|
||||
"""Adds logging prefix to newlines to align multi-line messages."""
|
||||
|
||||
def __init__(self, fmt, datefmt=None, style="%"):
|
||||
logging.Formatter.__init__(self, fmt, datefmt, style)
|
||||
|
||||
def format(self, record):
|
||||
msg = logging.Formatter.format(self, record)
|
||||
if record.message != "":
|
||||
parts = msg.split(record.message)
|
||||
msg = msg.replace("\n", "\r\n" + parts[0])
|
||||
return msg
|
||||
Reference in New Issue
Block a user