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

e122d677 · takatost · GitHub · 4c63cbf5 · e122d677
Unverified Commit e122d677 authored Oct 21, 2023 by takatost Committed by GitHub Oct 21, 2023
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

model_provider_factory.py api/core/model_providers/model_provider_factory.py +3 -0

No files found.
--- 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