feat: logging (#39)

* add logging

* simplify the logger

* formatting

* final touches

* fix format

* Model: Add log to metrics

Signed-off-by: kingbri <bdashore3@proton.me>

---------

Authored-by: AlpinDale <52078762+AlpinDale@users.noreply.github.com>
This commit is contained in:
AlpinDale
2023-12-23 04:33:31 +00:00
committed by GitHub
parent f5314fcdad
commit 6a5bbd217c
11 changed files with 170 additions and 74 deletions

View File

@@ -4,6 +4,10 @@ from typing import Optional
from pydantic import BaseModel
from logger import init_logger
logger = init_logger(__name__)
def load_progress(module, modules):
"""Wrapper callback for load progress."""
@@ -32,7 +36,7 @@ def get_generator_error(message: str):
generator_error = TabbyGeneratorError(error=error_message)
# Log and send the exception
print(f"\n{generator_error.error.trace}")
logger.error(generator_error.error.message)
return get_sse_packet(generator_error.model_dump_json())