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

2 years ago · e122d677ad
parent 4c63cbf5b1
commit e122d677ad
1 changed files with 3 additions and 0 deletions
--- a/api/core/model_providers/model_provider_factory.py
+++ b/api/core/model_providers/model_provider_factory.py
@ -211,6 +211,9 @@ class ModelProviderFactory:
                                Provider.quota_type == ProviderQuotaType.TRIAL.value
                            ).first()

+                        if provider.quota_limit == 0:
+                            return None
+
                        return provider

            no_system_provider = True