@metrio-ai/cli 1.0.0

package/dist/commands/shared.js

@@ -0,0 +1,328 @@
+"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;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.readMessagesFile = readMessagesFile;
+exports.buildMessages = buildMessages;
+exports.loadEnv = loadEnv;
+exports.resolveApiKey = resolveApiKey;
+exports.resolveBaseUrl = resolveBaseUrl;
+exports.parseJsonFlag = parseJsonFlag;
+exports.resolvePromptId = resolvePromptId;
+exports.parseJsonlObjects = parseJsonlObjects;
+exports.caseIdFor = caseIdFor;
+exports.readCasesSource = readCasesSource;
+exports.mapWithConcurrency = mapWithConcurrency;
+exports.emitBatchResults = emitBatchResults;
+/**
+ * Shared CLI plumbing: env-file loading, credential resolution, and the
+ * conversion from the user-facing messages file format into the SDK's message
+ * shape. Every `metrio` subcommand builds on these.
+ */
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+/** Maps common file extensions to a MetrioAI file content type and MIME type. */
+const FILE_TYPES = {
+    '.png': { type: 'image', mime: 'image/png' },
+    '.jpg': { type: 'image', mime: 'image/jpeg' },
+    '.jpeg': { type: 'image', mime: 'image/jpeg' },
+    '.gif': { type: 'image', mime: 'image/gif' },
+    '.webp': { type: 'image', mime: 'image/webp' },
+    '.pdf': { type: 'pdf', mime: 'application/pdf' },
+};
+/**
+ * Minimal `.env` loader so the CLI stays dependency-free. When `required` is
+ * true (an explicit --env file), a missing file is an error; otherwise the
+ * default `.env` is loaded if present and silently skipped if not.
+ */
+function loadDotEnv(envPath, required) {
+    if (!fs.existsSync(envPath)) {
+        if (required)
+            throw new Error(`Env file not found: ${envPath}`);
+        return;
+    }
+    let contents;
+    try {
+        contents = fs.readFileSync(envPath, 'utf8');
+    }
+    catch (err) {
+        // An explicit --env file must be readable; for the default .env, fall back
+        // to whatever is already in the environment instead of crashing.
+        if (required) {
+            throw new Error(`Cannot read env file: ${envPath} (${err.message})`);
+        }
+        return;
+    }
+    const lines = contents.split('\n');
+    for (const rawLine of lines) {
+        const line = rawLine.trim();
+        if (!line || line.startsWith('#'))
+            continue;
+        const eq = line.indexOf('=');
+        if (eq === -1)
+            continue;
+        const key = line.slice(0, eq).trim();
+        let value = line.slice(eq + 1).trim();
+        // Strip surrounding single or double quotes.
+        if ((value.startsWith('"') && value.endsWith('"')) ||
+            (value.startsWith("'") && value.endsWith("'"))) {
+            value = value.slice(1, -1);
+        }
+        // Existing environment variables take precedence over the .env file.
+        if (process.env[key] === undefined) {
+            process.env[key] = value;
+        }
+    }
+}
+/** Reads and validates the messages file, returning the raw input entries. */
+function readMessagesFile(file) {
+    let raw;
+    try {
+        raw = fs.readFileSync(file, 'utf8');
+    }
+    catch (err) {
+        throw new Error(`Cannot read messages file: ${file} (${err.message})`);
+    }
+    let parsed;
+    try {
+        parsed = JSON.parse(raw);
+    }
+    catch (err) {
+        throw new Error(`Messages file is not valid JSON: ${err.message}`);
+    }
+    if (!Array.isArray(parsed)) {
+        throw new Error('Messages file must contain a JSON array of messages');
+    }
+    return parsed;
+}
+/** Builds a file content block by reading and base64-encoding the file. */
+function buildFileContent(input, baseDir) {
+    const filePath = path.resolve(baseDir, input.file);
+    let data;
+    try {
+        data = fs.readFileSync(filePath);
+    }
+    catch (err) {
+        throw new Error(`Cannot read attached file: ${filePath} (${err.message})`);
+    }
+    const ext = path.extname(filePath).toLowerCase();
+    const known = FILE_TYPES[ext];
+    const type = input.fileType ?? known?.type ?? 'binary';
+    const mime = input.mime ?? known?.mime ?? 'application/octet-stream';
+    return { type, mime, data: data.toString('base64') };
+}
+/**
+ * Converts user-friendly input messages into the SDK's message shape. A single
+ * input entry may expand into two messages when it carries both text and a file.
+ */
+function buildMessages(inputs, baseDir) {
+    const messages = [];
+    inputs.forEach((input, index) => {
+        if (!input || typeof input !== 'object') {
+            throw new Error(`Message #${index} must be an object`);
+        }
+        if (input.role !== 'system' && input.role !== 'user' && input.role !== 'assistant') {
+            throw new Error(`Message #${index} has invalid role: ${JSON.stringify(input.role)}`);
+        }
+        const text = input.content ?? input.text;
+        const hasText = typeof text === 'string' && text.trim() !== '';
+        const hasFile = typeof input.file === 'string' && input.file !== '';
+        if (!hasText && !hasFile) {
+            throw new Error(`Message #${index} must have "content" text and/or a "file"`);
+        }
+        if (hasText) {
+            messages.push({ role: input.role, content: { type: 'text', text: text } });
+        }
+        if (hasFile) {
+            messages.push({ role: input.role, content: buildFileContent(input, baseDir) });
+        }
+    });
+    return messages;
+}
+/**
+ * Loads the env file. An explicit --env file must exist; the default `.env`
+ * is loaded if present and silently skipped otherwise.
+ */
+function loadEnv(envFile) {
+    const envPath = envFile ? path.resolve(envFile) : path.join(process.cwd(), '.env');
+    loadDotEnv(envPath, envFile !== undefined);
+}
+/** Resolves the API key from a flag then the environment, throwing if absent. */
+function resolveApiKey(flagKey) {
+    const apiKey = flagKey ||
+        process.env.METRIOAI_API_KEY ||
+        process.env.METRIO_API_KEY ||
+        process.env.METRIO_PROJECT_API_KEY;
+    if (!apiKey) {
+        throw new Error('No API key found. Set METRIOAI_API_KEY in .env or pass --api-key.');
+    }
+    return apiKey;
+}
+/** Resolves the base URL from a flag then the environment (undefined = default). */
+function resolveBaseUrl(flagUrl) {
+    return flagUrl || process.env.METRIOAI_BASE_URL || process.env.METRIO_BASE_URL;
+}
+/** Parses a JSON CLI flag, throwing a friendly error on malformed input. */
+function parseJsonFlag(raw, flag) {
+    try {
+        return JSON.parse(raw);
+    }
+    catch (err) {
+        throw new Error(`${flag} must be valid JSON: ${err.message}`);
+    }
+}
+/**
+ * Resolves the prompt id to run: the `--prompt-id`/`-p` flag if given,
+ * otherwise `METRIO_PROMPT_ID` or `METRIOAI_PROMPT_ID` from the environment
+ * (checked in that order). Throws if neither source provides one, or if the
+ * environment value is not a positive integer.
+ */
+function resolvePromptId(flagId) {
+    if (flagId !== undefined)
+        return flagId;
+    const raw = process.env.METRIO_PROMPT_ID ?? process.env.METRIOAI_PROMPT_ID;
+    if (raw !== undefined && raw.trim() !== '') {
+        const id = Number(raw);
+        if (!Number.isInteger(id) || id <= 0) {
+            throw new Error(`METRIO_PROMPT_ID must be a positive integer, got "${raw}"`);
+        }
+        return id;
+    }
+    throw new Error('Missing prompt id. Pass --prompt-id or set METRIO_PROMPT_ID in your env file.');
+}
+/**
+ * Parses a JSONL cases file into objects. Blank lines and `//` comment lines
+ * are ignored so case files stay hand-editable; ignored lines still count
+ * toward the line numbers used for error messages and generated ids.
+ *
+ * Field validation belongs to each command — this only guarantees every entry
+ * is a JSON object and knows where it came from.
+ */
+function parseJsonlObjects(raw) {
+    const entries = [];
+    raw.split('\n').forEach((rawLine, index) => {
+        const line = rawLine.trim();
+        if (!line || line.startsWith('//'))
+            return;
+        let parsed;
+        try {
+            parsed = JSON.parse(line);
+        }
+        catch (err) {
+            throw new Error(`Case on line ${index + 1} is not valid JSON: ${err.message}`);
+        }
+        if (!parsed || typeof parsed !== 'object' || Array.isArray(parsed)) {
+            throw new Error(`Case on line ${index + 1} must be a JSON object`);
+        }
+        entries.push({ line: index + 1, value: parsed });
+    });
+    if (entries.length === 0) {
+        throw new Error('Cases file contains no cases');
+    }
+    return entries;
+}
+/**
+ * The id a case is known by: its own `id` when it has a non-empty one,
+ * otherwise `case-<line>` so every output row stays addressable.
+ */
+function caseIdFor(value, line) {
+    return typeof value.id === 'string' && value.id !== '' ? value.id : `case-${line}`;
+}
+/** Reads all of stdin as UTF-8 — backs `--cases -`. */
+async function readStdin() {
+    const chunks = [];
+    for await (const chunk of process.stdin)
+        chunks.push(Buffer.from(chunk));
+    return Buffer.concat(chunks).toString('utf8');
+}
+/** Reads a `--cases` argument, where `-` means stdin. */
+async function readCasesSource(casesFile) {
+    if (casesFile === '-') {
+        return { raw: await readStdin(), baseDir: process.cwd() };
+    }
+    try {
+        return {
+            raw: fs.readFileSync(casesFile, 'utf8'),
+            baseDir: path.dirname(path.resolve(casesFile)),
+        };
+    }
+    catch (err) {
+        throw new Error(`Cannot read cases file: ${casesFile} (${err.message})`);
+    }
+}
+/**
+ * Runs `worker` over `items` with at most `limit` in flight, resolving to
+ * results in input order — batch output is diffed across runs, so completion
+ * order must not leak into it.
+ */
+async function mapWithConcurrency(items, limit, worker) {
+    const results = new Array(items.length);
+    let cursor = 0;
+    const runners = Array.from({ length: Math.max(1, Math.min(limit, items.length)) }, async () => {
+        while (cursor < items.length) {
+            const index = cursor++;
+            results[index] = await worker(items[index], index);
+        }
+    });
+    await Promise.all(runners);
+    return results;
+}
+/**
+ * Emits a batch's result rows as JSONL — to `--out` when given, otherwise to
+ * stdout — then reports the tally on stderr and sets the exit code.
+ *
+ * The batch has already run by this point, so a failed `--out` write falls
+ * back to stdout rather than throwing the results away. A failing case is a
+ * result, not a crash: every row is still emitted, and only the exit code
+ * (1 when any case failed) marks the batch as unsuccessful.
+ */
+function emitBatchResults(rows, out) {
+    const body = `${rows.map((row) => JSON.stringify(row)).join('\n')}\n`;
+    if (out) {
+        try {
+            fs.writeFileSync(path.resolve(out), body);
+        }
+        catch (err) {
+            process.stderr.write(`Warning: could not write --out file ${out} (${err.message}); printing results to stdout instead\n`);
+            process.stdout.write(body);
+        }
+    }
+    else {
+        process.stdout.write(body);
+    }
+    const failed = rows.filter((row) => !row.ok).length;
+    process.stderr.write(`${rows.length} cases: ${rows.length - failed} ok, ${failed} failed\n`);
+    if (failed > 0)
+        process.exitCode = 1;
+}

package/dist/commands/tools.js

@@ -0,0 +1,114 @@
+"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;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runToolsCommand = runToolsCommand;
+exports.runCallCommand = runCallCommand;
+const fs = __importStar(require("fs"));
+const client_1 = require("../console/client");
+const errors_1 = require("../errors");
+const TOOLS_HELP = `metrio tools - list the console tools your login can call
+
+Usage: metrio tools [--json]
+`;
+const CALL_HELP = `metrio call - call one console tool
+
+Usage: metrio call <tool> [-f <args.json> | -f - | --args '<json>']
+
+Arguments are a JSON object matching the tool's schema (see "metrio tools --json").
+With no arguments, {} is sent.
+`;
+async function runToolsCommand(argv) {
+    if (argv.includes('-h') || argv.includes('--help')) {
+        process.stdout.write(TOOLS_HELP);
+        return;
+    }
+    const asJson = argv.includes('--json');
+    const unknown = argv.find((a) => a !== '--json');
+    if (unknown)
+        throw new errors_1.CliError(`Unknown argument: ${unknown}`);
+    const tools = await new client_1.ConsoleClient(await (0, client_1.resolveConsoleAuth)()).listTools();
+    if (asJson) {
+        process.stdout.write(`${JSON.stringify(tools, null, 2)}\n`);
+        return;
+    }
+    const nameWidth = Math.max(...tools.map((t) => t.name.length), 0) + 2;
+    const scopeWidth = Math.max(...tools.map((t) => t.scope.length), 0) + 2;
+    for (const tool of tools) {
+        const summary = tool.description.split('\n')[0];
+        process.stdout.write(`${tool.name.padEnd(nameWidth)}${tool.scope.padEnd(scopeWidth)}${summary}\n`);
+    }
+}
+function parseArgsObject(raw, source) {
+    let value;
+    try {
+        value = JSON.parse(raw);
+    }
+    catch (err) {
+        throw new errors_1.CliError(`${source} must be valid JSON: ${err.message}`);
+    }
+    if (typeof value !== 'object' || value === null || Array.isArray(value)) {
+        throw new errors_1.CliError(`${source} must be a JSON object`);
+    }
+    return value;
+}
+async function runCallCommand(argv, readStdin = () => fs.readFileSync(0, 'utf8')) {
+    if (argv.includes('-h') || argv.includes('--help')) {
+        process.stdout.write(CALL_HELP);
+        return;
+    }
+    const [name, ...rest] = argv;
+    if (!name || name.startsWith('-'))
+        throw new errors_1.CliError('Missing tool name. Usage: metrio call <tool> [-f args.json]');
+    let file;
+    let inline;
+    for (let i = 0; i < rest.length; i += 1) {
+        const arg = rest[i];
+        if ((arg === '-f' || arg === '--file') && rest[i + 1] !== undefined)
+            file = rest[(i += 1)];
+        else if (arg === '--args' && rest[i + 1] !== undefined)
+            inline = rest[(i += 1)];
+        else
+            throw new errors_1.CliError(`Unknown argument: ${arg}`);
+    }
+    if (file !== undefined && inline !== undefined)
+        throw new errors_1.CliError('Use either -f or --args, not both');
+    const args = file !== undefined
+        ? parseArgsObject(file === '-' ? readStdin() : fs.readFileSync(file, 'utf8'), file === '-' ? 'stdin' : file)
+        : inline !== undefined
+            ? parseArgsObject(inline, '--args')
+            : {};
+    const data = await new client_1.ConsoleClient(await (0, client_1.resolveConsoleAuth)()).callTool(name, args);
+    process.stdout.write(`${JSON.stringify(data, null, 2)}\n`);
+}