Uddiii commited on
Commit
63726b6
·
1 Parent(s): 6c7cbc0

feat: add support for lowercase Hugging Face Space secrets

Browse files
ER_MAP/autoplay.py CHANGED
@@ -170,9 +170,9 @@ def main():
170
 
171
  from ER_MAP.envs.triage_env import TriageEnv
172
 
173
- nurse_key = os.environ.get("GROQ_NURSE_API_KEY", "")
174
- patient_key = os.environ.get("GROQ_PATIENT_API_KEY", "")
175
- doctor_key = os.environ.get("GROQ_DOCTOR_API_KEY", "") or patient_key
176
 
177
  if not nurse_key or not patient_key:
178
  print("ERROR: Set GROQ_NURSE_API_KEY and GROQ_PATIENT_API_KEY")
 
170
 
171
  from ER_MAP.envs.triage_env import TriageEnv
172
 
173
+ nurse_key = ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", ""))
174
+ patient_key = ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", ""))
175
+ doctor_key = ((os.environ.get("GROQ_DOCTOR_API_KEY") or os.environ.get("doctor")) or os.environ.get("doctor", "")) or patient_key
176
 
177
  if not nurse_key or not patient_key:
178
  print("ERROR: Set GROQ_NURSE_API_KEY and GROQ_PATIENT_API_KEY")
ER_MAP/cli_tester.py CHANGED
@@ -23,7 +23,7 @@ def print_header(title):
23
  print(f"{Fore.CYAN}{Style.BRIGHT}{'='*60}")
24
 
25
  def run_automated_cli(phase: int):
26
- groq_key = os.environ.get("GROQ_API_KEY") or os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("GROQ_PATIENT_API_KEY")
27
  if not groq_key or not GROQ_AVAILABLE:
28
  print(f"{Fore.RED}ERROR: GROQ_API_KEY environment variable required and 'groq' package must be installed.")
29
  return
 
23
  print(f"{Fore.CYAN}{Style.BRIGHT}{'='*60}")
24
 
25
  def run_automated_cli(phase: int):
26
+ groq_key = (os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or (os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or (os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient"))
27
  if not groq_key or not GROQ_AVAILABLE:
28
  print(f"{Fore.RED}ERROR: GROQ_API_KEY environment variable required and 'groq' package must be installed.")
29
  return
ER_MAP/dashboard.py CHANGED
@@ -259,10 +259,10 @@ def get_env():
259
  from ER_MAP.envs.triage_env import TriageEnv
260
  # Per-role keys (4 distinct Groq accounts in the demo config:
261
  # Nurse / Patient / Empathy Judge / Medical Judge).
262
- nurse_key = os.environ.get("GROQ_NURSE_API_KEY", "")
263
- patient_key = os.environ.get("GROQ_PATIENT_API_KEY", "")
264
- empathy_key = os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY", "")
265
- medical_key = os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY", "")
266
 
267
  # Default / fallback model — used by any role that doesn't
268
  # specify its own model.
@@ -292,14 +292,14 @@ def get_doctor():
292
  global DOCTOR
293
  if DOCTOR is None:
294
  # Doctor's primary key = its own dedicated Groq account.
295
- primary_key = os.environ.get("GROQ_DOCTOR_API_KEY", "") or os.environ.get("GROQ_PATIENT_API_KEY", "")
296
  # Full fallback chain: nurse → patient → empathy_judge → medical_judge.
297
  # Lets the demo survive 4-of-5 dead keys.
298
  fallback_keys = [
299
- os.environ.get("GROQ_NURSE_API_KEY", ""),
300
- os.environ.get("GROQ_PATIENT_API_KEY", ""),
301
- os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY", ""),
302
- os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY", ""),
303
  ]
304
  # Doctor model defaults to 8B (small/fast tier). Override with
305
  # ERMAP_DOCTOR_MODEL or legacy ERMAP_MODEL.
@@ -1474,11 +1474,11 @@ def _print_role_config_banner() -> None:
1474
  return k[:8] + "..." + k[-4:]
1475
 
1476
  rows = [
1477
- ("Doctor", os.environ.get("GROQ_DOCTOR_API_KEY", ""), os.environ.get("ERMAP_DOCTOR_MODEL", "llama-3.1-8b-instant")),
1478
- ("Nurse", os.environ.get("GROQ_NURSE_API_KEY", ""), os.environ.get("ERMAP_NURSE_MODEL", "llama-3.3-70b-versatile")),
1479
- ("Patient", os.environ.get("GROQ_PATIENT_API_KEY", ""), os.environ.get("ERMAP_PATIENT_MODEL", "llama-3.3-70b-versatile")),
1480
- ("Empathy Judge", os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY", ""), os.environ.get("ERMAP_EMPATHY_JUDGE_MODEL", "llama-3.3-70b-versatile")),
1481
- ("Medical Judge", os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY", ""), os.environ.get("ERMAP_MEDICAL_JUDGE_MODEL", "llama-3.3-70b-versatile")),
1482
  ]
1483
 
1484
  print("", flush=True)
 
259
  from ER_MAP.envs.triage_env import TriageEnv
260
  # Per-role keys (4 distinct Groq accounts in the demo config:
261
  # Nurse / Patient / Empathy Judge / Medical Judge).
262
+ nurse_key = ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", ""))
263
+ patient_key = ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", ""))
264
+ empathy_key = ((os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")) or os.environ.get("empathy", ""))
265
+ medical_key = ((os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")) or os.environ.get("medical", ""))
266
 
267
  # Default / fallback model — used by any role that doesn't
268
  # specify its own model.
 
292
  global DOCTOR
293
  if DOCTOR is None:
294
  # Doctor's primary key = its own dedicated Groq account.
295
+ primary_key = ((os.environ.get("GROQ_DOCTOR_API_KEY") or os.environ.get("doctor")) or os.environ.get("doctor", "")) or ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", ""))
296
  # Full fallback chain: nurse → patient → empathy_judge → medical_judge.
297
  # Lets the demo survive 4-of-5 dead keys.
298
  fallback_keys = [
299
+ ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", "")),
300
+ ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", "")),
301
+ ((os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")) or os.environ.get("empathy", "")),
302
+ ((os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")) or os.environ.get("medical", "")),
303
  ]
304
  # Doctor model defaults to 8B (small/fast tier). Override with
305
  # ERMAP_DOCTOR_MODEL or legacy ERMAP_MODEL.
 
1474
  return k[:8] + "..." + k[-4:]
1475
 
1476
  rows = [
1477
+ ("Doctor", ((os.environ.get("GROQ_DOCTOR_API_KEY") or os.environ.get("doctor")) or os.environ.get("doctor", "")), os.environ.get("ERMAP_DOCTOR_MODEL", "llama-3.1-8b-instant")),
1478
+ ("Nurse", ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", "")), os.environ.get("ERMAP_NURSE_MODEL", "llama-3.3-70b-versatile")),
1479
+ ("Patient", ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", "")), os.environ.get("ERMAP_PATIENT_MODEL", "llama-3.3-70b-versatile")),
1480
+ ("Empathy Judge", ((os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")) or os.environ.get("empathy", "")), os.environ.get("ERMAP_EMPATHY_JUDGE_MODEL", "llama-3.3-70b-versatile")),
1481
+ ("Medical Judge", ((os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")) or os.environ.get("medical", "")), os.environ.get("ERMAP_MEDICAL_JUDGE_MODEL", "llama-3.3-70b-versatile")),
1482
  ]
1483
 
1484
  print("", flush=True)
ER_MAP/envs/api_router.py CHANGED
@@ -140,11 +140,11 @@ class AgentRouter:
140
  }
141
 
142
  # Resolve per-role API keys (explicit > role-specific env > shared)
143
- shared_key = api_key or os.environ.get("GROQ_API_KEY", "")
144
- nurse_key = nurse_api_key or os.environ.get("GROQ_NURSE_API_KEY", "") or shared_key
145
- patient_key = patient_api_key or os.environ.get("GROQ_PATIENT_API_KEY", "") or shared_key
146
- empathy_key = empathy_judge_api_key or os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY", "") or ""
147
- medical_key = medical_judge_api_key or os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY", "") or ""
148
 
149
  # Create per-role Groq clients (4 logical roles).
150
  # If a judge has no dedicated key, its client is left None and the
 
140
  }
141
 
142
  # Resolve per-role API keys (explicit > role-specific env > shared)
143
+ shared_key = api_key or ((os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or os.environ.get("groq", ""))
144
+ nurse_key = nurse_api_key or ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", "")) or shared_key
145
+ patient_key = patient_api_key or ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", "")) or shared_key
146
+ empathy_key = empathy_judge_api_key or ((os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")) or os.environ.get("empathy", "")) or ""
147
+ medical_key = medical_judge_api_key or ((os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")) or os.environ.get("medical", "")) or ""
148
 
149
  # Create per-role Groq clients (4 logical roles).
150
  # If a judge has no dedicated key, its client is left None and the
ER_MAP/envs/openenv_triage/env.py CHANGED
@@ -72,11 +72,11 @@ def _stub_env_kwargs() -> Dict[str, Any]:
72
  empty keys is sufficient. The stub mode is documented in the README.
73
  """
74
  return {
75
- "groq_api_key": os.environ.get("GROQ_API_KEY", "") or None,
76
- "nurse_api_key": os.environ.get("GROQ_NURSE_API_KEY", "") or None,
77
- "patient_api_key": os.environ.get("GROQ_PATIENT_API_KEY", "") or None,
78
- "empathy_judge_api_key": os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY", "") or None,
79
- "medical_judge_api_key": os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY", "") or None,
80
  "model": os.environ.get("ERMAP_MODEL", "llama-3.3-70b-versatile"),
81
  }
82
 
 
72
  empty keys is sufficient. The stub mode is documented in the README.
73
  """
74
  return {
75
+ "groq_api_key": ((os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or os.environ.get("groq", "")) or None,
76
+ "nurse_api_key": ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", "")) or None,
77
+ "patient_api_key": ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", "")) or None,
78
+ "empathy_judge_api_key": ((os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")) or os.environ.get("empathy", "")) or None,
79
+ "medical_judge_api_key": ((os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")) or os.environ.get("medical", "")) or None,
80
  "model": os.environ.get("ERMAP_MODEL", "llama-3.3-70b-versatile"),
81
  }
82
 
ER_MAP/evaluate.py CHANGED
@@ -434,9 +434,9 @@ def main():
434
 
435
  from ER_MAP.envs.triage_env import TriageEnv
436
 
437
- nurse_key = os.environ.get("GROQ_NURSE_API_KEY", "")
438
- patient_key = os.environ.get("GROQ_PATIENT_API_KEY", "")
439
- doctor_key = os.environ.get("GROQ_DOCTOR_API_KEY", "") or patient_key
440
 
441
  if not nurse_key or not patient_key:
442
  print("ERROR: Set GROQ_NURSE_API_KEY and GROQ_PATIENT_API_KEY", flush=True)
 
434
 
435
  from ER_MAP.envs.triage_env import TriageEnv
436
 
437
+ nurse_key = ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", ""))
438
+ patient_key = ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", ""))
439
+ doctor_key = ((os.environ.get("GROQ_DOCTOR_API_KEY") or os.environ.get("doctor")) or os.environ.get("doctor", "")) or patient_key
440
 
441
  if not nurse_key or not patient_key:
442
  print("ERROR: Set GROQ_NURSE_API_KEY and GROQ_PATIENT_API_KEY", flush=True)
ER_MAP/evaluate_baseline.py CHANGED
@@ -273,11 +273,11 @@ def main() -> int:
273
  "(human-readable terminal). Set 0 for fast.")
274
  args = parser.parse_args()
275
 
276
- nurse_key = os.environ.get("GROQ_NURSE_API_KEY", "")
277
- patient_key = os.environ.get("GROQ_PATIENT_API_KEY", "")
278
- doctor_key = os.environ.get("GROQ_DOCTOR_API_KEY", "") or patient_key
279
- empathy_key = os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY", "") or nurse_key
280
- medical_key = os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY", "") or nurse_key
281
 
282
  if not nurse_key or not patient_key:
283
  print("ERROR: Set GROQ_NURSE_API_KEY and GROQ_PATIENT_API_KEY (and "
 
273
  "(human-readable terminal). Set 0 for fast.")
274
  args = parser.parse_args()
275
 
276
+ nurse_key = ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", ""))
277
+ patient_key = ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", ""))
278
+ doctor_key = ((os.environ.get("GROQ_DOCTOR_API_KEY") or os.environ.get("doctor")) or os.environ.get("doctor", "")) or patient_key
279
+ empathy_key = ((os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")) or os.environ.get("empathy", "")) or nurse_key
280
+ medical_key = ((os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")) or os.environ.get("medical", "")) or nurse_key
281
 
282
  if not nurse_key or not patient_key:
283
  print("ERROR: Set GROQ_NURSE_API_KEY and GROQ_PATIENT_API_KEY (and "
ER_MAP/play.py CHANGED
@@ -206,9 +206,9 @@ def main():
206
  except Exception as e:
207
  print(f" [TTS init failed: {e}] Running without voice.")
208
 
209
- nurse_key = os.environ.get("GROQ_NURSE_API_KEY", "")
210
- patient_key = os.environ.get("GROQ_PATIENT_API_KEY", "")
211
- shared_key = os.environ.get("GROQ_API_KEY", "")
212
 
213
  has_any_key = nurse_key or patient_key or shared_key
214
 
 
206
  except Exception as e:
207
  print(f" [TTS init failed: {e}] Running without voice.")
208
 
209
+ nurse_key = ((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", ""))
210
+ patient_key = ((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", ""))
211
+ shared_key = ((os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or os.environ.get("groq", ""))
212
 
213
  has_any_key = nurse_key or patient_key or shared_key
214
 
ER_MAP/server.py CHANGED
@@ -97,11 +97,11 @@ def _get_env() -> TriageEnv:
97
  global _ENV
98
  if _ENV is None:
99
  _ENV = TriageEnv(
100
- groq_api_key=os.environ.get("GROQ_API_KEY", ""),
101
- nurse_api_key=os.environ.get("GROQ_NURSE_API_KEY", ""),
102
- patient_api_key=os.environ.get("GROQ_PATIENT_API_KEY", ""),
103
- empathy_judge_api_key=os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY", ""),
104
- medical_judge_api_key=os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY", ""),
105
  model=os.environ.get("ERMAP_MODEL", "llama-3.3-70b-versatile"),
106
  nurse_model=os.environ.get("ERMAP_NURSE_MODEL"),
107
  patient_model=os.environ.get("ERMAP_PATIENT_MODEL"),
 
97
  global _ENV
98
  if _ENV is None:
99
  _ENV = TriageEnv(
100
+ groq_api_key=((os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or os.environ.get("groq", "")),
101
+ nurse_api_key=((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", "")),
102
+ patient_api_key=((os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")) or os.environ.get("patient", "")),
103
+ empathy_judge_api_key=((os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")) or os.environ.get("empathy", "")),
104
+ medical_judge_api_key=((os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")) or os.environ.get("medical", "")),
105
  model=os.environ.get("ERMAP_MODEL", "llama-3.3-70b-versatile"),
106
  nurse_model=os.environ.get("ERMAP_NURSE_MODEL"),
107
  patient_model=os.environ.get("ERMAP_PATIENT_MODEL"),
ER_MAP/training/train_grpo.py CHANGED
@@ -791,7 +791,7 @@ def train(
791
  f"kl_beta={kl_beta}"
792
  )
793
 
794
- groq_key = groq_api_key or os.environ.get("GROQ_API_KEY", "")
795
 
796
  scheduler = CurriculumScheduler()
797
  logger.info(f"Starting Phase: {scheduler.current_phase.name}")
 
791
  f"kl_beta={kl_beta}"
792
  )
793
 
794
+ groq_key = groq_api_key or ((os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or os.environ.get("groq", ""))
795
 
796
  scheduler = CurriculumScheduler()
797
  logger.info(f"Starting Phase: {scheduler.current_phase.name}")
ER_MAP/tts_engine.py CHANGED
@@ -461,7 +461,7 @@ class TTSEngine:
461
  """
462
 
463
  def __init__(self, elevenlabs_api_key: Optional[str] = None, groq_api_key: Optional[str] = None):
464
- self.api_key = elevenlabs_api_key or os.environ.get("ELEVENLABS_API_KEY", "")
465
  self.use_elevenlabs = False
466
  self._eleven_client = None
467
  self._pygame = None
@@ -493,12 +493,12 @@ class TTSEngine:
493
  # if one of the other Groq accounts is alive.
494
  candidate_keys = [
495
  groq_api_key,
496
- os.environ.get("GROQ_NURSE_API_KEY"),
497
- os.environ.get("GROQ_PATIENT_API_KEY"),
498
- os.environ.get("GROQ_DOCTOR_API_KEY"),
499
- os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY"),
500
- os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY"),
501
- os.environ.get("GROQ_API_KEY"),
502
  ]
503
  _groq_key = next((k for k in candidate_keys if k), "")
504
  if _groq_key:
 
461
  """
462
 
463
  def __init__(self, elevenlabs_api_key: Optional[str] = None, groq_api_key: Optional[str] = None):
464
+ self.api_key = elevenlabs_api_key or ((os.environ.get("ELEVENLABS_API_KEY") or os.environ.get("elevenlabs")) or os.environ.get("elevenlabs", ""))
465
  self.use_elevenlabs = False
466
  self._eleven_client = None
467
  self._pygame = None
 
493
  # if one of the other Groq accounts is alive.
494
  candidate_keys = [
495
  groq_api_key,
496
+ (os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")),
497
+ (os.environ.get("GROQ_PATIENT_API_KEY") or os.environ.get("patient")),
498
+ (os.environ.get("GROQ_DOCTOR_API_KEY") or os.environ.get("doctor")),
499
+ (os.environ.get("GROQ_EMPATHY_JUDGE_API_KEY") or os.environ.get("empathy")),
500
+ (os.environ.get("GROQ_MEDICAL_JUDGE_API_KEY") or os.environ.get("medical")),
501
+ (os.environ.get("GROQ_API_KEY") or os.environ.get("groq")),
502
  ]
503
  _groq_key = next((k for k in candidate_keys if k), "")
504
  if _groq_key:
kaggle/build_notebook.py CHANGED
@@ -653,8 +653,8 @@ metrics = train(
653
  num_episodes=NUM_EPISODES,
654
  group_size=GROUP_SIZE,
655
  model_name=MODEL_NAME,
656
- groq_api_key=os.environ.get("GROQ_NURSE_API_KEY", "")
657
- or os.environ.get("GROQ_API_KEY", ""),
658
  learning_rate=LEARNING_RATE,
659
  kl_beta=KL_BETA,
660
  use_wandb=USE_WANDB,
 
653
  num_episodes=NUM_EPISODES,
654
  group_size=GROUP_SIZE,
655
  model_name=MODEL_NAME,
656
+ groq_api_key=((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", ""))
657
+ or ((os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or os.environ.get("groq", "")),
658
  learning_rate=LEARNING_RATE,
659
  kl_beta=KL_BETA,
660
  use_wandb=USE_WANDB,
kaggle/clean_launch.py CHANGED
@@ -217,8 +217,8 @@ metrics = tg.train(
217
  num_episodes=NUM_EPISODES,
218
  group_size=GROUP_SIZE,
219
  model_name=MODEL_NAME,
220
- groq_api_key=os.environ.get("GROQ_NURSE_API_KEY", "")
221
- or os.environ.get("GROQ_API_KEY", ""),
222
  learning_rate=LEARNING_RATE,
223
  kl_beta=KL_BETA,
224
  use_wandb=False,
 
217
  num_episodes=NUM_EPISODES,
218
  group_size=GROUP_SIZE,
219
  model_name=MODEL_NAME,
220
+ groq_api_key=((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or os.environ.get("nurse", ""))
221
+ or ((os.environ.get("GROQ_API_KEY") or os.environ.get("groq")) or os.environ.get("groq", "")),
222
  learning_rate=LEARNING_RATE,
223
  kl_beta=KL_BETA,
224
  use_wandb=False,
kaggle/kaggle_helpers.py CHANGED
@@ -278,7 +278,7 @@ def kaggle_env_summary() -> None:
278
  except Exception:
279
  gpu, vram = "(torch import failed)", "—"
280
 
281
- has_groq = bool(os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("GROQ_API_KEY"))
282
  has_hf = bool(os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_TOKEN"))
283
  has_wandb = bool(os.environ.get("WANDB_API_KEY"))
284
 
 
278
  except Exception:
279
  gpu, vram = "(torch import failed)", "—"
280
 
281
+ has_groq = bool((os.environ.get("GROQ_NURSE_API_KEY") or os.environ.get("nurse")) or (os.environ.get("GROQ_API_KEY") or os.environ.get("groq")))
282
  has_hf = bool(os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_TOKEN"))
283
  has_wandb = bool(os.environ.get("WANDB_API_KEY"))
284