giga-code/dist/utils/added-models.js

A personal AI CLI assistant powered by Grok for local development.

"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.getOpenRouterProvider = exports.setOpenRouterProvider = exports.isModelAdded = exports.getAllAddedModels = exports.getAddedModelsForProvider = exports.deleteModel = exports.addModel = exports.saveAddedModels = exports.loadAddedModels = void 0;
const fs = __importStar(require("fs"));
const path = __importStar(require("path"));
const os = __importStar(require("os"));
const getStorageFile = () => {
    const homeDir = os.homedir();
    const gigaDir = path.join(homeDir, '.giga');
    // Create .giga directory if it doesn't exist
    if (!fs.existsSync(gigaDir)) {
        fs.mkdirSync(gigaDir, { mode: 0o700 });
    }
    return path.join(gigaDir, 'added-models.json');
};
function loadAddedModels() {
    try {
        const storageFile = getStorageFile();
        if (!fs.existsSync(storageFile)) {
            return [];
        }
        const data = JSON.parse(fs.readFileSync(storageFile, 'utf8'));
        return data.models || [];
    }
    catch (error) {
        console.error('Error loading added models:', error);
        return [];
    }
}
exports.loadAddedModels = loadAddedModels;
function saveAddedModels(models) {
    try {
        const storageFile = getStorageFile();
        const data = { models };
        fs.writeFileSync(storageFile, JSON.stringify(data, null, 2), { mode: 0o600 });
    }
    catch (error) {
        console.error('Error saving added models:', error);
    }
}
exports.saveAddedModels = saveAddedModels;
function addModel(modelName, providerName) {
    const models = loadAddedModels();
    // Check if model already exists
    const exists = models.some(m => m.modelName === modelName && m.providerName === providerName);
    if (!exists) {
        const newModel = {
            modelName,
            providerName,
            dateAdded: new Date().toISOString(),
        };
        models.push(newModel);
        saveAddedModels(models);
    }
}
exports.addModel = addModel;
function deleteModel(modelName, providerName) {
    const models = loadAddedModels();
    const initialLength = models.length;
    const filteredModels = models.filter(m => !(m.modelName === modelName && m.providerName === providerName));
    if (filteredModels.length < initialLength) {
        saveAddedModels(filteredModels);
        return true;
    }
    return false;
}
exports.deleteModel = deleteModel;
function getAddedModelsForProvider(providerName) {
    const models = loadAddedModels();
    return models.filter(m => m.providerName === providerName);
}
exports.getAddedModelsForProvider = getAddedModelsForProvider;
function getAllAddedModels() {
    return loadAddedModels();
}
exports.getAllAddedModels = getAllAddedModels;
function isModelAdded(modelName, providerName) {
    const models = loadAddedModels();
    return models.some(m => m.modelName === modelName && m.providerName === providerName);
}
exports.isModelAdded = isModelAdded;
function setOpenRouterProvider(modelName, openRouterProvider) {
    const models = loadAddedModels();
    console.log(`DEBUG: Looking for model ${modelName} with provider OpenRouter`);
    console.log(`DEBUG: Available models:`, models.map(m => `${m.modelName} (${m.providerName})`));
    const modelIndex = models.findIndex(m => m.modelName === modelName && m.providerName === 'OpenRouter');
    console.log(`DEBUG: Found model at index: ${modelIndex}`);
    if (modelIndex === -1) {
        console.log(`DEBUG: Model ${modelName} not found or not an OpenRouter model`);
        return false; // Model not found or not an OpenRouter model
    }
    console.log(`DEBUG: Setting openRouterProvider to ${openRouterProvider} for ${modelName}`);
    models[modelIndex].openRouterProvider = openRouterProvider;
    saveAddedModels(models);
    return true;
}
exports.setOpenRouterProvider = setOpenRouterProvider;
function getOpenRouterProvider(modelName) {
    const models = loadAddedModels();
    const model = models.find(m => m.modelName === modelName && m.providerName === 'OpenRouter');
    return model?.openRouterProvider || null;
}
exports.getOpenRouterProvider = getOpenRouterProvider;
//# sourceMappingURL=added-models.js.map