src/parameters.py

import os

from dotenv import load_dotenv

from src import custom_log

load_dotenv()

# AWS Resouce variables
os.environ["AWS_ACCESS_KEY_ID"] = os.environ.get("AWS_ACCESS_KEY_ID")
os.environ["AWS_SECRET_ACCESS_KEY"] = os.environ.get("AWS_SECRET_ACCESS_KEY")
os.environ["S3_BUCKET_NAME"] = os.environ.get("AWS_BUCKET")
AWS_BUCKET = os.environ.get("AWS_BUCKET")
AWS_REGION = os.environ.get("AWS_REGION")

# Local variables
MAX_WORKERS = int(os.environ.get("MAX_WORKERS"))
MAX_TRIES = int(os.environ.get("MAX_TRIES"))

# xAI AI API Key
XAI_API_KEY = os.environ.get("XAI_API_KEY")

# Custom logger
LOGGER = custom_log.setup_logger(__name__)