diff --git a/comet/main.py b/comet/main.py index e83d267..dc31268 100644 --- a/comet/main.py +++ b/comet/main.py @@ -15,9 +15,9 @@ from starlette.requests import Request from comet.api.core import main from comet.api.stream import streams +from comet.utils.db import setup_database, teardown_database, write_config from comet.utils.logger import logger from comet.utils.models import settings -from comet.utils.db import setup_database, teardown_database, write_config class LoguruMiddleware(BaseHTTPMiddleware): diff --git a/comet/utils/db.py b/comet/utils/db.py index 2bc1a4f..c6c0b86 100644 --- a/comet/utils/db.py +++ b/comet/utils/db.py @@ -1,8 +1,9 @@ import json import os + +from comet.utils.general import lang_code_map from comet.utils.logger import logger from comet.utils.models import database, settings -from comet.utils.general import lang_code_map async def setup_database():