From 9193ae52790f97c0855ac5220a601971cef66327 Mon Sep 17 00:00:00 2001 From: Junyan Qin Date: Wed, 9 Jul 2025 15:06:00 +0800 Subject: [PATCH] feat: create default autoupgrade strategy on tenant creating --- api/services/account_service.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/api/services/account_service.py b/api/services/account_service.py index c88e70e380..4c4510395e 100644 --- a/api/services/account_service.py +++ b/api/services/account_service.py @@ -29,6 +29,7 @@ from models.account import ( Tenant, TenantAccountJoin, TenantAccountRole, + TenantPluginAutoUpgradeStrategy, TenantStatus, ) from models.model import DifySetup @@ -828,6 +829,17 @@ class TenantService: db.session.add(tenant) db.session.commit() + plugin_upgrade_strategy = TenantPluginAutoUpgradeStrategy( + tenant_id=tenant.id, + strategy_setting=TenantPluginAutoUpgradeStrategy.StrategySetting.FIX_ONLY, + upgrade_time_of_day=0, + upgrade_mode=TenantPluginAutoUpgradeStrategy.UpgradeMode.EXCLUDE, + exclude_plugins=[], + include_plugins=[], + ) + db.session.add(plugin_upgrade_strategy) + db.session.commit() + tenant.encrypt_public_key = generate_key_pair(tenant.id) db.session.commit() return tenant