111 lines
3.6 KiB
TypeScript
111 lines
3.6 KiB
TypeScript
import type { WebGenerationPreviewTask } from "../types";
|
|
import { aiGenerationClient } from "./aiGenerationClient";
|
|
import { resolveVideoRequestModel } from "../utils/resolveVideoModel";
|
|
import { ENTERPRISE_DEFAULT_VIDEO_MODEL } from "../utils/enterpriseVideoPolicy";
|
|
|
|
function formatPreviewTaskTimestamp(date = new Date()): string {
|
|
const year = date.getFullYear();
|
|
const month = String(date.getMonth() + 1).padStart(2, "0");
|
|
const day = String(date.getDate()).padStart(2, "0");
|
|
const hours = String(date.getHours()).padStart(2, "0");
|
|
const minutes = String(date.getMinutes()).padStart(2, "0");
|
|
return `${year}-${month}-${day} ${hours}:${minutes}`;
|
|
}
|
|
|
|
export interface CreatePreviewTaskInput {
|
|
title: string;
|
|
type: WebGenerationPreviewTask["type"];
|
|
prompt: string;
|
|
params?: {
|
|
existingTaskId?: string;
|
|
projectId?: string;
|
|
conversationId?: number;
|
|
model?: string;
|
|
ratio?: string;
|
|
quality?: string;
|
|
resolution?: string;
|
|
gridMode?: string;
|
|
duration?: number;
|
|
frameMode?: string;
|
|
referenceUrls?: string[];
|
|
audioUrl?: string;
|
|
muted?: boolean;
|
|
hasReferenceVideo?: boolean;
|
|
};
|
|
}
|
|
|
|
export const webGenerationGateway = {
|
|
async createPreviewTask(input: CreatePreviewTaskInput): Promise<WebGenerationPreviewTask> {
|
|
const { type, params } = input;
|
|
const prompt = input.prompt.trim();
|
|
const title = input.title.trim() || "未命名任务";
|
|
const createdAt = formatPreviewTaskTimestamp();
|
|
|
|
try {
|
|
let taskId: string;
|
|
|
|
if (params?.existingTaskId) {
|
|
taskId = params.existingTaskId;
|
|
} else if (type === "image") {
|
|
const result = await aiGenerationClient.createImageTask({
|
|
projectId: params?.projectId,
|
|
conversationId: params?.conversationId,
|
|
model: params?.model || "nano-banana-pro",
|
|
prompt,
|
|
ratio: params?.ratio || "16:9",
|
|
quality: params?.quality || "1K",
|
|
gridMode: params?.gridMode || "single",
|
|
referenceUrls: params?.referenceUrls,
|
|
});
|
|
taskId = result.taskId;
|
|
} else if (type === "video") {
|
|
const refs = params?.referenceUrls;
|
|
let model: string = params?.model || ENTERPRISE_DEFAULT_VIDEO_MODEL;
|
|
model = resolveVideoRequestModel({ model, referenceUrls: refs });
|
|
const result = await aiGenerationClient.createVideoTask({
|
|
projectId: params?.projectId,
|
|
conversationId: params?.conversationId,
|
|
model,
|
|
prompt,
|
|
ratio: params?.ratio || "16:9",
|
|
duration: params?.duration || 5,
|
|
quality: params?.quality || params?.resolution || "1080P",
|
|
resolution: params?.resolution || params?.quality || "1080P",
|
|
frameMode: params?.frameMode || "start-end",
|
|
referenceUrls: params?.referenceUrls,
|
|
audioUrl: params?.audioUrl,
|
|
muted: params?.muted ?? false,
|
|
hasReferenceVideo: params?.hasReferenceVideo ?? false,
|
|
});
|
|
taskId = result.taskId;
|
|
} else {
|
|
taskId = `web-task-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
|
|
}
|
|
|
|
return {
|
|
id: taskId,
|
|
title,
|
|
type,
|
|
status: "queued",
|
|
progress: 5,
|
|
prompt,
|
|
createdAt,
|
|
source: "server",
|
|
projectId: params?.projectId,
|
|
};
|
|
} catch (err) {
|
|
return {
|
|
id: `web-task-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`,
|
|
title,
|
|
type,
|
|
status: "failed",
|
|
progress: 0,
|
|
prompt,
|
|
createdAt,
|
|
source: "server",
|
|
errorMessage: err instanceof Error ? err.message : "请求失败",
|
|
};
|
|
}
|
|
},
|
|
};
|