diff --git a/internal/logic/core/pulltaskinfologic.go b/internal/logic/core/pulltaskinfologic.go
index 4833e71c..2d62712d 100644
--- a/internal/logic/core/pulltaskinfologic.go
+++ b/internal/logic/core/pulltaskinfologic.go
@@ -75,7 +75,7 @@ func (l *PullTaskInfoLogic) PullTaskInfo(req *clientCore.PullTaskInfoReq) (*clie
 		}
 
 	case 1:
-		var aiModelList []models.Ai
+		var aiModelList []models.TaskAi
 		err := findModelList(req.AdapterId, l.svcCtx.DbEngin, &aiModelList)
 		if err != nil {
 			return nil, err