fix(pool): isolate dynamic model quota buckets and 429 scheduling

2026-10-08 02:17:46 +08:00 · 2026-09-02 15:23:23 +08:00
parent 2cd20da1ec
commit dbbe7b22ab
37 changed files with 1491 additions and 78 deletions
@@ -118,6 +118,30 @@ pub fn admin_pool_key_quota_hard_blocked(
    aether_provider_pool::provider_pool_key_quota_hard_blocked(key, provider_type)
 }

+pub fn admin_pool_key_model_quota_exhausted(
+    key: &StoredProviderCatalogKey,
+    provider_type: &str,
+    provider_model_name: &str,
+) -> Option<bool> {
+    aether_provider_pool::provider_pool_key_model_quota_exhausted(
+        key,
+        provider_type,
+        provider_model_name,
+    )
+}
+
+pub fn admin_pool_key_model_quota_hard_blocked(
+    key: &StoredProviderCatalogKey,
+    provider_type: &str,
+    provider_model_name: &str,
+) -> bool {
+    aether_provider_pool::provider_pool_key_model_quota_hard_blocked(
+        key,
+        provider_type,
+        provider_model_name,
+    )
+}
+
 fn admin_pool_has_proxy(key: &StoredProviderCatalogKey) -> bool {
    match key.proxy.as_ref() {
        Some(Value::Object(values)) => !values.is_empty(),