mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-04-20 14:28:54 +00:00
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:
6
utils.py
6
utils.py
@@ -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())
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user