config.ts

import vscode from 'vscode';

class Config {

    // Inference
    get inference() {
        let config = this.#config;

        // Load endpoint
        let endpoint = (config.get('endpoint') as string).trim();
        if (endpoint.endsWith('/')) {
            endpoint = endpoint.slice(0, endpoint.length - 1).trim();
        }
        if (endpoint === '') {
            endpoint = 'http://127.0.0.1:11434';
        }
        let bearerToken = config.get('bearerToken') as string;

        // Load general paremeters
        let maxLines = config.get('maxLines') as number;
        let maxTokens = config.get('maxTokens') as number;
        let temperature = config.get('temperature') as number;

        // Load model
        let modelName = config.get('model') as string;

        let delay = config.get('delay') as number;

        return {
            endpoint,
            bearerToken,
            maxLines,
            maxTokens,
            temperature,
            modelName,
            delay
        };
    }

    // Notebook
    get notebook() {
        let config = vscode.workspace.getConfiguration('notebook');

        let includeMarkup = config.get('includeMarkup') as boolean;
        let includeCellOutputs = config.get('includeCellOutputs') as boolean;
        let cellOutputLimit = config.get('cellOutputLimit') as number;
        return {
            includeMarkup,
            includeCellOutputs,
            cellOutputLimit,
        };
    }

    get #config() {
        return vscode.workspace.getConfiguration('inference');
    };
}

export const config = new Config();