Unverified Commit af7be9bd authored by takatost's avatar takatost Committed by GitHub

Feat/optimize entity construct (#1935)

parent 2cfd5568
...@@ -40,7 +40,7 @@ class ProviderCredentialsCache: ...@@ -40,7 +40,7 @@ class ProviderCredentialsCache:
:param credentials: provider credentials :param credentials: provider credentials
:return: :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: def delete(self) -> None:
""" """
......
...@@ -229,11 +229,18 @@ class ProviderManager: ...@@ -229,11 +229,18 @@ class ProviderManager:
return None return None
provider_instance = model_provider_factory.get_provider_instance(default_model.provider_name) provider_instance = model_provider_factory.get_provider_instance(default_model.provider_name)
provider_schema = provider_instance.get_provider_schema()
return DefaultModelEntity( return DefaultModelEntity(
model=default_model.model_name, model=default_model.model_name,
model_type=model_type, 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) \ def update_default_model_record(self, tenant_id: str, model_type: ModelType, provider: str, model: str) \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment