sql-ai-harness 0.0.0-stage → 0.1.0

package/dist/commands/init.js

@@ -0,0 +1,232 @@
+import { createInterface } from 'node:readline/promises';
+import { stdin, stdout } from 'node:process';
+import { readFile, writeFile, access, chmod } from 'node:fs/promises';
+import { constants } from 'node:fs';
+import { resolve } from 'node:path';
+import { MySqlAdapter } from '../database/mysql/adapter.js';
+const PROVIDERS = [
+    { value: 'openrouter', label: 'OpenRouter', model: '' },
+    { value: 'google', label: 'Google Gemini', model: '' },
+    { value: 'groq', label: 'Groq', model: '' },
+    { value: 'cerebras', label: 'Cerebras', model: '' },
+    { value: 'ollama', label: 'Ollama (local)', model: 'qwen2.5:7b' },
+];
+function required(value, label) {
+    const clean = value.trim();
+    if (!clean)
+        throw new Error(`${label} is required.`);
+    return clean;
+}
+async function askSecret(question) {
+    if (!stdin.isTTY || typeof stdin.setRawMode !== 'function') {
+        const rl = createInterface({ input: stdin, output: stdout });
+        try {
+            return await rl.question(`${question}: `);
+        }
+        finally {
+            rl.close();
+        }
+    }
+    stdout.write(`${question}: `);
+    stdin.setRawMode(true);
+    stdin.resume();
+    stdin.setEncoding('utf8');
+    return new Promise((resolvePromise, reject) => {
+        let value = '';
+        const cleanup = () => {
+            stdin.setRawMode(false);
+            stdin.pause();
+            stdin.removeListener('data', onData);
+        };
+        const onData = (chunk) => {
+            for (const char of chunk) {
+                if (char === '\u0003') {
+                    cleanup();
+                    stdout.write('\n');
+                    reject(new Error('Setup cancelled.'));
+                    return;
+                }
+                if (char === '\r' || char === '\n') {
+                    cleanup();
+                    stdout.write('\n');
+                    resolvePromise(value);
+                    return;
+                }
+                if (char === '\u007f' || char === '\b')
+                    value = value.slice(0, -1);
+                else if (char >= ' ')
+                    value += char;
+            }
+        };
+        stdin.on('data', onData);
+    });
+}
+async function askChoice(rl, question, options) {
+    stdout.write(`\n${question}\n`);
+    options.forEach((option, index) => stdout.write(`  ${index + 1}. ${option.label}\n`));
+    while (true) {
+        const answer = (await rl.question(`Choose 1-${options.length} [1]: `)).trim() || '1';
+        const index = Number(answer) - 1;
+        if (Number.isInteger(index) && options[index])
+            return options[index].value;
+        stdout.write(`Enter a number from 1 to ${options.length}.\n`);
+    }
+}
+async function askValue(rl, label, defaultValue) {
+    const answer = (await rl.question(`${label} [${defaultValue}]: `)).trim();
+    return answer || defaultValue;
+}
+function dotenvValue(value) {
+    return value === '' ? '' : JSON.stringify(value);
+}
+async function saveEnvFile(path, values) {
+    let existing = '';
+    try {
+        existing = await readFile(path, 'utf8');
+    }
+    catch (error) {
+        if (error.code !== 'ENOENT')
+            throw error;
+    }
+    const lines = existing.split(/\r?\n/);
+    const output = [];
+    const written = new Set();
+    for (const line of lines) {
+        const match = line.match(/^\s*(?:export\s+)?([A-Z][A-Z0-9_]*)\s*=/);
+        const key = match?.[1];
+        if (key && Object.hasOwn(values, key)) {
+            if (!written.has(key))
+                output.push(`${key}=${dotenvValue(values[key])}`);
+            written.add(key);
+        }
+        else {
+            output.push(line);
+        }
+    }
+    const additions = Object.entries(values)
+        .filter(([key]) => !written.has(key))
+        .map(([key, value]) => `${key}=${dotenvValue(value)}`);
+    const merged = [...output.filter((line, index, all) => line !== '' || index < all.length - 1), ...additions].join('\n').replace(/\n*$/, '\n');
+    await writeFile(path, merged, { encoding: 'utf8', mode: 0o600 });
+    await chmod(path, 0o600);
+}
+export async function runInit(args = []) {
+    if (!stdin.isTTY || !stdout.isTTY) {
+        throw new Error('The init command requires an interactive terminal. Configure environment variables for non-interactive use.');
+    }
+    const force = args.includes('--force');
+    const envPath = resolve(process.cwd(), '.env');
+    try {
+        await access(envPath, constants.F_OK);
+        if (!force) {
+            const check = createInterface({ input: stdin, output: stdout });
+            const answer = (await check.question('A .env file already exists. Continue and add harness settings? (y/N): ')).trim().toLowerCase();
+            check.close();
+            if (answer !== 'y' && answer !== 'yes')
+                throw new Error('Setup cancelled. Existing .env file was not changed. Use --force to update harness settings.');
+        }
+    }
+    catch (error) {
+        if (error.code !== 'ENOENT')
+            throw error;
+    }
+    let rl = createInterface({ input: stdin, output: stdout });
+    const values = {
+        DB_DRIVER: 'mysql',
+        DB_HOST: '127.0.0.1',
+        DB_PORT: '3306',
+        DB_NAME: '',
+        DB_USER: '',
+        DB_PASSWORD: '',
+        AI_PROVIDER: 'openrouter',
+        AI_MODEL: '',
+        OPENROUTER_API_KEY: '',
+        GROQ_API_KEY: '',
+        GOOGLE_GENERATIVE_AI_API_KEY: '',
+        CEREBRAS_API_KEY: '',
+        OLLAMA_BASE_URL: 'http://localhost:11434/v1',
+        AI_SITE_NAME: 'SQL AI Harness',
+        SQL_MAX_ROWS: '500',
+        SQL_QUERY_TIMEOUT_MS: '10000',
+        SQL_MAX_RESULT_BYTES: '1048576',
+        AGENT_MAX_STEPS: '8',
+        SCHEMA_CACHE_TTL_MS: '300000',
+    };
+    try {
+        stdout.write('\nSQL AI Harness — Setup\n\nDatabase\n');
+        values.DB_HOST = await askValue(rl, 'MySQL/MariaDB host', '127.0.0.1');
+        values.DB_PORT = await askValue(rl, 'Port', '3306');
+        if (!/^\d+$/.test(values.DB_PORT) || Number(values.DB_PORT) < 1 || Number(values.DB_PORT) > 65535) {
+            throw new Error('Database port must be a number between 1 and 65535.');
+        }
+        values.DB_NAME = required(await askValue(rl, 'Database name', ''), 'Database name');
+        values.DB_USER = required(await askValue(rl, 'Database username', ''), 'Database username');
+        rl.close();
+        values.DB_PASSWORD = await askSecret('Database password');
+        rl = createInterface({ input: stdin, output: stdout });
+        const provider = await askChoice(rl, 'AI provider', PROVIDERS.map(({ value, label }) => ({ value, label })));
+        values.AI_PROVIDER = provider;
+        const providerInfo = PROVIDERS.find((item) => item.value === provider);
+        values.AI_MODEL = required(await rl.question('Model ID (enter the exact model identifier supported by this provider): '), 'Model ID');
+        if (provider !== 'ollama') {
+            const keyNames = {
+                openrouter: 'OPENROUTER_API_KEY',
+                google: 'GOOGLE_GENERATIVE_AI_API_KEY',
+                groq: 'GROQ_API_KEY',
+                cerebras: 'CEREBRAS_API_KEY',
+            };
+            rl.close();
+            values[keyNames[provider]] = required(await askSecret(`${providerInfo.label} API key`), 'API key');
+            rl = createInterface({ input: stdin, output: stdout });
+        }
+        else {
+            values.OLLAMA_BASE_URL = await askValue(rl, 'Ollama OpenAI-compatible base URL', 'http://localhost:11434/v1');
+            try {
+                new URL(values.OLLAMA_BASE_URL);
+            }
+            catch {
+                throw new Error('Ollama base URL must be a valid URL.');
+            }
+        }
+        values.AGENT_MAX_STEPS = await askValue(rl, 'Maximum agent steps', '8');
+        values.SQL_MAX_ROWS = await askValue(rl, 'Maximum rows returned', '500');
+        values.SCHEMA_CACHE_TTL_MS = String(Number(await askValue(rl, 'Schema cache TTL in minutes', '5')) * 60_000);
+        if (!/^\d+$/.test(values.AGENT_MAX_STEPS) || Number(values.AGENT_MAX_STEPS) < 1 || Number(values.AGENT_MAX_STEPS) > 20) {
+            throw new Error('Maximum agent steps must be between 1 and 20.');
+        }
+        if (!/^\d+$/.test(values.SQL_MAX_ROWS) || Number(values.SQL_MAX_ROWS) < 1) {
+            throw new Error('Maximum rows returned must be a positive integer.');
+        }
+        if (!/^\d+$/.test(values.SCHEMA_CACHE_TTL_MS) || Number(values.SCHEMA_CACHE_TTL_MS) < 60_000) {
+            throw new Error('Schema cache TTL must be at least 1 minute.');
+        }
+        stdout.write('\nTesting MySQL/MariaDB connection...\n');
+        const database = new MySqlAdapter({
+            host: values.DB_HOST, port: Number(values.DB_PORT), database: values.DB_NAME,
+            user: values.DB_USER, password: values.DB_PASSWORD,
+        });
+        try {
+            const tables = await database.listTables();
+            stdout.write(`✓ Database connection successful (${tables.length} tables discovered).\n`);
+        }
+        finally {
+            await database.close();
+        }
+        // Keep only the selected provider's secret key in the generated configuration.
+        const selectedKey = provider === 'ollama' ? undefined : {
+            openrouter: 'OPENROUTER_API_KEY', google: 'GOOGLE_GENERATIVE_AI_API_KEY',
+            groq: 'GROQ_API_KEY', cerebras: 'CEREBRAS_API_KEY',
+        }[provider];
+        for (const key of ['OPENROUTER_API_KEY', 'GROQ_API_KEY', 'GOOGLE_GENERATIVE_AI_API_KEY', 'CEREBRAS_API_KEY']) {
+            if (key !== selectedKey)
+                delete values[key];
+        }
+        await saveEnvFile(envPath, values);
+        stdout.write(`\n✓ Configuration saved to ${envPath}\n`);
+        stdout.write('  Keep .env out of version control; it contains credentials.\n\nSetup complete. Try:\n  sql-ai-harness ask "How many records are in the database?"\n\n');
+    }
+    finally {
+        rl.close();
+    }
+}
+//# sourceMappingURL=init.js.map

package/dist/commands/init.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"init.js","sourceRoot":"","sources":["../../src/commands/init.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACtE,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AAG5D,MAAM,SAAS,GAAmE;IAChF,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,EAAE,EAAE;IACvD,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,EAAE;IACtD,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;IAC3C,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE;IACnD,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,YAAY,EAAE;CAClE,CAAC;AAEF,SAAS,QAAQ,CAAC,KAAa,EAAE,KAAa;IAC5C,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3B,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,GAAG,KAAK,eAAe,CAAC,CAAC;IACrD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,QAAgB;IACvC,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,OAAO,KAAK,CAAC,UAAU,KAAK,UAAU,EAAE,CAAC;QAC3D,MAAM,EAAE,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAC7D,IAAI,CAAC;YAAC,OAAO,MAAM,EAAE,CAAC,QAAQ,CAAC,GAAG,QAAQ,IAAI,CAAC,CAAC;QAAC,CAAC;gBAAS,CAAC;YAAC,EAAE,CAAC,KAAK,EAAE,CAAC;QAAC,CAAC;IAC5E,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,GAAG,QAAQ,IAAI,CAAC,CAAC;IAC9B,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IACvB,KAAK,CAAC,MAAM,EAAE,CAAC;IACf,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAE1B,OAAO,IAAI,OAAO,CAAC,CAAC,cAAc,EAAE,MAAM,EAAE,EAAE;QAC5C,IAAI,KAAK,GAAG,EAAE,CAAC;QACf,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACxB,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,KAAK,CAAC,cAAc,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC,CAAC;QACF,MAAM,MAAM,GAAG,CAAC,KAAa,EAAE,EAAE;YAC/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACtB,OAAO,EAAE,CAAC;oBAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAAC,MAAM,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC;oBAAC,OAAO;gBAC/E,CAAC;gBACD,IAAI,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;oBACnC,OAAO,EAAE,CAAC;oBAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBAAC,cAAc,CAAC,KAAK,CAAC,CAAC;oBAAC,OAAO;gBAC/D,CAAC;gBACD,IAAI,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI;oBAAE,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;qBAC9D,IAAI,IAAI,IAAI,GAAG;oBAAE,KAAK,IAAI,IAAI,CAAC;YACtC,CAAC;QACH,CAAC,CAAC;QACF,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,KAAK,UAAU,SAAS,CACtB,EAAsC,EACtC,QAAgB,EAChB,OAA2C;IAE3C,MAAM,CAAC,KAAK,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC;IAChC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,KAAK,GAAG,CAAC,KAAK,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;IACtF,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,MAAM,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,OAAO,CAAC,MAAM,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,GAAG,CAAC;QACrF,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,KAAK,CAAC;YAAE,OAAO,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC;QAC3E,MAAM,CAAC,KAAK,CAAC,4BAA4B,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC;IAChE,CAAC;AACH,CAAC;AAED,KAAK,UAAU,QAAQ,CACrB,EAAsC,EACtC,KAAa,EACb,YAAoB;IAEpB,MAAM,MAAM,GAAG,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,GAAG,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC1E,OAAO,MAAM,IAAI,YAAY,CAAC;AAChC,CAAC;AAED,SAAS,WAAW,CAAC,KAAa;IAChC,OAAO,KAAK,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AACnD,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,IAAY,EAAE,MAA8B;IACrE,IAAI,QAAQ,GAAG,EAAE,CAAC;IAClB,IAAI,CAAC;QAAC,QAAQ,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAAC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAC9D,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ;YAAE,MAAM,KAAK,CAAC;IACtE,CAAC;IAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,OAAO,GAAG,IAAI,GAAG,EAAU,CAAC;IAClC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;QACpE,MAAM,GAAG,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACvB,IAAI,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC;YACtC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,IAAI,WAAW,CAAC,MAAM,CAAC,GAAG,CAAE,CAAC,EAAE,CAAC,CAAC;YAC1E,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpB,CAAC;IACH,CAAC;IACD,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;SACrC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;SACpC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,IAAI,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACzD,MAAM,MAAM,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,KAAK,EAAE,IAAI,KAAK,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,GAAG,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAC9I,MAAM,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;IACjE,MAAM,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,OAAO,CAAC,IAAI,GAAa,EAAE;IAC/C,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,6GAA6G,CAAC,CAAC;IACjI,CAAC;IACD,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACvC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,MAAM,CAAC,CAAC;IAC/C,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,KAAK,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAChE,MAAM,MAAM,GAAG,CAAC,MAAM,KAAK,CAAC,QAAQ,CAAC,wEAAwE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACrI,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,MAAM,KAAK,GAAG,IAAI,MAAM,KAAK,KAAK;gBAAE,MAAM,IAAI,KAAK,CAAC,8FAA8F,CAAC,CAAC;QAC1J,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAK,KAA+B,CAAC,IAAI,KAAK,QAAQ;YAAE,MAAM,KAAK,CAAC;IACtE,CAAC;IAED,IAAI,EAAE,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IAC3D,MAAM,MAAM,GAA2B;QACrC,SAAS,EAAE,OAAO;QAClB,OAAO,EAAE,WAAW;QACpB,OAAO,EAAE,MAAM;QACf,OAAO,EAAE,EAAE;QACX,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,EAAE;QACf,WAAW,EAAE,YAAY;QACzB,QAAQ,EAAE,EAAE;QACZ,kBAAkB,EAAE,EAAE;QACtB,YAAY,EAAE,EAAE;QAChB,4BAA4B,EAAE,EAAE;QAChC,gBAAgB,EAAE,EAAE;QACpB,eAAe,EAAE,2BAA2B;QAC5C,YAAY,EAAE,gBAAgB;QAC9B,YAAY,EAAE,KAAK;QACnB,oBAAoB,EAAE,OAAO;QAC7B,oBAAoB,EAAE,SAAS;QAC/B,eAAe,EAAE,GAAG;QACpB,mBAAmB,EAAE,QAAQ;KAC9B,CAAC;IAEF,IAAI,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACvD,MAAM,CAAC,OAAO,GAAG,MAAM,QAAQ,CAAC,EAAE,EAAE,oBAAoB,EAAE,WAAW,CAAC,CAAC;QACvE,MAAM,CAAC,OAAO,GAAG,MAAM,QAAQ,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QACpD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,KAAK,EAAE,CAAC;YAClG,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QACD,MAAM,CAAC,OAAO,GAAG,QAAQ,CAAC,MAAM,QAAQ,CAAC,EAAE,EAAE,eAAe,EAAE,EAAE,CAAC,EAAE,eAAe,CAAC,CAAC;QACpF,MAAM,CAAC,OAAO,GAAG,QAAQ,CAAC,MAAM,QAAQ,CAAC,EAAE,EAAE,mBAAmB,EAAE,EAAE,CAAC,EAAE,mBAAmB,CAAC,CAAC;QAC5F,EAAE,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,CAAC,WAAW,GAAG,MAAM,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC1D,EAAE,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAEvD,MAAM,QAAQ,GAAG,MAAM,SAAS,CAAC,EAAE,EAAE,aAAa,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QAC7G,MAAM,CAAC,WAAW,GAAG,QAAQ,CAAC;QAC9B,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,KAAK,QAAQ,CAAE,CAAC;QACxE,MAAM,CAAC,QAAQ,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0EAA0E,CAAC,EAAE,UAAU,CAAC,CAAC;QAEtI,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAsD;gBAClE,UAAU,EAAE,oBAAoB;gBAChC,MAAM,EAAE,8BAA8B;gBACtC,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,kBAAkB;aAC7B,CAAC;YACF,EAAE,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,GAAG,QAAQ,CAAC,MAAM,SAAS,CAAC,GAAG,YAAY,CAAC,KAAK,UAAU,CAAC,EAAE,SAAS,CAAC,CAAC;YACnG,EAAE,GAAG,eAAe,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QACzD,CAAC;aAAM,CAAC;YACN,MAAM,CAAC,eAAe,GAAG,MAAM,QAAQ,CAAC,EAAE,EAAE,mCAAmC,EAAE,2BAA2B,CAAC,CAAC;YAC9G,IAAI,CAAC;gBAAC,IAAI,GAAG,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC;gBAAC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;YAAC,CAAC;QAC7G,CAAC;QAED,MAAM,CAAC,eAAe,GAAG,MAAM,QAAQ,CAAC,EAAE,EAAE,qBAAqB,EAAE,GAAG,CAAC,CAAC;QACxE,MAAM,CAAC,YAAY,GAAG,MAAM,QAAQ,CAAC,EAAE,EAAE,uBAAuB,EAAE,KAAK,CAAC,CAAC;QACzE,MAAM,CAAC,mBAAmB,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,QAAQ,CAAC,EAAE,EAAE,6BAA6B,EAAE,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC;QAC7G,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,GAAG,EAAE,EAAE,CAAC;YACvH,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1E,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,GAAG,MAAM,EAAE,CAAC;YAC7F,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,yCAAyC,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;YAChC,IAAI,EAAE,MAAM,CAAC,OAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,OAAQ;YAC9E,IAAI,EAAE,MAAM,CAAC,OAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,WAAY;SACrD,CAAC,CAAC;QACH,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;YAC3C,MAAM,CAAC,KAAK,CAAC,qCAAqC,MAAM,CAAC,MAAM,wBAAwB,CAAC,CAAC;QAC3F,CAAC;gBAAS,CAAC;YACT,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC;QAED,+EAA+E;QAC/E,MAAM,WAAW,GAAG,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAE;YACvD,UAAU,EAAE,oBAAoB,EAAE,MAAM,EAAE,8BAA8B;YACxE,IAAI,EAAE,cAAc,EAAE,QAAQ,EAAE,kBAAkB;SACzC,CAAC,QAAQ,CAAC,CAAC;QACtB,KAAK,MAAM,GAAG,IAAI,CAAC,oBAAoB,EAAE,cAAc,EAAE,8BAA8B,EAAE,kBAAkB,CAAC,EAAE,CAAC;YAC7G,IAAI,GAAG,KAAK,WAAW;gBAAE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,WAAW,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,8BAA8B,OAAO,IAAI,CAAC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,wJAAwJ,CAAC,CAAC;IACzK,CAAC;YAAS,CAAC;QACT,EAAE,CAAC,KAAK,EAAE,CAAC;IACb,CAAC;AACH,CAAC"}

package/dist/config/config.d.ts

@@ -0,0 +1,39 @@
+import 'dotenv/config';
+import { z } from 'zod';
+import type { SqlSafetyPolicy } from '../core/types/index.js';
+declare const providerSchema: z.ZodEnum<{
+    cerebras: "cerebras";
+    google: "google";
+    groq: "groq";
+    ollama: "ollama";
+    openrouter: "openrouter";
+}>;
+export type AiProviderName = z.infer<typeof providerSchema>;
+export interface AppConfig {
+    ai: {
+        provider: AiProviderName;
+        apiKey?: string;
+        model: string;
+        baseUrl?: string;
+        siteUrl?: string;
+        siteName: string;
+    };
+    database: {
+        driver: 'mysql' | 'postgres' | 'sqlite';
+        host: string;
+        port: number;
+        name: string;
+        user: string;
+        password: string;
+    };
+    safety: SqlSafetyPolicy;
+    agent: {
+        maxSteps: number;
+    };
+    schema: {
+        cacheTtlMs: number;
+    };
+}
+export declare function loadConfig(env?: NodeJS.ProcessEnv): AppConfig;
+export {};
+//# sourceMappingURL=config.d.ts.map

package/dist/config/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/config/config.ts"],"names":[],"mappings":"AACA,OAAO,eAAe,CAAC;AACvB,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAE9D,QAAA,MAAM,cAAc;;;;;;EAMlB,CAAC;AAEH,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,cAAc,CAAC,CAAC;AAiC5D,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE;QACF,QAAQ,EAAE,cAAc,CAAC;QACzB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,QAAQ,EAAE;QACR,MAAM,EAAE,OAAO,GAAG,UAAU,GAAG,QAAQ,CAAC;QACxC,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,MAAM,EAAE,eAAe,CAAC;IACxB,KAAK,EAAE;QACL,QAAQ,EAAE,MAAM,CAAC;KAClB,CAAC;IACF,MAAM,EAAE;QACN,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;CACH;AAED,wBAAgB,UAAU,CAAC,GAAG,GAAE,MAAM,CAAC,UAAwB,GAAG,SAAS,CA0D1E"}

package/dist/config/config.js

@@ -0,0 +1,84 @@
+import 'dotenv/config';
+import { z } from 'zod';
+const providerSchema = z.enum([
+    'openrouter',
+    'groq',
+    'google',
+    'cerebras',
+    'ollama',
+]);
+const envSchema = z.object({
+    AI_PROVIDER: providerSchema.default('openrouter'),
+    AI_MODEL: z.string().min(1, 'AI_MODEL is required'),
+    OPENROUTER_API_KEY: z.string().optional(),
+    GROQ_API_KEY: z.string().optional(),
+    GOOGLE_GENERATIVE_AI_API_KEY: z.string().optional(),
+    CEREBRAS_API_KEY: z.string().optional(),
+    OLLAMA_BASE_URL: z.string().url().default('http://localhost:11434/v1'),
+    AI_SITE_URL: z.preprocess((value) => value === '' ? undefined : value, z.string().url().optional()),
+    AI_SITE_NAME: z.string().default('SQL AI Harness'),
+    DB_DRIVER: z.enum(['mysql', 'postgres', 'sqlite']).default('mysql'),
+    DB_HOST: z.string().default('127.0.0.1'),
+    DB_PORT: z.coerce.number().int().positive().default(3306),
+    DB_NAME: z.string().default(''),
+    DB_USER: z.string().default(''),
+    DB_PASSWORD: z.string().default(''),
+    SQL_MAX_ROWS: z.coerce.number().int().positive().default(500),
+    SQL_QUERY_TIMEOUT_MS: z.coerce.number().int().positive().default(10_000),
+    SQL_MAX_RESULT_BYTES: z.coerce.number().int().positive().default(1_048_576),
+    AGENT_MAX_STEPS: z.coerce.number().int().min(1).max(20).default(8),
+    SCHEMA_CACHE_TTL_MS: z.coerce.number().int().positive().default(300_000),
+});
+export function loadConfig(env = process.env) {
+    const parsed = envSchema.parse(env);
+    const apiKeys = {
+        openrouter: parsed.OPENROUTER_API_KEY,
+        groq: parsed.GROQ_API_KEY,
+        google: parsed.GOOGLE_GENERATIVE_AI_API_KEY,
+        cerebras: parsed.CEREBRAS_API_KEY,
+        ollama: undefined,
+    };
+    const apiKey = apiKeys[parsed.AI_PROVIDER];
+    if (parsed.AI_PROVIDER !== 'ollama' && !apiKey?.trim()) {
+        const envNames = {
+            openrouter: 'OPENROUTER_API_KEY',
+            groq: 'GROQ_API_KEY',
+            google: 'GOOGLE_GENERATIVE_AI_API_KEY',
+            cerebras: 'CEREBRAS_API_KEY',
+        };
+        throw new Error(`${envNames[parsed.AI_PROVIDER]} is required when AI_PROVIDER=${parsed.AI_PROVIDER}`);
+    }
+    return {
+        ai: {
+            provider: parsed.AI_PROVIDER,
+            apiKey,
+            model: parsed.AI_MODEL,
+            baseUrl: parsed.AI_PROVIDER === 'ollama'
+                ? parsed.OLLAMA_BASE_URL
+                : undefined,
+            siteUrl: parsed.AI_SITE_URL,
+            siteName: parsed.AI_SITE_NAME,
+        },
+        database: {
+            driver: parsed.DB_DRIVER,
+            host: parsed.DB_HOST,
+            port: parsed.DB_PORT,
+            name: parsed.DB_NAME,
+            user: parsed.DB_USER,
+            password: parsed.DB_PASSWORD,
+        },
+        agent: {
+            maxSteps: parsed.AGENT_MAX_STEPS,
+        },
+        schema: {
+            cacheTtlMs: parsed.SCHEMA_CACHE_TTL_MS,
+        },
+        safety: {
+            readOnly: true,
+            maxRows: parsed.SQL_MAX_ROWS,
+            queryTimeoutMs: parsed.SQL_QUERY_TIMEOUT_MS,
+            maxResultBytes: parsed.SQL_MAX_RESULT_BYTES,
+        },
+    };
+}
+//# sourceMappingURL=config.js.map

package/dist/config/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config/config.ts"],"names":[],"mappings":"AACA,OAAO,eAAe,CAAC;AACvB,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,cAAc,GAAG,CAAC,CAAC,IAAI,CAAC;IAC5B,YAAY;IACZ,MAAM;IACN,QAAQ;IACR,UAAU;IACV,QAAQ;CACT,CAAC,CAAC;AAIH,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,CAAC;IACzB,WAAW,EAAE,cAAc,CAAC,OAAO,CAAC,YAAY,CAAC;IACjD,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,sBAAsB,CAAC;IAEnD,kBAAkB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACzC,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnC,4BAA4B,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACnD,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAEvC,eAAe,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,2BAA2B,CAAC;IAEtE,WAAW,EAAE,CAAC,CAAC,UAAU,CACvB,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,EAC3C,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAC5B;IACD,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC;IAElD,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;IACnE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,WAAW,CAAC;IACxC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;IACzD,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/B,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IAC/B,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC;IAEnC,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC;IAC7D,oBAAoB,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC;IACxE,oBAAoB,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC;IAC3E,eAAe,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;IAClE,mBAAmB,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;CACzE,CAAC,CAAC;AA4BH,MAAM,UAAU,UAAU,CAAC,GAAG,GAAsB,OAAO,CAAC,GAAG;IAC7D,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAEpC,MAAM,OAAO,GAA+C;QAC1D,UAAU,EAAE,MAAM,CAAC,kBAAkB;QACrC,IAAI,EAAE,MAAM,CAAC,YAAY;QACzB,MAAM,EAAE,MAAM,CAAC,4BAA4B;QAC3C,QAAQ,EAAE,MAAM,CAAC,gBAAgB;QACjC,MAAM,EAAE,SAAS;KAClB,CAAC;IAEF,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAE3C,IAAI,MAAM,CAAC,WAAW,KAAK,QAAQ,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC;QACvD,MAAM,QAAQ,GAAsD;YAClE,UAAU,EAAE,oBAAoB;YAChC,IAAI,EAAE,cAAc;YACpB,MAAM,EAAE,8BAA8B;YACtC,QAAQ,EAAE,kBAAkB;SAC7B,CAAC;QAEF,MAAM,IAAI,KAAK,CACb,GAAG,QAAQ,CAAC,MAAM,CAAC,WAAW,CAAC,iCAAiC,MAAM,CAAC,WAAW,EAAE,CACrF,CAAC;IACJ,CAAC;IAED,OAAO;QACL,EAAE,EAAE;YACF,QAAQ,EAAE,MAAM,CAAC,WAAW;YAC5B,MAAM;YACN,KAAK,EAAE,MAAM,CAAC,QAAQ;YACtB,OAAO,EAAE,MAAM,CAAC,WAAW,KAAK,QAAQ;gBACtC,CAAC,CAAC,MAAM,CAAC,eAAe;gBACxB,CAAC,CAAC,SAAS;YACb,OAAO,EAAE,MAAM,CAAC,WAAW;YAC3B,QAAQ,EAAE,MAAM,CAAC,YAAY;SAC9B;QACD,QAAQ,EAAE;YACR,MAAM,EAAE,MAAM,CAAC,SAAS;YACxB,IAAI,EAAE,MAAM,CAAC,OAAO;YACpB,IAAI,EAAE,MAAM,CAAC,OAAO;YACpB,IAAI,EAAE,MAAM,CAAC,OAAO;YACpB,IAAI,EAAE,MAAM,CAAC,OAAO;YACpB,QAAQ,EAAE,MAAM,CAAC,WAAW;SAC7B;QACD,KAAK,EAAE;YACL,QAAQ,EAAE,MAAM,CAAC,eAAe;SACjC;QACD,MAAM,EAAE;YACN,UAAU,EAAE,MAAM,CAAC,mBAAmB;SACvC;QACD,MAAM,EAAE;YACN,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,MAAM,CAAC,YAAY;YAC5B,cAAc,EAAE,MAAM,CAAC,oBAAoB;YAC3C,cAAc,EAAE,MAAM,CAAC,oBAAoB;SAC5C;KACF,CAAC;AACJ,CAAC"}

package/dist/context/db-context.d.ts

@@ -0,0 +1,12 @@
+import type { DatabaseContext } from '../core/types/index.js';
+export declare const DEFAULT_DATABASE_CONTEXT_FILE = "sql-ai.md";
+export interface DatabaseContextLoaderOptions {
+    cwd?: string;
+    fileName?: string;
+}
+/**
+ * Loads optional, user-authored business/domain context for the connected database.
+ * Missing context is normal and never prevents the harness from running.
+ */
+export declare function loadDatabaseContext(options?: DatabaseContextLoaderOptions): Promise<DatabaseContext>;
+//# sourceMappingURL=db-context.d.ts.map

package/dist/context/db-context.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"db-context.d.ts","sourceRoot":"","sources":["../../src/context/db-context.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAE9D,eAAO,MAAM,6BAA6B,cAAc,CAAC;AAEzD,MAAM,WAAW,4BAA4B;IAC3C,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,wBAAsB,mBAAmB,CACvC,OAAO,GAAE,4BAAiC,GACzC,OAAO,CAAC,eAAe,CAAC,CAqB1B"}

package/dist/context/db-context.js

@@ -0,0 +1,29 @@
+import { access, readFile } from 'node:fs/promises';
+import { constants } from 'node:fs';
+import { resolve } from 'node:path';
+export const DEFAULT_DATABASE_CONTEXT_FILE = 'sql-ai.md';
+/**
+ * Loads optional, user-authored business/domain context for the connected database.
+ * Missing context is normal and never prevents the harness from running.
+ */
+export async function loadDatabaseContext(options = {}) {
+    const cwd = options.cwd ?? process.cwd();
+    const fileName = options.fileName ?? DEFAULT_DATABASE_CONTEXT_FILE;
+    const path = resolve(cwd, fileName);
+    try {
+        await access(path, constants.R_OK);
+        const content = (await readFile(path, 'utf8')).trim();
+        if (!content)
+            return {};
+        return { content, source: path };
+    }
+    catch (error) {
+        const code = error && typeof error === 'object' && 'code' in error
+            ? error.code
+            : undefined;
+        if (code === 'ENOENT')
+            return {};
+        throw error;
+    }
+}
+//# sourceMappingURL=db-context.js.map

package/dist/context/db-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"db-context.js","sourceRoot":"","sources":["../../src/context/db-context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAGpC,MAAM,CAAC,MAAM,6BAA6B,GAAG,WAAW,CAAC;AAOzD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,mBAAmB,CACvC,OAAO,GAAiC,EAAE;IAE1C,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IACzC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,6BAA6B,CAAC;IACnE,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;IAEpC,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;QACnC,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAEtD,IAAI,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;QAExB,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACnC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,GAAG,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,IAAI,KAAK;YAChE,CAAC,CAAE,KAA2B,CAAC,IAAI;YACnC,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,IAAI,KAAK,QAAQ;YAAE,OAAO,EAAE,CAAC;QAEjC,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC"}

package/dist/core/agent/agent.d.ts

@@ -0,0 +1,32 @@
+import type { AiProvider } from "../../ai/provider.js";
+import type { DatabaseContext, ToolContext } from "../types/index.js";
+import { type AgentObserver } from "./observability.js";
+export type ResponseMode = "natural" | "technical" | "debug";
+export interface AgentOptions {
+    maxSteps?: number;
+    databaseContext?: DatabaseContext;
+    responseMode?: ResponseMode;
+    temperature?: number;
+    observer?: AgentObserver;
+}
+export interface AgentResult {
+    text: string;
+    steps: number;
+    toolCalls: number;
+    raw?: unknown;
+}
+/**
+ * Application-level agent. Provider-specific model calls stay behind AiProvider;
+ * database access stays behind ToolContext and the tool registry.
+ */
+export declare class Agent {
+    private readonly provider;
+    private readonly context;
+    private readonly options;
+    constructor(provider: AiProvider, context: ToolContext, options?: AgentOptions);
+    ask(question: string): Promise<AgentResult>;
+    private responseInstructions;
+    private buildTools;
+    private discoverRelationships;
+}
+//# sourceMappingURL=agent.d.ts.map

package/dist/core/agent/agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../../../src/core/agent/agent.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,KAAK,EACV,eAAe,EAEf,WAAW,EACZ,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAqB,KAAK,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAoC3E,MAAM,MAAM,YAAY,GAAG,SAAS,GAAG,WAAW,GAAG,OAAO,CAAC;AAE7D,MAAM,WAAW,YAAY;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,YAAY,CAAC,EAAE,YAAY,CAAC;IAE5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,QAAQ,CAAC,EAAE,aAAa,CAAC;CAC1B;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,OAAO,CAAC;CACf;AAsED;;;GAGG;AACH,qBAAa,KAAK;IAEd,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,OAAO;IACxB,OAAO,CAAC,QAAQ,CAAC,OAAO;IAH1B,YACmB,QAAQ,EAAE,UAAU,EACpB,OAAO,EAAE,WAAW,EACpB,OAAO,GAAE,YAAiB,EACzC;IAEE,GAAG,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAqEhD;IAED,OAAO,CAAC,oBAAoB;IAyB5B,OAAO,CAAC,UAAU;YAkHJ,qBAAqB;CAmDpC"}