LongArticleSearchAgent/src/core/dependency/dependencies.py

from dependency_injector import containers, providers

from src.config import LongArticlesSearchAgentConfig
from src.infra.database import AsyncMySQLPool
from src.infra.trace import LogService


class ServerContainer(containers.DeclarativeContainer):
    config = providers.Singleton(LongArticlesSearchAgentConfig)

    log_service = providers.Singleton(LogService, log_config=config.provided.aliyun_log)

    mysql_pool = providers.Singleton(
        AsyncMySQLPool,
        config=config,
        log_service=log_service,
    )


__all__ = [
    "ServerContainer",
]