import os
from src.core import DataBaseManager
from src.core.retriever import Retriever
from src.models import select_model
from src.config import Config
from src.utils import setup_logger

logger = setup_logger("Startup")


class Startup:
    def __init__(self):
        self.start()

    def start(self):
        self.config = Config()
        self.model = select_model(self.config)
        self.dbm = DataBaseManager(self.config)
        self.retriever = Retriever(self.config, self.dbm, self.model)

        self.model_lite = select_model(self.config,
                                       model_provider=self.config.model_provider_lite or "zhipu",
                                       model_name=self.config.model_name_lite or "glm-4-flash")

    def restart(self):
        logger.info("Restarting...")
        self.start()
        logger.info("Restarted")


startup = Startup()