refactor(util): 重构映射工具函数并优化异步任务轮询逻辑

service/model/model_service.go

@@ -91,12 +91,14 @@ func (s *modelService) Get(ctx context.Context, req *dto.GetModelReq) (*dto.GetM
 	if g.IsEmpty(req.ID) {
 		req.Creator = user.UserName
 	}
-	modelReq := new(entity.AsynchModel)
-	err = gconv.Struct(req, modelReq)
-	if err != nil {
-		return nil, err
-	}
-	model, err := dao.Model.Get(ctx, modelReq)
+	model, err := dao.Model.Get(ctx, &entity.AsynchModel{
+		SQLBaseDO: beans.SQLBaseDO{
+			Id:      req.ID,
+			Creator: user.UserName,
+		},
+		ModelName:   req.ModelName,
+		IsChatModel: req.IsChatModel,
+	})
 	if err != nil {
 		return nil, err
 	}

service/task/task_service.go

@@ -179,7 +179,7 @@ func (s *taskService) QueryPendingTasks(ctx context.Context, req *dto.QueryPendi
 		if err != nil || model == nil || model.QueryConfig == nil {
 			continue
 		}
-		result, err := util.PullTaskResult(ctx, t.TaskID, model.QueryConfig)
+		result, err := util.PullTaskResult(ctx, nil, model.QueryConfig, model.HeadMsg)
 		if err != nil {
 			g.Log().Warningf(ctx, "[轮询] 查询失败 taskID=%s err=%v", t.TaskID, err)
 			continue

service/task/worker.go

@@ -122,9 +122,7 @@ func (w *asyncWorker) handleOne(ctx context.Context, task *entity.AsynchTask, mo
 			w.failTask(ctx, task, err.Error())
 			return
 		}
-		// 拿到 task_id，启动轮询
-		taskID := gjson.New(body).Get(model.ResponseBody).String()
-		body, err = util.PullTaskResult(ctx, taskID, model.QueryConfig)
+		body, err = util.PullTaskResult(ctx, body, model.QueryConfig, model.HeadMsg)
 		if err != nil {
 			w.failTask(ctx, task, err.Error())
 			return