LiamKhoaLe commited on
Commit
7fbd1d7
·
1 Parent(s): 90c12a2

Upd env pref

Browse files
Files changed (1) hide show
  1. utils/cloud_llm.py +2 -1
utils/cloud_llm.py CHANGED
@@ -40,6 +40,7 @@ def snip(s: str, n: int = 12) -> str:
40
 
41
  class KeyRotator:
42
  def __init__(self, env_prefix: str, max_keys: int = 5):
 
43
  keys = []
44
  for i in range(1, max_keys + 1):
45
  v = os.getenv(f"{env_prefix}_{i}")
@@ -82,7 +83,7 @@ class KeyRotator:
82
  return k
83
 
84
  # All keys are dead or temporarily unavailable
85
- logger.warning(f"[LLM] All keys for {env_prefix} are unavailable")
86
  return None
87
 
88
  def mark_bad(self, key: Optional[str], error_type: str = "unknown"):
 
40
 
41
  class KeyRotator:
42
  def __init__(self, env_prefix: str, max_keys: int = 5):
43
+ self.env_prefix = env_prefix # Store as instance variable
44
  keys = []
45
  for i in range(1, max_keys + 1):
46
  v = os.getenv(f"{env_prefix}_{i}")
 
83
  return k
84
 
85
  # All keys are dead or temporarily unavailable
86
+ logger.warning(f"[LLM] All keys for {self.env_prefix} are unavailable")
87
  return None
88
 
89
  def mark_bad(self, key: Optional[str], error_type: str = "unknown"):