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

2023-10-22 03:02:38 +08:00 · 2023-10-22 03:02:38 +08:00 · e122d677ad
commit e122d677ad
parent 4c63cbf5b1
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