osr-mono/packages/osr-code-bot/models/index.js
2025-02-01 14:17:49 +01:00

134 lines
10 KiB
JavaScript

"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 () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__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.all = exports.models = exports.models_dist = void 0;
const chalk_1 = __importDefault(require("chalk"));
const path = __importStar(require("path"));
const read_1 = require("@plastichub/fs/read");
const exists_1 = require("@plastichub/fs/exists");
const index_1 = require("../index");
const openrouter_1 = require("./openrouter");
const openai_1 = require("./openai");
const openai_2 = require("../models/openai");
const osr_commons_1 = require("@plastichub/osr-commons");
const openai_3 = require("./cache/openai");
const openrouter_2 = require("./cache/openrouter");
const models_dist = () => {
let or_models = openrouter_2.models;
let oai_models = openai_3.models;
let deepseek_models = [
{
"id": "deepseek-chat",
"name": "deepseek-chat"
},
{
"id": "deepseek-reasoner",
"name": "deepseek-reasoner"
},
];
const modelsOpenAIPath = path.resolve((0, index_1.module_root)(), 'openai.json');
if ((0, exists_1.sync)(modelsOpenAIPath)) {
oai_models = (0, read_1.sync)(modelsOpenAIPath, 'json');
}
const modelsRouterPath = path.resolve((0, index_1.module_root)(), 'openrouter.json');
if ((0, exists_1.sync)(modelsRouterPath)) {
or_models = (0, read_1.sync)(modelsRouterPath, 'json');
}
const models = [];
models.push(chalk_1.default.magenta.bold('\n OpenRouter models:\n'));
models.push(...(0, openrouter_1.listModelsAsStrings)(or_models));
models.push(chalk_1.default.magenta.bold('\n OpenAI models:\n'));
models.push(...(0, openai_1.listModelsAsStrings)(oai_models));
models.push('-----\n');
models.push(chalk_1.default.magenta.bold('\n Deepseek models:\n'));
models.push(...(0, openai_1.listModelsAsStrings)(deepseek_models));
models.push('-----\n');
return models;
};
exports.models_dist = models_dist;
const models = () => {
const models = [];
const openRouterPath = path.resolve(openrouter_1.CACHE_PATH);
if (!(0, exists_1.sync)(openRouterPath)) {
(0, openrouter_1.fetchOpenRouterModels)();
}
if ((0, exists_1.sync)(openRouterPath)) {
const modelData = (0, read_1.sync)(openRouterPath, 'json');
models.push(chalk_1.default.magenta.bold('\n OpenRouter models:\n'));
models.push(...(0, openrouter_1.listModelsAsStrings)(modelData.models));
}
index_1.logger.debug('Openrouter models cache: ', openai_1.CACHE_PATH);
const openAIPath = path.resolve(openai_1.CACHE_PATH);
const config = (0, osr_commons_1.CONFIG_DEFAULT)();
if (!(0, exists_1.sync)(openAIPath) && config?.openai?.key) {
(0, openai_2.fetchOpenAIModels)(config.openai.key);
}
if ((0, exists_1.sync)(openAIPath)) {
const modelData = (0, read_1.sync)(openAIPath, 'json');
models.push(chalk_1.default.magenta.bold('\n OpenAI models:\n'));
models.push(...(0, openai_1.listModelsAsStrings)(modelData.models));
}
index_1.logger.debug('OpenAI models cache: ', openai_1.CACHE_PATH);
models.push('-----\n');
return models;
};
exports.models = models;
const all = () => {
let models = [];
const openRouterPath = path.resolve(openrouter_1.CACHE_PATH);
if (!(0, exists_1.sync)(openRouterPath)) {
(0, openrouter_1.fetchOpenRouterModels)();
}
if ((0, exists_1.sync)(openRouterPath)) {
const modelData = (0, read_1.sync)(openRouterPath, 'json');
models = models.concat(modelData.models);
}
const openAIPath = path.resolve(openai_1.CACHE_PATH);
const config = (0, osr_commons_1.CONFIG_DEFAULT)();
if (!(0, exists_1.sync)(openAIPath) && config?.openai?.key) {
(0, openai_2.fetchOpenAIModels)(config.openai.key);
}
if ((0, exists_1.sync)(openAIPath)) {
const modelData = (0, read_1.sync)(openAIPath, 'json');
models.push(chalk_1.default.magenta.bold('\n OpenAI models:\n'));
models = models.concat(modelData.models);
}
return models;
};
exports.all = all;
//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi9zcmMvbW9kZWxzL2luZGV4LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUNBLGtEQUF5QjtBQUN6QiwyQ0FBNEI7QUFFNUIsOENBQWtEO0FBQ2xELGtEQUFzRDtBQUV0RCxvQ0FBOEM7QUFDOUMsNkNBQXVMO0FBQ3ZMLHFDQUFnSjtBQUNoSiw2Q0FBb0Q7QUFDcEQseURBQXdEO0FBRXhELDJDQUF1RDtBQUN2RCxtREFBK0Q7QUFFeEQsTUFBTSxXQUFXLEdBQUcsR0FBRyxFQUFFO0lBRTlCLElBQUksU0FBUyxHQUFHLG1CQUFnQixDQUFBO0lBQ2hDLElBQUksVUFBVSxHQUFHLGVBQVksQ0FBQTtJQUM3QixJQUFJLGVBQWUsR0FBRztRQUNwQjtZQUNFLElBQUksRUFBRSxlQUFlO1lBQ3JCLE1BQU0sRUFBRSxlQUFlO1NBQ3hCO1FBQ0Q7WUFDRSxJQUFJLEVBQUUsbUJBQW1CO1lBQ3pCLE1BQU0sRUFBRSxtQkFBbUI7U0FDNUI7S0FDRixDQUFBO0lBRUQsTUFBTSxnQkFBZ0IsR0FBRyxJQUFJLENBQUMsT0FBTyxDQUFDLElBQUEsbUJBQVcsR0FBRSxFQUFFLGFBQWEsQ0FBQyxDQUFBO0lBQ25FLElBQUksSUFBQSxhQUFNLEVBQUMsZ0JBQWdCLENBQUMsRUFBRSxDQUFDO1FBQzdCLFVBQVUsR0FBRyxJQUFBLFdBQUksRUFBQyxnQkFBZ0IsRUFBRSxNQUFNLENBQVEsQ0FBQTtJQUNwRCxDQUFDO0lBRUQsTUFBTSxnQkFBZ0IsR0FBRyxJQUFJLENBQUMsT0FBTyxDQUFDLElBQUEsbUJBQVcsR0FBRSxFQUFFLGlCQUFpQixDQUFDLENBQUE7SUFDdkUsSUFBSSxJQUFBLGFBQU0sRUFBQyxnQkFBZ0IsQ0FBQyxFQUFFLENBQUM7UUFDN0IsU0FBUyxHQUFHLElBQUEsV0FBSSxFQUFDLGdCQUFnQixFQUFFLE1BQU0sQ0FBUSxDQUFBO0lBQ25ELENBQUM7SUFDRCxNQUFNLE1BQU0sR0FBYSxFQUFFLENBQUE7SUFDM0IsTUFBTSxDQUFDLElBQUksQ0FBQyxlQUFLLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyx5QkFBeUIsQ0FBQyxDQUFDLENBQUE7SUFDMUQsTUFBTSxDQUFDLElBQUksQ0FBQyxHQUFHLElBQUEsZ0NBQTZCLEVBQUMsU0FBZ0IsQ0FBQyxDQUFDLENBQUE7SUFFL0QsTUFBTSxDQUFDLElBQUksQ0FBQyxlQUFLLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxxQkFBcUIsQ0FBQyxDQUFDLENBQUE7SUFDdEQsTUFBTSxDQUFDLElBQUksQ0FBQyxHQUFHLElBQUEsNEJBQXlCLEVBQUMsVUFBaUIsQ0FBQyxDQUFDLENBQUE7SUFDNUQsTUFBTSxDQUFDLElBQUksQ0FBQyxTQUFTLENBQUMsQ0FBQTtJQUV0QixNQUFNLENBQUMsSUFBSSxDQUFDLGVBQUssQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLHVCQUF1QixDQUFDLENBQUMsQ0FBQTtJQUN4RCxNQUFNLENBQUMsSUFBSSxDQUFDLEdBQUcsSUFBQSw0QkFBeUIsRUFBQyxlQUFzQixDQUFDLENBQUMsQ0FBQTtJQUNqRSxNQUFNLENBQUMsSUFBSSxDQUFDLFNBQVMsQ0FBQyxDQUFBO0lBQ3RCLE9BQU8sTUFBTSxDQUFBO0FBQ2YsQ0FBQyxDQUFBO0FBcENZLFFBQUEsV0FBVyxlQW9DdkI7QUFJTSxNQUFNLE1BQU0sR0FBRyxHQUFHLEVBQUU7SUFDekIsTUFBTSxNQUFNLEdBQWEsRUFBRSxDQUFBO0lBQzNCLE1BQU0sY0FBYyxHQUFHLElBQUksQ0FBQyxPQUFPLENBQUMsdUJBQXFCLENBQUMsQ0FBQTtJQUMxRCxJQUFJLENBQUMsSUFBQSxhQUFNLEVBQUMsY0FBYyxDQUFDLEVBQUUsQ0FBQztRQUM1QixJQUFBLGtDQUFxQixHQUFFLENBQUE7SUFDekIsQ0FBQztJQUNELElBQUksSUFBQSxhQUFNLEVBQUMsY0FBYyxDQUFDLEVBQUUsQ0FBQztRQUMzQixNQUFNLFNBQVMsR0FBMkIsSUFBQSxXQUFJLEVBQUMsY0FBYyxFQUFFLE1BQU0sQ0FBMkIsQ0FBQTtRQUNoRyxNQUFNLENBQUMsSUFBSSxDQUFDLGVBQUssQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLHlCQUF5QixDQUFDLENBQUMsQ0FBQTtRQUMxRCxNQUFNLENBQUMsSUFBSSxDQUFDLEdBQUcsSUFBQSxnQ0FBNkIsRUFBQyxTQUFTLENBQUMsTUFBTSxDQUFDLENBQUMsQ0FBQTtJQUNqRSxDQUFDO0lBQ0QsY0FBTSxDQUFDLEtBQUssQ0FBQywyQkFBMkIsRUFBRSxtQkFBaUIsQ0FBQyxDQUFBO0lBRTVELE1BQU0sVUFBVSxHQUFHLElBQUksQ0FBQyxPQUFPLENBQUMsbUJBQWlCLENBQUMsQ0FBQTtJQUNsRCxNQUFNLE1BQU0sR0FBRyxJQUFBLDRCQUFjLEdBQVMsQ0FBQTtJQUN0QyxJQUFJLENBQUMsSUFBQSxhQUFNLEVBQUMsVUFBVSxDQUFDLElBQUksTUFBTSxFQUFFLE1BQU0sRUFBRSxHQUFHLEVBQUUsQ0FBQztRQUMvQyxJQUFBLDBCQUFpQixFQUFDLE1BQU0sQ0FBQyxNQUFNLENBQUMsR0FBRyxDQUFDLENBQUE7SUFDdEMsQ0FBQztJQUVELElBQUksSUFBQSxhQUFNLEVBQUMsVUFBVSxDQUFDLEVBQUUsQ0FBQztRQUN2QixNQUFNLFNBQVMsR0FBdUIsSUFBQSxXQUFJLEVBQUMsVUFBVSxFQUFFLE1BQU0sQ0FBdUIsQ0FBQTtRQUNwRixNQUFNLENBQUMsSUFBSSxDQUFDLGVBQUssQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLHFCQUFxQixDQUFDLENBQUMsQ0FBQTtRQUN0RCxNQUFNLENBQUMsSUFBSSxDQUFDLEdBQUcsSUFBQSw0QkFBeUIsRUFBQyxTQUFTLENBQUMsTUFBTSxDQUFDLENBQUMsQ0FBQTtJQUM3RCxDQUFDO0lBQ0QsY0FBTSxDQUFDLEtBQUssQ0FBQyx1QkFBdUIsRUFBRSxtQkFBaUIsQ0FBQyxDQUFBO0lBQ3hELE1BQU0sQ0FBQyxJQUFJLENBQUMsU0FBUyxDQUFDLENBQUE7SUFDdEIsT0FBTyxNQUFNLENBQUE7QUFDZixDQUFDLENBQUE7QUEzQlksUUFBQSxNQUFNLFVBMkJsQjtBQUVNLE1BQU0sR0FBRyxHQUFHLEdBQUcsRUFBRTtJQUN0QixJQUFJLE1BQU0sR0FBVSxFQUFFLENBQUE7SUFDdEIsTUFBTSxjQUFjLEdBQUcsSUFBSSxDQUFDLE9BQU8sQ0FBQyx1QkFBcUIsQ0FBQyxDQUFBO0lBQzFELElBQUksQ0FBQyxJQUFBLGFBQU0sRUFBQyxjQUFjLENBQUMsRUFBRSxDQUFDO1FBQzVCLElBQUEsa0NBQXFCLEdBQUUsQ0FBQTtJQUN6QixDQUFDO0lBQ0QsSUFBSSxJQUFBLGFBQU0sRUFBQyxjQUFjLENBQUMsRUFBRSxDQUFDO1FBQzNCLE1BQU0sU0FBUyxHQUEyQixJQUFBLFdBQUksRUFBQyxjQUFjLEVBQUUsTUFBTSxDQUEyQixDQUFBO1FBQ2hHLE1BQU0sR0FBRyxNQUFNLENBQUMsTUFBTSxDQUFDLFNBQVMsQ0FBQyxNQUFNLENBQUMsQ0FBQTtJQUMxQyxDQUFDO0lBQ0QsTUFBTSxVQUFVLEdBQUcsSUFBSSxDQUFDLE9BQU8sQ0FBQyxtQkFBaUIsQ0FBQyxDQUFBO0lBQ2xELE1BQU0sTUFBTSxHQUFHLElBQUEsNEJBQWMsR0FBUyxDQUFBO0lBQ3RDLElBQUksQ0FBQyxJQUFBLGFBQU0sRUFBQyxVQUFVLENBQUMsSUFBSSxNQUFNLEVBQUUsTUFBTSxFQUFFLEdBQUcsRUFBRSxDQUFDO1FBQy9DLElBQUEsMEJBQWlCLEVBQUMsTUFBTSxDQUFDLE1BQU0sQ0FBQyxHQUFHLENBQUMsQ0FBQTtJQUN0QyxDQUFDO0lBRUQsSUFBSSxJQUFBLGFBQU0sRUFBQyxVQUFVLENBQUMsRUFBRSxDQUFDO1FBQ3ZCLE1BQU0sU0FBUyxHQUF1QixJQUFBLFdBQUksRUFBQyxVQUFVLEVBQUUsTUFBTSxDQUF1QixDQUFBO1FBQ3BGLE1BQU0sQ0FBQyxJQUFJLENBQUMsZUFBSyxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMscUJBQXFCLENBQUMsQ0FBQyxDQUFBO1FBQ3RELE1BQU0sR0FBRyxNQUFNLENBQUMsTUFBTSxDQUFDLFNBQVMsQ0FBQyxNQUFNLENBQUMsQ0FBQTtJQUMxQyxDQUFDO0lBQ0QsT0FBTyxNQUFNLENBQUE7QUFDZixDQUFDLENBQUE7QUF0QlksUUFBQSxHQUFHLE9Bc0JmIn0=