mohsin-devs commited on
Commit
9d1006c
·
1 Parent(s): 2fe2727

Default HF storage mode for Spaces startup

Browse files
Files changed (1) hide show
  1. server/config.py +5 -3
server/config.py CHANGED
@@ -48,14 +48,16 @@ LOG_LEVEL = os.getenv("LOG_LEVEL", "INFO")
48
  HF_STORAGE_LABEL = "HF"
49
 
50
 
51
- def _required_env(name: str) -> str:
52
  value = os.getenv(name, "").strip()
 
 
53
  if not value:
54
  raise RuntimeError(f"Missing required environment variable: {name}")
55
  return value
56
 
57
 
58
- STORAGE_MODE = _required_env("STORAGE_MODE").upper()
59
  if STORAGE_MODE != HF_STORAGE_LABEL:
60
  raise RuntimeError(
61
  f"Unsupported STORAGE_MODE '{STORAGE_MODE}'. "
@@ -64,7 +66,7 @@ if STORAGE_MODE != HF_STORAGE_LABEL:
64
 
65
  HF_TOKEN = _required_env("HF_TOKEN")
66
  HF_REPO_ID = _required_env("HF_REPO_ID")
67
- HF_REPO_TYPE = _required_env("HF_REPO_TYPE")
68
  if HF_REPO_TYPE not in {"dataset", "model", "space"}:
69
  raise RuntimeError(
70
  "Invalid HF_REPO_TYPE. Supported values are: dataset, model, space."
 
48
  HF_STORAGE_LABEL = "HF"
49
 
50
 
51
+ def _required_env(name: str, default: str | None = None) -> str:
52
  value = os.getenv(name, "").strip()
53
+ if not value and default is not None:
54
+ value = default.strip()
55
  if not value:
56
  raise RuntimeError(f"Missing required environment variable: {name}")
57
  return value
58
 
59
 
60
+ STORAGE_MODE = _required_env("STORAGE_MODE", "HF").upper()
61
  if STORAGE_MODE != HF_STORAGE_LABEL:
62
  raise RuntimeError(
63
  f"Unsupported STORAGE_MODE '{STORAGE_MODE}'. "
 
66
 
67
  HF_TOKEN = _required_env("HF_TOKEN")
68
  HF_REPO_ID = _required_env("HF_REPO_ID")
69
+ HF_REPO_TYPE = _required_env("HF_REPO_TYPE", "dataset")
70
  if HF_REPO_TYPE not in {"dataset", "model", "space"}:
71
  raise RuntimeError(
72
  "Invalid HF_REPO_TYPE. Supported values are: dataset, model, space."