Integrate role asset studio into custom world agent flow

server-node/src/server.ts

@@ -5,16 +5,18 @@ import { type AppConfig, loadConfig } from './config.js';
 import type { AppContext } from './context.js';
 import { createDatabase } from './db.js';
 import { createLogger } from './logging.js';
-import { AuthIdentityRepository } from './repositories/authIdentityRepository.js';
 import { AuthAuditLogRepository } from './repositories/authAuditLogRepository.js';
+import { AuthIdentityRepository } from './repositories/authIdentityRepository.js';
 import { AuthRiskBlockRepository } from './repositories/authRiskBlockRepository.js';
 import { RuntimeRepository } from './repositories/runtimeRepository.js';
 import { SmsAuthEventRepository } from './repositories/smsAuthEventRepository.js';
 import { UserRepository } from './repositories/userRepository.js';
 import { UserSessionRepository } from './repositories/userSessionRepository.js';
+import { CaptchaChallengeStore } from './services/captchaChallengeStore.js';
+import { CustomWorldAgentOrchestrator } from './services/customWorldAgentOrchestrator.js';
+import { CustomWorldAgentSessionStore } from './services/customWorldAgentSessionStore.js';
 import { CustomWorldSessionStore } from './services/customWorldSessionStore.js';
 import { UpstreamLlmClient } from './services/llmClient.js';
-import { CaptchaChallengeStore } from './services/captchaChallengeStore.js';
 import { createSmsVerificationService } from './services/smsVerificationService.js';
 import { createWechatAuthService } from './services/wechatAuthService.js';
 import { WechatAuthStateStore } from './services/wechatAuthStateStore.js';
@@ -77,6 +79,10 @@ function describeDatabase(databaseUrl: string) {
 export async function createAppContext(config: AppConfig = loadConfig()) {
   const logger = createLogger(config);
   const db = await createDatabase(config);
+  const runtimeRepository = new RuntimeRepository(db);
+  const customWorldAgentSessions = new CustomWorldAgentSessionStore(
+    runtimeRepository,
+  );
   const context: AppContext = {
     config,
     logger,
@@ -87,9 +93,16 @@ export async function createAppContext(config: AppConfig = loadConfig()) {
     authRiskBlockRepository: new AuthRiskBlockRepository(db),
     smsAuthEventRepository: new SmsAuthEventRepository(db),
     userSessionRepository: new UserSessionRepository(db),
-    runtimeRepository: new RuntimeRepository(db),
+    runtimeRepository,
     llmClient: new UpstreamLlmClient(config, logger),
-    customWorldSessions: new CustomWorldSessionStore(),
+    customWorldSessions: new CustomWorldSessionStore(runtimeRepository),
+    customWorldAgentSessions,
+    customWorldAgentOrchestrator: new CustomWorldAgentOrchestrator(
+      customWorldAgentSessions,
+      config.llm.apiKey.trim()
+        ? new UpstreamLlmClient(config, logger)
+        : null,
+    ),
     smsVerificationService: createSmsVerificationService(config, logger),
     wechatAuthService: createWechatAuthService(config, logger),
     wechatAuthStates: new WechatAuthStateStore(),