briyah 1.0.7 → 1.0.9

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');
@@ -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

@@ -15,6 +15,7 @@ const sdk_1 = require("@anthropic-ai/sdk");
 const base_ai_service_1 = require("./base-ai.service");
 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() {
@@ -30,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;
         }
     }
@@ -70,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) {
@@ -241,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({
@@ -279,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));
                 }
@@ -290,7 +298,7 @@ let AnthropicAiService = class AnthropicAiService extends base_ai_service_1.Base
     computeMessageCost(agent, usage) {
         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;
@@ -309,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

@@ -17,6 +17,7 @@ const base_ai_service_1 = require("./base-ai.service");
 const zod_1 = require("zod");
 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() {
@@ -30,9 +31,10 @@ let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAi
             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;
         }
     }
@@ -66,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;
         }
     }
@@ -226,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;
         }
     }
@@ -246,7 +248,7 @@ let DeepSeekAiService = class DeepSeekAiService extends base_ai_service_1.BaseAi
             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;
@@ -259,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

@@ -15,6 +15,7 @@ const client_1 = require("@fal-ai/client");
 const base_ai_service_1 = require("./base-ai.service");
 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();
@@ -25,10 +26,10 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
                 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;
         }
     }
@@ -50,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();
@@ -65,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 [];
         }
     }
@@ -159,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.');
@@ -178,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,7 +194,7 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
             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);
@@ -205,10 +206,10 @@ 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);
@@ -220,14 +221,14 @@ let FalAiService = class FalAiService extends base_ai_service_1.BaseAiService {
             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);
         }
@@ -297,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

@@ -15,6 +15,7 @@ const genai_1 = require("@google/genai");
 const base_ai_service_1 = require("./base-ai.service");
 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() {
@@ -26,9 +27,10 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
                 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;
         }
     }
@@ -55,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) {
@@ -139,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) {
@@ -201,7 +203,7 @@ 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 };
         }
     }
@@ -210,7 +212,7 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
         const outputTokens = usage.candidatesTokenCount || 0;
         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;
@@ -221,7 +223,7 @@ 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);
         }
@@ -229,7 +231,7 @@ let GoogleAiService = class GoogleAiService extends base_ai_service_1.BaseAiServ
     computeImageCost(agent, 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;
@@ -237,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);
         }

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

@@ -16,6 +16,7 @@ const openai_2 = require("openai");
 const base_ai_service_1 = require("./base-ai.service");
 const model_prices_1 = require("../model_prices");
 const errors_1 = require("../../common/errors");
+const logger_1 = require("../../common/logger");
 let GrokAiService = class GrokAiService extends base_ai_service_1.BaseAiService {
     openai;
     constructor() {
@@ -33,9 +34,10 @@ let GrokAiService = class GrokAiService extends base_ai_service_1.BaseAiService
                 maxRetries: 3,
             };
             this.openai = new openai_1.OpenAI(options);
+            logger_1.logger.debug('Grok service initialized successfully');
         }
         catch (error) {
-            console.error('Error initializing Grok service:', error);
+            logger_1.logger.error('Error initializing Grok service:', error);
             this._isAvailable = false;
         }
     }
@@ -92,10 +94,10 @@ let GrokAiService = class GrokAiService extends base_ai_service_1.BaseAiService
                 metadata.serviceMetadata?.resourceId) {
                 try {
                     await this.openai.files.delete(metadata.serviceMetadata.resourceId);
-                    console.log(`Deleted Grok file ${metadata.serviceMetadata.resourceId} for agent ${agent.id}`);
+                    logger_1.logger.log(`Deleted Grok file ${metadata.serviceMetadata.resourceId} for agent ${agent.id}`);
                 }
                 catch (error) {
-                    console.error(`Error deleting Grok file ${metadata.serviceMetadata.resourceId}:`, error);
+                    logger_1.logger.error(`Error deleting Grok file ${metadata.serviceMetadata.resourceId}:`, error);
                 }
             }
         }
@@ -194,7 +196,7 @@ let GrokAiService = class GrokAiService extends base_ai_service_1.BaseAiService
             }
         }
         catch (error) {
-            console.error(error);
+            logger_1.logger.error(error);
             throw error;
         }
         if (saveResponse) {
@@ -230,7 +232,7 @@ let GrokAiService = class GrokAiService extends base_ai_service_1.BaseAiService
             modelInfo = (0, model_prices_1.getModelPrices)()[`xai/${agent.modelName}`];
         }
         if (!modelInfo) {
-            console.error(`No price info found for model ${agent.modelName} or xai/${agent.modelName}`);
+            logger_1.logger.error(`No price info found for model ${agent.modelName} or xai/${agent.modelName}`);
             return;
         }
         const uncachedCost = uncachedPromptTokens * modelInfo.input_cost_per_token;
@@ -243,9 +245,9 @@ let GrokAiService = class GrokAiService extends base_ai_service_1.BaseAiService
         agent.totalOutputTokens += completionTokens;
         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=${uncachedPromptTokens}`);
+            logger_1.logger.log(`Cache metrics for ${agent.agentName}: cached=${cachedTokens}, uncached=${uncachedPromptTokens}`);
         }
         if (agent.balanceService && cost > 0) {
             agent.balanceService.decrementBalance(cost, markup);

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

@@ -47,6 +47,7 @@ const common_1 = require("@nestjs/common");
 const base_ai_service_1 = require("./base-ai.service");
 const fs = __importStar(require("fs"));
 const model_prices_1 = require("../model_prices");
+const logger_1 = require("../../common/logger");
 let MockAiService = class MockAiService extends base_ai_service_1.BaseAiService {
     constructor() {
         super();
@@ -69,7 +70,7 @@ let MockAiService = class MockAiService extends base_ai_service_1.BaseAiService
     async preparedPrompt(agent, promptFileName, variables, saveResponse = true, _cacheConfig = null) {
         const mockFile = this.findPromptFile(agent, promptFileName, '.mock');
         if (!mockFile) {
-            console.warn(`Mock file '${promptFileName}.mock' not found for agent '${agent.agentName}'. Using generic response.`);
+            logger_1.logger.warn(`Mock file '${promptFileName}.mock' not found for agent '${agent.agentName}'. Using generic response.`);
             const preparedPrompt = this.getPreparedPrompt(agent, promptFileName, variables);
             const jsonSchema = this.getPreparedSchema(agent, promptFileName);
             return this.textPrompt(agent, preparedPrompt, jsonSchema, saveResponse);
@@ -88,7 +89,7 @@ let MockAiService = class MockAiService extends base_ai_service_1.BaseAiService
                 return parsedResponse;
             }
             catch (error) {
-                console.error(`Error parsing mock JSON response from ${mockFile}:`, error);
+                logger_1.logger.error(`Error parsing mock JSON response from ${mockFile}:`, error);
                 throw new Error(`Invalid JSON in mock file: ${mockFile}`);
             }
         }
@@ -122,7 +123,7 @@ let MockAiService = class MockAiService extends base_ai_service_1.BaseAiService
         const mockFileBaseName = this.getMockFile(promptFileName, prompt, variables);
         const mockFile = this.findPromptFile(agent, mockFileBaseName, '.mock');
         if (!mockFile) {
-            console.warn(`Mock file '${mockFile}' not found for agent '${agent.agentName}'. Using generic response.`);
+            logger_1.logger.warn(`Mock file '${mockFile}' not found for agent '${agent.agentName}'. Using generic response.`);
             return this.textPrompt(agent, prompt || 'Mock prompt', null, saveResponse, cacheConfig);
         }
         const mockResponse = fs.readFileSync(mockFile).toString('utf8');
@@ -138,7 +139,7 @@ let MockAiService = class MockAiService extends base_ai_service_1.BaseAiService
                 return parsedResponse;
             }
             catch (error) {
-                console.error(`Error parsing mock JSON response from ${mockFile}:`, error);
+                logger_1.logger.error(`Error parsing mock JSON response from ${mockFile}:`, error);
                 throw new Error(`Invalid JSON in mock file: ${mockFile}`);
             }
         }
@@ -167,7 +168,7 @@ let MockAiService = class MockAiService extends base_ai_service_1.BaseAiService
         const outputTokens = Math.max(50, Math.floor(response.length / 4));
         const modelInfo = (0, model_prices_1.getModelPrices)()['mock-fast-v1'];
         if (!modelInfo) {
-            console.warn('Mock model pricing not found in model_prices.ts');
+            logger_1.logger.warn('Mock model pricing not found in model_prices.ts');
             return;
         }
         const inputCost = inputTokens * modelInfo.input_cost_per_token;
@@ -178,7 +179,7 @@ let MockAiService = class MockAiService extends base_ai_service_1.BaseAiService
         agent.totalInputTokens += inputTokens;
         agent.totalOutputTokens += outputTokens;
         agent.totalMarkup += markup;
-        console.log(`Cost for ${agent.agentName} message: ${totalCost} (markup: ${markup.toFixed(4)})`);
+        logger_1.logger.log(`Cost for ${agent.agentName} message: ${totalCost} (markup: ${markup.toFixed(4)})`);
         if (agent.balanceService) {
             await agent.balanceService.decrementBalance(totalCost, markup);
         }