This commit is contained in:
superobk
2024-05-16 17:08:56 +08:00
parent 0612c065c0
commit 6b943d9f3d
3 changed files with 20 additions and 2 deletions

View File

@ -20,7 +20,7 @@ class Main():
def run(self): def run(self):
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logger.info("jarvis-models start", extra={"version": "0.0.1"}) 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__": if __name__ == "__main__":
injector = Injector() injector = Injector()

View File

@ -4,6 +4,8 @@ from typing import Any, Coroutine
from fastapi import Request, Response, status from fastapi import Request, Response, status
from fastapi.responses import JSONResponse from fastapi.responses import JSONResponse
from ..log.log import Logger
from ..asr.rapid_paraformer.utils import read_yaml from ..asr.rapid_paraformer.utils import read_yaml
from ..asr.rapid_paraformer import RapidParaformer from ..asr.rapid_paraformer import RapidParaformer
from .blackbox import Blackbox from .blackbox import Blackbox
@ -13,7 +15,8 @@ from injector import singleton, inject
class ASR(Blackbox): class ASR(Blackbox):
@inject @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) config = read_yaml(path)
self.paraformer = RapidParaformer(config) self.paraformer = RapidParaformer(config)

15
src/log/log.py Normal file
View File

@ -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'))