berget 2.2.6 → 2.2.7

package/dist/src/commands/chat.js

@@ -38,10 +38,10 @@ function confirm(question) {
             input: process.stdin,
             output: process.stdout,
         });
-        return new Promise((resolve) => {
-            rl.question(question, (answer) => {
+        return new Promise(resolve => {
+            rl.question(question, answer => {
                 rl.close();
-                resolve(answer.toLowerCase() === 'y' || answer.toLowerCase() === 'yes');
+                resolve(answer.toLowerCase() === "y" || answer.toLowerCase() === "yes");
             });
         });
     });
@@ -50,19 +50,17 @@ function confirm(question) {
  * 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)')
+        .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;
         try {
@@ -76,7 +74,7 @@ function registerChatCommands(program) {
                 console.log(chalk_1.default.dim(`Using API key from BERGET_API_KEY environment variable`));
                 apiKey = envApiKey;
                 // Debug the API key (first few characters only)
-                if (process.argv.includes('--debug')) {
+                if (process.argv.includes("--debug")) {
                     console.log(chalk_1.default.yellow(`DEBUG: API key from env starts with: ${envApiKey.substring(0, 4)}...`));
                 }
             }
@@ -103,22 +101,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();
                         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)));
                     }
                 }
@@ -128,7 +126,7 @@ function registerChatCommands(program) {
                 try {
                     const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
                     const keys = yield apiKeyService.list();
-                    const selectedKey = keys.find((key) => key.id.toString() === options.apiKeyId);
+                    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.`));
                     }
@@ -141,24 +139,24 @@ function registerChatCommands(program) {
                             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."));
                         }
                     }
                 }
                 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
@@ -166,13 +164,13 @@ function registerChatCommands(program) {
                 try {
                     auth_service_1.AuthService.getInstance();
                 }
-                catch (error) {
-                    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 (_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>`"));
                     return;
                 }
             }
@@ -181,20 +179,20 @@ function registerChatCommands(program) {
             // Add system message if provided
             if (options.system) {
                 messages.push({
-                    role: 'system',
+                    role: "system",
                     content: options.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 _d = true, _e = __asyncValues(process.stdin), _f; _f = yield _e.next(), _a = _f.done, !_a; _d = true) {
-                        _c = _f.value;
-                        _d = false;
+                    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);
                     }
@@ -202,11 +200,11 @@ function registerChatCommands(program) {
                 catch (e_1_1) { e_1 = { error: e_1_1 }; }
                 finally {
                     try {
-                        if (!_d && !_a && (_b = _e.return)) yield _b.call(_e);
+                        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) {
@@ -219,13 +217,13 @@ function registerChatCommands(program) {
             if (inputMessage) {
                 // Add user message
                 messages.push({
-                    role: 'user',
+                    role: "user",
                     content: inputMessage,
                 });
                 try {
                     // Call the API
                     const completionOptions = {
-                        model: options.model || 'openai/gpt-oss',
+                        model: options.model || "openai/gpt-oss",
                         messages: messages,
                         temperature: options.temperature !== undefined ? options.temperature : 0.7,
                         max_tokens: options.maxTokens || 4096,
@@ -237,7 +235,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 &&
@@ -250,7 +248,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;
                                     }
@@ -263,9 +261,9 @@ function registerChatCommands(program) {
                             yield 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);
@@ -286,9 +284,9 @@ 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;
@@ -302,7 +300,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));
                     }
@@ -315,25 +313,25 @@ function registerChatCommands(program) {
                 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: "), (input) => __awaiter(this, void 0, void 0, function* () {
                     // 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',
+                        role: "user",
                         content: input,
                     });
                     try {
                         // Call the API
                         const completionOptions = {
-                            model: options.model || 'openai/gpt-oss',
+                            model: options.model || "openai/gpt-oss",
                             messages: messages,
                             temperature: options.temperature !== undefined ? options.temperature : 0.7,
                             max_tokens: options.maxTokens || 4096,
@@ -345,8 +343,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 &&
@@ -359,7 +357,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;
                                         }
@@ -372,9 +370,9 @@ function registerChatCommands(program) {
                                 yield 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);
@@ -386,10 +384,10 @@ function registerChatCommands(program) {
                                     console.log(assistantResponse);
                                 }
                             }
-                            console.log('\n');
+                            console.log("\n");
                             // Add assistant response to messages
                             messages.push({
-                                role: 'assistant',
+                                role: "assistant",
                                 content: assistantResponse,
                             });
                             // Continue the conversation
@@ -399,7 +397,7 @@ function registerChatCommands(program) {
                         const response = yield 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
@@ -407,19 +405,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',
+                            role: "assistant",
                             content: assistantMessage,
                         });
                         // 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));
@@ -432,7 +430,7 @@ 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));
                         }
@@ -445,14 +443,14 @@ function registerChatCommands(program) {
             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')
+        .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* () {
         try {
             // If API key ID is provided, fetch the actual key
@@ -471,7 +469,7 @@ function registerChatCommands(program) {
                 try {
                     const apiKeyService = api_key_service_1.ApiKeyService.getInstance();
                     const keys = yield apiKeyService.list();
-                    const selectedKey = keys.find((key) => key.id.toString() === options.apiKeyId);
+                    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.`));
                     }
@@ -484,44 +482,44 @@ function registerChatCommands(program) {
                             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."));
                         }
                     }
                 }
                 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);
             // 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);
         }
     }));
 }

package/dist/src/commands/clusters.js

@@ -18,51 +18,51 @@ const error_handler_1 = require("../utils/error-handler");
 function registerClusterCommands(program) {
     const cluster = program
         .command(cluster_service_1.ClusterService.COMMAND_GROUP)
-        .description('Manage Berget clusters');
+        .description("Manage Berget clusters");
     cluster
         .command(cluster_service_1.ClusterService.COMMANDS.LIST)
-        .description('List all Berget clusters')
+        .description("List all Berget clusters")
         .action(() => __awaiter(this, void 0, void 0, function* () {
         try {
             const clusterService = cluster_service_1.ClusterService.getInstance();
             const clusters = yield clusterService.list();
-            console.log('NAME                   STATUS    NODES    CREATED');
+            console.log("NAME                   STATUS    NODES    CREATED");
             clusters.forEach((cluster) => {
                 console.log(`${cluster.name.padEnd(22)} ${cluster.status.padEnd(9)} ${String(cluster.nodes).padEnd(8)} ${cluster.created}`);
             });
         }
         catch (error) {
-            (0, error_handler_1.handleError)('Failed to list clusters', error);
+            (0, error_handler_1.handleError)("Failed to list clusters", error);
         }
     }));
     cluster
         .command(cluster_service_1.ClusterService.COMMANDS.GET_USAGE)
-        .description('Get usage metrics for a specific cluster')
-        .argument('<clusterId>', 'Cluster ID')
+        .description("Get usage metrics for a specific cluster")
+        .argument("<clusterId>", "Cluster ID")
         .action((clusterId) => __awaiter(this, void 0, void 0, function* () {
         try {
             const clusterService = cluster_service_1.ClusterService.getInstance();
             const usage = yield clusterService.getUsage(clusterId);
-            console.log('Cluster Usage:');
+            console.log("Cluster Usage:");
             console.log(JSON.stringify(usage, null, 2));
         }
         catch (error) {
-            (0, error_handler_1.handleError)('Failed to get cluster usage', error);
+            (0, error_handler_1.handleError)("Failed to get cluster usage", error);
         }
     }));
     cluster
         .command(cluster_service_1.ClusterService.COMMANDS.DESCRIBE)
-        .description('Get detailed information about a cluster')
-        .argument('<clusterId>', 'Cluster ID')
+        .description("Get detailed information about a cluster")
+        .argument("<clusterId>", "Cluster ID")
         .action((clusterId) => __awaiter(this, void 0, void 0, function* () {
         try {
             const clusterService = cluster_service_1.ClusterService.getInstance();
             const clusterInfo = yield clusterService.describe(clusterId);
-            console.log('Cluster Details:');
+            console.log("Cluster Details:");
             console.log(JSON.stringify(clusterInfo, null, 2));
         }
         catch (error) {
-            (0, error_handler_1.handleError)('Failed to describe cluster', error);
+            (0, error_handler_1.handleError)("Failed to describe cluster", error);
         }
     }));
 }