src/utils/_config.py

import os

from dotenv import load_dotenv
from loguru import logger

if os.getenv("PYTHON_ENV") and os.getenv("PYTHON_ENV").lower() == "test":
    load_dotenv(dotenv_path=".env.test", override=True)
else:
    load_dotenv()


LOG_FILE = os.getenv("LOG_FILE")
if LOG_FILE:
    logger.add(LOG_FILE, retention="10 days")

if (
    not os.getenv("SLACK_BOT_TOKEN")
    and os.getenv("PYTHON_ENV")
    and os.getenv("PYTHON_ENV").lower() == "test"
):
    logger.error("SLACK_BOT_TOKEN environment not set.")
    raise ValueError("SLACK_BOT_TOKEN environment not set.")

# if not os.getenv("SERVICE_ACCOUNT_CREDENTIAL_FILE_PATH"):
#     logger.error("SERVICE_ACCOUNT_CREDENTIAL_FILE_PATH environment not set.")
#     raise ValueError("SERVICE_ACCOUNT_CREDENTIAL_FILE_PATH environment not set.")

# if not os.getenv("SPREAD_SHEET_ID"):
#     logger.error("SPREAD_SHEET_ID environment not set.")
#     raise ValueError("SPREAD_SHEET_ID environment not set.")

# if not os.getenv("WORKSHEET_NAME"):
#     logger.error("WORKSHEET_NAME environment not set.")
#     raise ValueError("WORKSHEET_NAME environment not set.")

if not os.getenv("OPENAI_API_KEY"):
    logger.error("OPENAI_API_KEY environment not set.")
    raise ValueError("OPENAI_API_KEY environment not set.")

if not os.getenv("PINECONE_API_KEY"):
    logger.error("PINECONE_API_KEY environment not set.")
    raise ValueError("PINECONE_API_KEY environment not set.")

if not os.getenv("LLAMAPARSE_API_KEY"):
    logger.error("LLAMAPARSE_API_KEY environment not set.")
    raise ValueError("LLAMAPARSE_API_KEY environment not set.")

if not os.getenv("SQLALCHEMY_DATABASE_URI"):
    logger.error("SQLALCHEMY_DATABASE_URI environment not set.")
    raise ValueError("SQLALCHEMY_DATABASE_URI environment not set.")

if not os.getenv("SLACK_CLIENT_ID"):
    logger.error("SLACK_CLIENT_ID environment not set.")
    raise ValueError("SLACK_CLIENT_ID environment not set.")

if not os.getenv("SLACK_CLIENT_SECRET"):
    logger.error("SLACK_CLIENT_SECRET environment not set.")
    raise ValueError("SLACK_CLIENT_SECRET environment not set.")

if not os.getenv("SLACK_REDIRECT_URI"):
    logger.error("SLACK_REDIRECT_URI environment not set.")
    raise ValueError("SLACK_REDIRECT_URI environment not set.")