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", 0) > now or \
148
- key_data.get("model_cooldowns", {}).get(model, 0) > now:
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)