jalr/tabbyAPI-ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Logging: Fix inconsistent formatting

Browse source 
Some colorization was incorrect and the separator insertion has become
more robust.

Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
kingbri 2024-03-08 00:51:07 -05:00 • committed by Brian Dashore
parent 228c227c1e
commit ef2dc326f5
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
common/logger.py
View file

@ -46,6 +46,9 @@ def _log_formatter(record: dict) -> str:
}
level = record.get("level")
level_color = color_map.get(level.name, "cyan")
colored_level = f"[{level_color}]{level.name}[/{level_color}]:"
separator = " " * (9 - len(level.name))
message = unwrap(record.get("message"), "")
lines = message.splitlines()
@ -57,12 +60,12 @@ def _log_formatter(record: dict) -> str:
if len(lines) > 1:
fmt = "\n".join(
[
f"[{level_color}]{level.name + ':' :<10}[/{level_color}]{line}"
f"{colored_level}{separator}{line}"
for line in lines
]
)
else:
fmt = f"[{level_color}]{level.name + ':' :<10}[/{level_color}]{message}"
fmt = f"{colored_level}{separator}{message}"
return fmt