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 17:09:09 +08:00
parent ccf4bd8555
commit 3bfb26d571
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/core/provider_manager.py
View File

@ -338,11 +338,7 @@ 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:
if provider.provider_name in provider_name_to_provider_records_dict: provider_name_to_provider_records_dict[provider.provider_name].append(provider)
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
@ -667,7 +663,7 @@ class ProviderManager:
# Convert provider_records to dict # Convert provider_records to dict
quota_type_to_provider_records_dict = {} quota_type_to_provider_records_dict = {}
for provider_record in provider_records: for provider_record in provider_records:
if provider_record.provider_type != ProviderType.SYSTEM.value: if provider_record.provider_type != ProviderType.SYSTEM.value and provider_record.quota_type != ProviderQuotaType.PAID.value:
continue continue
quota_type_to_provider_records_dict[ProviderQuotaType.value_of(provider_record.quota_type)] \ quota_type_to_provider_records_dict[ProviderQuotaType.value_of(provider_record.quota_type)] \