briyah 1.1.4 → 1.1.6

package/data/common/prompts/story_moderator/moderate.prompt

@@ -53,6 +53,11 @@ This request will *not* become part of the readable story, only the narrator's r
 3. Responder: The one character you choose to respond next to best maintain the flow of the conversation and further the story.
 4. Targets: List of all characters who would realistically be able to hear the message (even if they shouldn't respond).
 - When processing a message from the Narrator, the targets array should only include characters who are present for the situation being narrated.
+5. Update Portrait: List the names of any characters whose visible appearance has just changed in a way that should be reflected in their portrait image.
+- Include a character's name when the current message describes a meaningful change to their visible appearance: new or different clothing/outfit, significant injury (wounds, blood, bandages), physical transformation, or a notable item they are now visibly carrying or wearing.
+- Use the character's exact name as it appears in the Available Characters list.
+- Leave the array empty if no visible appearance changes occurred.
+- Minor or invisible changes (emotional state alone, unseen actions) do NOT warrant a portrait update.
 
 ## Content Moderation
 - Do *not* allow profanity or swear words like 'hell', 'shit', and 'damn' or 'damned'.
@@ -67,7 +72,8 @@ You must respond with a valid JSON object in the following format:
   "action": "relay",  // or "relay_silent" for internal processing
   "content": "relay",  // or modified message body
   "name": "CharacterName1",  // Name of character who should respond next
-  "targets": ["CharacterName1", "CharacterName2"]   // List of all characters who would realistically be able to hear the message
+  "targets": ["CharacterName1", "CharacterName2"],  // List of all characters who would realistically be able to hear the message
+  "update_portrait": []  // Names of characters whose visible appearance changed; empty if none
 }
 
 ## Conversation Rules
@@ -81,10 +87,10 @@ You must respond with a valid JSON object in the following format:
 - **Important**:Avoid lengthy conversations that exclude the Human Character completely. Give the Human Character a chance to interact with the story.
 
 ## Introductions
-- Only set "name" to the exact character name (nickname) of one of the characters in **Available Characters** list.
-- However, if a message from the Human Character directly addresses someone in the story who is not listed in the **Available Characters** list, you may introduce that character like this:
-  - Set the **name** field to a single-word character name (nickname) of the referenced character.
-  - Doing this will give the system the opportunity to create a new character profile for that character using the name you provide so that they will appear in the **Available Characters** list in the future.
+- Only set "name" to the exact character name (nickname) of one of the characters in the **Available Characters** list.
+- If a message from the Human Character (or any character) directly addresses someone who is NOT listed in the **Available Characters** list, route the message to the Narrator. The Narrator has access to the secret plot plan and can identify whether the addressed person is a planned character; do NOT guess at a name yourself.
+  - Set "name" to "Narrator" and "targets" to ["Narrator"].
+  - Use "action": "relay" so the dialog still becomes part of the readable story.
 - If you receive a message from the system like "INTRODUCE [Character name]", relay the INTRODUCE [Character name] message to the Narrator so they may properly introduce the new character.
 
 ## Example Responses
@@ -105,7 +111,8 @@ Correct Response:
   "action": "relay_silent",
   "content": "PASS",
   "name": "Narrator",
-  "targets": ["Narrator"]
+  "targets": ["Narrator"],
+  "update_portrait": []
 }
 - Mike is choosing not to act or speak, so select another appropriate character (or the narrator) to speak next instead.
 - Use "relay_silent" because PASS is a meta-action and shouldn't appear in the story
@@ -118,7 +125,8 @@ Correct Response:
   "action": "relay",
   "content": "relay",
   "name": "Mike",
-  "targets": ["Mike", "Bob"]
+  "targets": ["Mike", "Bob"],
+  "update_portrait": []
 }
 - According to the current situation, only Mike, Jessica, and Bob are within hearing distance of each other.
 - Jessica is speaking, and only Mike and Bob can hear. Sergei and Carter are not present.
@@ -134,7 +142,8 @@ Correct Response:
   "action": "relay",
   "content": "relay",
   "name": "Jessica",
-  "targets": ["Jessica", "Bob"]
+  "targets": ["Jessica", "Bob"],
+  "update_portrait": []
 }
 - According to the current situation, only Mike, Jessica, and Bob are within hearing distance of each other.
 - Mike is speaking, and only Jessica and Bob can hear. Sergei and Carter are not present.
@@ -150,7 +159,8 @@ Correct Response:
   "action": "relay_silent",
   "content": "relay",
   "name": "Narrator",
-  "targets": ["Narrator"]
+  "targets": ["Narrator"],
+  "update_portrait": []
 }
 - The sender is attempting to perform an action with the message
 - Only the Narrator is able to handle the character's request to perform an action.
@@ -167,7 +177,8 @@ Correct Response:
   "action": "relay",
   "content": "relay",
   "name": "Mike",
-  "targets": ["Mike", "Jessica", "Bob"]
+  "targets": ["Mike", "Jessica", "Bob"],
+  "update_portrait": []
 }
 - This is not dialog and there is no obvious responder, so choose the human character to respond because they are involved in the current situation.
 - All characters involved in the current situation should be targets of the message
@@ -182,25 +193,28 @@ Correct Response:
   "action": "relay",
   "content": "relay",
   "name": "Sergei",
-  "targets": ["Sergei", "Carter"]
+  "targets": ["Sergei", "Carter"],
+  "update_portrait": []
 }
 - This is not dialog and there is no obvious responder, so choose an character involved in the current situation to be the responder
 - All characters involved in the current situation should be targets of the message
 - Mike, Jessica, and Bob should not be targets of this message because they are not involved in the situation.
 
 
-### Example 6: Character addresses another character who is not listed in the Available Characters list
+### Example 6: Character addresses someone not listed in the Available Characters list
 Recent Message Sender: Mike (human character)
 Recent Message: Hello, Albert Flanagan!
 Correct Response:
 {
   "action": "relay",
   "content": "relay",
-  "name": "Albert",
-  "targets": ["Albert"]
+  "name": "Narrator",
+  "targets": ["Narrator"],
+  "update_portrait": []
 }
-- If Human Character directly addresses someone not listed in the **Available Characters** list, create a single-word name for the referenced character.
-- The single-word name you choose may appear in the **Available Characters** list in the future if the user decides to introduce them as a new character.
+- "Albert Flanagan" is not in the Available Characters list.
+- Route to the Narrator. The Narrator can consult the plot plan to identify the addressed character (and emit an appropriate INTRODUCE situation) or narrate an alternative response.
+- Use "relay" so Mike's dialog still appears in the readable story.
 
 ### Example 7: Message contains explicit content
 Recent Message Sender: Mike
@@ -210,7 +224,8 @@ Correct Response:
   "action": "relay",
   "content": "[message content modified to remove explicit content]",
   "name": "Narrator",
-  "targets": ["Narrator"]
+  "targets": ["Narrator"],
+  "update_portrait": []
 }
 - If the message contains explicit content, just remove the explicit content if possible or modify the message to something more appropriate.
 
@@ -222,8 +237,38 @@ Correct Response:
   "action": "relay_silent",
   "content": "relay",
   "name": "",
-  "targets": []
+  "targets": [],
+  "update_portrait": []
 }
 - Use the relay_silent action and leave the name field blank and set the targets list to an empty array.
 - The message will be silently relayed to the Narrator to perform the introduction.
 
+
+### Example 9: Narrator describes a visible appearance change
+Recent Message Sender: Narrator
+Recent Message: The arrow grazes Jessica's arm, tearing her sleeve and leaving a bloody gash. She wraps a strip of cloth around the wound and keeps moving.
+Correct Response:
+{
+  "action": "relay",
+  "content": "relay",
+  "name": "Mike",
+  "targets": ["Mike", "Jessica", "Bob"],
+  "update_portrait": ["Jessica"]
+}
+- Jessica's visible appearance has changed (torn sleeve, wound, makeshift bandage), so her portrait should be updated.
+- All characters present in the current situation are targets.
+
+
+### Example 10: Character action with no visible change
+Recent Message Sender: Bob
+Recent Message: Bob wields his new sword proudly above his head
+Correct Response:
+{
+  "action": "relay_silent",
+  "content": "relay",
+  "name": "Narrator",
+  "targets": ["Narrator"],
+  "update_portrait": ["Bob"]
+}
+- Bob just got a new weapon (or other visible item) that we would want to include in his portrait.
+

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

@@ -4,12 +4,6 @@ import { ArtifactService } from '../artifact.service';
 import { AgentStoreService } from '../agent-store.service';
 import { BalanceService } from '../../app/balance.service';
 import { ModelInfo } from '../../../../shared/types/app.types';
-export interface ImageGenerationOptions {
-    model?: string;
-    size?: '1024x1024' | '1536x1024' | '1024x1536' | 'auto';
-    quality?: 'auto' | 'high' | 'medium' | 'low';
-    format?: 'png' | 'jpeg' | 'webp';
-}
 export declare class BaseAiService {
     protected modelsCache: ModelInfo[];
     protected _isAvailable: boolean;
@@ -28,7 +22,11 @@ export declare class BaseAiService {
     protected fetchModelsFromApi(): Promise<ModelInfo[]>;
     createAgent(configService: ConfigurationService, agentStoreService: AgentStoreService, balanceService: BalanceService, artifactService: ArtifactService, agentName: string, agentNickname: string, description: string, promptFolder: string, modelName: string, storageDir?: string): Agent;
     textPrompt(agent: Agent, prompt: string, _jsonSchema?: any, _saveResponse?: boolean, _promptInstructions?: string, _cacheConfig?: any, _maxOutputChars?: number): Promise<any>;
-    generateImage(_agent: Agent, _prompt: string, _width: number, _height: number, _quality: string, _referenceImageArtifactIds?: string[], _options?: ImageGenerationOptions): 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/base-ai.service.js

@@ -177,12 +177,18 @@ let BaseAiService = class BaseAiService {
         }
         throw new Error('Method not implemented in base class');
     }
-    async generateImage(_agent, _prompt, _width, _height, _quality, _referenceImageArtifactIds, _options) {
+    async generateImage(_agent, _prompt, _imageProperties) {
         if (!this.isAvailable) {
             return Promise.reject(new Error(`${this.getServiceName()} service is not available. Check API key configuration.`));
         }
         throw new Error(`generateImage method not implemented for ${this.getServiceName()}`);
     }
+    async editImage(_agent, _prompt, _imageProperties, _referenceImageArtifactIds) {
+        if (!this.isAvailable) {
+            return Promise.reject(new Error(`${this.getServiceName()} service is not available. Check API key configuration.`));
+        }
+        throw new Error(`editImage method not implemented for ${this.getServiceName()}`);
+    }
     async loadReferenceImages(agent, artifactIds) {
         const images = [];
         for (const artifactId of artifactIds) {

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

@@ -11,7 +11,11 @@ export declare class FalAiService 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/fal.service.js

@@ -218,25 +218,37 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
             agent.balanceService.decrementBalance(cost, markup);
         }
     }
-    computeImageCost(agent) {
+    computeImageCost(agent, centsPerImage) {
         let modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (!modelInfo && !agent.modelName.startsWith('fal_ai/')) {
             modelInfo = (0, model_prices_1.getModelPrices)()[`fal_ai/${agent.modelName}`];
         }
-        if (!modelInfo || !modelInfo.output_cost_per_image) {
-            logger_1.logger.warn(`No image price info found for ${agent.modelName}. Skipping cost tracking.`);
+        let cost;
+        if (modelInfo?.output_cost_per_image != null) {
+            cost = modelInfo.output_cost_per_image;
+        }
+        else if (centsPerImage != null) {
+            cost = centsPerImage / 100;
+        }
+        else {
+            logger_1.logger.warn(`No price info for model ${agent.modelName}. Skipping cost tracking.`);
             return;
         }
-        const cost = modelInfo.output_cost_per_image;
-        const markup = 0;
         agent.totalCost += cost;
-        agent.totalMarkup += markup;
-        logger_1.logger.log(`Cost for image generation: ${cost.toFixed(4)} (markup: ${markup.toFixed(4)})`);
+        logger_1.logger.log(`Image cost for ${agent.agentName}: $${cost.toFixed(4)}`);
         if (agent.balanceService && cost > 0) {
-            agent.balanceService.decrementBalance(cost, markup);
+            agent.balanceService.decrementBalance(cost, 0);
         }
     }
-    async generateImage(agent, prompt, width, height, quality, referenceImageArtifactIds) {
+    async generateImage(agent, prompt, imageProperties) {
+        const { centsPerImage, ...apiProperties } = imageProperties ?? {};
+        let modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
+        if (!modelInfo && !agent.modelName.startsWith('fal_ai/')) {
+            modelInfo = (0, model_prices_1.getModelPrices)()[`fal_ai/${agent.modelName}`];
+        }
+        if (modelInfo?.output_cost_per_image == null && 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) {
@@ -249,23 +261,14 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
                 endpoint = `fal-ai/${endpoint}`;
             }
             const input = {
-                prompt: prompt,
-                image_size: {
-                    width: width,
-                    height: height,
-                },
+                prompt,
                 num_images: 1,
-                output_format: 'png',
+                ...apiProperties,
             };
-            if (referenceImageArtifactIds && referenceImageArtifactIds.length > 0) {
-                input.image_urls = await this.loadReferenceImages(agent, referenceImageArtifactIds);
-                if (!endpoint.endsWith('/edit'))
-                    endpoint += '/edit';
-            }
             const result = (await client_1.fal.subscribe(endpoint, {
                 input: input,
             }));
-            this.computeImageCost(agent);
+            this.computeImageCost(agent, centsPerImage);
             let base64Data;
             if (result.data.images && result.data.images[0]) {
                 const image = result.data.images[0];
@@ -305,6 +308,75 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
             return { error };
         }
     }
+    async editImage(agent, prompt, imageProperties, referenceImageArtifactIds) {
+        const { centsPerImage, ...apiProperties } = imageProperties ?? {};
+        let modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
+        if (!modelInfo && !agent.modelName.startsWith('fal_ai/')) {
+            modelInfo = (0, model_prices_1.getModelPrices)()[`fal_ai/${agent.modelName}`];
+        }
+        if (modelInfo?.output_cost_per_image == null && 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 {
+            let endpoint = agent.modelName;
+            if (!endpoint.startsWith('fal-ai/')) {
+                endpoint = `fal-ai/${endpoint}`;
+            }
+            const input = {
+                prompt,
+                num_images: 1,
+                ...apiProperties,
+            };
+            if (referenceImageArtifactIds && referenceImageArtifactIds.length > 0) {
+                input.image_urls = await this.loadReferenceImages(agent, referenceImageArtifactIds);
+            }
+            const result = (await client_1.fal.subscribe(endpoint, { input }));
+            this.computeImageCost(agent, centsPerImage);
+            let base64Data;
+            if (result.data.images && result.data.images[0]) {
+                const image = result.data.images[0];
+                if (image.url) {
+                    const imageResponse = await fetch(image.url);
+                    const imageBuffer = Buffer.from(await imageResponse.arrayBuffer());
+                    base64Data = imageBuffer.toString('base64');
+                }
+                else if (image.content) {
+                    base64Data = image.content;
+                }
+                else {
+                    throw new Error('No image data found in response');
+                }
+            }
+            else if (result.data.image) {
+                if (result.data.image.url) {
+                    const imageResponse = await fetch(result.data.image.url);
+                    const imageBuffer = Buffer.from(await imageResponse.arrayBuffer());
+                    base64Data = imageBuffer.toString('base64');
+                }
+                else if (result.data.image.content) {
+                    base64Data = result.data.image.content;
+                }
+                else {
+                    throw new Error('No image data found in response');
+                }
+            }
+            else {
+                throw new Error('Unexpected response format from Fal AI');
+            }
+            const artifactId = await this.saveImageAsArtifact(agent, base64Data, prompt);
+            return { artifactId };
+        }
+        catch (error) {
+            logger_1.logger.error('Error editing image with Fal AI:', error.message || error);
+            return { error };
+        }
+    }
 };
 exports.FalAiService = FalAiService;
 exports.FalAiService = FalAiService = __decorate([

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

@@ -10,7 +10,7 @@ export declare class GoogleAiService extends BaseAiService {
     fetchModelsFromApi(): Promise<ModelInfo[]>;
     addToConversationHistory(agent: Agent, message: string, fromSelf?: boolean): void;
     textPrompt(agent: Agent, prompt: string, jsonSchema?: any, saveResponse?: boolean, promptInstructions?: string, _cacheConfig?: any, maxOutputChars?: number): Promise<any>;
-    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;
     }>;

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

@@ -153,7 +153,12 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
         }
         return responseText;
     }
-    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 || !modelInfo.output_cost_per_image) && 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) {
@@ -161,32 +166,10 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
             }
         }
         try {
-            const modelName = agent.modelName;
-            let aspectRatio = '1:1';
-            if (width && height) {
-                const ratio = width / height;
-                if (ratio > 1.4)
-                    aspectRatio = '16:9';
-                else if (ratio > 1.2)
-                    aspectRatio = '3:2';
-                else if (ratio < 0.7)
-                    aspectRatio = '9:16';
-                else if (ratio < 0.8)
-                    aspectRatio = '2:3';
-            }
-            const config = {
-                numberOfImages: 1,
-                aspectRatio: aspectRatio,
-            };
-            if (quality && quality !== 'auto') {
-                config.outputOptions = {
-                    compressionQuality: quality === 'high' ? 90 : quality === 'medium' ? 70 : 50,
-                };
-            }
             const response = await this.googleAI.models.generateImages({
-                model: modelName,
-                prompt: prompt,
-                config: config,
+                model: agent.modelName,
+                prompt,
+                config: { ...apiProperties },
             });
             let base64Data;
             if (response.generatedImages?.[0]?.image?.imageBytes) {
@@ -198,7 +181,7 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
             else {
                 throw new Error('No image data in response');
             }
-            this.computeImageCost(agent, modelName);
+            this.computeImageCost(agent, agent.modelName, centsPerImage);
             const artifactId = await this.saveImageAsArtifact(agent, base64Data, prompt);
             return { artifactId };
         }
@@ -228,10 +211,18 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
             agent.balanceService.decrementBalance(cost, markup);
         }
     }
-    computeImageCost(agent, modelName) {
+    computeImageCost(agent, modelName, centsPerImage) {
         const modelInfo = (0, model_prices_1.getModelPrices)()[modelName];
         if (!modelInfo || !modelInfo.output_cost_per_image) {
-            logger_1.logger.error(`No image price info found for model ${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 image price info found for model ${modelName}. Skipping cost tracking.`);
             return;
         }
         const outputCost = modelInfo.output_cost_per_image;

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

@@ -14,7 +14,11 @@ export declare class OpenAiService extends BaseAiService {
     addToConversationHistory(agent: Agent, message: string, fromSelf?: boolean, developer?: boolean): void;
     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/openai.service.js

@@ -315,60 +315,31 @@ let OpenAiService = class OpenAiService extends base_ai_service_1.BaseAiService
             agent.balanceService.decrementBalance(cost, markup);
         }
     }
-    computeImageCost(agent, usage) {
+    computeImageCost(agent, centsPerImage) {
         const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
-        if (!modelInfo) {
-            logger_1.logger.error(`No price info found for model ${agent.modelName}`);
-            return;
+        let cost;
+        if (modelInfo?.output_cost_per_image != null) {
+            cost = modelInfo.output_cost_per_image;
         }
-        const inputImageTokens = usage.input_tokens_details?.image_tokens || 0;
-        const inputTextTokens = usage.input_tokens_details?.text_tokens || 0;
-        const outputImageTokens = usage.output_tokens_details?.image_tokens || 0;
-        const outputTextTokens = usage.output_tokens_details?.text_tokens || 0;
-        const totalInputTokens = usage.input_tokens || 0;
-        const totalOutputTokens = usage.output_tokens || 0;
-        agent.totalInputTokens += totalInputTokens;
-        agent.totalOutputTokens += totalOutputTokens;
-        let inputCost = 0;
-        if (inputImageTokens > 0) {
-            const cachedImageTokens = usage.input_tokens_details?.cached_image_tokens || 0;
-            const uncachedImageTokens = inputImageTokens - cachedImageTokens;
-            inputCost += uncachedImageTokens * (modelInfo.input_cost_per_image_token || 0);
-            if (cachedImageTokens > 0) {
-                inputCost +=
-                    cachedImageTokens *
-                        (modelInfo.cache_read_input_image_token_cost ||
-                            modelInfo.input_cost_per_image_token * 0.5);
-            }
+        else if (centsPerImage != null) {
+            cost = centsPerImage / 100;
         }
-        if (inputTextTokens > 0) {
-            const cachedTextTokens = usage.input_tokens_details?.cached_text_tokens || 0;
-            const uncachedTextTokens = inputTextTokens - cachedTextTokens;
-            inputCost += uncachedTextTokens * (modelInfo.input_cost_per_token || 0);
-            if (cachedTextTokens > 0) {
-                inputCost +=
-                    cachedTextTokens *
-                        (modelInfo.cache_read_input_text_token_cost ||
-                            modelInfo.input_cost_per_token * 0.5);
-            }
-        }
-        let outputCost = 0;
-        if (outputImageTokens > 0) {
-            outputCost += outputImageTokens * (modelInfo.output_cost_per_image_token || 0);
-        }
-        if (outputTextTokens > 0) {
-            outputCost += outputTextTokens * (modelInfo.output_cost_per_token || 0);
+        else {
+            logger_1.logger.warn(`No price info for model ${agent.modelName}. Skipping cost tracking.`);
+            return;
         }
-        const markup = outputCost * 0;
-        const cost = inputCost + outputCost + markup;
         agent.totalCost += cost;
-        agent.totalMarkup += markup;
-        logger_1.logger.log(`Cost for image generation: ${cost} (markup: ${markup.toFixed(4)})`);
+        logger_1.logger.log(`Image cost for ${agent.agentName}: $${cost.toFixed(4)}`);
         if (agent.balanceService && cost > 0) {
-            agent.balanceService.decrementBalance(cost, markup);
+            agent.balanceService.decrementBalance(cost, 0);
         }
     }
-    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?.output_cost_per_image == null && 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) {
             const hasSufficientBalance = agent.balanceService.hasSufficientBalance();
             if (!hasSufficientBalance) {
@@ -376,31 +347,13 @@ let OpenAiService = class OpenAiService extends base_ai_service_1.BaseAiService
             }
         }
         try {
-            let imageResponse;
-            const size = `${width}x${height}`;
-            if (referenceImageArtifactIds && referenceImageArtifactIds.length > 0) {
-                const referenceImages = await this.loadReferenceImages(agent, referenceImageArtifactIds);
-                imageResponse = await this.openai.images.edit({
-                    model: agent.modelName,
-                    prompt: prompt,
-                    image: referenceImages.length === 1 ? referenceImages[0] : referenceImages,
-                    size: size,
-                    n: 1,
-                });
-            }
-            else {
-                imageResponse = await this.openai.images.generate({
-                    model: agent.modelName,
-                    prompt: prompt,
-                    n: 1,
-                    size: size,
-                    quality: quality || 'auto',
-                    moderation: 'low',
-                });
-            }
-            if (imageResponse.usage) {
-                this.computeImageCost(agent, imageResponse.usage);
-            }
+            const imageResponse = await this.openai.images.generate({
+                model: agent.modelName,
+                prompt,
+                n: 1,
+                ...apiProperties,
+            });
+            this.computeImageCost(agent, centsPerImage);
             const base64Data = imageResponse.data[0].b64_json;
             const artifactId = await this.saveImageAsArtifact(agent, base64Data, prompt);
             return { artifactId };
@@ -411,6 +364,40 @@ let OpenAiService = class OpenAiService extends base_ai_service_1.BaseAiService
             return { error };
         }
     }
+    async editImage(agent, prompt, imageProperties, referenceImageArtifactIds) {
+        const { centsPerImage, ...apiProperties } = imageProperties ?? {};
+        const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
+        if (modelInfo?.output_cost_per_image == null && 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) {
+            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.openai.images.edit({
+                model: agent.modelName,
+                prompt,
+                image: referenceImages.length === 1 ? referenceImages[0] : referenceImages,
+                n: 1,
+                ...apiProperties,
+            });
+            this.computeImageCost(agent, 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);
+            logger_1.logger.log(prompt);
+            return { error };
+        }
+    }
 };
 exports.OpenAiService = OpenAiService;
 exports.OpenAiService = OpenAiService = __decorate([