berget 1.0.0 → 1.1.0

package/dist/src/constants/command-structure.js

@@ -17,6 +17,7 @@ exports.COMMAND_GROUPS = {
     FLUX: 'flux',
     USERS: 'users',
     BILLING: 'billing',
+    CHAT: 'chat',
 };
 // Subcommands for each group
 exports.SUBCOMMANDS = {
@@ -26,6 +27,11 @@ exports.SUBCOMMANDS = {
         LOGOUT: 'logout',
         WHOAMI: 'whoami',
     },
+    // Chat commands
+    CHAT: {
+        RUN: 'run',
+        LIST: 'list',
+    },
     // API Keys commands
     API_KEYS: {
         LIST: 'list',
@@ -33,6 +39,8 @@ exports.SUBCOMMANDS = {
         DELETE: 'delete',
         ROTATE: 'rotate',
         DESCRIBE: 'describe',
+        SET_DEFAULT: 'set-default',
+        GET_DEFAULT: 'get-default',
     },
     // Clusters commands
     CLUSTERS: {
@@ -102,6 +110,8 @@ exports.COMMAND_DESCRIPTIONS = {
     [`${exports.COMMAND_GROUPS.API_KEYS} ${exports.SUBCOMMANDS.API_KEYS.DELETE}`]: 'Delete an API key',
     [`${exports.COMMAND_GROUPS.API_KEYS} ${exports.SUBCOMMANDS.API_KEYS.ROTATE}`]: 'Rotate an API key',
     [`${exports.COMMAND_GROUPS.API_KEYS} ${exports.SUBCOMMANDS.API_KEYS.DESCRIBE}`]: 'Get usage statistics for an API key',
+    [`${exports.COMMAND_GROUPS.API_KEYS} ${exports.SUBCOMMANDS.API_KEYS.SET_DEFAULT}`]: 'Set an API key as the default for chat commands',
+    [`${exports.COMMAND_GROUPS.API_KEYS} ${exports.SUBCOMMANDS.API_KEYS.GET_DEFAULT}`]: 'Show the current default API key',
     // Clusters group
     [exports.COMMAND_GROUPS.CLUSTERS]: 'Manage Kubernetes clusters',
     [`${exports.COMMAND_GROUPS.CLUSTERS} ${exports.SUBCOMMANDS.CLUSTERS.LIST}`]: 'List all clusters',
@@ -147,4 +157,8 @@ exports.COMMAND_DESCRIPTIONS = {
     [`${exports.COMMAND_GROUPS.BILLING} ${exports.SUBCOMMANDS.BILLING.ADD_PAYMENT_METHOD}`]: 'Add a new payment method',
     [`${exports.COMMAND_GROUPS.BILLING} ${exports.SUBCOMMANDS.BILLING.REMOVE_PAYMENT_METHOD}`]: 'Remove a payment method',
     [`${exports.COMMAND_GROUPS.BILLING} ${exports.SUBCOMMANDS.BILLING.UPDATE_SUBSCRIPTION}`]: 'Update subscription plan',
+    // Chat group
+    [exports.COMMAND_GROUPS.CHAT]: 'Interact with AI chat models',
+    [`${exports.COMMAND_GROUPS.CHAT} ${exports.SUBCOMMANDS.CHAT.RUN}`]: 'Run a chat session with a specified model',
+    [`${exports.COMMAND_GROUPS.CHAT} ${exports.SUBCOMMANDS.CHAT.LIST}`]: 'List available chat models',
 };

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

@@ -55,6 +55,21 @@ class AuthService {
         }
         return AuthService.instance;
     }
+    whoami() {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const { data: profile, error } = yield this.client.GET('/v1/users/me');
+                if (error) {
+                    throw new Error(error ? JSON.stringify(error) : 'Failed to get user profile');
+                }
+                return profile;
+            }
+            catch (error) {
+                (0, error_handler_1.handleError)('Failed to get user profile', error);
+                return null;
+            }
+        });
+    }
     login() {
         return __awaiter(this, void 0, void 0, function* () {
             try {
@@ -68,16 +83,20 @@ class AuthService {
                         ? JSON.stringify(deviceError)
                         : 'Failed to get device authorization data');
                 }
+                // Type assertion for deviceData
+                const typedDeviceData = deviceData;
                 // Display information to user
                 console.log(chalk_1.default.cyan('\nTo complete login:'));
-                console.log(chalk_1.default.cyan(`1. Open this URL: ${chalk_1.default.bold(deviceData.verification_url || 'https://auth.berget.ai/device')}`));
-                console.log(chalk_1.default.cyan(`2. Enter this code: ${chalk_1.default.bold(deviceData.user_code || '')}\n`));
+                console.log(chalk_1.default.cyan(`1. Open this URL: ${chalk_1.default.bold(typedDeviceData.verification_url ||
+                    'https://keycloak.berget.ai/device')}`));
+                if (!typedDeviceData.verification_url)
+                    console.log(chalk_1.default.cyan(`2. Enter this code: ${chalk_1.default.bold(typedDeviceData.user_code || '')}\n`));
                 // Try to open browser automatically
                 try {
-                    if (deviceData.verification_url) {
+                    if (typedDeviceData.verification_url) {
                         // Use dynamic import for the 'open' package
-                        const open = yield Promise.resolve().then(() => __importStar(require('open'))).then(m => m.default);
-                        yield open(deviceData.verification_url);
+                        const open = yield Promise.resolve().then(() => __importStar(require('open'))).then((m) => m.default);
+                        yield open(typedDeviceData.verification_url);
                         console.log(chalk_1.default.dim("Browser opened automatically. If it didn't open, please use the URL above."));
                     }
                 }
@@ -87,9 +106,11 @@ class AuthService {
                 console.log(chalk_1.default.dim('\nWaiting for authentication to complete...'));
                 // Step 2: Poll for completion
                 const startTime = Date.now();
-                const expiresIn = deviceData.expires_in !== undefined ? deviceData.expires_in : 900;
+                const expiresIn = typedDeviceData.expires_in !== undefined
+                    ? typedDeviceData.expires_in
+                    : 900;
                 const expiresAt = startTime + expiresIn * 1000;
-                let pollInterval = (deviceData.interval || 5) * 1000;
+                let pollInterval = (typedDeviceData.interval || 5) * 1000;
                 const spinner = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏'];
                 let spinnerIdx = 0;
                 while (Date.now() < expiresAt) {
@@ -99,7 +120,7 @@ class AuthService {
                     process.stdout.write(`\r${chalk_1.default.blue(spinner[spinnerIdx])} Waiting for authentication...`);
                     spinnerIdx = (spinnerIdx + 1) % spinner.length;
                     // Check if authentication is complete
-                    const deviceCode = deviceData.device_code || '';
+                    const deviceCode = typedDeviceData.device_code || '';
                     const { data: tokenData, error: tokenError } = yield client_1.apiClient.POST('/v1/auth/device/token', {
                         body: {
                             device_code: deviceCode,
@@ -147,16 +168,27 @@ class AuthService {
                             continue;
                         }
                     }
-                    else if (tokenData && tokenData.token) {
-                        // Success!
-                        (0, client_1.saveAuthToken)(tokenData.token);
-                        process.stdout.write('\r' + ' '.repeat(50) + '\r'); // Clear the spinner line
-                        console.log(chalk_1.default.green('✓ Successfully logged in to Berget'));
-                        if (tokenData.user) {
-                            const user = tokenData.user;
-                            console.log(chalk_1.default.green(`Logged in as ${user.name || user.email || 'User'}`));
+                    else if (tokenData) {
+                        // Type assertion for tokenData
+                        const typedTokenData = tokenData;
+                        if (typedTokenData.token) {
+                            // Success!
+                            (0, client_1.saveAuthToken)(typedTokenData.token, typedTokenData.refresh_token || '', typedTokenData.expires_in || 3600);
+                            if (process.argv.includes('--debug')) {
+                                console.log(chalk_1.default.yellow('DEBUG: Token data received:'));
+                                console.log(chalk_1.default.yellow(JSON.stringify({
+                                    expires_in: typedTokenData.expires_in,
+                                    refresh_expires_in: typedTokenData.refresh_expires_in,
+                                }, null, 2)));
+                            }
+                            process.stdout.write('\r' + ' '.repeat(50) + '\r'); // Clear the spinner line
+                            console.log(chalk_1.default.green('✓ Successfully logged in to Berget'));
+                            if (typedTokenData.user) {
+                                const user = typedTokenData.user;
+                                console.log(chalk_1.default.green(`Logged in as ${user.name || user.email || 'User'}`));
+                            }
+                            return true;
                         }
-                        return true;
                     }
                 }
                 console.log(chalk_1.default.red('\n\nAuthentication timed out. Please try again.'));
@@ -168,36 +200,6 @@ class AuthService {
             }
         });
     }
-    isAuthenticated() {
-        return __awaiter(this, void 0, void 0, function* () {
-            try {
-                // Call an API endpoint that requires authentication
-                const { data, error } = yield this.client.GET('/v1/users/me');
-                return !!data && !error;
-            }
-            catch (_a) {
-                return false;
-            }
-        });
-    }
-    /**
-     * Get current user profile
-     * Command: berget auth whoami
-     */
-    whoami() {
-        return __awaiter(this, void 0, void 0, function* () {
-            try {
-                const { data, error } = yield this.client.GET('/v1/users/me');
-                if (error)
-                    throw new Error(JSON.stringify(error));
-                return data;
-            }
-            catch (error) {
-                (0, error_handler_1.handleError)('Failed to get user profile', error);
-                throw error;
-            }
-        });
-    }
 }
 exports.AuthService = AuthService;
 // Command group name for this service

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

@@ -0,0 +1,177 @@
+"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 __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    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"));
+/**
+ * Service for interacting with the chat API
+ * Command group: chat
+ */
+class ChatService {
+    constructor() {
+        this.client = (0, client_1.createAuthenticatedClient)();
+    }
+    static getInstance() {
+        if (!ChatService.instance) {
+            ChatService.instance = new ChatService();
+        }
+        return ChatService.instance;
+    }
+    /**
+     * Create a chat completion
+     * Command: berget chat completion
+     */
+    createCompletion(options) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                const headers = {};
+                // Check if debug is enabled
+                const isDebug = process.argv.includes('--debug');
+                if (isDebug) {
+                    console.log(chalk_1.default.yellow('DEBUG: Chat completion options:'));
+                    console.log(chalk_1.default.yellow(JSON.stringify(options, null, 2)));
+                }
+                // If an API key is provided, use it for this request
+                if (options.apiKey) {
+                    headers['Authorization'] = `Bearer ${options.apiKey}`;
+                    // Remove apiKey from options before sending to API
+                    const { apiKey } = options, requestOptions = __rest(options, ["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)));
+                    }
+                    const { data, error } = yield this.client.POST('/v1/chat/completions', {
+                        body: requestOptions,
+                        headers
+                    });
+                    if (isDebug) {
+                        console.log(chalk_1.default.yellow('DEBUG: API response:'));
+                        console.log(chalk_1.default.yellow(JSON.stringify({ data, error }, 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(data, null, 2)));
+                    }
+                    if (error)
+                        throw new Error(JSON.stringify(error));
+                    return data;
+                }
+                else {
+                    // Use the default authenticated client
+                    if (isDebug) {
+                        console.log(chalk_1.default.yellow('DEBUG: Using default authentication'));
+                    }
+                    const { data, error } = yield this.client.POST('/v1/chat/completions', {
+                        body: options
+                    });
+                    if (isDebug) {
+                        console.log(chalk_1.default.yellow('DEBUG: API response:'));
+                        console.log(chalk_1.default.yellow(JSON.stringify({ data, error }, 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(data, null, 2)));
+                    }
+                    if (error)
+                        throw new Error(JSON.stringify(error));
+                    return data;
+                }
+            }
+            catch (error) {
+                // Improved error handling
+                let errorMessage = 'Failed to create chat completion';
+                if (error instanceof Error) {
+                    try {
+                        // Try to parse the error message as JSON
+                        const parsedError = JSON.parse(error.message);
+                        if (parsedError.error && parsedError.error.message) {
+                            errorMessage = `Chat error: ${parsedError.error.message}`;
+                        }
+                    }
+                    catch (e) {
+                        // If parsing fails, use the original error message
+                        errorMessage = `Chat error: ${error.message}`;
+                    }
+                }
+                console.error(chalk_1.default.red(errorMessage));
+                throw new Error(errorMessage);
+            }
+        });
+    }
+    /**
+     * List available models
+     * Command: berget chat list
+     */
+    listModels(apiKey) {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                if (apiKey) {
+                    const headers = {
+                        'Authorization': `Bearer ${apiKey}`
+                    };
+                    const { data, error } = yield this.client.GET('/v1/models', { headers });
+                    if (error)
+                        throw new Error(JSON.stringify(error));
+                    return data;
+                }
+                else {
+                    const { data, error } = yield this.client.GET('/v1/models');
+                    if (error)
+                        throw new Error(JSON.stringify(error));
+                    return data;
+                }
+            }
+            catch (error) {
+                // Improved error handling
+                let errorMessage = 'Failed to list models';
+                if (error instanceof Error) {
+                    try {
+                        // 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))}`;
+                        }
+                    }
+                    catch (e) {
+                        // If parsing fails, use the original error message
+                        errorMessage = `Models error: ${error.message}`;
+                    }
+                }
+                console.error(chalk_1.default.red(errorMessage));
+                throw new Error(errorMessage);
+            }
+        });
+    }
+}
+exports.ChatService = ChatService;
+// Command group name for this service
+ChatService.COMMAND_GROUP = 'chat';
+// Subcommands for this service
+ChatService.COMMANDS = {
+    RUN: 'run',
+    LIST: 'list'
+};

package/dist/src/utils/config-checker.js

@@ -0,0 +1,50 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.checkBergetConfig = void 0;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+/**
+ * Check for .bergetconfig file and handle cluster switching
+ */
+function checkBergetConfig() {
+    const configPath = path.join(process.cwd(), '.bergetconfig');
+    if (fs.existsSync(configPath)) {
+        try {
+            const config = fs.readFileSync(configPath, 'utf8');
+            const match = config.match(/cluster:\s*(.+)/);
+            if (match && match[1]) {
+                const clusterName = match[1].trim();
+                console.log(`🔄 Berget: Switched to cluster "${clusterName}"`);
+                console.log('✓ kubectl config updated');
+                console.log('');
+            }
+        }
+        catch (error) {
+            // Silently ignore errors reading config
+        }
+    }
+}
+exports.checkBergetConfig = checkBergetConfig;

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

@@ -0,0 +1,111 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+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"));
+/**
+ * Manages the default API key for chat commands
+ */
+class DefaultApiKeyManager {
+    constructor() {
+        this.defaultApiKey = null;
+        // Set up config file path in user's home directory
+        const bergetDir = path.join(os.homedir(), '.berget');
+        if (!fs.existsSync(bergetDir)) {
+            fs.mkdirSync(bergetDir, { recursive: true });
+        }
+        this.configFilePath = path.join(bergetDir, 'default-api-key.json');
+        this.loadConfig();
+    }
+    static getInstance() {
+        if (!DefaultApiKeyManager.instance) {
+            DefaultApiKeyManager.instance = new DefaultApiKeyManager();
+        }
+        return DefaultApiKeyManager.instance;
+    }
+    /**
+     * Load default API key from file
+     */
+    loadConfig() {
+        try {
+            if (fs.existsSync(this.configFilePath)) {
+                const data = fs.readFileSync(this.configFilePath, 'utf8');
+                this.defaultApiKey = JSON.parse(data);
+            }
+        }
+        catch (error) {
+            console.error(chalk_1.default.dim('Failed to load default API key configuration'));
+            this.defaultApiKey = null;
+        }
+    }
+    /**
+     * Save default API key to file
+     */
+    saveConfig() {
+        try {
+            if (this.defaultApiKey) {
+                fs.writeFileSync(this.configFilePath, JSON.stringify(this.defaultApiKey, null, 2));
+                // Set file permissions to be readable only by the owner
+                fs.chmodSync(this.configFilePath, 0o600);
+            }
+            else {
+                // If default API key is null, remove the file
+                if (fs.existsSync(this.configFilePath)) {
+                    fs.unlinkSync(this.configFilePath);
+                }
+            }
+        }
+        catch (error) {
+            console.error(chalk_1.default.dim('Failed to save default API key configuration'));
+        }
+    }
+    /**
+     * Set the default API key
+     */
+    setDefaultApiKey(id, name, prefix) {
+        this.defaultApiKey = { id, name, prefix };
+        this.saveConfig();
+    }
+    /**
+     * Get the default API key
+     */
+    getDefaultApiKey() {
+        return this.defaultApiKey;
+    }
+    /**
+     * Clear the default API key
+     */
+    clearDefaultApiKey() {
+        this.defaultApiKey = null;
+        this.saveConfig();
+    }
+}
+exports.DefaultApiKeyManager = DefaultApiKeyManager;