Spaces:
Sleeping
Sleeping
Commit
·
7fbd1d7
1
Parent(s):
90c12a2
Upd env pref
Browse files- 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"):
|