fix: return wrong when init 0 quota in trial provider (#1394)

langgenius · Oct 21, 2023 · e122d67 · e122d67
1 parent 4c63cbf
commit e122d67
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/api/core/model_providers/model_provider_factory.py b/api/core/model_providers/model_provider_factory.py
@@ -211,6 +211,9 @@ def _get_preferred_provider(cls, tenant_id: str, model_provider_name: str):
                                 Provider.quota_type == ProviderQuotaType.TRIAL.value
                             ).first()
 
+                        if provider.quota_limit == 0:
+                            return None
+
                         return provider
 
             no_system_provider = True