naisys/dist/config.js

Node.js Autonomous Intelligence System

import { program } from "commander";
import dotenv from "dotenv";
import * as fs from "fs";
import { readFile } from "fs/promises";
import yaml from "js-yaml";
import path from "path";
import { CommandProtection } from "./utils/enums.js";
import * as pathService from "./utils/pathService.js";
import { valueFromString } from "./utils/utilities.js";
program.argument("<agent-path>", "Path to agent configuration file").parse();
dotenv.config();
/** The system name that shows after the @ in the command prompt */
export const hostname = "naisys";
export const shellCommand = {
    /** Limits the size of files that can be read/wrote */
    outputTokenMax: 5000,
    /** The time NAISYS will wait for new shell output before giving up */
    timeoutSeconds: 15,
    /** These commands have their own timeout so the LLM doesn't have to continually waste tokens on wait commands */
    longRunningCommands: ["nmap", "traceroute", "tracepath", "mtr"],
    longRunningTimeoutSeconds: 120,
};
export const agent = loadAgentConfig();
/** Web pages loaded with llmynx will be reduced down to around this number of tokens */
export const webTokenMax = 3000;
/** Allows the LLM to end it's own session */
export const endSessionEnabled = true;
/** Inter agent communication */
export const mailEnabled = agent.mailEnabled || false;
/** The LLM optimized browser */
export const webEnabled = agent.webEnabled || false;
/** Experimental, live updating spot in the context for the LLM to put files, to avoid having to continually cat */
export const workspacesEnabled = false;
/** Experimental, allow LLM to trim it's own session context to avoid having to restart the session */
export const trimSessionEnabled = false;
/* .env is used for global configs across naisys, while agent configs are for the specific agent */
export const naisysFolder = getEnv("NAISYS_FOLDER", true);
export const websiteFolder = getEnv("WEBSITE_FOLDER");
export const localLlmUrl = getEnv("LOCAL_LLM_URL");
export const localLlmName = getEnv("LOCAL_LLM_NAME");
export const openaiApiKey = getEnv("OPENAI_API_KEY");
export const googleApiKey = getEnv("GOOGLE_API_KEY");
export const anthropicApiKey = getEnv("ANTHROPIC_API_KEY");
export const openRouterApiKey = getEnv("OPENROUTER_API_KEY");
function loadAgentConfig() {
    const config = yaml.load(fs.readFileSync(program.args[0], "utf8"));
    config.hostpath = path.resolve(program.args[0]);
    // throw if any property is undefined
    for (const key of [
        "username",
        "title",
        "shellModel",
        "agentPrompt",
        "spendLimitDollars",
        "tokenMax",
        // other properties can be undefined
    ]) {
        if (!valueFromString(config, key)) {
            throw `Agent config: Error, ${key} is not defined`;
        }
    }
    // Sanitize input
    if (!config.initialCommands) {
        config.initialCommands = [];
    }
    else if (!Array.isArray(config.initialCommands)) {
        throw `Agent config: Error, 'initialCommands' is not an array`;
    }
    config.debugPauseSeconds = config.debugPauseSeconds
        ? Number(config.debugPauseSeconds)
        : 0;
    config.wakeOnMessage = Boolean(config.wakeOnMessage);
    config.webModel || (config.webModel = config.shellModel);
    config.dreamModel || (config.dreamModel = config.shellModel);
    if (!config.commandProtection) {
        config.commandProtection = CommandProtection.None;
    }
    if (!Object.values(CommandProtection).includes(config.commandProtection)) {
        throw `Agent config: Error, 'commandProtection' is not a valid value`;
    }
    return config;
}
export const packageVersion = await getVersion();
export const binPath = getBinPath();
/** Can only get version from env variable when naisys is started with npm,
 * otherwise need to rip it from the package ourselves relative to where this file is located */
async function getVersion() {
    try {
        /* Removed for compatibility with https://bundlephobia.com/package/naisys
        const packageJson = await import(packageJsonUrl.href, {
          assert: { type: "json" },
        });*/
        const installPath = pathService.getInstallPath();
        const packageJsonPath = path.join(installPath.getHostPath(), "package.json");
        const packageJson = JSON.parse(await readFile(packageJsonPath, "utf8"));
        return packageJson.version;
    }
    catch (e) {
        return "Error getting NAISYS verison";
    }
}
function getEnv(key, required) {
    const value = process.env[key];
    if (!value && required) {
        throw `Config: Error, .env ${key} is not defined`;
    }
    return value;
}
export function resolveConfigVars(templateString) {
    let resolvedString = templateString;
    resolvedString = resolveTemplateVars(resolvedString, "agent", agent);
    resolvedString = resolveTemplateVars(resolvedString, "env", process.env);
    return resolvedString;
}
function resolveTemplateVars(templateString, allowedVarString, mappedVar) {
    const pattern = new RegExp(`\\$\\{${allowedVarString}\\.([^}]+)\\}`, "g");
    return templateString.replace(pattern, (match, key) => {
        const value = valueFromString(mappedVar, key);
        if (value === undefined) {
            throw `Agent config: Error, ${key} is not defined`;
        }
        return value;
    });
}
function getBinPath() {
    // C:/git/naisys/dist/config.js
    let binPath = new URL("../bin", import.meta.url).pathname;
    if (binPath.startsWith("/C:")) {
        binPath = "/mnt/c" + binPath.substring(3);
    }
    return binPath;
}
//# sourceMappingURL=config.js.map