From 5e7a7cc0c7a03e9d56030875d738a1b3e35085b8 Mon Sep 17 00:00:00 2001
From: Junyan Qin <rockchinq@gmail.com>
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 2ba6f4345b..d9a9c4f280 100644
--- a/api/services/account_service.py
+++ b/api/services/account_service.py
@@ -28,6 +28,7 @@ from models.account import (
     Tenant,
     TenantAccountJoin,
     TenantAccountRole,
+    TenantPluginAutoUpgradeStrategy,
     TenantStatus,
 )
 from models.model import DifySetup
@@ -611,6 +612,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