Skip to content

Commit 75873b3

Browse files
committed
fix: use consistent firmware/_quota model name in quota groups
1 parent 8360d8c commit 75873b3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/rotator_library/providers/firmware_provider.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ class FirmwareProvider(FirmwareQuotaTracker, ProviderInterface):
3535
"""
3636

3737
# Quota groups for tracking 5-hour rolling window limits
38-
# Uses a virtual model "_quota" for credential-level quota tracking
38+
# Uses a virtual model "firmware/_quota" for credential-level quota tracking
3939
model_quota_groups = {
40-
"firmware_global": ["_quota"],
40+
"firmware_global": ["firmware/_quota"],
4141
}
4242

4343
def __init__(self, *args, **kwargs):
@@ -80,7 +80,7 @@ def get_models_in_quota_group(self, group: str) -> List[str]:
8080
"""
8181
Get all models in a quota group.
8282
83-
For Firmware.ai, we use a virtual model "_quota" to track the
83+
For Firmware.ai, we use a virtual model "firmware/_quota" to track the
8484
credential-level 5-hour rolling window quota.
8585
8686
Args:
@@ -90,7 +90,7 @@ def get_models_in_quota_group(self, group: str) -> List[str]:
9090
List of model names in the group
9191
"""
9292
if group == "firmware_global":
93-
return ["_quota"]
93+
return ["firmware/_quota"]
9494
return []
9595

9696
def get_usage_reset_config(self, credential: str) -> Optional[Dict[str, Any]]:

0 commit comments

Comments
 (0)