Integrate role asset studio into custom world agent flow

server-node/src/routes/runtimeRoutes.ts

@@ -1,6 +1,7 @@
 import { Router } from 'express';
 import { z } from 'zod';
 
+import type { ListCustomWorldWorksResponse } from '../../../packages/shared/src/contracts/customWorldAgent.js';
 import type {
   AnswerCustomWorldSessionQuestionRequest,
   CreateCustomWorldSessionRequest,
@@ -27,6 +28,8 @@ import {
   prepareEventStreamResponse,
   sendApiResponse,
 } from '../http.js';
+import { requireJwtAuth } from '../middleware/auth.js';
+import { routeMeta } from '../middleware/routeMeta.js';
 import {
   generateCharacterChatSuggestionsFromOrchestrator,
   generateCharacterChatSummaryFromOrchestrator,
@@ -34,8 +37,6 @@ import {
   streamNpcChatDialogueFromOrchestrator,
   streamNpcRecruitDialogueFromOrchestrator,
 } from '../modules/ai/chatOrchestrator.js';
-import { requireJwtAuth } from '../middleware/auth.js';
-import { routeMeta } from '../middleware/routeMeta.js';
 import {
   hydrateSavedSnapshot,
   normalizeSavedSnapshotPayload,
@@ -48,6 +49,9 @@ import {
   npcRecruitDialogueRequestSchema,
 } from '../services/chatService.js';
 import { generateCustomWorldProfile } from '../services/customWorldGenerationService.js';
+import {
+  listCustomWorldWorkSummaries,
+} from '../services/customWorldWorkSummaryService.js';
 import { generateQuestForNpcEncounter } from '../services/questService.js';
 import { generateRuntimeItemIntents } from '../services/runtimeItemService.js';
 import {
@@ -59,6 +63,7 @@ import {
   generateHighQualityNextStory,
   parseStoryRequest,
 } from '../services/storyService.js';
+import { createCustomWorldAgentRoutes } from './customWorldAgent.js';
 
 const jsonObjectSchema = z.record(z.string(), z.unknown());
 
@@ -109,6 +114,10 @@ export function createRuntimeRoutes(context: AppContext) {
   const requireAuth = requireJwtAuth(context.config, context.userRepository);
 
   router.use(requireAuth);
+  router.use(
+    '/runtime/custom-world/agent',
+    createCustomWorldAgentRoutes(context),
+  );
 
   router.post(
     '/llm/chat/completions',
@@ -198,6 +207,19 @@ export function createRuntimeRoutes(context: AppContext) {
     }),
   );
 
+  router.get(
+    '/runtime/custom-world/works',
+    routeMeta({ operation: 'runtime.customWorldWorks.list' }),
+    asyncHandler(async (request, response) => {
+      sendApiResponse<ListCustomWorldWorksResponse>(response, {
+        items: await listCustomWorldWorkSummaries(request.userId!, {
+          runtimeRepository: context.runtimeRepository,
+          customWorldAgentSessions: context.customWorldAgentSessions,
+        }),
+      });
+    }),
+  );
+
   router.get(
     '/runtime/custom-world-library',
     routeMeta({ operation: 'runtime.customWorldLibrary.list' }),
@@ -356,7 +378,7 @@ export function createRuntimeRoutes(context: AppContext) {
       ) as CreateCustomWorldSessionRequest;
       sendApiResponse(
         response,
-        context.customWorldSessions.create(
+        await context.customWorldSessions.create(
           request.userId!,
           payload.settingText,
           payload.creatorIntent,
@@ -370,7 +392,7 @@ export function createRuntimeRoutes(context: AppContext) {
     '/runtime/custom-world/sessions/:sessionId',
     routeMeta({ operation: 'runtime.customWorldSession.get' }),
     asyncHandler(async (request, response) => {
-      const session = context.customWorldSessions.get(
+      const session = await context.customWorldSessions.get(
         request.userId!,
         readParam(request.params.sessionId),
       );
@@ -388,7 +410,7 @@ export function createRuntimeRoutes(context: AppContext) {
       const payload = customWorldAnswerSchema.parse(
         request.body,
       ) as AnswerCustomWorldSessionQuestionRequest;
-      const session = context.customWorldSessions.answer(
+      const session = await context.customWorldSessions.answer(
         request.userId!,
         readParam(request.params.sessionId),
         payload.questionId,
@@ -405,7 +427,7 @@ export function createRuntimeRoutes(context: AppContext) {
     '/runtime/custom-world/sessions/:sessionId/generate/stream',
     routeMeta({ operation: 'runtime.customWorldSession.generateStream' }),
     asyncHandler(async (request, response) => {
-      const session = context.customWorldSessions.get(
+      const session = await context.customWorldSessions.get(
         request.userId!,
         readParam(request.params.sessionId),
       );
@@ -426,7 +448,7 @@ export function createRuntimeRoutes(context: AppContext) {
       };
 
       writeEvent('progress', { phase: 'preparing', progress: 10 });
-      context.customWorldSessions.updateStatus(
+      await context.customWorldSessions.updateStatus(
         request.userId!,
         readParam(request.params.sessionId),
         'generating',
@@ -443,7 +465,7 @@ export function createRuntimeRoutes(context: AppContext) {
             );
           },
         });
-        context.customWorldSessions.setResult(
+        await context.customWorldSessions.setResult(
           request.userId!,
           readParam(request.params.sessionId),
           profile,
@@ -456,7 +478,7 @@ export function createRuntimeRoutes(context: AppContext) {
           error instanceof Error
             ? error.message
             : 'custom world generation failed';
-        context.customWorldSessions.updateStatus(
+        await context.customWorldSessions.updateStatus(
           request.userId!,
           readParam(request.params.sessionId),
           'generation_error',