瀏覽代碼

Feat/optimize entity construct (#1935)

tags/0.4.4
takatost 1 年之前
父節點
當前提交
af7be9bdd7
沒有連結到貢獻者的電子郵件帳戶。
共有 2 個檔案被更改,包括 9 行新增2 行删除
  1. 1
    1
      api/core/helper/model_provider_cache.py
  2. 8
    1
      api/core/provider_manager.py

+ 1
- 1
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:
"""

+ 8
- 1
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) \

Loading…
取消
儲存