11

src/services/aiService.ts

@@ -30,6 +30,9 @@ import { parseApiErrorMessage } from '../../packages/shared/src/http';
 import type {
   AIResponse,
   Character,
+  CustomWorldLandmark,
+  CustomWorldNpc,
+  CustomWorldPlayableNpc,
   CharacterChatTurn,
   CustomWorldProfile,
   Encounter,
@@ -49,6 +52,7 @@ import { type CharacterChatTargetStatus } from './characterChatPrompt';
 import { parseLineListContent } from './llmParsers';
 
 const RUNTIME_API_BASE = '/api/runtime';
+const CUSTOM_WORLD_API_BASE = '/api';
 
 type LegacyAiModule = typeof import('./ai');
 
@@ -490,6 +494,74 @@ export async function generateCustomWorldSceneImage(
   return aiClient.generateCustomWorldSceneImage(...args);
 }
 
+export async function generateCustomWorldSceneNpc(payload: {
+  profile: CustomWorldProfile;
+  landmarkId: string;
+}) {
+  const response = await requestPostJson<{ npc: CustomWorldNpc }>(
+    `${CUSTOM_WORLD_API_BASE}/custom-world/scene-npc`,
+    payload,
+    '生成场景 NPC 失败',
+  );
+
+  return response.npc;
+}
+
+async function requestCustomWorldEntity<T>(
+  payload: {
+    profile: CustomWorldProfile;
+    kind: 'playable' | 'story' | 'landmark';
+  },
+  fallbackMessage: string,
+) {
+  return requestPostJson<{
+    kind: 'playable' | 'story' | 'landmark';
+    entity: T;
+  }>(`${CUSTOM_WORLD_API_BASE}/custom-world/entity`, payload, fallbackMessage);
+}
+
+export async function generateCustomWorldPlayableNpc(payload: {
+  profile: CustomWorldProfile;
+}) {
+  const response = await requestCustomWorldEntity<CustomWorldPlayableNpc>(
+    {
+      ...payload,
+      kind: 'playable',
+    },
+    '生成可扮演角色失败',
+  );
+
+  return response.entity;
+}
+
+export async function generateCustomWorldStoryNpc(payload: {
+  profile: CustomWorldProfile;
+}) {
+  const response = await requestCustomWorldEntity<CustomWorldNpc>(
+    {
+      ...payload,
+      kind: 'story',
+    },
+    '生成场景角色失败',
+  );
+
+  return response.entity;
+}
+
+export async function generateCustomWorldLandmark(payload: {
+  profile: CustomWorldProfile;
+}) {
+  const response = await requestCustomWorldEntity<CustomWorldLandmark>(
+    {
+      ...payload,
+      kind: 'landmark',
+    },
+    '生成场景失败',
+  );
+
+  return response.entity;
+}
+
 export async function createCustomWorldSession(payload: {
   settingText: string;
   creatorIntent?: Record<string, unknown> | null;