From ccf4bd8555e2cb9571c001698d0e5c8aed43a930 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Mon, 12 Aug 2024 15:57:53 +0800
Subject: [PATCH] fix model runtime quota

---
 api/core/provider_manager.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/api/core/provider_manager.py b/api/core/provider_manager.py
index c0b3746e18..a4f77dd020 100644
--- a/api/core/provider_manager.py
+++ b/api/core/provider_manager.py
@@ -338,7 +338,11 @@ class ProviderManager:
 
         provider_name_to_provider_records_dict = defaultdict(list)
         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