briyah 1.0.6 → 1.0.8

package/README.md

@@ -34,10 +34,10 @@ const appService = briyah.getAppService('user-123');
 // Call agent.save() when you want to persist it to disk.
 const agent = appService.createAgent(
     'Anthropic',
-    'Assistant',
-    'Assistant',
+    'AI Assistant',
+    'James',
     'A helpful AI assistant',
-    'claude-3-5-sonnet-20241022'
+    'claude-haiku-4-5'
 );
 
 if (!agent.id) throw new Error('Agent creation failed');
@@ -58,7 +58,7 @@ await briyah.shutdown();
 interface BriyahConfigOptions {
     /**
      * Base directory for Briyah data storage
-     * Default: './data' relative to process.cwd()
+     * Default: './briyah-data' relative to process.cwd()
      */
     dataPath?: string;
 
@@ -84,11 +84,6 @@ interface BriyahConfigOptions {
 
 ### Environment Variables
 
-Briyah requires certain environment variables to be set:
-
-**Required:**
-- `STARTING_BALANCE` - Starting balance for new users (e.g., "10.00")
-
 **LLM Provider API Keys** (at least one required):
 - `ANTHROPIC_API_KEY` - For Claude models
 - `OPENAI_API_KEY` - For GPT models
@@ -98,9 +93,32 @@ Briyah requires certain environment variables to be set:
 - `TOGETHER_API_KEY` - For Together AI models
 
 **Optional:**
+- `STARTING_BALANCE` - Starting balance for new users (e.g., "10.00")
 - `SERVER_DATA_PATH` - Global default data path (overridden by constructor `dataPath`)
 - `USER_SERVICE_CACHE_TIMEOUT_MINUTES` - Cache timeout (overridden by constructor option)
 
+### Logging
+
+Briyah writes logs to `{dataPath}/logs/briyah.log` by default. Configure or disable via the `logging` option:
+
+```typescript
+const briyah = new Briyah({
+    dataPath: './my-data',
+    logging: {
+        enabled: true,                        // default
+        logFile: './my-data/logs/app.log',    // default: {dataPath}/logs/briyah.log
+        console: false,                       // set true to also log to stdout/stderr
+        level: 'warn',                        // 'debug' | 'log' | 'warn' | 'error' — default: 'log'
+    }
+});
+
+// Log to console only (no file):
+const briyah = new Briyah({ logging: { console: true, logFile: null } });
+
+// Disable logging entirely:
+const briyah = new Briyah({ logging: { enabled: false } });
+```
+
 ## API Reference
 
 ### Briyah Class
@@ -282,10 +300,10 @@ const appService = briyah.getAppService('user-123');
 // Create an agent
 const agent = appService.createAgent(
     'Anthropic',           // provider
-    'Assistant',           // name
-    'Assistant',           // nickname
-    'Helpful AI',          // description
-    'claude-3-5-sonnet-20241022'  // model
+    'AI Assistant',        // name
+    'James',               // nickname
+    'A helpful AI assistant',          // description
+    'claude-haiku-4-5'  // model
 );
 
 if (!agent.id) throw new Error('Agent creation failed');
@@ -309,7 +327,7 @@ const appService = briyah.getAppService('user-123');
 
 // Create agents
 const analyst = appService.createAgent('OpenAI', 'Analyst', 'Analyst', 'Data analyst', 'gpt-4');
-const writer = appService.createAgent('Anthropic', 'Writer', 'Writer', 'Content writer', 'claude-3-5-sonnet-20241022');
+const writer = appService.createAgent('Anthropic', 'Writer', 'Writer', 'Content writer', 'claude-haiku-4-5');
 
 if (!analyst.id || !writer.id) throw new Error('Agent creation failed');
 
@@ -375,10 +393,6 @@ const user2Service = briyah2.getAppService('user-123');
 
 **Note:** Singleton services (LLM providers, message emitters) are shared across all Briyah instances within the same process. This is safe because they are stateless or keyed by ID.
 
-## Backward Compatibility
-
-The Briyah SDK is fully backward compatible with the existing NestJS application. The NestJS app continues to work without any changes, while new applications can use the SDK for simplified integration.
-
 ## Troubleshooting
 
 ### "Must call init() before getAppService()"
@@ -396,7 +410,3 @@ Set the `STARTING_BALANCE` environment variable or pass `startingBalance` in the
 ### Cache not working as expected
 
 Check the cache statistics with `getCacheStats()` and adjust `userServiceCacheTimeoutMinutes` in the constructor options.
-
-## Support
-
-For issues or questions, please refer to the main Briyah documentation or create an issue in the repository.

package/dist/server/src/ai/LLM/anthropic.service.js

@@ -8,16 +8,14 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 var __metadata = (this && this.__metadata) || function (k, v) {
     if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AnthropicAiService = void 0;
 const common_1 = require("@nestjs/common");
 const sdk_1 = require("@anthropic-ai/sdk");
 const base_ai_service_1 = require("./base-ai.service");
-const model_prices_1 = __importDefault(require("../model_prices"));
+const model_prices_1 = require("../model_prices");
 const errors_1 = require("../../common/errors");
+const logger_1 = require("../../common/logger");
 let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.BaseAiService {
     anthropic;
     constructor() {
@@ -25,7 +23,6 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
         try {
             const apiKey = process.env['ANTHROPIC_API_KEY'];
             if (!apiKey) {
-                console.warn('ANTHROPIC_API_KEY environment variable not set. Anthropic service will not be available.');
                 this._isAvailable = false;
                 return;
             }
@@ -34,9 +31,10 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
                 timeout: 1200000,
                 maxRetries: 3,
             });
+            logger_1.logger.debug('Anthropic service initialized successfully');
         }
         catch (error) {
-            console.error('Error initializing Anthropic service:', error);
+            logger_1.logger.error('Error initializing Anthropic service:', error);
             this._isAvailable = false;
         }
     }
@@ -74,7 +72,7 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
             }));
         }
         catch (error) {
-            console.error('Error fetching Anthropic models:', error);
+            logger_1.logger.error('Error fetching Anthropic models:', error);
         }
     }
     addToConversationHistory(agent, message, fromSelf = false) {
@@ -193,7 +191,7 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
             thinkingTokens = 2048;
         else if (agent.reasoningEffort === 'high')
             thinkingTokens = 4096;
-        const modelInfo = model_prices_1.default[agent.modelName];
+        const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (modelInfo && !modelInfo.supports_reasoning)
             thinkingTokens = 0;
         let maxOutputTokens;
@@ -245,8 +243,8 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
                     }
                     catch (parseError) {
                         lastError = parseError;
-                        console.warn(`JSON parse attempt ${attempt}/${maxRetries} failed:`, parseError);
-                        console.warn('Raw response:', responseText);
+                        logger_1.logger.warn(`JSON parse attempt ${attempt}/${maxRetries} failed:`, parseError);
+                        logger_1.logger.warn('Raw response:', responseText);
                         if (attempt < maxRetries) {
                             const retryMessage = `Your previous response was not valid JSON. Please respond with ONLY valid JSON that can be parsed. No explanatory text, no markdown, no formatting - just pure JSON that conforms to the schema. Here was the parsing error: ${parseError.message}`;
                             messages.push({
@@ -283,7 +281,13 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
             }
             catch (apiError) {
                 lastError = apiError;
-                console.error(`API call attempt ${attempt}/${maxRetries} failed:`, apiError);
+                if (apiError.status === 404) {
+                    const innerError = apiError.error?.error;
+                    const errorType = innerError?.type ?? 'not_found_error';
+                    const errorMessage = innerError?.message ?? apiError.message;
+                    throw new Error(`${errorType}: ${errorMessage}`);
+                }
+                logger_1.logger.error(`API call attempt ${attempt}/${maxRetries} failed:`, apiError);
                 if (attempt < maxRetries) {
                     await new Promise((resolve) => setTimeout(resolve, 1000 * attempt));
                 }
@@ -292,9 +296,9 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
         throw new Error(`Failed to get valid response after ${maxRetries} attempts. Last error: ${lastError?.message || 'Unknown error'}`);
     }
     computeMessageCost(agent, usage) {
-        const modelInfo = model_prices_1.default[agent.modelName];
+        const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (!modelInfo) {
-            console.error(`No price info found for model ${agent.modelName}`);
+            logger_1.logger.error(`No price info found for model ${agent.modelName}`);
             return 0;
         }
         const cacheCreationTokens = usage.cache_creation_input_tokens || 0;
@@ -313,12 +317,12 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
         agent.totalOutputTokens += outputTokens;
         agent.totalCost += cost;
         agent.totalMarkup += markup;
-        console.log(`Cost for ${agent.agentName} message: ${cost.toFixed(4)}, markup: ${markup.toFixed(4)}`);
+        logger_1.logger.log(`Cost for ${agent.agentName} message: ${cost.toFixed(4)}, markup: ${markup.toFixed(4)}`);
         if (agent.balanceService && cost > 0) {
             agent.balanceService.decrementBalance(cost, markup);
         }
         if (cacheCreationTokens > 0 || cacheReadTokens > 0) {
-            console.log(`Cache metrics for ${agent.agentName}: creation=${cacheCreationTokens}, read=${cacheReadTokens}, standard=${standardInputTokens}`);
+            logger_1.logger.log(`Cache metrics for ${agent.agentName}: creation=${cacheCreationTokens}, read=${cacheReadTokens}, standard=${standardInputTokens}`);
         }
     }
 };

package/dist/server/src/ai/LLM/base-ai.service.js

@@ -50,6 +50,7 @@ const fs_1 = require("fs");
 const fs = __importStar(require("fs"));
 const handlebars_1 = require("handlebars");
 const errors_1 = require("../../common/errors");
+const logger_1 = require("../../common/logger");
 let BaseAiService = class BaseAiService {
     modelsCache = null;
     _isAvailable = true;
@@ -112,7 +113,7 @@ let BaseAiService = class BaseAiService {
             return JSON.parse(jsonSchemaString);
         }
         catch (error) {
-            console.error(`Error parsing JSON schema from ${jsonSchemaFile}:`, error);
+            logger_1.logger.error(`Error parsing JSON schema from ${jsonSchemaFile}:`, error);
             return null;
         }
     }
@@ -129,7 +130,7 @@ let BaseAiService = class BaseAiService {
             return models;
         }
         catch (error) {
-            console.error(`Error fetching models for ${this.getServiceName()}:`, error);
+            logger_1.logger.error(`Error fetching models for ${this.getServiceName()}:`, error);
             throw error;
         }
     }
@@ -154,7 +155,7 @@ let BaseAiService = class BaseAiService {
                 systemInstructionPrompt = (0, fs_1.readFileSync)(systemInstructionFile).toString('utf8');
             }
             catch (error) {
-                console.error(`Error reading system instruction from ${systemInstructionFile}:`, error);
+                logger_1.logger.error(`Error reading system instruction from ${systemInstructionFile}:`, error);
                 systemInstructionPrompt = null;
             }
         }
@@ -261,7 +262,7 @@ let BaseAiService = class BaseAiService {
             return `Document attached: ${fileName}`;
         }
         catch (error) {
-            console.error(`Error attaching document in ${this.getServiceName()} service:`, error);
+            logger_1.logger.error(`Error attaching document in ${this.getServiceName()} service:`, error);
             throw new Error(`Error attaching document: ${error.message}`);
         }
     }

package/dist/server/src/ai/LLM/deepseek.service.js

@@ -8,9 +8,6 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 var __metadata = (this && this.__metadata) || function (k, v) {
     if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DeepSeekAiService = void 0;
 const common_1 = require("@nestjs/common");
@@ -18,8 +15,9 @@ const deepseek_1 = require("@ai-sdk/deepseek");
 const ai_1 = require("ai");
 const base_ai_service_1 = require("./base-ai.service");
 const zod_1 = require("zod");
-const model_prices_1 = __importDefault(require("../model_prices"));
+const model_prices_1 = require("../model_prices");
 const errors_1 = require("../../common/errors");
+const logger_1 = require("../../common/logger");
 let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAiService {
     deepseekClient;
     constructor() {
@@ -27,16 +25,16 @@ let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAi
         try {
             const apiKey = process.env['DEEPSEEK_API_KEY'];
             if (!apiKey) {
-                console.warn('DEEPSEEK_API_KEY environment variable not set. DeepSeek service will not be available.');
                 this._isAvailable = false;
                 return;
             }
             this.deepseekClient = (0, deepseek_1.createDeepSeek)({
                 apiKey: apiKey,
             });
+            logger_1.logger.debug('DeepSeek service initialized successfully');
         }
         catch (error) {
-            console.error('Error initializing DeepSeek service:', error);
+            logger_1.logger.error('Error initializing DeepSeek service:', error);
             this._isAvailable = false;
         }
     }
@@ -70,7 +68,7 @@ let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAi
             });
         }
         catch (error) {
-            console.error('Error fetching DeepSeek models from API:', error);
+            logger_1.logger.error('Error fetching DeepSeek models from API:', error);
             throw error;
         }
     }
@@ -230,7 +228,7 @@ let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAi
             }
         }
         catch (error) {
-            console.error('Error generating response from DeepSeek:', error);
+            logger_1.logger.error('Error generating response from DeepSeek:', error);
             throw error;
         }
     }
@@ -245,12 +243,12 @@ let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAi
         const cachedTokens = usage.cachedInputTokens || 0;
         const uncachedInputTokens = totalInputTokens - cachedTokens;
         const outputTokens = usage.outputTokens || 0;
-        let modelInfo = model_prices_1.default[agent.modelName];
+        let modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (!modelInfo && !agent.modelName.startsWith('deepseek/')) {
-            modelInfo = model_prices_1.default[`deepseek/${agent.modelName}`];
+            modelInfo = (0, model_prices_1.getModelPrices)()[`deepseek/${agent.modelName}`];
         }
         if (!modelInfo) {
-            console.error(`No price info found for model ${agent.modelName} or deepseek/${agent.modelName}`);
+            logger_1.logger.error(`No price info found for model ${agent.modelName} or deepseek/${agent.modelName}`);
             return;
         }
         const uncachedCost = uncachedInputTokens * modelInfo.input_cost_per_token;
@@ -263,9 +261,9 @@ let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAi
         agent.totalOutputTokens += outputTokens;
         agent.totalCost += cost;
         agent.totalMarkup += markup;
-        console.log(`Cost for ${agent.agentName} message: ${cost} (markup: ${markup.toFixed(4)})`);
+        logger_1.logger.log(`Cost for ${agent.agentName} message: ${cost} (markup: ${markup.toFixed(4)})`);
         if (cachedTokens > 0) {
-            console.log(`Cache metrics for ${agent.agentName}: cached=${cachedTokens}, uncached=${uncachedInputTokens}`);
+            logger_1.logger.log(`Cache metrics for ${agent.agentName}: cached=${cachedTokens}, uncached=${uncachedInputTokens}`);
         }
         if (agent.balanceService && cost > 0) {
             agent.balanceService.decrementBalance(cost, markup);

package/dist/server/src/ai/LLM/fal.service.js

@@ -8,31 +8,28 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 var __metadata = (this && this.__metadata) || function (k, v) {
     if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.FalAiService = void 0;
 const common_1 = require("@nestjs/common");
 const client_1 = require("@fal-ai/client");
 const base_ai_service_1 = require("./base-ai.service");
-const model_prices_1 = __importDefault(require("../model_prices"));
+const model_prices_1 = require("../model_prices");
 const errors_1 = require("../../common/errors");
+const logger_1 = require("../../common/logger");
 let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
     constructor() {
         super();
         try {
             const apiKey = process.env['FAL_API_KEY'];
             if (!apiKey) {
-                console.warn('FAL_API_KEY environment variable not set. Fal AI service will not be available.');
                 this._isAvailable = false;
                 return;
             }
             client_1.fal.config({ credentials: apiKey });
-            console.log('Fal AI service initialized successfully');
+            logger_1.logger.debug('FalAI service initialized successfully');
         }
         catch (error) {
-            console.error('Error initializing Fal AI service:', error);
+            logger_1.logger.error('Error initializing Fal AI service:', error);
             this._isAvailable = false;
         }
     }
@@ -46,7 +43,6 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
         try {
             const apiKey = process.env['FAL_API_KEY'];
             if (!apiKey) {
-                console.warn('FAL_API_KEY not set, cannot fetch models');
                 return [];
             }
             const response = await fetch('https://api.fal.ai/v1/models', {
@@ -55,7 +51,7 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
                 },
             });
             if (!response.ok) {
-                console.error(`Fal API error: ${response.status} ${response.statusText}`);
+                logger_1.logger.error(`Fal API error: ${response.status} ${response.statusText}`);
                 return [];
             }
             const data = await response.json();
@@ -70,11 +66,11 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
                     });
                 }
             }
-            console.log(`Fetched ${models.length} active models from Fal AI`);
+            logger_1.logger.log(`Fetched ${models.length} active models from Fal AI`);
             return models;
         }
         catch (error) {
-            console.error('Error fetching Fal AI models:', error);
+            logger_1.logger.error('Error fetching Fal AI models:', error);
             return [];
         }
     }
@@ -164,7 +160,7 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
             this.computeMessageCost(agent, usage, estimatedInputTokens, estimatedOutputTokens);
         }
         catch (error) {
-            console.error('Error calling Fal AI:', error);
+            logger_1.logger.error('Error calling Fal AI:', error);
             if (error.message?.includes('authentication') ||
                 error.message?.includes('unauthorized')) {
                 throw new Error('Fal AI authentication failed. Check FAL_API_KEY.');
@@ -183,8 +179,8 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
                 return JSON.parse(responseText);
             }
             catch (parseError) {
-                console.error('Failed to parse JSON response:', parseError);
-                console.error('Response text:', responseText);
+                logger_1.logger.error('Failed to parse JSON response:', parseError);
+                logger_1.logger.error('Response text:', responseText);
                 throw new Error('Failed to parse JSON response from Fal AI');
             }
         }
@@ -193,12 +189,12 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
     computeMessageCost(agent, usage, estimatedInputTokens, estimatedOutputTokens) {
         const inputTokens = usage?.prompt_tokens || estimatedInputTokens || 0;
         const outputTokens = usage?.completion_tokens || estimatedOutputTokens || 0;
-        let modelInfo = model_prices_1.default[agent.modelName];
+        let modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (!modelInfo && !agent.modelName.startsWith('fal_ai/')) {
-            modelInfo = model_prices_1.default[`fal_ai/${agent.modelName}`];
+            modelInfo = (0, model_prices_1.getModelPrices)()[`fal_ai/${agent.modelName}`];
         }
         if (!modelInfo) {
-            console.warn(`No price info found for model ${agent.modelName}. Skipping cost tracking.`);
+            logger_1.logger.warn(`No price info found for model ${agent.modelName}. Skipping cost tracking.`);
             return;
         }
         const inputCost = inputTokens * (modelInfo.input_cost_per_token || 0);
@@ -210,29 +206,29 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
         agent.totalCost += cost;
         agent.totalMarkup += markup;
         if (!usage?.prompt_tokens) {
-            console.log(`Cost for ${agent.agentName} message (estimated): ${cost.toFixed(4)} (markup: ${markup.toFixed(4)})`);
+            logger_1.logger.log(`Cost for ${agent.agentName} message (estimated): ${cost.toFixed(4)} (markup: ${markup.toFixed(4)})`);
         }
         else {
-            console.log(`Cost for ${agent.agentName} message: ${cost.toFixed(4)} (markup: ${markup.toFixed(4)})`);
+            logger_1.logger.log(`Cost for ${agent.agentName} message: ${cost.toFixed(4)} (markup: ${markup.toFixed(4)})`);
         }
         if (agent.balanceService && cost > 0) {
             agent.balanceService.decrementBalance(cost, markup);
         }
     }
     computeImageCost(agent) {
-        let modelInfo = model_prices_1.default[agent.modelName];
+        let modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (!modelInfo && !agent.modelName.startsWith('fal_ai/')) {
-            modelInfo = model_prices_1.default[`fal_ai/${agent.modelName}`];
+            modelInfo = (0, model_prices_1.getModelPrices)()[`fal_ai/${agent.modelName}`];
         }
         if (!modelInfo || !modelInfo.output_cost_per_image) {
-            console.warn(`No image price info found for ${agent.modelName}. Skipping cost tracking.`);
+            logger_1.logger.warn(`No image price info found for ${agent.modelName}. Skipping cost tracking.`);
             return;
         }
         const cost = modelInfo.output_cost_per_image;
         const markup = 0;
         agent.totalCost += cost;
         agent.totalMarkup += markup;
-        console.log(`Cost for image generation: ${cost.toFixed(4)} (markup: ${markup.toFixed(4)})`);
+        logger_1.logger.log(`Cost for image generation: ${cost.toFixed(4)} (markup: ${markup.toFixed(4)})`);
         if (agent.balanceService && cost > 0) {
             agent.balanceService.decrementBalance(cost, markup);
         }
@@ -302,7 +298,7 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
             return { artifactId };
         }
         catch (error) {
-            console.error('Error generating image with Fal AI:', error.message || error);
+            logger_1.logger.error('Error generating image with Fal AI:', error.message || error);
             return { error };
         }
     }

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

@@ -8,16 +8,14 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 var __metadata = (this && this.__metadata) || function (k, v) {
     if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GoogleAiService = void 0;
 const common_1 = require("@nestjs/common");
 const genai_1 = require("@google/genai");
 const base_ai_service_1 = require("./base-ai.service");
-const model_prices_1 = __importDefault(require("../model_prices"));
+const model_prices_1 = require("../model_prices");
 const errors_1 = require("../../common/errors");
+const logger_1 = require("../../common/logger");
 let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiService {
     googleAI = null;
     constructor() {
@@ -25,14 +23,14 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
         try {
             const apiKey = process.env['GOOGLE_GENAI_API_KEY'];
             if (!apiKey) {
-                console.warn('GOOGLE_GENAI_API_KEY environment variable not set. Google AI service will not be available.');
                 this._isAvailable = false;
                 return;
             }
             this.googleAI = new genai_1.GoogleGenAI({ apiKey });
+            logger_1.logger.debug('Google AI service initialized successfully');
         }
         catch (error) {
-            console.error('Error initializing Google AI service:', error);
+            logger_1.logger.error('Error initializing Google AI service:', error);
             this._isAvailable = false;
         }
     }
@@ -59,7 +57,7 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
             });
         }
         catch (error) {
-            console.warn('Could not fetch models from Google AI API, using fallback list:', error.message);
+            logger_1.logger.warn('Could not fetch models from Google AI API, using fallback list:', error.message);
         }
     }
     addToConversationHistory(agent, message, fromSelf = false) {
@@ -143,7 +141,7 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
             }
         }
         catch (error) {
-            console.error('Error calling Google AI:', error);
+            logger_1.logger.error('Error calling Google AI:', error);
             throw error;
         }
         if (saveResponse) {
@@ -205,16 +203,16 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
             return { artifactId };
         }
         catch (error) {
-            console.error('Error generating image with Google AI:', error);
+            logger_1.logger.error('Error generating image with Google AI:', error);
             return { error };
         }
     }
     computeMessageCost(agent, usage) {
         const inputTokens = usage.promptTokenCount || 0;
         const outputTokens = usage.candidatesTokenCount || 0;
-        const modelInfo = model_prices_1.default[agent.modelName];
+        const modelInfo = (0, model_prices_1.getModelPrices)()[agent.modelName];
         if (!modelInfo) {
-            console.error(`No price info found for model ${agent.modelName}`);
+            logger_1.logger.error(`No price info found for model ${agent.modelName}`);
             return;
         }
         const inputCost = inputTokens * modelInfo.input_cost_per_token;
@@ -225,15 +223,15 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
         agent.totalOutputTokens += outputTokens;
         agent.totalCost += cost;
         agent.totalMarkup += markup;
-        console.log(`Cost for ${agent.agentName} message: ${cost} (markup: ${markup.toFixed(4)})`);
+        logger_1.logger.log(`Cost for ${agent.agentName} message: ${cost} (markup: ${markup.toFixed(4)})`);
         if (agent.balanceService && cost > 0) {
             agent.balanceService.decrementBalance(cost, markup);
         }
     }
     computeImageCost(agent, modelName) {
-        const modelInfo = model_prices_1.default[modelName];
+        const modelInfo = (0, model_prices_1.getModelPrices)()[modelName];
         if (!modelInfo || !modelInfo.output_cost_per_image) {
-            console.error(`No image price info found for model ${modelName}`);
+            logger_1.logger.error(`No image price info found for model ${modelName}`);
             return;
         }
         const outputCost = modelInfo.output_cost_per_image;
@@ -241,7 +239,7 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
         const cost = outputCost + markup;
         agent.totalCost += cost;
         agent.totalMarkup += markup;
-        console.log(`Cost for image generation: ${cost} (markup: ${markup.toFixed(4)})`);
+        logger_1.logger.log(`Cost for image generation: ${cost} (markup: ${markup.toFixed(4)})`);
         if (agent.balanceService && cost > 0) {
             agent.balanceService.decrementBalance(cost, markup);
         }