rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

logger file

Browse Source
This commit is contained in:
Divided by Zer0
2022-09-11 22:59:19 +02:00
parent 432af79fa5
commit ce2d1ff654
2 changed files with 30 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
logger.py Normal file
View File

@ -0,0 +1,23 @@
import sys
from functools import partialmethod
from loguru import logger
def is_stdout_log(record):
if record["level"].name == "GENERATION":
return(True)
return(False)
def is_stderr_log(record):
if record["level"].name != "GENERATION":
return(True)
return(False)
logfmt = "<level>{level: <10}</level> | <green>{name}</green>:<green>{function}</green>:<green>{line}</green> - <level>{message}</level>"
genfmt = "<level>{level: <10}</level> @ <green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{message}</level>"
new_level = logger.level("GENERATION", no=21, color="<cyan>")
logger.__class__.generation = partialmethod(logger.__class__.log, "GENERATION")
config = {
"handlers": [
{"sink": sys.stderr, "format": logfmt, "colorize":True, "filter": is_stderr_log},
{"sink": sys.stdout, "format": genfmt, "level": "GENERATION", "colorize":True, "filter": is_stdout_log},
],
}
logger.configure(**config)