From 3bfb26d571198dc8a93d1cb20d9110d02bab58ab Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Mon, 12 Aug 2024 17:09:09 +0800 Subject: [PATCH] fix model runtime quota --- api/core/provider_manager.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/api/core/provider_manager.py b/api/core/provider_manager.py index a4f77dd020..9e52dfe15e 100644 --- a/api/core/provider_manager.py +++ b/api/core/provider_manager.py @@ -338,11 +338,7 @@ class ProviderManager: provider_name_to_provider_records_dict = defaultdict(list) for provider in providers: - 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) + provider_name_to_provider_records_dict[provider.provider_name].append(provider) return provider_name_to_provider_records_dict @@ -667,7 +663,7 @@ class ProviderManager: # Convert provider_records to dict quota_type_to_provider_records_dict = {} 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 quota_type_to_provider_records_dict[ProviderQuotaType.value_of(provider_record.quota_type)] \