Spaces:
Paused
Paused
Mirrowel commited on
Commit ·
5fb237f
1
Parent(s): 50f9490
fix: Handle missing cooldown values in UsageManager to prevent key lockout type errors
Browse files
src/rotator_library/usage_manager.py
CHANGED
|
@@ -144,8 +144,8 @@ class UsageManager:
|
|
| 144 |
key_data = self._usage_data.get(key, {})
|
| 145 |
|
| 146 |
# Skip keys on global or model-specific cooldown
|
| 147 |
-
if key_data.get("key_cooldown_until"
|
| 148 |
-
key_data.get("model_cooldowns", {}).get(model
|
| 149 |
continue
|
| 150 |
|
| 151 |
usage_count = key_data.get("daily", {}).get("models", {}).get(model, {}).get("success_count", 0)
|
|
|
|
| 144 |
key_data = self._usage_data.get(key, {})
|
| 145 |
|
| 146 |
# Skip keys on global or model-specific cooldown
|
| 147 |
+
if (key_data.get("key_cooldown_until") or 0) > now or \
|
| 148 |
+
(key_data.get("model_cooldowns", {}).get(model) or 0) > now:
|
| 149 |
continue
|
| 150 |
|
| 151 |
usage_count = key_data.get("daily", {}).get("models", {}).get(model, {}).get("success_count", 0)
|