berget 1.2.0 → 1.3.0

package/dist/src/services/chat-service.js

@@ -42,13 +42,10 @@ var __rest = (this && this.__rest) || function (s, e) {
         }
     return t;
 };
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatService = void 0;
 const client_1 = require("../client");
-const chalk_1 = __importDefault(require("chalk"));
+const logger_1 = require("../utils/logger");
 /**
  * Service for interacting with the chat API
  * Command group: chat
@@ -70,154 +67,95 @@ class ChatService {
     createCompletion(options) {
         return __awaiter(this, void 0, void 0, function* () {
             try {
-                console.log(chalk_1.default.yellow('DEBUG: Starting createCompletion method'));
-                // Check if options is defined
-                if (!options) {
-                    console.log(chalk_1.default.red('ERROR: options is undefined'));
-                    throw new Error('Chat completion options are undefined');
+                logger_1.logger.debug('Starting createCompletion method');
+                // Initialize options if undefined
+                const optionsCopy = options ? Object.assign({}, options) : { messages: [] };
+                // Check if messages are defined
+                if (!optionsCopy.messages || !Array.isArray(optionsCopy.messages)) {
+                    logger_1.logger.error('messages is undefined or not an array');
+                    optionsCopy.messages = [];
                 }
-                // Log the raw options object
-                console.log(chalk_1.default.yellow('DEBUG: Raw options:'), typeof options, options ? 'defined' : 'undefined');
-                const headers = {};
-                // Check if debug is enabled
-                const isDebug = process.argv.includes('--debug');
-                if (isDebug) {
-                    console.log(chalk_1.default.yellow('DEBUG: Starting createCompletion with options:'));
-                    try {
-                        console.log(chalk_1.default.yellow(JSON.stringify(Object.assign(Object.assign({}, options), { apiKey: options.apiKey ? '***' : undefined, messages: options.messages ? `${options.messages.length} messages` : undefined }), null, 2)));
-                    }
-                    catch (error) {
-                        console.log(chalk_1.default.red('ERROR: Failed to stringify options:'), error);
-                    }
+                // Log the options object
+                logger_1.logger.debug('Starting createCompletion with options:');
+                try {
+                    logger_1.logger.debug(JSON.stringify(Object.assign(Object.assign({}, optionsCopy), { apiKey: optionsCopy.apiKey ? '***' : undefined, messages: optionsCopy.messages
+                            ? `${optionsCopy.messages.length} messages`
+                            : '0 messages' }), null, 2));
                 }
-                // Create a copy of options to avoid modifying the original
-                const optionsCopy = Object.assign({}, options);
-                if (isDebug) {
-                    console.log(chalk_1.default.yellow('DEBUG: Checking for API key'));
-                    console.log(chalk_1.default.yellow(`DEBUG: optionsCopy.apiKey exists: ${!!optionsCopy.apiKey}`));
+                catch (error) {
+                    logger_1.logger.error('Failed to stringify options:', error);
                 }
+                const headers = {};
                 // Check for environment variables first - prioritize this over everything else
                 const envApiKey = process.env.BERGET_API_KEY;
                 if (envApiKey) {
-                    if (isDebug) {
-                        console.log(chalk_1.default.yellow('DEBUG: Using API key from BERGET_API_KEY environment variable'));
-                    }
+                    logger_1.logger.debug('Using API key from BERGET_API_KEY environment variable');
                     optionsCopy.apiKey = envApiKey;
+                    // Skip the default API key logic if we already have a key
+                    return this.executeCompletion(optionsCopy, headers);
+                }
+                // If API key is already provided, use it directly
+                else if (optionsCopy.apiKey) {
+                    logger_1.logger.debug('Using API key provided in options');
+                    // Skip the default API key logic if we already have a key
+                    return this.executeCompletion(optionsCopy, headers);
                 }
                 // Only try to get the default API key if no API key is provided and no env var is set
-                else if (!optionsCopy.apiKey) {
-                    if (isDebug) {
-                        console.log(chalk_1.default.yellow('DEBUG: No API key provided, trying to get default'));
-                    }
+                else {
+                    logger_1.logger.debug('No API key provided, trying to get default');
                     try {
                         // Import the DefaultApiKeyManager directly
-                        if (isDebug) {
-                            console.log(chalk_1.default.yellow('DEBUG: Importing DefaultApiKeyManager'));
-                        }
+                        logger_1.logger.debug('Importing DefaultApiKeyManager');
                         const DefaultApiKeyManager = (yield Promise.resolve().then(() => __importStar(require('../utils/default-api-key')))).DefaultApiKeyManager;
                         const defaultApiKeyManager = DefaultApiKeyManager.getInstance();
-                        if (isDebug) {
-                            console.log(chalk_1.default.yellow('DEBUG: Got DefaultApiKeyManager instance'));
-                        }
+                        logger_1.logger.debug('Got DefaultApiKeyManager instance');
                         // Try to get the default API key
-                        if (isDebug) {
-                            console.log(chalk_1.default.yellow('DEBUG: Calling promptForDefaultApiKey'));
-                        }
+                        logger_1.logger.debug('Calling promptForDefaultApiKey');
                         const defaultApiKeyData = defaultApiKeyManager.getDefaultApiKeyData();
-                        const apiKey = (defaultApiKeyData === null || defaultApiKeyData === void 0 ? void 0 : defaultApiKeyData.key) || (yield defaultApiKeyManager.promptForDefaultApiKey());
-                        if (isDebug) {
-                            console.log(chalk_1.default.yellow(`DEBUG: Default API key data exists: ${!!defaultApiKeyData}`));
-                            console.log(chalk_1.default.yellow(`DEBUG: promptForDefaultApiKey returned: ${apiKey ? 'a key' : 'null'}`));
-                        }
+                        const apiKey = (defaultApiKeyData === null || defaultApiKeyData === void 0 ? void 0 : defaultApiKeyData.key) ||
+                            (yield defaultApiKeyManager.promptForDefaultApiKey());
+                        logger_1.logger.debug(`Default API key data exists: ${!!defaultApiKeyData}`);
+                        logger_1.logger.debug(`promptForDefaultApiKey returned: ${apiKey ? 'a key' : 'null'}`);
                         if (apiKey) {
-                            if (isDebug) {
-                                console.log(chalk_1.default.yellow('DEBUG: Using API key from default API key manager'));
-                            }
+                            logger_1.logger.debug('Using API key from default API key manager');
                             optionsCopy.apiKey = apiKey;
                         }
                         else {
-                            console.log(chalk_1.default.yellow('No API key available. You need to either:'));
-                            console.log(chalk_1.default.yellow('1. Create an API key with: berget api-keys create --name "My Key"'));
-                            console.log(chalk_1.default.yellow('2. Set a default API key with: berget api-keys set-default <id>'));
-                            console.log(chalk_1.default.yellow('3. Provide an API key with the --api-key option'));
-                            console.log(chalk_1.default.yellow('4. Set the BERGET_API_KEY environment variable'));
-                            console.log(chalk_1.default.yellow('\nExample:'));
-                            console.log(chalk_1.default.yellow('  export BERGET_API_KEY=your_api_key_here'));
-                            console.log(chalk_1.default.yellow('  # or for a single command:'));
-                            console.log(chalk_1.default.yellow('  BERGET_API_KEY=your_api_key_here berget chat run google/gemma-3-27b-it'));
+                            logger_1.logger.warn('No API key available. You need to either:');
+                            logger_1.logger.warn('1. Create an API key with: berget api-keys create --name "My Key"');
+                            logger_1.logger.warn('2. Set a default API key with: berget api-keys set-default <id>');
+                            logger_1.logger.warn('3. Provide an API key with the --api-key option');
+                            logger_1.logger.warn('4. Set the BERGET_API_KEY environment variable');
+                            logger_1.logger.warn('\nExample:');
+                            logger_1.logger.warn('  export BERGET_API_KEY=your_api_key_here');
+                            logger_1.logger.warn('  # or for a single command:');
+                            logger_1.logger.warn('  BERGET_API_KEY=your_api_key_here berget chat run google/gemma-3-27b-it');
                             throw new Error('No API key provided and no default API key set');
                         }
                         // Set the API key in the options
-                        if (isDebug) {
-                            console.log(chalk_1.default.yellow('DEBUG: Setting API key in options'));
-                        }
+                        logger_1.logger.debug('Setting API key in options');
                         // Only set the API key if it's not null
                         if (apiKey) {
                             optionsCopy.apiKey = apiKey;
                         }
                     }
                     catch (error) {
-                        console.log(chalk_1.default.red('Error getting API key:'));
+                        logger_1.logger.error('Error getting API key:');
                         if (error instanceof Error) {
-                            console.log(chalk_1.default.red(error.message));
+                            logger_1.logger.error(error.message);
                         }
-                        console.log(chalk_1.default.yellow('Please create an API key with: berget api-keys create --name "My Key"'));
+                        logger_1.logger.warn('Please create an API key with: berget api-keys create --name "My Key"');
                         throw new Error('Failed to get API key');
                     }
                 }
-                if (isDebug) {
-                    console.log(chalk_1.default.yellow('DEBUG: Chat completion options:'));
-                    console.log(chalk_1.default.yellow(JSON.stringify(Object.assign(Object.assign({}, optionsCopy), { apiKey: optionsCopy.apiKey ? '***' : undefined // Hide the actual API key in debug output
-                     }), null, 2)));
-                }
-                // If an API key is provided, use it for this request
-                if (optionsCopy.apiKey) {
-                    headers['Authorization'] = `Bearer ${optionsCopy.apiKey}`;
-                    // Remove apiKey from options before sending to API
-                    const { apiKey } = optionsCopy, requestOptions = __rest(optionsCopy, ["apiKey"]);
-                    if (isDebug) {
-                        console.log(chalk_1.default.yellow('DEBUG: Using provided API key'));
-                        console.log(chalk_1.default.yellow('DEBUG: Request options:'));
-                        console.log(chalk_1.default.yellow(JSON.stringify(requestOptions, null, 2)));
-                    }
-                    try {
-                        const response = yield this.client.POST('/v1/chat/completions', {
-                            body: requestOptions,
-                            headers
-                        });
-                        // Check if response has an error property
-                        const responseAny = response;
-                        if (responseAny && responseAny.error)
-                            throw new Error(JSON.stringify(responseAny.error));
-                        if (isDebug) {
-                            console.log(chalk_1.default.yellow('DEBUG: API response:'));
-                            console.log(chalk_1.default.yellow(JSON.stringify(response, null, 2)));
-                            // Output the complete response data for debugging
-                            console.log(chalk_1.default.yellow('DEBUG: Complete response data:'));
-                            console.log(chalk_1.default.yellow(JSON.stringify(response.data, null, 2)));
-                        }
-                        return response.data;
-                    }
-                    catch (requestError) {
-                        if (process.argv.includes('--debug')) {
-                            console.log(chalk_1.default.red(`DEBUG: Request error: ${requestError instanceof Error ? requestError.message : String(requestError)}`));
-                        }
-                        throw requestError;
-                    }
-                }
-                else {
-                    // We've exhausted all options for getting an API key
-                    console.log(chalk_1.default.yellow('No API key available. You need to either:'));
-                    console.log(chalk_1.default.yellow('1. Create an API key with: berget api-keys create --name "My Key"'));
-                    console.log(chalk_1.default.yellow('2. Set a default API key with: berget api-keys set-default <id>'));
-                    console.log(chalk_1.default.yellow('3. Provide an API key with the --api-key option'));
-                    console.log(chalk_1.default.yellow('4. Set the BERGET_API_KEY environment variable'));
-                    console.log(chalk_1.default.yellow('\nExample:'));
-                    console.log(chalk_1.default.yellow('  export BERGET_API_KEY=your_api_key_here'));
-                    console.log(chalk_1.default.yellow('  # or for a single command:'));
-                    console.log(chalk_1.default.yellow('  BERGET_API_KEY=your_api_key_here berget chat run google/gemma-3-27b-it'));
-                    throw new Error('No API key available. Please provide an API key or set a default API key.');
+                // Set default model if not provided
+                if (!optionsCopy.model) {
+                    logger_1.logger.debug('No model specified, using default: google/gemma-3-27b-it');
+                    optionsCopy.model = 'google/gemma-3-27b-it';
                 }
+                logger_1.logger.debug('Chat completion options:');
+                logger_1.logger.debug(JSON.stringify(Object.assign(Object.assign({}, optionsCopy), { apiKey: optionsCopy.apiKey ? '***' : undefined }), null, 2));
+                return this.executeCompletion(optionsCopy, headers);
             }
             catch (error) {
                 // Improved error handling
@@ -235,11 +173,178 @@ class ChatService {
                         errorMessage = `Chat error: ${error.message}`;
                     }
                 }
-                console.error(chalk_1.default.red(errorMessage));
+                logger_1.logger.error(errorMessage);
                 throw new Error(errorMessage);
             }
         });
     }
+    /**
+     * Execute the completion request with the provided options
+     * @param options The completion options
+     * @param headers Additional headers to include
+     * @returns The completion response
+     */
+    executeCompletion(options, headers = {}) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                // If an API key is provided, use it for this request
+                if (options.apiKey) {
+                    // API keys should be sent directly, not with Bearer prefix
+                    headers['Authorization'] = options.apiKey;
+                }
+                // Remove apiKey and onChunk from options before sending to API
+                const { apiKey, onChunk } = options, requestOptions = __rest(options, ["apiKey", "onChunk"]);
+                logger_1.logger.debug('Request options:');
+                logger_1.logger.debug(JSON.stringify(Object.assign(Object.assign({}, requestOptions), { messages: requestOptions.messages
+                        ? `${requestOptions.messages.length} messages`
+                        : '0 messages' }), null, 2));
+                // Handle streaming responses differently
+                if (requestOptions.stream && onChunk) {
+                    return yield this.handleStreamingResponse(Object.assign(Object.assign({}, requestOptions), { onChunk }), headers);
+                }
+                else {
+                    // Ensure model is always defined for the API call
+                    const requestBody = Object.assign(Object.assign({}, requestOptions), { model: requestOptions.model || 'google/gemma-3-27b-it' });
+                    // Debug the headers being sent
+                    logger_1.logger.debug('Headers being sent:');
+                    logger_1.logger.debug(JSON.stringify(headers, null, 2));
+                    const response = yield this.client.POST('/v1/chat/completions', {
+                        body: requestBody,
+                        headers,
+                    });
+                    // Check if response has an error property
+                    const responseAny = response;
+                    if (responseAny && responseAny.error)
+                        throw new Error(JSON.stringify(responseAny.error));
+                    logger_1.logger.debug('API response:');
+                    logger_1.logger.debug(JSON.stringify(response, null, 2));
+                    // Output the complete response data for debugging
+                    logger_1.logger.debug('Complete response data:');
+                    logger_1.logger.debug(JSON.stringify(response.data, null, 2));
+                    return response.data;
+                }
+            }
+            catch (requestError) {
+                logger_1.logger.debug(`Request error: ${requestError instanceof Error
+                    ? requestError.message
+                    : String(requestError)}`);
+                throw requestError;
+            }
+        });
+    }
+    /**
+     * Handle the case when no API key is available
+     */
+    handleNoApiKey() {
+        // We've exhausted all options for getting an API key
+        logger_1.logger.warn('No API key available. You need to either:');
+        logger_1.logger.warn('1. Create an API key with: berget api-keys create --name "My Key"');
+        logger_1.logger.warn('2. Set a default API key with: berget api-keys set-default <id>');
+        logger_1.logger.warn('3. Provide an API key with the --api-key option');
+        logger_1.logger.warn('4. Set the BERGET_API_KEY environment variable');
+        logger_1.logger.warn('\nExample:');
+        logger_1.logger.warn('  export BERGET_API_KEY=your_api_key_here');
+        logger_1.logger.warn('  # or for a single command:');
+        logger_1.logger.warn('  BERGET_API_KEY=your_api_key_here berget chat run google/gemma-3-27b-it');
+        throw new Error('No API key available. Please provide an API key or set a default API key.');
+    }
+    /**
+     * Handle streaming response from the API
+     * @param options Request options
+     * @param headers Request headers
+     * @returns A promise that resolves when the stream is complete
+     */
+    handleStreamingResponse(options, headers) {
+        return __awaiter(this, void 0, void 0, function* () {
+            logger_1.logger.debug('Handling streaming response');
+            // Create URL with query parameters
+            const url = new URL(`${client_1.API_BASE_URL}/v1/chat/completions`);
+            // Debug the headers and options
+            logger_1.logger.debug('Streaming headers:');
+            logger_1.logger.debug(JSON.stringify(headers, null, 2));
+            logger_1.logger.debug('Streaming options:');
+            logger_1.logger.debug(JSON.stringify(Object.assign(Object.assign({}, options), { onChunk: options.onChunk ? 'function present' : 'no function' }), null, 2));
+            try {
+                // Make fetch request directly to handle streaming
+                const response = yield fetch(url.toString(), {
+                    method: 'POST',
+                    headers: Object.assign({ 'Content-Type': 'application/json', Accept: 'text/event-stream' }, headers),
+                    body: JSON.stringify(options),
+                });
+                if (!response.ok) {
+                    const errorText = yield response.text();
+                    logger_1.logger.error(`Stream request failed: ${response.status} ${response.statusText}`);
+                    logger_1.logger.debug(`Error response: ${errorText}`);
+                    throw new Error(`Stream request failed: ${response.status} ${response.statusText}`);
+                }
+                if (!response.body) {
+                    throw new Error('No response body received');
+                }
+                // Process the stream
+                const reader = response.body.getReader();
+                const decoder = new TextDecoder();
+                let fullContent = '';
+                let fullResponse = null;
+                while (true) {
+                    const { done, value } = yield reader.read();
+                    if (done)
+                        break;
+                    const chunk = decoder.decode(value, { stream: true });
+                    logger_1.logger.debug(`Received chunk: ${chunk.length} bytes`);
+                    // Process the chunk - it may contain multiple SSE events
+                    const lines = chunk.split('\n');
+                    for (const line of lines) {
+                        if (line.startsWith('data:')) {
+                            const jsonData = line.slice(5).trim();
+                            // Skip empty data or [DONE] marker
+                            if (jsonData === '' || jsonData === '[DONE]')
+                                continue;
+                            try {
+                                const parsedData = JSON.parse(jsonData);
+                                // Call the onChunk callback with the parsed data
+                                if (options.onChunk) {
+                                    options.onChunk(parsedData);
+                                }
+                                // Keep track of the full response
+                                if (!fullResponse) {
+                                    fullResponse = parsedData;
+                                }
+                                else if (parsedData.choices &&
+                                    parsedData.choices[0] &&
+                                    parsedData.choices[0].delta) {
+                                    // Accumulate content for the full response
+                                    if (parsedData.choices[0].delta.content) {
+                                        fullContent += parsedData.choices[0].delta.content;
+                                    }
+                                }
+                            }
+                            catch (e) {
+                                logger_1.logger.error(`Error parsing chunk: ${e}`);
+                                logger_1.logger.debug(`Problematic chunk: ${jsonData}`);
+                            }
+                        }
+                    }
+                }
+                // Construct the final response object similar to non-streaming response
+                if (fullResponse) {
+                    if (fullContent) {
+                        fullResponse.choices[0].message = {
+                            role: 'assistant',
+                            content: fullContent,
+                        };
+                    }
+                    return fullResponse;
+                }
+                return {
+                    choices: [{ message: { role: 'assistant', content: fullContent } }],
+                };
+            }
+            catch (error) {
+                logger_1.logger.error(`Streaming error: ${error instanceof Error ? error.message : String(error)}`);
+                throw error;
+            }
+        });
+    }
     /**
      * List available models
      * Command: berget chat list
@@ -252,7 +357,7 @@ class ChatService {
                 const effectiveApiKey = envApiKey || apiKey;
                 if (effectiveApiKey) {
                     const headers = {
-                        'Authorization': `Bearer ${effectiveApiKey}`
+                        Authorization: effectiveApiKey,
                     };
                     const { data, error } = yield this.client.GET('/v1/models', { headers });
                     if (error)
@@ -274,9 +379,9 @@ class ChatService {
                         // Try to parse the error message as JSON
                         const parsedError = JSON.parse(error.message);
                         if (parsedError.error) {
-                            errorMessage = `Models error: ${typeof parsedError.error === 'string' ?
-                                parsedError.error :
-                                (parsedError.error.message || JSON.stringify(parsedError.error))}`;
+                            errorMessage = `Models error: ${typeof parsedError.error === 'string'
+                                ? parsedError.error
+                                : parsedError.error.message || JSON.stringify(parsedError.error)}`;
                         }
                     }
                     catch (e) {
@@ -284,7 +389,7 @@ class ChatService {
                         errorMessage = `Models error: ${error.message}`;
                     }
                 }
-                console.error(chalk_1.default.red(errorMessage));
+                logger_1.logger.error(errorMessage);
                 throw new Error(errorMessage);
             }
         });
@@ -296,5 +401,5 @@ ChatService.COMMAND_GROUP = 'chat';
 // Subcommands for this service
 ChatService.COMMANDS = {
     RUN: 'run',
-    LIST: 'list'
+    LIST: 'list',
 };

package/dist/src/utils/default-api-key.js

@@ -39,9 +39,9 @@ exports.DefaultApiKeyManager = void 0;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 const os = __importStar(require("os"));
-const chalk_1 = __importDefault(require("chalk"));
 const api_key_service_1 = require("../services/api-key-service");
 const readline_1 = __importDefault(require("readline"));
+const logger_1 = require("./logger");
 /**
  * Manages the default API key for chat commands
  */
@@ -73,9 +73,7 @@ class DefaultApiKeyManager {
             }
         }
         catch (error) {
-            if (process.argv.includes('--debug')) {
-                console.error(chalk_1.default.dim('Failed to load default API key configuration'));
-            }
+            logger_1.logger.debug('Failed to load default API key configuration');
             this.defaultApiKey = null;
         }
     }
@@ -97,9 +95,7 @@ class DefaultApiKeyManager {
             }
         }
         catch (error) {
-            if (process.argv.includes('--debug')) {
-                console.error(chalk_1.default.dim('Failed to save default API key configuration'));
-            }
+            logger_1.logger.debug('Failed to save default API key configuration');
         }
     }
     /**
@@ -136,34 +132,23 @@ class DefaultApiKeyManager {
     promptForDefaultApiKey() {
         return __awaiter(this, void 0, void 0, function* () {
             try {
-                const isDebug = process.argv.includes('--debug');
-                if (isDebug) {
-                    console.log(chalk_1.default.yellow('DEBUG: promptForDefaultApiKey called'));
-                }
+                logger_1.logger.debug('promptForDefaultApiKey called');
                 // If we already have a default API key, return it
                 if (this.defaultApiKey) {
-                    if (isDebug) {
-                        console.log(chalk_1.default.yellow('DEBUG: Using existing default API key'));
-                    }
+                    logger_1.logger.debug('Using existing default API key');
                     return this.defaultApiKey.key;
                 }
-                if (isDebug) {
-                    console.log(chalk_1.default.yellow('DEBUG: No default API key found, getting ApiKeyService'));
-                }
+                logger_1.logger.debug('No default API key found, getting ApiKeyService');
                 const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
                 // Get all API keys
                 let apiKeys;
                 try {
-                    if (isDebug) {
-                        console.log(chalk_1.default.yellow('DEBUG: Calling apiKeyService.list()'));
-                    }
+                    logger_1.logger.debug('Calling apiKeyService.list()');
                     apiKeys = yield apiKeyService.list();
-                    if (isDebug) {
-                        console.log(chalk_1.default.yellow(`DEBUG: Got ${apiKeys ? apiKeys.length : 0} API keys`));
-                    }
+                    logger_1.logger.debug(`Got ${apiKeys ? apiKeys.length : 0} API keys`);
                     if (!apiKeys || apiKeys.length === 0) {
-                        console.log(chalk_1.default.yellow('No API keys found. Create one with:'));
-                        console.log(chalk_1.default.blue('  berget api-keys create --name "My Key"'));
+                        logger_1.logger.warn('No API keys found. Create one with:');
+                        logger_1.logger.info('  berget api-keys create --name "My Key"');
                         return null;
                     }
                 }
@@ -174,24 +159,22 @@ class DefaultApiKeyManager {
                         errorMessage.includes('Authentication failed') ||
                         errorMessage.includes('AUTH_FAILED');
                     if (isAuthError) {
-                        console.log(chalk_1.default.yellow('Authentication required. Please run `berget auth login` first.'));
+                        logger_1.logger.warn('Authentication required. Please run `berget auth login` first.');
                     }
                     else {
-                        console.log(chalk_1.default.red('Error fetching API keys:'));
+                        logger_1.logger.error('Error fetching API keys:');
                         if (error instanceof Error) {
-                            console.log(chalk_1.default.red(error.message));
-                            if (isDebug) {
-                                console.log(chalk_1.default.yellow(`DEBUG: API key list error: ${error.message}`));
-                                console.log(chalk_1.default.yellow(`DEBUG: Stack: ${error.stack}`));
-                            }
+                            logger_1.logger.error(error.message);
+                            logger_1.logger.debug(`API key list error: ${error.message}`);
+                            logger_1.logger.debug(`Stack: ${error.stack}`);
                         }
                     }
                     return null;
                 }
-                console.log(chalk_1.default.blue('Select an API key to use as default:'));
+                logger_1.logger.info('Select an API key to use as default:');
                 // Display available API keys
                 apiKeys.forEach((key, index) => {
-                    console.log(`  ${index + 1}. ${key.name} (${key.prefix}...)`);
+                    logger_1.logger.log(`  ${index + 1}. ${key.name} (${key.prefix}...)`);
                 });
                 // Create readline interface for user input
                 const rl = readline_1.default.createInterface({
@@ -212,7 +195,7 @@ class DefaultApiKeyManager {
                     });
                 });
                 if (selection === -1) {
-                    console.log(chalk_1.default.yellow('No API key selected'));
+                    logger_1.logger.warn('No API key selected');
                     return null;
                 }
                 const selectedKey = apiKeys[selection];
@@ -224,11 +207,11 @@ class DefaultApiKeyManager {
                 // Save the new key as default
                 this.setDefaultApiKey(newKey.id.toString(), newKey.name, newKey.key.substring(0, 8), // Use first 8 chars as prefix
                 newKey.key);
-                console.log(chalk_1.default.green(`✓ Default API key set to: ${newKey.name}`));
+                logger_1.logger.success(`✓ Default API key set to: ${newKey.name}`);
                 return newKey.key;
             }
             catch (error) {
-                console.error(chalk_1.default.red('Failed to set default API key:'), error);
+                logger_1.logger.error('Failed to set default API key:', error);
                 return null;
             }
         });