1

src/services/ai.ts

@@ -15,6 +15,26 @@ import {
   getDefaultFunctionIdsForContext,
   resolveFunctionOption,
 } from '../data/stateFunctions';
+import {
+  buildCustomWorldActorNarrativeProfileBatchJsonRepairPrompt,
+  buildCustomWorldActorNarrativeProfileBatchPrompt,
+  buildCustomWorldFrameworkJsonRepairPrompt,
+  buildCustomWorldFrameworkPrompt,
+  buildCustomWorldLandmarkNetworkBatchJsonRepairPrompt,
+  buildCustomWorldLandmarkNetworkBatchPrompt,
+  buildCustomWorldLandmarkSeedBatchJsonRepairPrompt,
+  buildCustomWorldLandmarkSeedBatchPrompt,
+  buildCustomWorldRoleBatchJsonRepairPrompt,
+  buildCustomWorldRoleBatchPrompt,
+  buildCustomWorldRoleOutlineBatchJsonRepairPrompt,
+  buildCustomWorldRoleOutlineBatchPrompt,
+  buildCustomWorldSceneImagePrompt,
+  buildCustomWorldStoryGraphJsonRepairPrompt,
+  buildCustomWorldStoryGraphPrompt,
+  buildCustomWorldThemePackJsonRepairPrompt,
+  buildCustomWorldThemePackPrompt,
+  DEFAULT_CUSTOM_WORLD_SCENE_IMAGE_NEGATIVE_PROMPT,
+} from '../prompts/customWorldPrompts';
 import {
   AIResponse,
   Character,
@@ -40,6 +60,8 @@ import {
   buildOfflineNpcRecruitDialogue as buildOfflineNpcRecruitDialogueFromFallback,
 } from './aiFallbacks';
 import type {
+  CustomWorldSceneImageRequest,
+  CustomWorldSceneImageResult,
   StoryGenerationContext,
   StoryRequestOptions,
   TextStreamOptions,
@@ -56,30 +78,12 @@ import {
   CharacterChatTargetStatus,
 } from './characterChatPrompt';
 import {
-  buildCustomWorldActorNarrativeProfileBatchJsonRepairPrompt,
-  buildCustomWorldActorNarrativeProfileBatchPrompt,
-  buildCustomWorldFrameworkJsonRepairPrompt,
-  buildCustomWorldFrameworkPrompt,
-  buildCustomWorldLandmarkNetworkBatchJsonRepairPrompt,
-  buildCustomWorldLandmarkNetworkBatchPrompt,
-  buildCustomWorldLandmarkSeedBatchJsonRepairPrompt,
-  buildCustomWorldLandmarkSeedBatchPrompt,
   buildCustomWorldRawProfileFromFramework,
-  buildCustomWorldRoleBatchJsonRepairPrompt,
-  buildCustomWorldRoleBatchPrompt,
-  buildCustomWorldRoleOutlineBatchJsonRepairPrompt,
-  buildCustomWorldRoleOutlineBatchPrompt,
-  buildCustomWorldSceneImagePrompt,
-  buildCustomWorldStoryGraphJsonRepairPrompt,
-  buildCustomWorldStoryGraphPrompt,
-  buildCustomWorldThemePackJsonRepairPrompt,
-  buildCustomWorldThemePackPrompt,
   type CustomWorldGenerationFramework,
   type CustomWorldGenerationLandmarkOutline,
   type CustomWorldGenerationRoleBatchStage,
   type CustomWorldGenerationRoleBatchType,
   type CustomWorldGenerationRoleOutline,
-  DEFAULT_CUSTOM_WORLD_SCENE_IMAGE_NEGATIVE_PROMPT,
   MIN_CUSTOM_WORLD_LANDMARK_COUNT,
   MIN_CUSTOM_WORLD_PLAYABLE_NPC_COUNT,
   MIN_CUSTOM_WORLD_STORY_NPC_COUNT,
@@ -137,6 +141,8 @@ export type {
   GenerateCustomWorldProfileOptions,
 } from '../../packages/shared/src/contracts/runtime';
 export type {
+  CustomWorldSceneImageRequest,
+  CustomWorldSceneImageResult,
   StoryGenerationContext,
   StoryRequestOptions,
   TextStreamOptions,
@@ -154,8 +160,7 @@ type MergeableCustomWorldRoleEntry = {
 };
 
 const CUSTOM_WORLD_SCENE_IMAGE_API_BASE_URL =
-  ENV.VITE_SCENE_IMAGE_PROXY_BASE_URL ||
-  '/api/custom-world/scene-image';
+  ENV.VITE_SCENE_IMAGE_PROXY_BASE_URL || '/api/custom-world/scene-image';
 const CUSTOM_WORLD_JSON_REPAIR_SYSTEM_PROMPT = `你是 JSON 修复器。
 你会收到一段本应为单个 JSON 对象的文本。
 你的唯一任务是把它修复成能被 JSON.parse 直接解析的单个 JSON 对象。
@@ -179,38 +184,6 @@ const CUSTOM_WORLD_SCENE_IMAGE_REQUEST_TIMEOUT_MS = (() => {
   return Number.isFinite(rawValue) && rawValue > 0 ? rawValue : 150000;
 })();
 
-export interface CustomWorldSceneImageRequest {
-  profile: Pick<
-    CustomWorldProfile,
-    | 'id'
-    | 'name'
-    | 'subtitle'
-    | 'summary'
-    | 'tone'
-    | 'playerGoal'
-    | 'settingText'
-  >;
-  landmark: Pick<
-    CustomWorldProfile['landmarks'][number],
-    'id' | 'name' | 'description' | 'dangerLevel'
-  >;
-  userPrompt?: string;
-  prompt?: string;
-  negativePrompt?: string;
-  size?: string;
-  referenceImageSrc?: string;
-}
-
-export interface CustomWorldSceneImageResult {
-  imageSrc: string;
-  assetId: string;
-  model: string;
-  size: string;
-  taskId: string;
-  prompt: string;
-  actualPrompt?: string;
-}
-
 const CUSTOM_WORLD_GENERATION_STAGE_DEFINITIONS = [
   {
     id: 'framework',
@@ -1992,23 +1965,26 @@ export async function generateCustomWorldSceneImage({
   );
 
   try {
-    const response = await fetchWithApiAuth(CUSTOM_WORLD_SCENE_IMAGE_API_BASE_URL, {
-      method: 'POST',
-      headers: { 'Content-Type': 'application/json' },
-      body: JSON.stringify({
-        profileId: profile.id,
-        worldName: profile.name,
-        landmarkId: landmark.id,
-        landmarkName: landmark.name,
-        prompt: resolvedPrompt,
-        negativePrompt: resolvedNegativePrompt,
-        size,
-        ...(referenceImageSrc?.trim()
-          ? { referenceImageSrc: referenceImageSrc.trim() }
-          : {}),
-      }),
-      signal: controller.signal,
-    });
+    const response = await fetchWithApiAuth(
+      CUSTOM_WORLD_SCENE_IMAGE_API_BASE_URL,
+      {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({
+          profileId: profile.id,
+          worldName: profile.name,
+          landmarkId: landmark.id,
+          landmarkName: landmark.name,
+          prompt: resolvedPrompt,
+          negativePrompt: resolvedNegativePrompt,
+          size,
+          ...(referenceImageSrc?.trim()
+            ? { referenceImageSrc: referenceImageSrc.trim() }
+            : {}),
+        }),
+        signal: controller.signal,
+      },
+    );
     const responseText = await response.text();
 
     if (!response.ok) {