Kaynağa Gözat

Feat/optimize entity construct (#1935)

tags/0.4.4
takatost 1 yıl önce
ebeveyn
işleme
af7be9bdd7
No account linked to committer's email address

+ 1
- 1
api/core/helper/model_provider_cache.py Dosyayı Görüntüle

@@ -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:
"""

+ 8
- 1
api/core/provider_manager.py Dosyayı Görüntüle

@@ -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) \

Loading…
İptal
Kaydet