awsl-project · ymkiux · Mar 17, 2026 · Mar 17, 2026 · Mar 17, 2026 · coderabbitai
diff --git a/internal/domain/model.go b/internal/domain/model.go
@@ -238,6 +238,9 @@ type Provider struct {
 	// 如果配置了，在 Route 匹配时会检查前置映射后的模型是否在支持列表中
 	// 空数组表示支持所有模型
 	SupportModels []string `json:"supportModels,omitempty"`
+
+	// 为 true 时，该 provider 不参与导出/备份
+	ExcludeFromExport bool `json:"excludeFromExport,omitempty"`
 }
 
 type Project struct {

diff --git a/internal/handler/admin_import_export_test.go b/internal/handler/admin_import_export_test.go
@@ -123,9 +123,10 @@ func TestAdminHandler_ProvidersImport_WithTrailingSlash(t *testing.T) {
 func TestAdminHandler_ProvidersExport_WithTrailingSlash(t *testing.T) {
 	providerRepo := &adminTestProviderRepo{
 		providers: []*domain.Provider{{
-			ID:   1,
-			Name: "exported-provider",
-			Type: "custom",
+			ID:                1,
+			Name:              "exported-provider",
+			Type:              "custom",
+			ExcludeFromExport: true,
 		}},
 	}
 	h := newAdminHandlerForProviderImportExportTests(providerRepo)
@@ -151,7 +152,7 @@ func TestAdminHandler_ProvidersExport_WithTrailingSlash(t *testing.T) {
 		t.Fatalf("decode response: %v", err)
 	}
 
-	if len(providers) != 1 || providers[0].Name != "exported-provider" {
-		t.Fatalf("providers = %+v, want one exported-provider", providers)
+	if len(providers) != 0 {
+		t.Fatalf("providers = %+v, want excluded providers to be omitted", providers)
 	}
 }
diff --git a/internal/repository/sqlite/migrations.go b/internal/repository/sqlite/migrations.go
@@ -207,6 +207,15 @@ var migrations = []Migration{
 		Down: func(db *gorm.DB) error {
 			return nil
 		},
+	}, {
+		Version:     7,
+		Description: "Backfill providers.exclude_from_export defaults to 0",
+		Up: func(db *gorm.DB) error {
+			return db.Exec("UPDATE providers SET exclude_from_export = 0 WHERE exclude_from_export IS NULL").Error
+		},
+		Down: func(db *gorm.DB) error {
+			return nil
+		},
 	},
 }
 

diff --git a/internal/repository/sqlite/models.go b/internal/repository/sqlite/models.go
@@ -95,6 +95,7 @@ type Provider struct {
 	Config               LongText
 	SupportedClientTypes LongText
 	SupportModels        LongText
+	ExcludeFromExport    int `gorm:"default:0"`
 }
 
 func (Provider) TableName() string { return "providers" }

diff --git a/internal/repository/sqlite/provider.go b/internal/repository/sqlite/provider.go
@@ -87,6 +87,7 @@ func (r *ProviderRepository) toModel(p *domain.Provider) *Provider {
 		Config:               LongText(toJSON(p.Config)),
 		SupportedClientTypes: LongText(toJSON(p.SupportedClientTypes)),
 		SupportModels:        LongText(toJSON(p.SupportModels)),
+		ExcludeFromExport:    boolToInt(p.ExcludeFromExport),
 	}
 }
 
@@ -104,5 +105,6 @@ func (r *ProviderRepository) toDomain(m *Provider) *domain.Provider {
 		Config:               fromJSON[*domain.ProviderConfig](string(m.Config)),
 		SupportedClientTypes: fromJSON[[]domain.ClientType](string(m.SupportedClientTypes)),
 		SupportModels:        fromJSON[[]string](string(m.SupportModels)),
+		ExcludeFromExport:    m.ExcludeFromExport != 0,
 	}
 }
diff --git a/internal/service/admin.go b/internal/service/admin.go
@@ -166,8 +166,14 @@ func (s *AdminService) ExportProviders(tenantID uint64) ([]*domain.Provider, err
 	if err != nil {
 		return nil, err
 	}
-	// Return as-is, the handler will handle JSON serialization
-	return providers, nil
+	filtered := make([]*domain.Provider, 0, len(providers))
+	for _, provider := range providers {
+		if provider.ExcludeFromExport {
+			continue
+		}
+		filtered = append(filtered, provider)
+	}
+	return filtered, nil
 }
 
 // ImportProviders imports providers from exported data

diff --git a/internal/service/backup.go b/internal/service/backup.go
@@ -108,6 +108,9 @@ func (s *BackupService) Export(tenantID uint64) (*domain.BackupFile, error) {
 		return nil, fmt.Errorf("failed to export providers: %w", err)
 	}
 	for _, p := range providers {
+		if p.ExcludeFromExport {
+			continue
+		}
 		providerIDToName[p.ID] = p.Name
 		backup.Data.Providers = append(backup.Data.Providers, domain.BackupProvider{
 			Name:                 p.Name,

diff --git a/tests/e2e/backup_test.go b/tests/e2e/backup_test.go
@@ -201,3 +201,35 @@ func TestBackupImport_SkipStrategy(t *testing.T) {
 		t.Fatalf("Expected import with skip strategy to succeed, got %v", result["success"])
 	}
 }
+
+func TestBackupExport_ExcludedProviderIsOmitted(t *testing.T) {
+	env := NewTestEnv(t)
+
+	provider := map[string]any{
+		"name":              "excluded-provider",
+		"type":              "custom",
+		"excludeFromExport": true,
+		"config": map[string]any{
+			"custom": map[string]any{
+				"baseURL": "https://api.example.com",
+				"apiKey":  "sk-hidden-key",
+			},
+		},
+		"supportedClientTypes": []string{"claude"},
+	}
+	resp := env.AdminPost("/api/admin/providers", provider)
+	AssertStatus(t, resp, http.StatusCreated)
+	resp.Body.Close()
+
+	resp = env.AdminGet("/api/admin/backup/export")
+	AssertStatus(t, resp, http.StatusOK)
+
+	var backup map[string]any
+	DecodeJSON(t, resp, &backup)
+
+	data := backup["data"].(map[string]any)
+	providers, _ := data["providers"].([]any)
+	if len(providers) != 0 {
+		t.Fatalf("expected excluded providers to be omitted from backup export, got %d entries", len(providers))
+	}
+}
diff --git a/web/src/lib/transport/types.ts b/web/src/lib/transport/types.ts
@@ -100,6 +100,7 @@ export interface Provider {
   config: ProviderConfig | null;
   supportedClientTypes: ClientType[];
   supportModels?: string[]; // 支持的模型列表（通配符模式），空数组表示支持所有模型
+  excludeFromExport?: boolean; // 为 true 时不参与导出/备份
 }
 
 // supportedClientTypes 可选，后端会根据 provider type 自动设置

diff --git a/web/src/locales/en.json b/web/src/locales/en.json
@@ -1218,7 +1218,9 @@
     "endpointOverride": "Endpoint Override",
     "errorCooldownTitle": "3. Error Cooldown",
     "disableErrorCooldown": "Disable Error Cooldown",
-    "disableErrorCooldownDesc": "When enabled, errors won't trigger automatic cooldown. Manual freezes and explicit cooldown times still apply."
+    "disableErrorCooldownDesc": "When enabled, errors won't trigger automatic cooldown. Manual freezes and explicit cooldown times still apply.",
+    "excludeFromExport": "Exclude from export and backup",
+    "excludeFromExportDesc": "When enabled, this provider stays local to this system and will not be included in exported files or backup packages."
   },
   "cooldown": {
     "title": "Cooldown Protection",

diff --git a/web/src/locales/zh.json b/web/src/locales/zh.json
@@ -1218,7 +1218,9 @@
     "endpointOverride": "端点覆盖",
     "errorCooldownTitle": "3. 错误冷冻",
     "disableErrorCooldown": "禁用错误冷冻",
-    "disableErrorCooldownDesc": "开启后，错误将不再触发自动冷冻；手动冷冻与上游明确冷冻时间仍会生效。"
+    "disableErrorCooldownDesc": "开启后，错误将不再触发自动冷冻；手动冷冻与上游明确冷冻时间仍会生效。",
+    "excludeFromExport": "从导出与备份中排除",
+    "excludeFromExportDesc": "开启后，该提供商仅保留在当前系统中，不会被写入导出文件或系统备份。"
   },
   "cooldown": {
     "title": "冷却保护中",

diff --git a/web/src/pages/providers/components/custom-config-step.tsx b/web/src/pages/providers/components/custom-config-step.tsx
@@ -80,6 +80,7 @@ export function CustomConfigStep() {
           },
         },
         supportedClientTypes,
+        excludeFromExport: !!formData.excludeFromExport,
       };
 
       const provider = await createProvider.mutateAsync(data);
@@ -224,19 +225,36 @@ export function CustomConfigStep() {
             <h3 className="text-lg font-semibold text-text-primary border-b border-border pb-2">
               {t('provider.errorCooldownTitle')}
             </h3>
-            <div className="flex items-center justify-between p-4 bg-card border border-border rounded-xl">
-              <div className="pr-4">
-                <div className="text-sm font-medium text-foreground">
-                  {t('provider.disableErrorCooldown')}
+            <div className="space-y-4">
+              <div className="flex items-center justify-between p-4 bg-card border border-border rounded-xl">
+                <div className="pr-4">
+                  <div className="text-sm font-medium text-foreground">
+                    {t('provider.disableErrorCooldown')}
+                  </div>
+                  <p className="text-xs text-muted-foreground mt-1">
+                    {t('provider.disableErrorCooldownDesc')}
+                  </p>
                 </div>
-                <p className="text-xs text-muted-foreground mt-1">
-                  {t('provider.disableErrorCooldownDesc')}
-                </p>
+                <Switch
+                  checked={!!formData.disableErrorCooldown}
+                  onCheckedChange={(checked) => updateFormData({ disableErrorCooldown: checked })}
+                />
+              </div>
+
+              <div className="flex items-center justify-between p-4 bg-card border border-border rounded-xl">
+                <div className="pr-4">
+                  <div className="text-sm font-medium text-foreground">
+                    {t('provider.excludeFromExport')}
+                  </div>
+                  <p className="text-xs text-muted-foreground mt-1">
+                    {t('provider.excludeFromExportDesc')}
+                  </p>
+                </div>
+                <Switch
+                  checked={!!formData.excludeFromExport}
+                  onCheckedChange={(checked) => updateFormData({ excludeFromExport: checked })}
+                />
               </div>
-              <Switch
-                checked={!!formData.disableErrorCooldown}
-                onCheckedChange={(checked) => updateFormData({ disableErrorCooldown: checked })}
-              />
             </div>
           </div>
 

diff --git a/web/src/pages/providers/components/provider-edit-flow.tsx b/web/src/pages/providers/components/provider-edit-flow.tsx
@@ -385,6 +385,7 @@ export function ProviderEditFlow({ provider, onClose }: ProviderEditFlowProps) {
         },
         supportedClientTypes,
         supportModels: formData.supportModels.length > 0 ? formData.supportModels : undefined,
+        excludeFromExport: !!provider.excludeFromExport,
       };
 
       await updateProvider.mutateAsync({ id: Number(provider.id), data });
@@ -446,6 +447,7 @@ export function ProviderEditFlow({ provider, onClose }: ProviderEditFlowProps) {
         },
         supportedClientTypes,
         supportModels: formData.supportModels.length > 0 ? formData.supportModels : undefined,
+        excludeFromExport: !!provider.excludeFromExport,
       };
 
       const newProvider = await createProvider.mutateAsync(data);

diff --git a/web/src/pages/providers/context/provider-form-context.tsx b/web/src/pages/providers/context/provider-form-context.tsx
@@ -41,6 +41,7 @@ const initialFormData: ProviderFormData = {
   cloakStrictMode: false,
   cloakSensitiveWords: '',
   disableErrorCooldown: false,
+  excludeFromExport: false,
 };
 
 export function ProviderFormProvider({ children }: { children: ReactNode }) {

diff --git a/web/src/pages/providers/types.ts b/web/src/pages/providers/types.ts
@@ -234,6 +234,7 @@ export type ProviderFormData = {
   modelMappings?: TemplateModelMapping[]; // 模型映射
   logo?: string; // Logo URL
   disableErrorCooldown?: boolean;
+  excludeFromExport?: boolean;
 };
 
 // Create step type