berget 2.2.7 → 2.2.8

package/dist/src/commands/chat.js

@@ -1,81 +1,47 @@
 "use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __asyncValues = (this && this.__asyncValues) || function (o) {
-    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
-    var m = o[Symbol.asyncIterator], i;
-    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
-    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
-    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
-};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.registerChatCommands = void 0;
 const chalk_1 = __importDefault(require("chalk"));
-const readline_1 = __importDefault(require("readline"));
+const node_readline_1 = __importDefault(require("node:readline"));
 const command_structure_1 = require("../constants/command-structure");
-const chat_service_1 = require("../services/chat-service");
 const api_key_service_1 = require("../services/api-key-service");
 const auth_service_1 = require("../services/auth-service");
-const error_handler_1 = require("../utils/error-handler");
+const chat_service_1 = require("../services/chat-service");
 const default_api_key_1 = require("../utils/default-api-key");
+const error_handler_1 = require("../utils/error-handler");
 const markdown_renderer_1 = require("../utils/markdown-renderer");
-/**
- * Helper function to get user confirmation
- */
-function confirm(question) {
-    return __awaiter(this, void 0, void 0, function* () {
-        const rl = readline_1.default.createInterface({
-            input: process.stdin,
-            output: process.stdout,
-        });
-        return new Promise(resolve => {
-            rl.question(question, answer => {
-                rl.close();
-                resolve(answer.toLowerCase() === "y" || answer.toLowerCase() === "yes");
-            });
-        });
-    });
-}
 /**
  * Register chat commands
  */
 function registerChatCommands(program) {
-    const chat = program.command(command_structure_1.COMMAND_GROUPS.CHAT).description("Interact with AI chat models");
+    const chat = program.command(command_structure_1.COMMAND_GROUPS.CHAT).description('Interact with AI chat models');
     chat
         .command(command_structure_1.SUBCOMMANDS.CHAT.RUN)
-        .description("Run a chat session with a specified model")
-        .argument("[message]", "Message to send directly (skips interactive mode)")
-        .option("-m, --model <model>", "Model to use (default: glm-4.7)")
-        .option("-t, --temperature <temp>", "Temperature (0-1)", parseFloat)
-        .option("--max-tokens <tokens>", "Maximum tokens to generate", parseInt)
-        .option("-k, --api-key <key>", "API key to use for this chat session")
-        .option("--api-key-id <id>", "ID of the API key to use from your saved keys")
-        .option("--no-stream", "Disable streaming (streaming is enabled by default)")
-        .action((message, options) => __awaiter(this, void 0, void 0, function* () {
-        var _a, e_1, _b, _c;
+        .description('Run a chat session with a specified model')
+        .argument('[message]', 'Message to send directly (skips interactive mode)')
+        .option('-m, --model <model>', 'Model to use (default: glm-4.7)')
+        .option('-t, --temperature <temp>', 'Temperature (0-1)', Number.parseFloat)
+        .option('--max-tokens <tokens>', 'Maximum tokens to generate', Number.parseInt)
+        .option('-k, --api-key <key>', 'API key to use for this chat session')
+        .option('--api-key-id <id>', 'ID of the API key to use from your saved keys')
+        .option('--no-stream', 'Disable streaming (streaming is enabled by default)')
+        .action(async (message, options) => {
         try {
             const chatService = chat_service_1.ChatService.getInstance();
             // Check if we have an API key or need to get one
             let apiKey = options.apiKey;
             let apiKeyId = options.apiKeyId;
             // Check for environment variable first
-            const envApiKey = process.env.BERGET_API_KEY;
-            if (envApiKey) {
+            const environmentApiKey = process.env.BERGET_API_KEY;
+            if (environmentApiKey) {
                 console.log(chalk_1.default.dim(`Using API key from BERGET_API_KEY environment variable`));
-                apiKey = envApiKey;
+                apiKey = environmentApiKey;
                 // Debug the API key (first few characters only)
-                if (process.argv.includes("--debug")) {
-                    console.log(chalk_1.default.yellow(`DEBUG: API key from env starts with: ${envApiKey.substring(0, 4)}...`));
+                if (process.argv.includes('--debug')) {
+                    console.log(chalk_1.default.yellow(`DEBUG: API key from env starts with: ${environmentApiKey.slice(0, 4)}...`));
                 }
             }
             // If API key is already provided via command line, use it
@@ -101,22 +67,22 @@ function registerChatCommands(program) {
                     }
                     else {
                         // No default API key, prompt the user to create one
-                        console.log(chalk_1.default.yellow("No default API key set."));
+                        console.log(chalk_1.default.yellow('No default API key set.'));
                         // Try to prompt for a default API key
-                        apiKey = yield defaultApiKeyManager.promptForDefaultApiKey();
+                        apiKey = await defaultApiKeyManager.promptForDefaultApiKey();
                         if (!apiKey) {
-                            console.log(chalk_1.default.red("Error: An API key is required to use the chat command."));
-                            console.log(chalk_1.default.yellow("You can:"));
+                            console.log(chalk_1.default.red('Error: An API key is required to use the chat command.'));
+                            console.log(chalk_1.default.yellow('You can:'));
                             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('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'));
                             return;
                         }
                     }
                 }
                 catch (error) {
-                    if (process.argv.includes("--debug")) {
-                        console.log(chalk_1.default.yellow("DEBUG: Error checking default API key:"));
+                    if (process.argv.includes('--debug')) {
+                        console.log(chalk_1.default.yellow('DEBUG: Error checking default API key:'));
                         console.log(chalk_1.default.yellow(String(error)));
                     }
                 }
@@ -125,38 +91,38 @@ function registerChatCommands(program) {
             if (!apiKey && apiKeyId) {
                 try {
                     const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
-                    const keys = yield apiKeyService.list();
-                    const selectedKey = keys.find(key => key.id.toString() === options.apiKeyId);
-                    if (!selectedKey) {
-                        console.log(chalk_1.default.yellow(`API key with ID ${options.apiKeyId} not found. Using default authentication.`));
-                    }
-                    else {
+                    const keys = await apiKeyService.list();
+                    const selectedKey = keys.find((key) => key.id.toString() === options.apiKeyId);
+                    if (selectedKey) {
                         console.log(chalk_1.default.dim(`Using API key: ${selectedKey.name}`));
                         // We need to rotate the key to get the actual key value
-                        if (yield confirm(chalk_1.default.yellow(`To use API key "${selectedKey.name}", it needs to be rotated. This will invalidate the current key. Continue? (y/n)`))) {
-                            const rotatedKey = yield apiKeyService.rotate(options.apiKeyId);
+                        if (await confirm(chalk_1.default.yellow(`To use API key "${selectedKey.name}", it needs to be rotated. This will invalidate the current key. Continue? (y/n)`))) {
+                            const rotatedKey = await apiKeyService.rotate(options.apiKeyId);
                             apiKey = rotatedKey.key;
                             console.log(chalk_1.default.green(`API key "${selectedKey.name}" rotated successfully.`));
                         }
                         else {
-                            console.log(chalk_1.default.yellow("Using default authentication instead."));
+                            console.log(chalk_1.default.yellow('Using default authentication instead.'));
                         }
                     }
+                    else {
+                        console.log(chalk_1.default.yellow(`API key with ID ${options.apiKeyId} not found. Using default authentication.`));
+                    }
                 }
                 catch (error) {
                     // Check if this is an authentication error
                     const errorMessage = error instanceof Error ? error.message : String(error);
-                    const isAuthError = errorMessage.includes("Unauthorized") ||
-                        errorMessage.includes("Authentication failed") ||
-                        errorMessage.includes("AUTH_FAILED");
+                    const isAuthError = errorMessage.includes('Unauthorized') ||
+                        errorMessage.includes('Authentication failed') ||
+                        errorMessage.includes('AUTH_FAILED');
                     if (isAuthError) {
-                        console.log(chalk_1.default.yellow("Authentication required. Please run `berget auth login` first."));
+                        console.log(chalk_1.default.yellow('Authentication required. Please run `berget auth login` first.'));
                     }
                     else {
-                        console.error(chalk_1.default.red("Error fetching API key:"));
+                        console.error(chalk_1.default.red('Error fetching API key:'));
                         console.error(error);
                     }
-                    console.log(chalk_1.default.yellow("Using default authentication instead."));
+                    console.log(chalk_1.default.yellow('Using default authentication instead.'));
                 }
             }
             // Verify we have authentication before starting chat
@@ -164,13 +130,13 @@ function registerChatCommands(program) {
                 try {
                     auth_service_1.AuthService.getInstance();
                 }
-                catch (_d) {
-                    console.log(chalk_1.default.red("Error: Authentication required for chat"));
-                    console.log(chalk_1.default.yellow("Please either:"));
-                    console.log(chalk_1.default.yellow("1. Log in with `berget auth login`"));
-                    console.log(chalk_1.default.yellow("2. Provide an API key with `--api-key`"));
-                    console.log(chalk_1.default.yellow("3. Provide an API key ID with `--api-key-id`"));
-                    console.log(chalk_1.default.yellow("4. Set a default API key with `berget api-keys set-default <id>`"));
+                catch {
+                    console.log(chalk_1.default.red('Error: Authentication required for chat'));
+                    console.log(chalk_1.default.yellow('Please either:'));
+                    console.log(chalk_1.default.yellow('1. Log in with `berget auth login`'));
+                    console.log(chalk_1.default.yellow('2. Provide an API key with `--api-key`'));
+                    console.log(chalk_1.default.yellow('3. Provide an API key ID with `--api-key-id`'));
+                    console.log(chalk_1.default.yellow('4. Set a default API key with `berget api-keys set-default <id>`'));
                     return;
                 }
             }
@@ -179,32 +145,20 @@ function registerChatCommands(program) {
             // Add system message if provided
             if (options.system) {
                 messages.push({
-                    role: "system",
                     content: options.system,
+                    role: 'system',
                 });
             }
             // Check if input is being piped in
             let inputMessage = message;
-            let stdinContent = "";
+            let stdinContent = '';
             if (!process.stdin.isTTY) {
                 // Read from stdin (piped input)
                 const chunks = [];
-                try {
-                    for (var _e = true, _f = __asyncValues(process.stdin), _g; _g = yield _f.next(), _a = _g.done, !_a; _e = true) {
-                        _c = _g.value;
-                        _e = false;
-                        const chunk = _c;
-                        chunks.push(chunk);
-                    }
+                for await (const chunk of process.stdin) {
+                    chunks.push(chunk);
                 }
-                catch (e_1_1) { e_1 = { error: e_1_1 }; }
-                finally {
-                    try {
-                        if (!_e && !_a && (_b = _f.return)) yield _b.call(_f);
-                    }
-                    finally { if (e_1) throw e_1.error; }
-                }
-                stdinContent = Buffer.concat(chunks).toString("utf8").trim();
+                stdinContent = Buffer.concat(chunks).toString('utf8').trim();
             }
             // Combine stdin content with message if both exist
             if (stdinContent && message) {
@@ -217,17 +171,17 @@ function registerChatCommands(program) {
             if (inputMessage) {
                 // Add user message
                 messages.push({
-                    role: "user",
                     content: inputMessage,
+                    role: 'user',
                 });
                 try {
                     // Call the API
                     const completionOptions = {
-                        model: options.model || "openai/gpt-oss",
-                        messages: messages,
-                        temperature: options.temperature !== undefined ? options.temperature : 0.7,
                         max_tokens: options.maxTokens || 4096,
+                        messages: messages,
+                        model: options.model || 'openai/gpt-oss',
                         stream: options.stream !== false,
+                        temperature: options.temperature === undefined ? 0.7 : options.temperature,
                     };
                     // Only add apiKey if it actually exists
                     if (apiKey) {
@@ -235,7 +189,7 @@ function registerChatCommands(program) {
                     }
                     // Add streaming support (now default)
                     if (completionOptions.stream) {
-                        let assistantResponse = "";
+                        let assistantResponse = '';
                         // Stream the response in real-time
                         completionOptions.onChunk = (chunk) => {
                             if (chunk.choices &&
@@ -248,7 +202,7 @@ function registerChatCommands(program) {
                                 }
                                 catch (error) {
                                     // Handle EPIPE errors gracefully (when pipe is closed)
-                                    if (error.code === "EPIPE") {
+                                    if (error.code === 'EPIPE') {
                                         // Stop streaming if the pipe is closed
                                         return;
                                     }
@@ -258,15 +212,15 @@ function registerChatCommands(program) {
                             }
                         };
                         try {
-                            yield chatService.createCompletion(completionOptions);
+                            await chatService.createCompletion(completionOptions);
                         }
                         catch (streamError) {
-                            console.error(chalk_1.default.red("\nStreaming error:"), streamError);
+                            console.error(chalk_1.default.red('\nStreaming error:'), streamError);
                             // Fallback to non-streaming if streaming fails
-                            console.log(chalk_1.default.yellow("Falling back to non-streaming mode..."));
+                            console.log(chalk_1.default.yellow('Falling back to non-streaming mode...'));
                             completionOptions.stream = false;
                             delete completionOptions.onChunk;
-                            const response = yield chatService.createCompletion(completionOptions);
+                            const response = await chatService.createCompletion(completionOptions);
                             if (response &&
                                 response.choices &&
                                 response.choices[0] &&
@@ -278,15 +232,15 @@ function registerChatCommands(program) {
                         console.log(); // Add newline at the end
                         return;
                     }
-                    const response = yield chatService.createCompletion(completionOptions);
+                    const response = await chatService.createCompletion(completionOptions);
                     // Check if response has the expected structure
                     if (!response ||
                         !response.choices ||
                         !response.choices[0] ||
                         !response.choices[0].message) {
-                        console.error(chalk_1.default.red("Error: Unexpected response format from API"));
-                        console.error(chalk_1.default.red("Response:", JSON.stringify(response, null, 2)));
-                        throw new Error("Unexpected response format from API");
+                        console.error(chalk_1.default.red('Error: Unexpected response format from API'));
+                        console.error(chalk_1.default.red('Response:', JSON.stringify(response, null, 2)));
+                        throw new Error('Unexpected response format from API');
                     }
                     // Get assistant's response
                     const assistantMessage = response.choices[0].message.content;
@@ -300,7 +254,7 @@ function registerChatCommands(program) {
                     return;
                 }
                 catch (error) {
-                    console.error(chalk_1.default.red("Error: Failed to get response"));
+                    console.error(chalk_1.default.red('Error: Failed to get response'));
                     if (error instanceof Error) {
                         console.error(chalk_1.default.red(error.message));
                     }
@@ -308,34 +262,34 @@ function registerChatCommands(program) {
                 }
             }
             // Set up readline interface for user input (only for interactive mode)
-            const rl = readline_1.default.createInterface({
+            const rl = node_readline_1.default.createInterface({
                 input: process.stdin,
                 output: process.stdout,
             });
             console.log(chalk_1.default.cyan('Chat with Berget AI (type "exit" to quit)'));
-            console.log(chalk_1.default.cyan("----------------------------------------"));
+            console.log(chalk_1.default.cyan('----------------------------------------'));
             // Start the conversation loop
             const askQuestion = () => {
-                rl.question(chalk_1.default.green("You: "), (input) => __awaiter(this, void 0, void 0, function* () {
+                rl.question(chalk_1.default.green('You: '), async (input) => {
                     // Check if user wants to exit
-                    if (input.toLowerCase() === "exit") {
-                        console.log(chalk_1.default.cyan("Goodbye!"));
+                    if (input.toLowerCase() === 'exit') {
+                        console.log(chalk_1.default.cyan('Goodbye!'));
                         rl.close();
                         return;
                     }
                     // Add user message
                     messages.push({
-                        role: "user",
                         content: input,
+                        role: 'user',
                     });
                     try {
                         // Call the API
                         const completionOptions = {
-                            model: options.model || "openai/gpt-oss",
-                            messages: messages,
-                            temperature: options.temperature !== undefined ? options.temperature : 0.7,
                             max_tokens: options.maxTokens || 4096,
+                            messages: messages,
+                            model: options.model || 'openai/gpt-oss',
                             stream: options.stream !== false,
+                            temperature: options.temperature === undefined ? 0.7 : options.temperature,
                         };
                         // Only add apiKey if it actually exists
                         if (apiKey) {
@@ -343,8 +297,8 @@ function registerChatCommands(program) {
                         }
                         // Add streaming support (now default)
                         if (completionOptions.stream) {
-                            let assistantResponse = "";
-                            console.log(chalk_1.default.blue("Assistant: "));
+                            let assistantResponse = '';
+                            console.log(chalk_1.default.blue('Assistant: '));
                             // Stream the response in real-time
                             completionOptions.onChunk = (chunk) => {
                                 if (chunk.choices &&
@@ -357,7 +311,7 @@ function registerChatCommands(program) {
                                     }
                                     catch (error) {
                                         // Handle EPIPE errors gracefully (when pipe is closed)
-                                        if (error.code === "EPIPE") {
+                                        if (error.code === 'EPIPE') {
                                             // Stop streaming if the pipe is closed
                                             return;
                                         }
@@ -367,15 +321,15 @@ function registerChatCommands(program) {
                                 }
                             };
                             try {
-                                yield chatService.createCompletion(completionOptions);
+                                await chatService.createCompletion(completionOptions);
                             }
                             catch (streamError) {
-                                console.error(chalk_1.default.red("\nStreaming error:"), streamError);
+                                console.error(chalk_1.default.red('\nStreaming error:'), streamError);
                                 // Fallback to non-streaming if streaming fails
-                                console.log(chalk_1.default.yellow("Falling back to non-streaming mode..."));
+                                console.log(chalk_1.default.yellow('Falling back to non-streaming mode...'));
                                 completionOptions.stream = false;
                                 delete completionOptions.onChunk;
-                                const response = yield chatService.createCompletion(completionOptions);
+                                const response = await chatService.createCompletion(completionOptions);
                                 if (response &&
                                     response.choices &&
                                     response.choices[0] &&
@@ -384,20 +338,20 @@ function registerChatCommands(program) {
                                     console.log(assistantResponse);
                                 }
                             }
-                            console.log("\n");
+                            console.log('\n');
                             // Add assistant response to messages
                             messages.push({
-                                role: "assistant",
                                 content: assistantResponse,
+                                role: 'assistant',
                             });
                             // Continue the conversation
                             askQuestion();
                             return;
                         }
-                        const response = yield chatService.createCompletion(completionOptions);
+                        const response = await chatService.createCompletion(completionOptions);
                         // Debug output
                         if (program.opts().debug) {
-                            console.log(chalk_1.default.yellow("DEBUG: Full response:"));
+                            console.log(chalk_1.default.yellow('DEBUG: Full response:'));
                             console.log(chalk_1.default.yellow(JSON.stringify(response, null, 2)));
                         }
                         // Check if response has the expected structure
@@ -405,19 +359,19 @@ function registerChatCommands(program) {
                             !response.choices ||
                             !response.choices[0] ||
                             !response.choices[0].message) {
-                            console.error(chalk_1.default.red("Error: Unexpected response format from API"));
-                            console.error(chalk_1.default.red("Response:", JSON.stringify(response, null, 2)));
-                            throw new Error("Unexpected response format from API");
+                            console.error(chalk_1.default.red('Error: Unexpected response format from API'));
+                            console.error(chalk_1.default.red('Response:', JSON.stringify(response, null, 2)));
+                            throw new Error('Unexpected response format from API');
                         }
                         // Get assistant's response
                         const assistantMessage = response.choices[0].message.content;
                         // Add to messages array
                         messages.push({
-                            role: "assistant",
                             content: assistantMessage,
+                            role: 'assistant',
                         });
                         // Display the response
-                        console.log(chalk_1.default.blue("Assistant: "));
+                        console.log(chalk_1.default.blue('Assistant: '));
                         // Check if the response contains markdown and render it if it does
                         if ((0, markdown_renderer_1.containsMarkdown)(assistantMessage)) {
                             console.log((0, markdown_renderer_1.renderMarkdown)(assistantMessage));
@@ -430,28 +384,28 @@ function registerChatCommands(program) {
                         askQuestion();
                     }
                     catch (error) {
-                        console.error(chalk_1.default.red("Error: Failed to get response"));
+                        console.error(chalk_1.default.red('Error: Failed to get response'));
                         if (error instanceof Error) {
                             console.error(chalk_1.default.red(error.message));
                         }
                         // Continue despite error
                         askQuestion();
                     }
-                }));
+                });
             };
             // Start the conversation
             askQuestion();
         }
         catch (error) {
-            (0, error_handler_1.handleError)("Failed to create chat completion", error);
+            (0, error_handler_1.handleError)('Failed to create chat completion', error);
         }
-    }));
+    });
     chat
         .command(command_structure_1.SUBCOMMANDS.CHAT.LIST)
-        .description("List available chat models")
-        .option("-k, --api-key <key>", "API key to use for this request")
-        .option("--api-key-id <id>", "ID of the API key to use from your saved keys")
-        .action((options) => __awaiter(this, void 0, void 0, function* () {
+        .description('List available chat models')
+        .option('-k, --api-key <key>', 'API key to use for this request')
+        .option('--api-key-id <id>', 'ID of the API key to use from your saved keys')
+        .action(async (options) => {
         try {
             // If API key ID is provided, fetch the actual key
             let apiKey = options.apiKey;
@@ -468,59 +422,74 @@ function registerChatCommands(program) {
             if (apiKeyId && !apiKey) {
                 try {
                     const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
-                    const keys = yield apiKeyService.list();
-                    const selectedKey = keys.find(key => key.id.toString() === options.apiKeyId);
-                    if (!selectedKey) {
-                        console.log(chalk_1.default.yellow(`API key with ID ${options.apiKeyId} not found. Using default authentication.`));
-                    }
-                    else {
+                    const keys = await apiKeyService.list();
+                    const selectedKey = keys.find((key) => key.id.toString() === options.apiKeyId);
+                    if (selectedKey) {
                         console.log(chalk_1.default.dim(`Using API key: ${selectedKey.name}`));
                         // We need to rotate the key to get the actual key value
-                        if (yield confirm(chalk_1.default.yellow(`To use API key "${selectedKey.name}", it needs to be rotated. This will invalidate the current key. Continue? (y/n)`))) {
-                            const rotatedKey = yield apiKeyService.rotate(options.apiKeyId);
+                        if (await confirm(chalk_1.default.yellow(`To use API key "${selectedKey.name}", it needs to be rotated. This will invalidate the current key. Continue? (y/n)`))) {
+                            const rotatedKey = await apiKeyService.rotate(options.apiKeyId);
                             apiKey = rotatedKey.key;
                             console.log(chalk_1.default.green(`API key "${selectedKey.name}" rotated successfully.`));
                         }
                         else {
-                            console.log(chalk_1.default.yellow("Using default authentication instead."));
+                            console.log(chalk_1.default.yellow('Using default authentication instead.'));
                         }
                     }
+                    else {
+                        console.log(chalk_1.default.yellow(`API key with ID ${options.apiKeyId} not found. Using default authentication.`));
+                    }
                 }
                 catch (error) {
-                    console.error(chalk_1.default.red("Error fetching API key:"));
+                    console.error(chalk_1.default.red('Error fetching API key:'));
                     console.error(error);
-                    console.log(chalk_1.default.yellow("Using default authentication instead."));
+                    console.log(chalk_1.default.yellow('Using default authentication instead.'));
                 }
             }
             const chatService = chat_service_1.ChatService.getInstance();
-            const models = yield chatService.listModels(apiKey);
+            const models = await chatService.listModels(apiKey);
             // Debug output
             if (program.opts().debug) {
-                console.log(chalk_1.default.yellow("DEBUG: Models response:"));
+                console.log(chalk_1.default.yellow('DEBUG: Models response:'));
                 console.log(chalk_1.default.yellow(JSON.stringify(models, null, 2)));
             }
-            console.log(chalk_1.default.bold("Available Chat Models:"));
-            console.log(chalk_1.default.dim("─".repeat(70)));
-            console.log(chalk_1.default.dim("MODEL ID".padEnd(40)) + chalk_1.default.dim("CAPABILITIES"));
-            console.log(chalk_1.default.dim("─".repeat(70)));
+            console.log(chalk_1.default.bold('Available Chat Models:'));
+            console.log(chalk_1.default.dim('─'.repeat(70)));
+            console.log(chalk_1.default.dim('MODEL ID'.padEnd(40)) + chalk_1.default.dim('CAPABILITIES'));
+            console.log(chalk_1.default.dim('─'.repeat(70)));
             // Filter to only show active models
             const activeModels = models.data.filter((model) => model.active === true);
             activeModels.forEach((model) => {
                 const capabilities = [];
                 if (model.capabilities.vision)
-                    capabilities.push("vision");
+                    capabilities.push('vision');
                 if (model.capabilities.function_calling)
-                    capabilities.push("function_calling");
+                    capabilities.push('function_calling');
                 if (model.capabilities.json_mode)
-                    capabilities.push("json_mode");
+                    capabilities.push('json_mode');
                 // Format model ID in Huggingface compatible format (owner/model)
                 const modelId = `${model.owned_by.toLowerCase()}/${model.id}`.padEnd(40);
-                console.log(modelId + capabilities.join(", "));
+                console.log(modelId + capabilities.join(', '));
             });
         }
         catch (error) {
-            (0, error_handler_1.handleError)("Failed to list chat models", error);
+            (0, error_handler_1.handleError)('Failed to list chat models', error);
         }
-    }));
+    });
 }
 exports.registerChatCommands = registerChatCommands;
+/**
+ * Helper function to get user confirmation
+ */
+async function confirm(question) {
+    const rl = node_readline_1.default.createInterface({
+        input: process.stdin,
+        output: process.stdout,
+    });
+    return new Promise((resolve) => {
+        rl.question(question, (answer) => {
+            rl.close();
+            resolve(answer.toLowerCase() === 'y' || answer.toLowerCase() === 'yes');
+        });
+    });
+}