briyah 1.1.4 → 1.1.6

package/dist-sdk/server/src/ai/LLM/together.service.d.ts

@@ -12,7 +12,11 @@ export declare class TogetherAiService extends BaseAiService {
     textPrompt(agent: Agent, prompt: string, jsonSchema?: any, saveResponse?: boolean, promptInstructions?: string, _cacheConfig?: any, maxOutputChars?: number): Promise<any>;
     private computeMessageCost;
     private computeImageCost;
-    generateImage(agent: Agent, prompt: string, width: number, height: number, quality: string, referenceImageArtifactIds?: string[]): Promise<{
+    generateImage(agent: Agent, prompt: string, imageProperties?: Record<string, any>): Promise<{
+        artifactId?: string;
+        error?: any;
+    }>;
+    editImage(agent: Agent, prompt: string, imageProperties?: Record<string, any>, referenceImageArtifactIds?: string[]): Promise<{
         artifactId?: string;
         error?: any;
     }>;

package/dist-sdk/server/src/ai/LLM/together.service.js

@@ -182,10 +182,18 @@ let TogetherAiService = class TogetherAiService extends base_ai_service_1.BaseAi
             agent.balanceService.decrementBalance(cost, markup);
         }
     }
-    computeImageCost(agent, usage) {
+    computeImageCost(agent, usage, centsPerImage) {
         const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (!modelInfo) {
-            logger_1.logger.error(`No price info found for model ${agent.modelName}`);
+            if (centsPerImage != null) {
+                const cost = centsPerImage / 100;
+                agent.totalCost += cost;
+                logger_1.logger.log(`Image cost for ${agent.agentName}: $${cost.toFixed(4)}`);
+                if (agent.balanceService && cost > 0)
+                    agent.balanceService.decrementBalance(cost, 0);
+                return;
+            }
+            logger_1.logger.warn(`No price info found for model ${agent.modelName}. Skipping cost tracking.`);
             return;
         }
         const inputImageTokens = usage.input_tokens_details?.image_tokens || 0;
@@ -235,7 +243,12 @@ let TogetherAiService = class TogetherAiService extends base_ai_service_1.BaseAi
             agent.balanceService.decrementBalance(cost, markup);
         }
     }
-    async generateImage(agent, prompt, width, height, quality, referenceImageArtifactIds) {
+    async generateImage(agent, prompt, imageProperties) {
+        const { centsPerImage, ...apiProperties } = imageProperties ?? {};
+        const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
+        if (!modelInfo && centsPerImage == null) {
+            return { error: `No pricing configured for image model "${agent.modelName}". Add a modelPrices entry or set centsPerImage in image_models.json.` };
+        }
         if (agent.balanceService && !agent.disableBalanceCheck) {
             const hasSufficientBalance = agent.balanceService.hasSufficientBalance();
             if (!hasSufficientBalance) {
@@ -243,34 +256,13 @@ let TogetherAiService = class TogetherAiService extends base_ai_service_1.BaseAi
             }
         }
         try {
-            let imageResponse;
-            if (referenceImageArtifactIds && referenceImageArtifactIds.length > 0) {
-                const referenceImages = await this.loadReferenceImages(agent, referenceImageArtifactIds);
-                imageResponse = await this.togetherClient.images.generate({
-                    reference_image: referenceImages.length === 1 ? referenceImages[0] : referenceImages,
-                    model: agent.modelName,
-                    prompt: prompt,
-                    height: height,
-                    width: width,
-                    n: 1,
-                    disable_safety_checker: true,
-                    response_format: 'base64',
-                });
-            }
-            else {
-                imageResponse = await this.togetherClient.images.generate({
-                    model: agent.modelName,
-                    prompt: prompt,
-                    height: height,
-                    width: width,
-                    n: 1,
-                    disable_safety_checker: true,
-                    response_format: 'base64',
-                });
-            }
-            if (imageResponse.usage) {
-                this.computeImageCost(agent, imageResponse.usage);
-            }
+            const imageResponse = await this.togetherClient.images.generate({
+                ...apiProperties,
+                model: agent.modelName,
+                prompt,
+                n: 1,
+            });
+            this.computeImageCost(agent, imageResponse.usage, centsPerImage);
             const base64Data = imageResponse.data[0].b64_json;
             const artifactId = await this.saveImageAsArtifact(agent, base64Data, prompt);
             return { artifactId };
@@ -281,6 +273,40 @@ let TogetherAiService = class TogetherAiService extends base_ai_service_1.BaseAi
             return { error };
         }
     }
+    async editImage(agent, prompt, imageProperties, referenceImageArtifactIds) {
+        const { centsPerImage, ...apiProperties } = imageProperties ?? {};
+        const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
+        if (!modelInfo && centsPerImage == null) {
+            return { error: `No pricing configured for image model "${agent.modelName}". Add a modelPrices entry or set centsPerImage in image_models.json.` };
+        }
+        if (agent.balanceService && !agent.disableBalanceCheck) {
+            const hasSufficientBalance = agent.balanceService.hasSufficientBalance();
+            if (!hasSufficientBalance) {
+                throw new errors_1.InsufficientBalanceError('Insufficient balance for image generation. Please add funds to continue.');
+            }
+        }
+        try {
+            const referenceImages = referenceImageArtifactIds?.length
+                ? await this.loadReferenceImages(agent, referenceImageArtifactIds)
+                : [];
+            const imageResponse = await this.togetherClient.images.generate({
+                ...apiProperties,
+                reference_image: referenceImages.length === 1 ? referenceImages[0] : referenceImages,
+                model: agent.modelName,
+                prompt,
+                n: 1,
+            });
+            this.computeImageCost(agent, imageResponse.usage, centsPerImage);
+            const base64Data = imageResponse.data[0].b64_json;
+            const artifactId = await this.saveImageAsArtifact(agent, base64Data, prompt);
+            return { artifactId };
+        }
+        catch (error) {
+            logger_1.logger.error('Error editing image:', error.error?.message || error.message);
+            logger_1.logger.log(prompt);
+            return { error };
+        }
+    }
 };
 exports.TogetherAiService = TogetherAiService;
 exports.TogetherAiService = TogetherAiService = __decorate([

package/dist-sdk/server/src/ai/agent-config.d.ts

@@ -7,6 +7,7 @@ export interface AgentConfig {
     promptFolder: string;
     systemInstruction: string;
     modelName: string;
+    smallModelName?: string;
     serviceName: string;
     reasoningEffort: 'low' | 'medium' | 'high' | null;
     allowSearch: boolean;

package/dist-sdk/server/src/ai/agent-store.service.js

@@ -143,6 +143,7 @@ let AgentStoreService = class AgentStoreService {
         instance.maxHistoryMessages = template.maxHistoryMessages;
         instance.promptCacheTTL = template.promptCacheTTL;
         instance.markupRate = template.markupRate;
+        instance.smallModelName = template.smallModelName;
         this.storeAgent(instance);
         return instanceId;
     }
@@ -174,6 +175,7 @@ let AgentStoreService = class AgentStoreService {
         instance.promptFolder = template.promptFolder;
         instance.systemInstruction = template.systemInstruction;
         instance.modelName = template.modelName;
+        instance.smallModelName = template.smallModelName;
         instance.serviceName = template.serviceName;
         instance.reasoningEffort = template.reasoningEffort;
         instance.allowSearch = template.allowSearch;
@@ -234,6 +236,7 @@ let AgentStoreService = class AgentStoreService {
                     const agent = new agent_1.Agent(config.id, this.configService, this, this.balanceService, finalArtifactService, config.agentName, config.agentNickname || config.agentName, config.description, config.promptFolder, config.systemInstruction, config.history || [], config.modelName, config.serviceName, aiService, config.reasoningEffort, config.allowSearch, config.maxOutputTokens, config.beginInstruction || '', storageDir);
                     agent.createdAt = new Date(config.createdAt);
                     agent.configService = this.configService;
+                    agent.smallModelName = config.smallModelName;
                     agent.totalInputTokens = config.totalInputTokens || 0;
                     agent.totalOutputTokens = config.totalOutputTokens || 0;
                     agent.totalCost = config.totalCost || 0;
@@ -305,6 +308,7 @@ let AgentStoreService = class AgentStoreService {
                 promptFolder: agent.promptFolder,
                 systemInstruction: agent.systemInstruction,
                 modelName: agent.modelName,
+                smallModelName: agent.smallModelName,
                 serviceName: agent.serviceName,
                 reasoningEffort: agent.reasoningEffort,
                 allowSearch: agent.allowSearch,

package/dist-sdk/server/src/ai/agent.d.ts

@@ -38,7 +38,9 @@ export declare class Agent {
     systemInstruction: string;
     maxHistoryMessages?: number;
     modelName: string;
+    smallModelName?: string;
     serviceName: string;
+    imageProperties?: Record<string, any>;
     reasoningEffort: 'low' | 'medium' | 'high' | null;
     allowSearch: boolean;
     maxOutputTokens: number;
@@ -67,6 +69,7 @@ export declare class Agent {
     trimHistoryIfNeeded(threshold?: number): void;
     instructedPrompt(prompt: string, instructionFileName: string, variables: any, saveResponse?: boolean, cacheMessage?: boolean, maxOutputChars?: number): Promise<any>;
     preparedPrompt(promptFileName: string, variables: any, saveResponse?: boolean, cacheMessage?: boolean): Promise<any>;
+    withSmallModel<T>(fn: () => Promise<T>): Promise<T>;
     compact(): Promise<{
         success: boolean;
         originalLength: number;
@@ -75,7 +78,11 @@ export declare class Agent {
         summary?: string;
     }>;
     textPrompt(prompt: string, jsonSchema?: any, saveResponse?: boolean, promptInstructions?: string, cacheMessage?: boolean, maxOutputChars?: number): Promise<any>;
-    generateImage(prompt: string, width: number, height: number, quality: string, referenceImageArtifactIds?: string[]): Promise<{
+    generateImage(prompt: string): Promise<{
+        artifactId?: string;
+        error?: any;
+    }>;
+    editImage(prompt: string, referenceImageArtifactIds?: string[]): Promise<{
         artifactId?: string;
         error?: any;
     }>;

package/dist-sdk/server/src/ai/agent.js

@@ -36,7 +36,9 @@ class Agent {
     systemInstruction;
     maxHistoryMessages;
     modelName;
+    smallModelName;
     serviceName;
+    imageProperties;
     reasoningEffort;
     allowSearch;
     maxOutputTokens = 0;
@@ -164,6 +166,19 @@ class Agent {
     async preparedPrompt(promptFileName, variables, saveResponse = true, cacheMessage = false) {
         return await this.aiService.preparedPrompt(this, promptFileName, variables, saveResponse, this.aiService.getCacheConfig(this, cacheMessage));
     }
+    async withSmallModel(fn) {
+        if (!this.smallModelName || this.smallModelName === this.modelName) {
+            return fn();
+        }
+        const original = this.modelName;
+        this.modelName = this.smallModelName;
+        try {
+            return await fn();
+        }
+        finally {
+            this.modelName = original;
+        }
+    }
     async compact() {
         const originalHistory = [...(this.history || [])];
         const originalLength = originalHistory.length;
@@ -178,7 +193,7 @@ class Agent {
         const formattedHistory = this.getFormattedMessages(0);
         try {
             this.history = [];
-            const summaryResponse = await this.preparedPrompt('compact_agent', { agentName: this.agentNickname, formattedHistory }, false);
+            const summaryResponse = await this.withSmallModel(() => this.preparedPrompt('compact_agent', { agentName: this.agentNickname, formattedHistory }, false));
             const summary = summaryResponse?.summary || summaryResponse;
             this.addToConversationHistory(`[CONVERSATION HISTORY SUMMARY]\n\n${summary}`, true);
             const newLength = this.history.length;
@@ -199,8 +214,11 @@ class Agent {
     async textPrompt(prompt, jsonSchema = null, saveResponse = true, promptInstructions = null, cacheMessage = false, maxOutputChars = 0) {
         return await this.aiService.textPrompt(this, prompt, jsonSchema, saveResponse, promptInstructions, this.aiService.getCacheConfig(this, cacheMessage), maxOutputChars);
     }
-    async generateImage(prompt, width, height, quality, referenceImageArtifactIds) {
-        return await this.aiService.generateImage(this, prompt, width, height, quality, referenceImageArtifactIds);
+    async generateImage(prompt) {
+        return await this.aiService.generateImage(this, prompt, this.imageProperties);
+    }
+    async editImage(prompt, referenceImageArtifactIds) {
+        return await this.aiService.editImage(this, prompt, this.imageProperties, referenceImageArtifactIds);
     }
 }
 exports.Agent = Agent;

package/dist-sdk/server/src/ai/model_prices.js

@@ -39,6 +39,24 @@ exports.isKnownModel = isKnownModel;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 let _modelPrices = null;
+let _imageModels = null;
+function getImageModels() {
+    if (_imageModels !== null)
+        return _imageModels;
+    const dataDir = process.env.BRIYAH_DATA_PATH || path.resolve(process.cwd(), 'briyah-data');
+    const imageModelsPath = path.join(dataDir, 'common', 'config', 'image_models.json');
+    if (!fs.existsSync(imageModelsPath)) {
+        _imageModels = [];
+        return _imageModels;
+    }
+    try {
+        _imageModels = JSON.parse(fs.readFileSync(imageModelsPath, 'utf8'));
+    }
+    catch {
+        _imageModels = [];
+    }
+    return _imageModels;
+}
 function getModelPrices() {
     if (_modelPrices !== null)
         return _modelPrices;
@@ -69,5 +87,7 @@ function resolveModelInfo(modelName, prefix) {
     return null;
 }
 function isKnownModel(modelName, prefix) {
-    return resolveModelInfo(modelName, prefix) !== null;
+    if (resolveModelInfo(modelName, prefix) !== null)
+        return true;
+    return getImageModels().some((m) => m.model === modelName && m.centsPerImage != null);
 }

package/dist-sdk/server/src/app.controller.d.ts

@@ -129,6 +129,7 @@ export declare class AppController {
     getPublishedRoomInstances(req: ExpressRequest, templateId: string): Promise<RoomInfo[]>;
     resetPublishedRoomInstance(req: ExpressRequest, instanceId: string): Promise<void>;
     resetStory(req: ExpressRequest, storyId: string): Promise<void>;
+    revertStoryChapter(req: ExpressRequest, storyId: string): Promise<void>;
     getRoomArtifacts(req: ExpressRequest, roomId: string): Promise<{
         artifacts: any[];
     }>;
@@ -150,9 +151,11 @@ export declare class AppController {
         idea: string;
         userCharacterDesc: string;
         otherCharactersDesc: string;
+        illustrateStory: boolean;
         storyModel?: string;
         isImport?: boolean;
         imageModelName?: string;
+        imageEditModelName?: string;
     }): Promise<StoryInfo>;
     importStoryZip(req: ExpressRequest, file: Express.Multer.File): Promise<StoryInfo>;
     deleteStory(req: ExpressRequest, storyId: string): Promise<void>;
@@ -164,6 +167,7 @@ export declare class AppController {
     respondToStory(req: ExpressRequest, storyId: string, body: {
         content: string;
     }): Promise<void>;
+    interruptStory(req: ExpressRequest, storyId: string): Promise<void>;
     introduceCharacter(req: ExpressRequest, storyId: string, body: {
         name: string;
         description?: string;

package/dist-sdk/server/src/app.controller.js

@@ -132,6 +132,7 @@ let AppController = class AppController {
         return {
             selectedStoryModel: process.env.DEFAULT_STORY_MODEL,
             selectedImageModel: process.env.DEFAULT_IMAGE_MODEL,
+            selectedImageEditModel: process.env.DEFAULT_IMAGE_EDIT_MODEL,
             enforceDefaultModels: process.env.ENFORCE_DEFAULT_MODELS === 'true',
         };
     }
@@ -1040,6 +1041,25 @@ let AppController = class AppController {
             throw new common_1.HttpException('Failed to reset story', common_1.HttpStatus.INTERNAL_SERVER_ERROR);
         }
     }
+    async revertStoryChapter(req, storyId) {
+        if (!storyId) {
+            throw new common_1.BadRequestException('Story ID is required');
+        }
+        try {
+            const appService = this.getAppService(req);
+            await appService.revertStoryChapter(storyId);
+        }
+        catch (error) {
+            logger_1.logger.error(`[${req.user?.sub}] Error reverting chapter for story ${storyId}:`, error);
+            if (error instanceof errors_1.NotFoundError) {
+                throw new common_1.NotFoundException(error.message);
+            }
+            if (error instanceof errors_1.OperationFailedError) {
+                throw new common_1.HttpException(error.message, common_1.HttpStatus.INTERNAL_SERVER_ERROR);
+            }
+            throw new common_1.HttpException('Failed to revert chapter', common_1.HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+    }
     async getRoomArtifacts(req, roomId) {
         try {
             const appService = this.getAppService(req, undefined, roomId);
@@ -1134,7 +1154,7 @@ let AppController = class AppController {
         }
         try {
             const appService = this.getAppService(req);
-            return appService.createStory(body.name, body.idea, body.userCharacterDesc, body.otherCharactersDesc, body.storyModel, body.isImport, body.imageModelName);
+            return appService.createStory(body.name, body.idea, body.userCharacterDesc, body.otherCharactersDesc, body.illustrateStory ?? true, body.storyModel, body.isImport, body.imageModelName, body.imageEditModelName);
         }
         catch (error) {
             logger_1.logger.error(`[${req.user?.sub}] Error creating story ${body.name}:`, error);
@@ -1255,6 +1275,19 @@ let AppController = class AppController {
             throw new common_1.HttpException(error.message || 'Failed to respond to story', common_1.HttpStatus.INTERNAL_SERVER_ERROR);
         }
     }
+    async interruptStory(req, storyId) {
+        try {
+            const appService = this.getAppService(req);
+            await appService.interruptStory(storyId);
+        }
+        catch (error) {
+            logger_1.logger.error(`[${req.user?.sub}] Error interrupting story ${storyId}:`, error);
+            if (error instanceof common_1.HttpException) {
+                throw error;
+            }
+            throw new common_1.HttpException(error.message || 'Failed to interrupt story', common_1.HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+    }
     async introduceCharacter(req, storyId, body) {
         if (!body.name) {
             throw new common_1.BadRequestException('Name is a required field');
@@ -2325,6 +2358,15 @@ __decorate([
     __metadata("design:paramtypes", [Object, String]),
     __metadata("design:returntype", Promise)
 ], AppController.prototype, "resetStory", null);
+__decorate([
+    (0, common_1.Post)('stories/:storyId/revert'),
+    (0, common_1.HttpCode)(common_1.HttpStatus.NO_CONTENT),
+    __param(0, (0, common_1.Request)()),
+    __param(1, (0, common_1.Param)('storyId')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, String]),
+    __metadata("design:returntype", Promise)
+], AppController.prototype, "revertStoryChapter", null);
 __decorate([
     (0, room_access_decorator_1.RoomReadAccess)(),
     (0, common_1.Get)('rooms/:roomId/artifacts'),
@@ -2448,6 +2490,14 @@ __decorate([
     __metadata("design:paramtypes", [Object, String, Object]),
     __metadata("design:returntype", Promise)
 ], AppController.prototype, "respondToStory", null);
+__decorate([
+    (0, common_1.Post)('stories/:storyId/interrupt'),
+    __param(0, (0, common_1.Request)()),
+    __param(1, (0, common_1.Param)('storyId')),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, String]),
+    __metadata("design:returntype", Promise)
+], AppController.prototype, "interruptStory", null);
 __decorate([
     (0, common_1.Post)('stories/:storyId/characters'),
     (0, common_1.HttpCode)(common_1.HttpStatus.CREATED),

package/dist-sdk/server/src/app.service.d.ts

@@ -115,7 +115,8 @@ export declare class AppService {
     compactAgentConversation(agentId: string): Promise<void>;
     resetRoom(roomId: string): Promise<void>;
     resetStory(storyId: string): Promise<void>;
-    createStory(name: string, idea: string, userCharacterDesc: string, otherCharactersDesc: string, storyModel?: string, isImport?: boolean, imageModelName?: string): Promise<StoryInfo>;
+    revertStoryChapter(storyId: string): Promise<void>;
+    createStory(name: string, idea: string, userCharacterDesc: string, otherCharactersDesc: string, illustrateStory: boolean, storyModel?: string, isImport?: boolean, imageModelName?: string, imageEditModelName?: string): Promise<StoryInfo>;
     importStoryFromZip(zipBuffer: Buffer): Promise<StoryInfo>;
     getStoryProgressEmitter(storyId: string): EventEmitter<any>;
     getStoryMessageEmitter(storyId: string): EventEmitter<any>;
@@ -140,6 +141,7 @@ export declare class AppService {
     getStoryState(storyId: string): Promise<StoryState>;
     getStoryInfo(storyId: string): Promise<StoryInfo | undefined>;
     respondToStory(storyId: string, content: string): Promise<void>;
+    interruptStory(storyId: string): Promise<void>;
     introduceCharacterToStory(storyId: string, name: string, description: string, storyModel?: string, fromNarratorSuggestion?: boolean): Promise<void>;
     declineCharacter(storyId: string, characterName: string): Promise<void>;
     deleteCharacterFromStory(storyId: string, characterName: string): Promise<void>;

package/dist-sdk/server/src/app.service.js

@@ -694,8 +694,7 @@ class AppService {
             const allFiles = fs.readdirSync(targetDir);
             const promptFiles = allFiles.filter((file) => file.endsWith('.prompt') || file.endsWith('.json'));
             promptFiles.forEach((file) => {
-                const baseName = file.replace(/\.(prompt|json)$/, '');
-                fileMap.set(baseName, scope);
+                fileMap.set(file, scope);
             });
         };
         scanDirectory(this.configService.getCommonPromptsDir(), 'common');
@@ -749,17 +748,20 @@ class AppService {
         fs.writeFileSync(filePath, content);
     }
     deletePromptFile(folderName, fileName) {
-        let targetDir = this.configService.getPromptsDir();
-        if (folderName !== 'shared') {
-            targetDir = path.join(this.configService.getPromptsDir(), folderName);
-        }
-        const promptPath = path.join(targetDir, fileName + '.prompt');
-        if (fs.existsSync(promptPath)) {
-            fs.unlinkSync(promptPath);
+        const baseDir = this.configService.getUserPromptsDir();
+        const targetDir = folderName === 'shared' ? baseDir : path.join(baseDir, folderName);
+        const hasExtension = fileName.endsWith('.prompt') || fileName.endsWith('.json');
+        if (hasExtension) {
+            const filePath = path.join(targetDir, fileName);
+            if (fs.existsSync(filePath))
+                fs.unlinkSync(filePath);
         }
-        const jsonPath = path.join(targetDir, fileName + '.json');
-        if (fs.existsSync(jsonPath)) {
-            fs.unlinkSync(jsonPath);
+        else {
+            for (const ext of ['.prompt', '.json']) {
+                const filePath = path.join(targetDir, fileName + ext);
+                if (fs.existsSync(filePath))
+                    fs.unlinkSync(filePath);
+            }
         }
     }
     async getRoomArtifacts(roomId) {
@@ -967,8 +969,11 @@ class AppService {
             logger_1.logger.warn(`Story "${storyName}" reset with warnings:`, warnings);
         }
     }
-    async createStory(name, idea, userCharacterDesc, otherCharactersDesc, storyModel, isImport, imageModelName) {
-        return this.storyService.createStory(name, idea, userCharacterDesc, otherCharactersDesc, storyModel, isImport, imageModelName);
+    async revertStoryChapter(storyId) {
+        return this.storyService.revertStoryChapter(storyId);
+    }
+    async createStory(name, idea, userCharacterDesc, otherCharactersDesc, illustrateStory, storyModel, isImport, imageModelName, imageEditModelName) {
+        return this.storyService.createStory(name, idea, userCharacterDesc, otherCharactersDesc, illustrateStory, storyModel, isImport, imageModelName, imageEditModelName);
     }
     async importStoryFromZip(zipBuffer) {
         return this.storyService.importStoryFromZip(zipBuffer);
@@ -1102,6 +1107,9 @@ class AppService {
     async respondToStory(storyId, content) {
         return this.storyService.respondToStory(storyId, content);
     }
+    async interruptStory(storyId) {
+        return this.storyService.interruptStory(storyId);
+    }
     async introduceCharacterToStory(storyId, name, description, storyModel, fromNarratorSuggestion) {
         await this.storyService.introduceCharacter(storyId, name, description, storyModel, fromNarratorSuggestion);
     }

package/dist-sdk/server/src/room/message.d.ts

@@ -7,6 +7,7 @@ export declare class RoomMessage implements Message {
     timestamp: Date;
     name?: string;
     situation?: string;
+    updatePortrait?: string[];
     index?: number;
     constructor(sender: string, action: MessageAction, content: string, targets?: string[], name?: string, situation?: string);
     getSender(): string;

package/dist-sdk/server/src/room/message.js

@@ -10,6 +10,7 @@ class RoomMessage {
     timestamp;
     name;
     situation;
+    updatePortrait;
     index;
     constructor(sender, action = app_types_1.MessageAction.SPEAK, content, targets = [], name, situation) {
         this.sender = sender;

package/dist-sdk/server/src/room/room-store.service.d.ts

@@ -34,5 +34,6 @@ export declare class RoomStoreService {
     saveArtifactFile(roomId: string, storageDir: string, artifact: Artifact): void;
     deleteArtifactFile(roomId: string, storageDir: string, artifactName: string): boolean;
     deleteAllArtifactFiles(roomId: string, storageDir: string): void;
+    loadArtifactFiles(roomId: string, storageDir: string): Artifact[];
     private deleteRoomFile;
 }

package/dist-sdk/server/src/room/room-store.service.js

@@ -318,6 +318,9 @@ let RoomStoreService = class RoomStoreService {
     deleteAllArtifactFiles(roomId, storageDir) {
         this.artifactStoreService.deleteAllArtifacts(roomId, storageDir);
     }
+    loadArtifactFiles(roomId, storageDir) {
+        return this.artifactStoreService.loadArtifacts(roomId, storageDir);
+    }
     deleteRoomFile(roomId, storageDir, deleteAgents = false, room) {
         try {
             if (deleteAgents && room) {

package/dist-sdk/server/src/room/room.d.ts

@@ -23,6 +23,7 @@ export declare class Room {
     private onError?;
     private currentSituation;
     private onSituationUpdate?;
+    private onModeratorResponseCallback?;
     private templateRoomId;
     private storageDir;
     private roomStoreService;
@@ -93,6 +94,7 @@ export declare class Room {
     getSituation(): string;
     setSituation(situation: string): void;
     setOnSituationUpdate(callback: (situation: string) => void): void;
+    setOnModeratorResponse(callback: (message: RoomMessage) => void): void;
     private handleSituationUpdate;
     handlePublishMessages(messages: RoomMessage[]): void;
     handleExecuteMessages(messages: RoomMessage[]): void;
@@ -102,6 +104,7 @@ export declare class Room {
     isProcessingInProgress(): boolean;
     pause(): void;
     resume(): void;
+    interrupt(): void;
     checkAgentNicknameMatch(existingNickname: string, targetNickname: string): boolean;
     private handleModeratorResponse;
     private processPendingMessages;

package/dist-sdk/server/src/room/room.js

@@ -29,6 +29,7 @@ class Room {
     onError;
     currentSituation = '';
     onSituationUpdate;
+    onModeratorResponseCallback;
     templateRoomId;
     storageDir;
     roomStoreService;
@@ -242,7 +243,11 @@ class Room {
             return new message_1.RoomMessage(agent.agentNickname, app_types_1.MessageAction.MODERATE, content, [], null, situation);
         }
         situation = response.situation || undefined;
-        return new message_1.RoomMessage(agent.agentNickname, normalizedAction, response.content, targets, response.name, situation);
+        const msg = new message_1.RoomMessage(agent.agentNickname, normalizedAction, response.content, targets, response.name, situation);
+        if (Array.isArray(response.update_portrait) && response.update_portrait.length > 0) {
+            msg.updatePortrait = response.update_portrait;
+        }
+        return msg;
     }
     static coerceMessageAction(raw) {
         if (typeof raw !== 'string')
@@ -484,6 +489,9 @@ Recent Message: ${message.getContent()}
     setOnSituationUpdate(callback) {
         this.onSituationUpdate = callback;
     }
+    setOnModeratorResponse(callback) {
+        this.onModeratorResponseCallback = callback;
+    }
     handleSituationUpdate(situation) {
         logger_1.logger.log(`Situation updated: ${situation}`);
         this.currentSituation = situation;
@@ -610,6 +618,10 @@ Recent Message: ${message.getContent()}
             this.processPendingMessages();
         }
     }
+    interrupt() {
+        this.pendingMessages = [];
+        this.pause();
+    }
     checkAgentNicknameMatch(existingNickname, targetNickname) {
         if (existingNickname.toLowerCase() === targetNickname.toLowerCase())
             return true;
@@ -625,6 +637,9 @@ Recent Message: ${message.getContent()}
             const targets = moderatorMessage.targets || [];
             if (!targets.includes(roomLeaderName))
                 targets.push(roomLeaderName);
+            if (this.onModeratorResponseCallback) {
+                this.onModeratorResponseCallback(moderatorMessage);
+            }
             let content = moderatorMessage.content;
             if (content && content.startsWith('relay'))
                 content = originalMessage.getContent();
@@ -659,13 +674,8 @@ Recent Message: ${message.getContent()}
                 let roomLeaderAgent = this.agents.find((a) => a.agentNickname === this.getRoomLeader());
                 if (!roomLeaderAgent)
                     throw new Error(`Room leader agent not found`);
-                logger_1.logger.warn(`Responder agent not found: ${responderNickname}`);
+                logger_1.logger.warn(`Responder agent "${responderNickname}" not found; falling back to room leader: ${roomLeaderAgent.agentNickname}`);
                 responderAgent = roomLeaderAgent;
-                if (responderNickname) {
-                    if (this.onSituationUpdate && responderNickname.length > 0) {
-                        this.onSituationUpdate(`INTRODUCE: ${responderNickname}`);
-                    }
-                }
             }
             if (responderAgent.isControlledByHuman) {
                 logger_1.logger.log(`Waiting for human agent ${responderNickname} to respond`);
@@ -678,6 +688,11 @@ Recent Message: ${message.getContent()}
             }
             this.setCurrentSpeaker(responderNickname);
             this.perceive(responderAgent, originalMessage, (response) => {
+                if (this.isPaused) {
+                    this.processingInProgress = false;
+                    this.notifyStateChange();
+                    return;
+                }
                 if (response !== null) {
                     this.pendingMessages = [response, ...this.pendingMessages];
                 }
@@ -738,6 +753,11 @@ Recent Message: ${message.getContent()}
                 this.pendingMessages = [...moderateMessages.slice(1), ...this.pendingMessages];
             }
             this.moderate(moderator, originalMessage, (response) => {
+                if (this.isPaused) {
+                    this.processingInProgress = false;
+                    this.notifyStateChange();
+                    return;
+                }
                 if (response !== null) {
                     this.handleModeratorResponse(response, originalMessage);
                 }

package/dist-sdk/server/src/story/story-store.service.d.ts

@@ -19,5 +19,6 @@ export declare class StoryStoreService {
     createChapterBackup(storyId: string, chapterIndex: number, chapterTitle: string): boolean;
     saveChapterContent(storyId: string, chapterIndex: number, content: string): boolean;
     deleteChapter(storyId: string, chapterIndex: number): boolean;
+    restoreRoomFilesFromChapter(storyId: string, chapterIndex: number): number;
     restoreAgentsFromChapter(storyId: string, chapterIndex: number, agentStore: any): Promise<string[]>;
 }