From 6b943d9f3dd1a0352b362a1ac680186a9928b82f Mon Sep 17 00:00:00 2001 From: superobk Date: Thu, 16 May 2024 17:08:56 +0800 Subject: [PATCH] wip --- main.py | 2 +- src/blackbox/asr.py | 5 ++++- src/log/log.py | 15 +++++++++++++++ 3 files changed, 20 insertions(+), 2 deletions(-) create mode 100644 src/log/log.py diff --git a/main.py b/main.py index 83ace40..8e51269 100644 --- a/main.py +++ b/main.py @@ -20,7 +20,7 @@ class Main(): def run(self): logger = logging.getLogger(__name__) logger.info("jarvis-models start", extra={"version": "0.0.1"}) - uvicorn.run("server:app", host="0.0.0.0", port=8000, log_level="info") + uvicorn.run("server:app", host="0.0.0.0", port=8001, log_level="info") if __name__ == "__main__": injector = Injector() diff --git a/src/blackbox/asr.py b/src/blackbox/asr.py index 5338ec2..6b0efcc 100644 --- a/src/blackbox/asr.py +++ b/src/blackbox/asr.py @@ -4,6 +4,8 @@ from typing import Any, Coroutine from fastapi import Request, Response, status from fastapi.responses import JSONResponse +from ..log.log import Logger + from ..asr.rapid_paraformer.utils import read_yaml from ..asr.rapid_paraformer import RapidParaformer from .blackbox import Blackbox @@ -13,7 +15,8 @@ from injector import singleton, inject class ASR(Blackbox): @inject - def __init__(self,path = ".env.yaml") -> None: + def __init__(self,logger: Logger,path = ".env.yaml") -> None: + logger.info("ASR init") config = read_yaml(path) self.paraformer = RapidParaformer(config) diff --git a/src/log/log.py b/src/log/log.py new file mode 100644 index 0000000..2c7c0b1 --- /dev/null +++ b/src/log/log.py @@ -0,0 +1,15 @@ +import logging +from injector import singleton, inject + +from ..configuration import LogConf + +@singleton +class Logger(logging.Logger): + + @inject + def __init__(self, config: LogConf) -> None: + super().__init__(__name__) + self.setLevel(config.level) + self.addHandler(logging.StreamHandler()) + self.addHandler(logging.FileHandler('log.log')) +