diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go
index d4c162544..dde89dfab 100644
--- a/routers/repo/aisafety.go
+++ b/routers/repo/aisafety.go
@@ -978,6 +978,9 @@ func modelSafetyNewDataPrepare(ctx *context.Context) error {
 		ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.GPUCombatDataSetName
 		ctx.Data["CombatDataSetUUID"] = setting.ModelSafetyTest.GPUCombatDataSetUUID
 	} else {
+		ctx.Data["engine_id"] = ctx.QueryInt("engine_id")
+		ctx.Data["pool_id"] = ctx.Query("pool_id")
+
 		ctx.Data["type"] = models.TypeCloudBrainTwo
 		ctx.Data["compute_resource"] = models.NPUResource
 		ctx.Data["datasetType"] = models.TypeCloudBrainTwo