@trazum/cli 1.45.0 → 1.46.0

package/dist/index.js

@@ -1,19 +1,19 @@
 #!/usr/bin/env node
-import { readdir, readFile, stat, writeFile } from 'node:fs/promises';
+import { open, readdir, readFile, stat, writeFile } from 'node:fs/promises';
 import { join, resolve as resolvePath } from 'node:path';
 import { gunzipSync } from 'node:zlib';
-import { applyRewrites, BASELINE_FILENAME, BASELINE_VERSION, breaches, cacheableMinimum, analyzeCachePrefix, billLevers, bucketedCacheEconomics, bucketedProfile, buildHistory, buildPlan, connectorFor, CONNECTORS, normalizeAnthropicUsage, normalizeOpenAIUsage, bucketsFromRecords, evaluateWatch, firedKey, pruneRecords, recordsFromBuckets, storeInventory, storedReportFrom, verifyPlan, cacheEconomics, cacheHitRate, contextPressure, comparePrompts, compareToBaseline, computeSavings, countTokensAnthropic, DEFAULT_USAGE, detectFromSource, coverageDrift, driversBetween, explainGateFailure, assignSources, fleetRollup, labelCoverage, measuredUsage, gateMargin, GATE_MARGIN_TIGHT, estimateTokens, evaluate, extractPrompts, findExamples, formatBaseline, formatSignedUsd, formatUsd, getMessages, getModel, hasMarker, LOCALES, MAX_BASELINE_BYTES, moneyIsComparable, mostSpecificMatch, nearestName, optimize, parseBaseline, PHRASE_LANGUAGES, plannedCalls, profilePrompt, profileToCsv, profileUsage, promptId, providerFromEnv, pruneExamples, refineWithLlm, rejectionText, reorderForCache, repriceProfile, reviewAgeDays, reviewExamples, RULES, sharedPrefixes, sharesOf, SOURCE_EXTENSIONS, suggestRewrites, toOtlpMetrics, toPromptfoo, TTL_1H_MS, UNLABELLED, withExactTokenCounts, } from '@trazum/core';
+import { applyRewrites, BASELINE_FILENAME, BASELINE_VERSION, breaches, cacheableMinimum, analyzeCachePrefix, billLevers, bucketedCacheEconomics, bucketedProfile, buildHistory, buildPlan, connectorFor, CONNECTORS, normalizeAnthropicUsage, normalizeOpenAIUsage, bucketsFromRecords, evaluateWatch, firedKey, pruneRecords, recordsFromBuckets, storeInventory, storedReportFrom, verifyPlan, cacheEconomics, cacheHitRate, contextPressure, comparePrompts, compareToBaseline, computeSavings, countTokensAnthropic, DEFAULT_USAGE, detectFromSource, matchLocale, proposeInit, MIN_RATE_DAYS, parseConfig, coverageDrift, driversBetween, explainGateFailure, assignSources, fleetRollup, labelCoverage, measuredUsage, gateMargin, GATE_MARGIN_TIGHT, estimateTokens, evaluate, extractPrompts, findExamples, formatBaseline, formatSignedUsd, formatUsd, getMessages, getModel, hasMarker, LOCALES, MAX_BASELINE_BYTES, moneyIsComparable, mostSpecificMatch, nearestName, optimize, parseBaseline, PHRASE_LANGUAGES, plannedCalls, profilePrompt, profileToCsv, profileUsage, promptId, providerFromEnv, pruneExamples, refineWithLlm, rejectionText, reorderForCache, repriceProfile, reviewAgeDays, reviewExamples, RULES, sharedPrefixes, sharesOf, SOURCE_EXTENSIONS, suggestRewrites, toOtlpMetrics, toPromptfoo, TTL_1H_MS, UNLABELLED, withExactTokenCounts, } from '@trazum/core';
 import { cacheDir, cacheStats, cachingProvider, clearCache } from './suggest-cache.js';
 import { dayOf, formatGap, median, spanDays } from './time.js';
 // Everything that reads the filesystem, on its own entry point so the web
 // bundle cannot reach it. See packages/core/src/node.ts.
 import { CONFIG_FILENAME, DEFAULT_EXTENSIONS, budgetFor, BUNDLED_CATALOGUE, SAFE_FETCH_INIT, applyPricingOverlay, catalogueFromOverlay, checkedEndpoint, openrouterOverlay, detectHost, loadConfig, walkPrompts, } from '@trazum/core/node';
 import { contentAt, gitAvailable, namesByRevision, pathInRepository, repositoryRoot, revisionsFor, } from './git.js';
-import { fetchProviderUsage } from './connect.js';
+import { fetchProviderUsage, findCredential } from './connect.js';
 import { STORE_DIR, appendRecords, readStore, rewriteStore } from './store-fs.js';
 import { DEFAULT_PORT, buildServer, listen } from './serve.js';
 import { WATCH_STATE_VERSION, checkWebhook, postWebhook, readWatchState, writeWatchState, } from './watch-run.js';
-import { detectLocale, getCliMessages } from './i18n/index.js';
+import { LOCALE_ENV_VARS, detectLocale, getCliMessages } from './i18n/index.js';
 import { MAX_SUMMARY_CHARS, fitWithin, renderBlameMarkdown, renderCheckMarkdown, renderDiffMarkdown, renderRankMarkdown, renderProfileMarkdown, } from './markdown.js';
 // --------------------------------------------------------------------------
 // Presentation
@@ -309,6 +309,7 @@ const COMMAND_FLAGS = {
     diff: ['level', 'model', 'calls', 'output-tokens', 'batch', 'max-growth', 'optimized', 'markdown-out', 'all', 'prompt'],
     models: [],
     rank: ['level', 'model', 'calls', 'output-tokens', 'batch', 'disable', 'prompt', 'markdown-out'],
+    init: ['dry-run', 'yes', 'json', 'pricing', 'pricing-live'],
     where: [],
     rules: [],
     blame: ['limit', 'model', 'calls', 'output-tokens', 'batch', 'prompt', 'markdown-out'],
@@ -1007,6 +1008,378 @@ async function commandWhere(args, config, pricing, t) {
     console.log(`  ${getModel(effective).displayName} ${c.dim(reason)}`);
     console.log();
 }
+/**
+ * Where `init` looks for a usage log before it gives up and says so.
+ *
+ * A short list of the names people actually use, checked in order — not a
+ * glob over the whole tree. A first run that finds a log by searching two
+ * thousand directories has spent the patience it was given, and a log found
+ * in `vendor/fixtures/` is more likely to be somebody's test data than their
+ * bill.
+ */
+const INIT_LOG_CANDIDATES = [
+    'usage.jsonl',
+    'usage.ndjson',
+    'usage.log',
+    'logs/usage.jsonl',
+    'logs',
+    '.trazum/usage.jsonl',
+];
+/**
+ * Extensions worth reading for a provider sighting.
+ *
+ * `SOURCE_EXTENSIONS`, the same list `rank` and `doctor` walk, rather than a
+ * second copy that drifts — a language added for extraction is a language
+ * `init` should be able to detect a provider in, and one list is how that
+ * stays true. Documentation is deliberately not on it: a `.md` file quoting
+ * `from 'openai'` inside a code fence would be read as evidence, and `where`
+ * answers for a file somebody named while this answers for a repository
+ * nobody has vouched for.
+ */
+/** How many source files, and how large each may be. Both reported when they bite. */
+const INIT_MAX_SOURCE_FILES = 400;
+const INIT_MAX_SOURCE_BYTES = 256 * 1024;
+/**
+ * The first run, printed.
+ *
+ * **The arithmetic comes before the figure**, everywhere below. A tool that
+ * opens with a dollar amount nobody can check gets closed, and the reader has
+ * no reason yet to believe anything this command says — so the headline shows
+ * the calls, the model and the rate it is being compared against, and only
+ * then the money.
+ */
+function renderInit(proposal, ctx) {
+    const { t } = ctx;
+    const n = (value) => value.toLocaleString(t.numberLocale);
+    console.log();
+    console.log(c.bold(t.init.heading()));
+    console.log();
+    // 1. Where this is running.
+    console.log(`  ${t.init.host(ctx.host.displayName)}`);
+    if (ctx.host.billing === 'subscription') {
+        console.log(`  ${c.yellow(t.where.subscription(ctx.host.displayName))}`);
+    }
+    // 2. The prompts.
+    console.log(`  ${ctx.prompts.files.length === 0 ? c.dim(t.init.noPrompts()) : t.init.prompts(ctx.prompts.files.length)}`);
+    if (ctx.truncated)
+        console.log(`  ${c.dim(t.init.sourcesTruncated(INIT_MAX_SOURCE_FILES))}`);
+    // 3. The usage, or the two ways there is none.
+    if (ctx.unreadable !== null) {
+        console.log(`  ${c.red(t.init.usageUnreadable(ctx.unreadable.where, ctx.unreadable.because))}`);
+    }
+    else if (ctx.usage.length === 0) {
+        console.log(`  ${c.dim(t.init.noUsage())}`);
+    }
+    else {
+        for (const sighting of ctx.usage) {
+            console.log(`  ${t.init.usageFound(sighting.kind, sighting.where)}`);
+        }
+    }
+    console.log();
+    // 4. What the config would say, and what it will not.
+    console.log(c.bold(t.init.configHeading()));
+    if (proposal.justified.length === 0) {
+        console.log(`  ${c.dim(t.init.nothingJustified())}`);
+    }
+    for (const why of proposal.justified) {
+        console.log(`  ${c.green('+')} ${c.bold(why.key)}  ${initJustification(why, t)}`);
+    }
+    for (const decline of proposal.declined) {
+        console.log(`  ${c.dim('·')} ${c.dim(decline.key)}  ${c.dim(initDecline(decline, t))}`);
+    }
+    if (proposal.overwrites !== null && proposal.overwrites.keys.length > 0) {
+        console.log();
+        console.log(`  ${c.yellow(t.init.wouldOverwrite(proposal.overwrites.keys.join(', ')))}`);
+    }
+    console.log();
+    // 5. The single most valuable thing found — arithmetic first.
+    console.log(c.bold(t.init.findingHeading()));
+    if (proposal.headline === null) {
+        console.log(`  ${c.dim(t.init.noFinding(proposal.noHeadline ?? 'nothing-measured'))}`);
+        console.log();
+        return;
+    }
+    const { slice, lever, savingUsd, days } = proposal.headline;
+    console.log(`  ${t.init.findingCalls(n(slice.calls), slice.label, slice.modelName, days)}`);
+    console.log(`  ${t.init.findingSpent(slice.spentUsd.toFixed(2))}`);
+    if (lever !== 'batch' && slice.route !== null) {
+        console.log(`  ${t.init.findingRoute(slice.route.candidate.displayName)}`);
+    }
+    if (lever !== 'route' && slice.batch !== null) {
+        console.log(`  ${t.init.findingBatch()}`);
+    }
+    console.log(`  ${c.bold(t.init.findingTotal(savingUsd.toFixed(2), days))}`);
+    console.log(`  ${c.dim(t.init.findingNext())}`);
+    console.log();
+}
+/** Why a key was written, in one line a person reads. */
+function initJustification(why, t) {
+    switch (why.key) {
+        case 'locale':
+            return c.dim(t.init.whyLocale(why.value));
+        case 'extensions':
+            return c.dim(t.init.whyExtensions(why.value.join(' '), why.files));
+        case 'usage.model':
+            return c.dim(why.from === 'measured'
+                ? t.init.whyModelMeasured(why.value, Math.round(why.share * 100))
+                : t.init.whyModelSource(why.value, why.file, why.line));
+        case 'usage.callsPerMonth':
+            return c.dim(t.init.whyCalls(why.value, why.calls, why.days));
+        case 'usage.avgOutputTokens':
+            return c.dim(t.init.whyOutput(why.value, why.outputTokens, why.calls));
+        case 'usage.cacheHitRate':
+            return c.dim(t.init.whyCache(why.value, why.cacheReadTokens, why.inputTokens));
+    }
+}
+/** Why a key was not written, and what would settle it. */
+function initDecline(decline, t) {
+    switch (decline.why) {
+        case 'no-evidence':
+            return t.init.noModelEvidence();
+        case 'conflicting-evidence':
+            return t.init.modelConflict(decline.files.join(', '));
+        case 'provider-only':
+            return t.init.modelProviderOnly(decline.provider, decline.file);
+        case 'nothing-measured':
+            return t.init.nothingMeasured();
+        case 'window-too-short':
+            return t.init.windowTooShort(decline.days, MIN_RATE_DAYS);
+        case 'undated-calls':
+            return t.init.undatedCalls(decline.undated, decline.calls);
+        case 'not-recorded':
+            return t.init.cacheNotRecorded();
+        case 'only-you-know':
+            return t.init.batchOnlyYouKnow();
+        case 'unprovable':
+            return t.init.labelsUnprovable(decline.labels);
+        case 'a-budget-is-a-policy':
+            return decline.measuredUsd === null
+                ? t.init.budgetIsPolicy()
+                : t.init.budgetIsPolicyMeasured(decline.measuredUsd.toFixed(2), decline.days ?? 0);
+    }
+}
+/**
+ * `trazum init [dir]` — the first five minutes.
+ *
+ * The floor, not the ceiling. Everything else in this tool assumes you know
+ * which of twenty-two commands answers your question; this one assumes you
+ * have just typed `npx @trazum/cli` and have thirty seconds of patience left.
+ *
+ * It is a **detection, not a wizard**. Nothing is asked. Each step reports
+ * what it found and moves on, and the only decision is whether to write the
+ * file — which `--yes` skips and `--dry-run` refuses. A first run that
+ * interrogates somebody is a first run that gets abandoned halfway.
+ *
+ * The judgement lives in `proposeInit`, in the core, with no filesystem
+ * anywhere near it. This function's whole job is to *look*: walk for prompts,
+ * read a few source files, notice a log or a credential, and hand the lot over
+ * as data. That split is why `--dry-run` cannot drift from the real thing —
+ * they are the same call, and one of them stops before `writeFile`.
+ */
+async function commandInit(args, config, pricing, t) {
+    const root = args.positional[0] ?? '.';
+    const dryRun = args.flags.get('dry-run') === true;
+    const asJson = args.flags.get('json') === true;
+    // --- what is here -------------------------------------------------------
+    const host = detectHost();
+    const prompts = await walkPrompts(root, {
+        extensions: config.extensions ?? DEFAULT_EXTENSIONS,
+    });
+    /**
+     * Source files, read only to be asked which provider they call.
+     *
+     * Capped hard and deliberately low. `where` reads one file because somebody
+     * named it; this reads whatever is lying around, and a first run that spends
+     * forty seconds walking a monorepo has already lost. The cap is reported
+     * when it bites, because "no provider found" and "stopped looking" are
+     * different sentences.
+     */
+    const sourceWalk = await walkPrompts(root, {
+        extensions: SOURCE_EXTENSIONS,
+        maxFiles: INIT_MAX_SOURCE_FILES,
+    });
+    const sightings = [];
+    for (const relative of sourceWalk.files) {
+        const path = join(root, relative);
+        let source;
+        /**
+         * Measured and read through **one open handle**, not by path twice.
+         *
+         * A bundle or a lockfile named `.js` is not worth reading, and reading it
+         * is how this command becomes slow on exactly the repositories that need
+         * it most — so the size is checked first. Checking it with `stat(path)`
+         * and then reading `path` is two lookups of the same name, and what
+         * arrives the second time need not be what was measured the first: the
+         * bound would be enforced against a file that is no longer there. One
+         * handle, stat'ed and read, is the same inode by construction.
+         */
+        let handle;
+        try {
+            handle = await open(path, 'r');
+        }
+        catch {
+            continue;
+        }
+        try {
+            const info = await handle.stat();
+            if (info.size > INIT_MAX_SOURCE_BYTES)
+                continue;
+            source = await handle.readFile('utf8');
+        }
+        catch {
+            continue;
+        }
+        finally {
+            await handle.close();
+        }
+        const detection = detectFromSource(source, { models: pricing.models });
+        if (detection.provider !== null || detection.model !== null || detection.conflicts.length > 0) {
+            sightings.push({ file: relative, detection });
+        }
+    }
+    // --- where the usage is, if it is anywhere ------------------------------
+    const usage = [];
+    for (const candidate of INIT_LOG_CANDIDATES) {
+        /**
+         * An existence check and nothing more — what is recorded is the *name*
+         * that was tried, and whether it is a file or a directory. Anything read
+         * later is opened then, on its own terms, so there is no measurement here
+         * for a later read to disagree with.
+         */
+        try {
+            const info = await stat(join(root, candidate));
+            usage.push({
+                kind: info.isDirectory() ? 'log-directory' : 'log-file',
+                where: candidate,
+                provider: null,
+            });
+        }
+        catch {
+            // Absent is the common case and not an error.
+        }
+    }
+    try {
+        const info = await stat(join(root, STORE_DIR));
+        if (info.isDirectory()) {
+            usage.push({ kind: 'store', where: STORE_DIR, provider: null });
+        }
+    }
+    catch {
+        // No store yet.
+    }
+    /**
+     * A credential is named by its **variable**, never read.
+     *
+     * `findCredential` returns the value as well because the connector needs it;
+     * this takes the name and drops the rest on the floor. A first-run summary
+     * is the single most likely output in this product to be pasted into a chat
+     * window, and the rule that has held since 1.41 holds here.
+     */
+    for (const connector of CONNECTORS) {
+        const found = findCredential(connector, process.env);
+        if (found !== null) {
+            usage.push({
+                kind: 'connector-credential',
+                where: found.source.variable,
+                provider: connector.id,
+            });
+        }
+    }
+    // --- read what can be read ----------------------------------------------
+    let measured = null;
+    let unreadable = null;
+    const readable = usage.find((u) => u.kind === 'log-file' || u.kind === 'log-directory');
+    if (readable !== undefined) {
+        try {
+            const files = readable.kind === 'log-file'
+                ? [join(root, readable.where)]
+                : (await readdir(join(root, readable.where)))
+                    .filter((name) => LOG_EXTENSIONS.some((extension) => name.endsWith(extension)))
+                    .sort()
+                    .map((name) => join(root, readable.where, name));
+            if (files.length > 0) {
+                const texts = await Promise.all(files.map((file) => readUsageLog(file, t)));
+                measured = profileUsage(texts.join('\n'), { catalogue: pricing });
+            }
+        }
+        catch (error) {
+            // Named, never swallowed. A log that is there and cannot be read is the
+            // single most useful thing this command can tell somebody, and treating
+            // it as "no usage found" would send them to configure a connector they
+            // do not need.
+            unreadable = {
+                where: readable.where,
+                because: error instanceof Error ? error.message : String(error),
+            };
+        }
+    }
+    // --- the config already there -------------------------------------------
+    const configPath = join(root, CONFIG_FILENAME);
+    let existing = null;
+    try {
+        existing = { path: configPath, config: parseConfig(await readFile(configPath, 'utf8'), configPath) };
+    }
+    catch {
+        // Absent, or unparseable. Either way there is nothing to compare against,
+        // and `init` refuses to overwrite below rather than reasoning about it.
+    }
+    let unparseable = false;
+    if (existing === null) {
+        try {
+            await stat(configPath);
+            unparseable = true;
+        }
+        catch {
+            // Genuinely absent.
+        }
+    }
+    const askedLocale = matchLocale(LOCALE_ENV_VARS.map((name) => process.env[name]).find((value) => matchLocale(value)));
+    const proposal = proposeInit({
+        host,
+        sightings,
+        promptFiles: prompts.files,
+        usage,
+        measured,
+        locale: askedLocale ?? null,
+        existing,
+    }, { catalogue: pricing });
+    if (asJson) {
+        console.log(JSON.stringify({ ...proposal, unreadable, truncated: sourceWalk.truncated }, null, 2));
+        return;
+    }
+    renderInit(proposal, { host, prompts, usage, unreadable, truncated: sourceWalk.truncated, t, pricing });
+    // --- writing it ---------------------------------------------------------
+    //
+    // Three ways this ends and they are kept apart: nothing to write, refused to
+    // overwrite, written. "Nothing happened" with no reason is the output that
+    // makes somebody run the command twice.
+    if (Object.keys(proposal.config).length === 0) {
+        console.log(c.dim(t.init.nothingToWrite()));
+        console.log();
+        return;
+    }
+    const body = `${JSON.stringify(proposal.config, null, 2)}\n`;
+    if (dryRun) {
+        console.log(c.bold(t.init.wouldWrite(configPath)));
+        console.log();
+        console.log(body.trimEnd());
+        console.log();
+        return;
+    }
+    if (unparseable) {
+        console.log(c.yellow(t.init.existingUnparseable(configPath)));
+        console.log();
+        return;
+    }
+    if (existing !== null && args.flags.get('yes') !== true) {
+        console.log(c.yellow(t.init.existingRefused(configPath)));
+        console.log();
+        return;
+    }
+    await writeFile(configPath, body, 'utf8');
+    console.log(c.green(t.init.wrote(configPath)));
+    console.log();
+}
 function commandModels(t, pricing) {
     const n = (value) => value.toLocaleString(t.numberLocale);
     const col = t.models.columns;
@@ -5651,7 +6024,35 @@ async function main() {
     // flag validation so a typo is reported before any file is touched. An
     // invalid config throws here rather than quietly reverting to defaults —
     // "defaults" for a budget means "no budget", which means a green build.
-    const loaded = await loadConfig({ explicit: stringFlag(args, 'config') });
+    /**
+     * `init` is the exception, and finding out why was worth the release.
+     *
+     * A malformed `trazum.config.json` throws here — correctly, for every other
+     * command, because "defaults" for a budget means "no budget" and a silent
+     * revert to defaults is a green build that should have been red. But `init`
+     * is the command somebody runs *because* their setup is broken, and it was
+     * the one command a broken setup could stop from running. The refusal to
+     * overwrite an unparseable config, written two hours earlier in this same
+     * release, was unreachable code standing behind a throw.
+     *
+     * So `init` loads the config the same way and survives the failure, with
+     * nothing carried forward: no keys, no budgets, no locale. It then refuses
+     * to write over the file it could not read, and says so.
+     */
+    let loaded;
+    try {
+        loaded = await loadConfig({ explicit: stringFlag(args, 'config') });
+    }
+    catch (error) {
+        if (args.command !== 'init')
+            throw error;
+        loaded = {
+            config: {},
+            path: null,
+            pricing: BUNDLED_CATALOGUE,
+            pricingPath: null,
+        };
+    }
     const { config } = loaded;
     const pricing = await pricingFor(args, loaded, t);
     // The config only gets to choose the locale when nothing more explicit did.
@@ -5708,6 +6109,9 @@ async function main() {
         case 'models':
             commandModels(t, pricing);
             break;
+        case 'init':
+            await commandInit(args, config, pricing, t);
+            break;
         case 'where':
             await commandWhere(args, config, pricing, t);
             break;