Initial commit: OmniAI Web Frontend

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-06-02 12:38:01 +08:00
commit bedee3ba8d
183 changed files with 94805 additions and 0 deletions
@@ -0,0 +1,110 @@
+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 : "请求失败",
+      };
+    }
+  },
+};