11

src/services/ai.ts

@@ -1,3 +1,9 @@
+import type {
+  CustomWorldGenerationStep,
+  GenerateCustomWorldProfileInput,
+  GenerateCustomWorldProfileOptions,
+} from '../../packages/shared/src/contracts/runtime';
+import { unwrapApiResponse } from '../../packages/shared/src/http';
 import { createSceneHostileNpcsFromEncounters } from '../data/hostileNpcs';
 import {
   buildEncounterFromSceneNpc,
@@ -26,12 +32,6 @@ import {
   WorldStoryGraph,
   WorldType,
 } from '../types';
-import type {
-  CustomWorldGenerationStep,
-  CustomWorldGenerationProgress,
-  GenerateCustomWorldProfileInput,
-  GenerateCustomWorldProfileOptions,
-} from '../../packages/shared/src/contracts/runtime';
 import {
   buildOfflineCharacterPanelChatReply as buildOfflineCharacterPanelChatReplyFromFallback,
   buildOfflineCharacterPanelChatSuggestions as buildOfflineCharacterPanelChatSuggestionsFromFallback,
@@ -136,7 +136,6 @@ export type {
   GenerateCustomWorldProfileInput,
   GenerateCustomWorldProfileOptions,
 } from '../../packages/shared/src/contracts/runtime';
-
 export type {
   StoryGenerationContext,
   StoryRequestOptions,
@@ -2018,8 +2017,8 @@ export async function generateCustomWorldSceneImage({
       );
     }
 
-    const data = JSON.parse(
-      responseText,
+    const data = unwrapApiResponse(
+      JSON.parse(responseText) as Partial<CustomWorldSceneImageResult>,
     ) as Partial<CustomWorldSceneImageResult>;
     if (
       !data.imageSrc ||