settings.py

import os
from typing import Optional

from pydantic import BaseSettings, root_validator

from backend.types import MetadataStoreConfig, VectorDBConfig


class Settings(BaseSettings):
    """
    Settings class to hold all the environment variables
    """

    class Config:
        extra = "allow"

    MODELS_CONFIG_PATH: str
    METADATA_STORE_CONFIG: MetadataStoreConfig
    VECTOR_DB_CONFIG: VectorDBConfig

    LOCAL: bool = False

    TFY_HOST: str = ""
    TFY_API_KEY: str = ""
    JOB_FQN: str = ""
    JOB_COMPONENT_NAME: str = ""

    LOG_LEVEL: str = "info"
    TFY_SERVICE_ROOT_PATH: str = ""

    # TODO: This will be removed in future releases - after fixing multimodal parser
    TFY_LLM_GATEWAY_URL: str = ""

    @root_validator(pre=True)
    def _validate_values(cls, values):
        models_config_path = values.get("MODELS_CONFIG_PATH")
        if not os.path.isabs(models_config_path):
            this_dir = os.path.abspath(os.path.dirname(__file__))
            root_dir = os.path.dirname(this_dir)
            models_config_path = os.path.join(root_dir, models_config_path)

        if not models_config_path:
            raise Exception(
                f"{models_config_path} does not exist. "
                f"You can copy models_config.sample.yaml to {settings.MODELS_CONFIG_PATH} to bootstrap config"
            )

        values["MODELS_CONFIG_PATH"] = models_config_path

        tfy_host = values.get("TFY_HOST")
        tfy_llm_gateway_url = values.get("TFY_LLM_GATEWAY_URL")
        if tfy_host and not tfy_llm_gateway_url:
            tfy_llm_gateway_url = f"{tfy_host.rstrip('/')}/api/llm"
            values["TFY_LLM_GATEWAY_URL"] = tfy_llm_gateway_url

        return values


settings = Settings()