feat(web): 支持文档页真实提交请求

apps/web/src/lib/run-task.ts

@@ -9,101 +9,97 @@ import {
   createVideoGenerationTask,
   getAPITask,
 } from '../api';
-import type { TaskForm } from '../types';
+import type { TaskForm, TaskSubmissionMode } from '../types';
 
 export interface RunTaskResponse {
   localOnly?: boolean;
   next?: Record<string, string>;
+  submissionMode: TaskSubmissionMode;
   task: GatewayTask;
 }
 
-export async function runTask(token: string, task: TaskForm): Promise<RunTaskResponse> {
+export interface RunTaskOptions {
+  requestBody?: Record<string, unknown>;
+  submissionMode?: TaskSubmissionMode;
+}
+
+const simulationParameterKeys = ['runMode', 'run_mode', 'simulation', 'testMode', 'test_mode'] as const;
+
+export function applyTaskSubmissionMode(
+  input: Record<string, unknown>,
+  submissionMode: TaskSubmissionMode,
+): Record<string, unknown> {
+  const body = { ...input };
+  for (const key of simulationParameterKeys) delete body[key];
+  if (submissionMode === 'simulation') {
+    body.runMode = 'simulation';
+    body.simulation = true;
+  }
+  return body;
+}
+
+export async function runTask(token: string, task: TaskForm, options: RunTaskOptions = {}): Promise<RunTaskResponse> {
+  const submissionMode = options.submissionMode ?? 'simulation';
+  const requestBody = task.kind === 'tasks.retrieve'
+    ? { taskId: task.taskId }
+    : applyTaskSubmissionMode(options.requestBody ?? defaultRequestBody(task), submissionMode);
+
   if (task.kind === 'chat.completions') {
-    const result = await createCompatibleChatCompletion(token, {
-      model: task.model,
-      runMode: 'simulation',
-      simulation: true,
-      stream: false,
-      messages: [{ role: 'user', content: task.prompt }],
-    });
-    return { localOnly: true, task: compatibleTask(task, result) };
+    const result = await createCompatibleChatCompletion(
+      token,
+      requestBody as Parameters<typeof createCompatibleChatCompletion>[1],
+    );
+    return { localOnly: true, submissionMode, task: compatibleTask(task, result, requestBody, submissionMode) };
   }
   if (task.kind === 'responses') {
-    const result = await createResponse(token, {
-      model: task.model,
-      input: task.prompt,
-      instructions: task.instructions,
-      previous_response_id: task.previousResponseId,
-      runMode: 'simulation',
-      simulation: true,
-      store: true,
-      stream: false,
-    });
-    return { localOnly: true, task: compatibleTask(task, result) };
+    const result = await createResponse(token, requestBody as Parameters<typeof createResponse>[1]);
+    return { localOnly: true, submissionMode, task: compatibleTask(task, result, requestBody, submissionMode) };
   }
   if (task.kind === 'embeddings') {
-    const result = await createEmbedding(token, {
-      model: task.model,
-      input: embeddingInput(task.prompt),
-      dimensions: task.dimensions,
-      runMode: 'simulation',
-      simulation: true,
-    });
-    return { localOnly: true, task: compatibleTask(task, result) };
+    const result = await createEmbedding(token, requestBody as Parameters<typeof createEmbedding>[1]);
+    return { localOnly: true, submissionMode, task: compatibleTask(task, result, requestBody, submissionMode) };
   }
   if (task.kind === 'reranks') {
-    const result = await createRerank(token, {
-      model: task.model,
-      query: task.prompt,
-      documents: rerankDocuments(task.documents),
-      top_n: task.topN,
-      runMode: 'simulation',
-      simulation: true,
-    });
-    return { localOnly: true, task: compatibleTask(task, result) };
+    const result = await createRerank(token, requestBody as Parameters<typeof createRerank>[1]);
+    return { localOnly: true, submissionMode, task: compatibleTask(task, result, requestBody, submissionMode) };
   }
   if (task.kind === 'images.generations') {
-    return createImageGenerationTask(token, {
-      model: task.model,
-      prompt: task.prompt,
-      quality: 'medium',
-      runMode: 'simulation',
-      simulation: true,
-      size: '1024x1024',
-    });
+    const response = await createImageGenerationTask(
+      token,
+      requestBody as Parameters<typeof createImageGenerationTask>[1],
+    );
+    return { ...response, submissionMode };
   }
   if (task.kind === 'images.edits') {
-    return createImageEditTask(token, {
-      model: task.model,
-      prompt: task.prompt,
-      image: task.image,
-      mask: task.mask,
-      runMode: 'simulation',
-      simulation: true,
-    });
+    const response = await createImageEditTask(token, requestBody as Parameters<typeof createImageEditTask>[1]);
+    return { ...response, submissionMode };
   }
   if (task.kind === 'videos.generations') {
-    return createVideoGenerationTask(token, {
-      model: task.model,
-      content: [{ type: 'text', text: task.prompt }],
-      aspect_ratio: task.aspectRatio ?? '16:9',
-      resolution: task.resolution ?? '720p',
-      duration: task.duration ?? 5,
-      audio: task.outputAudio ?? true,
-      runMode: 'simulation',
-      simulation: true,
-    });
+    const response = await createVideoGenerationTask(
+      token,
+      requestBody as unknown as Parameters<typeof createVideoGenerationTask>[1],
+    );
+    return { ...response, submissionMode };
   }
   if (task.kind === 'tasks.retrieve') {
     const taskId = task.taskId?.trim();
     if (!taskId) throw new Error('请输入要取回的 Task ID');
     const result = await getAPITask(token, taskId);
-    return { localOnly: true, task: compatibleTask(task, result as unknown as Record<string, unknown>) };
+    return {
+      localOnly: true,
+      submissionMode: 'production',
+      task: compatibleTask(task, result as unknown as Record<string, unknown>, requestBody, 'production'),
+    };
   }
   throw new Error(`Unsupported task kind: ${task.kind}`);
 }
 
-function compatibleTask(task: TaskForm, result: Record<string, unknown>): GatewayTask {
+function compatibleTask(
+  task: TaskForm,
+  result: Record<string, unknown>,
+  requestBody: Record<string, unknown>,
+  submissionMode: TaskSubmissionMode,
+): GatewayTask {
   const now = new Date().toISOString();
   return {
     id: `docs-${task.kind}-${Date.now()}`,
@@ -111,26 +107,31 @@ function compatibleTask(task: TaskForm, result: Record<string, unknown>): Gatewa
     createdAt: now,
     finishedAt: now,
     kind: task.kind,
-    model: task.model,
+    model: typeof requestBody.model === 'string' ? requestBody.model : task.model,
     modelType: modelTypeForKind(task.kind),
-    request: requestSnapshot(task),
+    request: requestBody,
     result,
-    runMode: 'simulation',
+    runMode: submissionMode,
     status: 'succeeded',
     updatedAt: now,
     userId: 'docs-runner',
   };
 }
 
-function requestSnapshot(task: TaskForm): Record<string, unknown> {
+function defaultRequestBody(task: TaskForm): Record<string, unknown> {
+  if (task.kind === 'chat.completions') {
+    return {
+      model: task.model,
+      messages: [{ role: 'user', content: task.prompt }],
+      stream: false,
+    };
+  }
   if (task.kind === 'responses') {
     return {
       model: task.model,
       input: task.prompt,
       instructions: task.instructions,
       previous_response_id: task.previousResponseId,
-      runMode: 'simulation',
-      simulation: true,
       store: true,
       stream: false,
     };
@@ -140,8 +141,6 @@ function requestSnapshot(task: TaskForm): Record<string, unknown> {
       model: task.model,
       input: embeddingInput(task.prompt),
       dimensions: task.dimensions,
-      runMode: 'simulation',
-      simulation: true,
     };
   }
   if (task.kind === 'reranks') {
@@ -150,8 +149,22 @@ function requestSnapshot(task: TaskForm): Record<string, unknown> {
       query: task.prompt,
       documents: rerankDocuments(task.documents),
       top_n: task.topN,
-      runMode: 'simulation',
-      simulation: true,
+    };
+  }
+  if (task.kind === 'images.generations') {
+    return {
+      model: task.model,
+      prompt: task.prompt,
+      quality: 'medium',
+      size: '1024x1024',
+    };
+  }
+  if (task.kind === 'images.edits') {
+    return {
+      model: task.model,
+      prompt: task.prompt,
+      image: task.image,
+      mask: task.mask,
     };
   }
   if (task.kind === 'videos.generations') {
@@ -162,18 +175,10 @@ function requestSnapshot(task: TaskForm): Record<string, unknown> {
       resolution: task.resolution ?? '720p',
       duration: task.duration ?? 5,
       audio: task.outputAudio ?? true,
-      runMode: 'simulation',
-      simulation: true,
     };
   }
   if (task.kind === 'tasks.retrieve') return { taskId: task.taskId };
-  return {
-    model: task.model,
-    messages: [{ role: 'user', content: task.prompt }],
-    runMode: 'simulation',
-    simulation: true,
-    stream: false,
-  };
+  return { model: task.model };
 }
 
 function embeddingInput(prompt: string) {