From af7be9bdd72b25ba8733f216756cfff399a910dc Mon Sep 17 00:00:00 2001 From: takatost Date: Fri, 5 Jan 2024 09:43:41 +0800 Subject: [PATCH] Feat/optimize entity construct (#1935) --- api/core/helper/model_provider_cache.py | 2 +- api/core/provider_manager.py | 9 ++++++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/api/core/helper/model_provider_cache.py b/api/core/helper/model_provider_cache.py index 64908f9501..81e589f65b 100644 --- a/api/core/helper/model_provider_cache.py +++ b/api/core/helper/model_provider_cache.py @@ -40,7 +40,7 @@ class ProviderCredentialsCache: :param credentials: provider credentials :return: """ - redis_client.setex(self.cache_key, 3600, json.dumps(credentials)) + redis_client.setex(self.cache_key, 86400, json.dumps(credentials)) def delete(self) -> None: """ diff --git a/api/core/provider_manager.py b/api/core/provider_manager.py index aadc5344e0..8edf1df8d6 100644 --- a/api/core/provider_manager.py +++ b/api/core/provider_manager.py @@ -229,11 +229,18 @@ class ProviderManager: return None provider_instance = model_provider_factory.get_provider_instance(default_model.provider_name) + provider_schema = provider_instance.get_provider_schema() return DefaultModelEntity( model=default_model.model_name, model_type=model_type, - provider=DefaultModelProviderEntity(**provider_instance.get_provider_schema().to_simple_provider().dict()) + provider=DefaultModelProviderEntity( + provider=provider_schema.provider, + label=provider_schema.label, + icon_small=provider_schema.icon_small, + icon_large=provider_schema.icon_large, + supported_model_types=provider_schema.supported_model_types + ) ) def update_default_model_record(self, tenant_id: str, model_type: ModelType, provider: str, model: str) \