nicechat
Version:
An extensible AI chat framework for OpenAi's models
172 lines (171 loc) • 7.68 kB
JavaScript
;
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 __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 __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.readLine = readLine;
exports.logger = logger;
const chalk_1 = __importDefault(require("chalk"));
const readline = __importStar(require("node:readline/promises"));
const openai_1 = __importDefault(require("openai"));
const settings_1 = require("./helpers/settings");
const anthropic_1 = require("./providers/anthropic");
const deepseek_1 = require("./providers/deepseek");
const openai_2 = require("./providers/openai");
const replicate_1 = require("./providers/replicate");
const EXIT_COMMANDS = ["exit", "quit", "q", "bye"];
const OPENAI_API_KEY = process.env.OPENAI_API_KEY;
const REPLICATE_API_KEY = process.env.REPLICATE_API_KEY;
const ANTHROPIC_API_KEY = process.env.ANTHROPIC_API_KEY;
const DEEPSEEK_API_KEY = process.env.DEEPSEEK_API_KEY;
const OPENROUTER_API_KEY = process.env.OPENROUTER_API_KEY;
function throwMissingEnv(envVar) {
throw new Error(`${envVar} environment variable is not set. Please set it to use this feature.`);
}
const NiceChat = {
run() {
return __awaiter(this, void 0, void 0, function* () {
const settings = yield (0, settings_1.readSettings)();
const args = process.argv.slice(2);
const command = !args[0] || args[0].startsWith("--") ? "chat" : args[0];
if (command === "list-models") {
if (!OPENAI_API_KEY) {
return throwMissingEnv("OPENAI_API_KEY");
}
yield listModels(OPENAI_API_KEY);
}
else if (command === "chat") {
const profileName = args[1] || "default";
const profile = settings.profiles[profileName];
if (!profile) {
throw new Error(`Profile ${profileName} not found. Possible values: ` +
Object.keys(settings.profiles).join(", "));
}
if (profile.vendor === "replicate") {
if (!REPLICATE_API_KEY) {
return throwMissingEnv("REPLICATE_API_KEY");
}
yield (0, replicate_1.chat)(REPLICATE_API_KEY, profile.model, profile.system);
}
else if (profile.vendor === "anthropic") {
if (!ANTHROPIC_API_KEY) {
return throwMissingEnv("ANTHROPIC_API_KEY");
}
yield (0, anthropic_1.chat)(ANTHROPIC_API_KEY, profile.model, profile.system);
}
else if (profile.vendor === "deepseek") {
if (!DEEPSEEK_API_KEY) {
return throwMissingEnv("DEEPSEEK_API_KEY");
}
yield (0, deepseek_1.chatDeepSeek)(DEEPSEEK_API_KEY, profile.model, profile.system);
}
else if (profile.vendor === "openai") {
if (!OPENAI_API_KEY) {
return throwMissingEnv("OPENAI_API_KEY");
}
yield (0, openai_2.chatOpenai)(OPENAI_API_KEY, profile.model, profile.system);
}
else if (profile.vendor === "openrouter") {
if (!OPENROUTER_API_KEY) {
return throwMissingEnv("OPENROUTER_API_KEY");
}
yield (0, openai_2.chatOpenai)(OPENROUTER_API_KEY, profile.model, profile.system, "https://openrouter.ai/api/v1");
}
else {
throw new Error("Unknown vendor: " + profile.vendor);
}
}
else if (command === "help") {
printHelp();
}
else {
console.log("Unknown command: " + command);
}
process.exit(0);
function printHelp() {
console.log("Available commands:");
const c = chalk_1.default.bold;
console.log("");
console.log(` ${c("chat [profile]")} start chat with the assistant (default)`);
console.log("");
console.log(` ${c("list-models")} list available models`);
console.log(` ${c("help")} show this help`);
}
function listModels(apiKey) {
return __awaiter(this, void 0, void 0, function* () {
const openai = new openai_1.default({ apiKey });
let models = null;
while (models === null || models.hasNextPage()) {
models = yield openai.models.list();
for (let m of models.data) {
console.log(m.id);
}
}
});
}
});
},
};
function readLine() {
return __awaiter(this, void 0, void 0, function* () {
const rl = readline.createInterface({
input: process.stdin,
output: process.stdout,
});
const textUntrimmed = yield rl.question(chalk_1.default.yellow("> "));
rl.close();
const text = textUntrimmed.trim();
if (EXIT_COMMANDS.includes(text)) {
process.exit(0);
}
return text;
});
}
exports.default = NiceChat;
function logger(prefix) {
return function log(msg) {
console.log(`${chalk_1.default.blueBright(prefix)} ${msg}`);
};
}