Spaces:
Sleeping
Sleeping
File size: 2,704 Bytes
3e25ded e29625a 3e25ded 99be2dc 3e25ded 3a1c55b 1438a00 3a1c55b 1438a00 3e25ded 99be2dc bc67f56 073de93 2037a88 3a1c55b 1c0aad9 3a1c55b 80c3ede 3a1c55b 905e1ea 3a1c55b 905e1ea 3a1c55b 905e1ea 5f72bf5 bc67f56 5f72bf5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 |
import os
from blossomtune_gradio import util
# BlossomTune yaml config path
BLOSSOMTUNE_CONFIG = os.getenv("BLOSSOMTUNE_CONFIG", None)
# HF Space ID
SPACE_ID = os.getenv("SPACE_ID", "ethicalabs/BlossomTune-Orchestrator")
SPACE_OWNER = os.getenv("SPACE_OWNER", SPACE_ID.split("/")[0] if SPACE_ID else None)
# Use persistent storage if available
DB_PATH = (
"/data/federation.db" if os.path.isdir("/data/") else "./data/db/federation.db"
)
SQLALCHEMY_URL = os.getenv("SQLALCHEMY_URL", f"sqlite:///{os.path.abspath(DB_PATH)}")
MAX_NUM_NODES = int(os.getenv("MAX_NUM_NODES", "20"))
SMTP_SENDER = os.getenv("SMTP_SENDER", "hello@ethicalabs.ai")
SMTP_SERVER = os.getenv("SMTP_SERVER", "localhost")
SMTP_PORT = int(os.getenv("SMTP_PORT", "1025"))
SMTP_REQUIRE_TLS = util.strtobool(os.getenv("SMTP_REQUIRE_TLS", "false"))
SMTP_USER = os.getenv("SMTP_USER", "")
SMTP_PASSWORD = os.getenv("SMTP_PASSWORD", "")
EMAIL_PROVIDER = os.getenv("EMAIL_PROVIDER", "smtp")
SUPERLINK_HOST = os.getenv("SUPERLINK_HOST", "127.0.0.1")
SUPERLINK_PORT = int(os.getenv("SUPERLINK_PORT", 9092))
SUPERLINK_CONTROL_API_PORT = int(os.getenv("SUPERLINK_CONTROL_API_PORT", 9093))
SUPERLINK_MODE = os.getenv("SUPERLINK_MODE", "internal").lower() # Or external
RUN_MIGRATIONS_ON_STARTUP = util.strtobool(
os.getenv("RUN_MIGRATIONS_ON_STARTUP", "true")
) # Set to false in prod.
# TLS root cert path. For production only.
TLS_CERT_DIR = os.getenv("TLS_CERT_DIR", "./certs/")
TLS_CA_KEY_PATH = os.getenv("TLS_CA_KEY_PATH", False)
TLS_CA_CERT_PATH = os.getenv("TLS_CA_CERT_PATH", False)
PROJECT_PATH = os.path.realpath(
os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")
)
# BlossomTune cert - To be distributed to the participants (supernodes).
BLOSSOMTUNE_TLS_CERT_PATH = os.getenv(
"BLOSSOMTUNE_TLS_CERT_PATH",
"/data/certs"
if os.path.isdir("/data/certs")
else os.path.join(PROJECT_PATH, "./data/certs"),
)
BLOSSOMTUNE_TLS_CA_CERTFILE = os.path.join(BLOSSOMTUNE_TLS_CERT_PATH, "ca.crt")
BLOSSOMTUNE_TLS_CERTFILE = os.path.join(BLOSSOMTUNE_TLS_CERT_PATH, "server.pem")
BLOSSOMTUNE_TLS_KEYFILE = os.path.join(BLOSSOMTUNE_TLS_CERT_PATH, "server.key")
# EC Auth - Keys
AUTH_KEYS_DIR = os.getenv(
"AUTH_KEYS_DIR",
"/data/keys/"
if os.path.isdir("/data/keys")
else os.path.join(PROJECT_PATH, "./data/keys/"),
)
# EC Auth - CSV File
AUTH_KEYS_CSV_PATH = os.getenv(
"AUTH_KEYS_CSV_PATH",
os.path.join(AUTH_KEYS_DIR, "authorized_supernodes.csv"),
)
# Flower Apps
FLOWER_APPS = os.getenv("FLOWER_APPS", ["flower_apps.quickstart_huggingface"])
FLOWER_APPS = (
FLOWER_APPS
if isinstance(FLOWER_APPS, list)
else [app.strip() for app in FLOWER_APPS.split(",")]
)
|