outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix model runtime quota

Browse Source
This commit is contained in:
jyong 2024-08-12 15:57:53 +08:00
parent a68df696ec
commit ccf4bd8555
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/provider_manager.py
View File

@ -338,7 +338,11 @@ class ProviderManager:
provider_name_to_provider_records_dict = defaultdict(list) provider_name_to_provider_records_dict = defaultdict(list)
for provider in providers: for provider in providers:
provider_name_to_provider_records_dict[provider.provider_name].append(provider) if provider.provider_name in provider_name_to_provider_records_dict:
if provider.quota_type == ProviderQuotaType.PAID.value or provider.quota_type == ProviderQuotaType.FREE.value:
provider_name_to_provider_records_dict[provider.provider_name].append(provider)
else:
provider_name_to_provider_records_dict[provider.provider_name].append(provider)
return provider_name_to_provider_records_dict return provider_name_to_provider_records_dict