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):
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()

View File

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

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