shuzhiren-comfyui/任务队列后端/outside/outPlatforms/runninghub.js

47 lines
1.1 KiB
JavaScript

export function getGenerateUrl() {
return process.env.RunningHub_URL
}
export function getGenerateHeader(){
return {
'Content-Type': 'application/json',
'Host': 'www.runninghub.cn'
};
}
export function getGenerateBody(task) {
const taskData = JSON.parse(task.payload)
const payload = {...taskData,apiKey:task.apikey,webhookUrl: process.env.CALLBACK_URL}
return JSON.stringify(payload);
}
export function getQueryUrl() {
return process.env.CALLBACK_URL
}
export function getTaskStatus() {
if(response.task_status === 'SUCCESS') return true;
}
export async function getSuccessTasks(response) {
const res = await response.json()
if(res.msg === 'success' && res.code === 0) {
return res.data.taskId
} else {
console.error('[runninghub] 返回错误:', res);
return {message:res, type: 2}
}
}
export async function getTaskResult(response) {
const res = await JSON.parse(response)
const files = []
if(res.msg === 'success' && res.code === 0) {
for(const file of res.data)
files.push(file.fileUrl)
return {files: files[0], type: 1}
} else {
return {message:res.msg, type: 2}
}
}