refactor: 移除 createTask 透传层,taskPolling 直接读 data.body

This commit is contained in:
王佑琳 2026-06-09 18:09:04 +08:00
parent fe1ce00f66
commit 2cd3f8fad6
2 changed files with 4 additions and 13 deletions

View File

@ -1,5 +0,0 @@
// 所有平台 descriptor 的 buildTaskBody() 已直接返回扁平 modelParams
// 此文件仅做透传,后续可直接移除
export async function createTask(data) {
return data.body
}

View File

@ -1,9 +1,8 @@
import { ElNotification } from 'element-plus'
import { h } from 'vue'
import { useDisplayStore, useUserStore } from '@/stores'
import { createTask } from '@/utils/createTask'
import { userError } from '@/utils/tokenError'
import { requestCreateTask, requestTaskStatus } from '@/apis/display'
import { useDisplayStore, useUserStore } from '@/stores'
import { userError } from '@/utils/tokenError'
export function getChargeType(chargeType) {
switch (chargeType) {
@ -57,7 +56,6 @@ export function websocketSuccess() {
})
}
// 当前活跃的轮询定时器集合,用于页面卸载时清理
const activePollIntervals = new Set()
@ -90,8 +88,7 @@ export async function generate(data, generateData) {
}
try {
// 通过 createTask 获取 body 内容RunningHub workflow payload
const body = await createTask(data)
const body = data.body
// 构造请求体
const requestBody = {
@ -140,7 +137,7 @@ export async function generate(data, generateData) {
useDisplay.isSubGerenate = false
// 提取结果 URL
const urls = taskData.outputs?.map(img => img.url) || []
const urls = taskData.outputs?.map((img) => img.url) || []
if (urls.length > 0) {
useDisplay.updateItemToSuccess(taskId, urls)
websocketSuccess()
@ -165,7 +162,6 @@ export async function generate(data, generateData) {
// 5 秒后先做第一次轮询
setTimeout(pollTask, 5000)
} catch (error) {
console.error('创建任务失败:', error)
useDisplay.isSubGerenate = false