@infrarix/locopilot 1.1.0

package/dist/shared/db/pool.js

@@ -0,0 +1,83 @@
+'use strict';
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getDatabase = getDatabase;
+exports.query = query;
+exports.withTransaction = withTransaction;
+exports.shutdown = shutdown;
+const path_1 = require("path");
+const os_1 = require("os");
+const sqlite3_1 = __importDefault(require("sqlite3"));
+const DEFAULT_SQLITE_PATH = (0, path_1.join)((0, os_1.homedir)(), '.locopilot', 'db.sqlite');
+class SQLiteDatabase {
+    db;
+    constructor(filename) {
+        this.db = new sqlite3_1.default.Database(filename);
+        console.log(`[db] Using SQLite at ${filename}`);
+    }
+    async query(text, params) {
+        const isSelect = text.trim().toUpperCase().startsWith('SELECT');
+        return new Promise((resolve, reject) => {
+            if (isSelect) {
+                this.db.all(text, params ?? [], (err, rows) => {
+                    if (err)
+                        return reject(err);
+                    resolve({ rows: rows, rowCount: rows.length });
+                });
+            }
+            else {
+                this.db.run(text, params ?? [], function (err) {
+                    if (err)
+                        return reject(err);
+                    resolve({ rows: [], rowCount: this.changes });
+                });
+            }
+        });
+    }
+    async withTransaction(fn) {
+        await this.query('BEGIN');
+        try {
+            await fn(this);
+            await this.query('COMMIT');
+        }
+        catch (err) {
+            await this.query('ROLLBACK');
+            throw err;
+        }
+    }
+    async shutdown() {
+        return new Promise((resolve, reject) => {
+            this.db.close((err) => {
+                if (err)
+                    return reject(err);
+                resolve();
+            });
+        });
+    }
+    async checkHealth() {
+        await this.query('SELECT 1');
+    }
+}
+let _instance = null;
+function getDatabase() {
+    if (!_instance) {
+        const sqlitePath = process.env.SQLITE_PATH || DEFAULT_SQLITE_PATH;
+        _instance = new SQLiteDatabase(sqlitePath);
+    }
+    return _instance;
+}
+async function query(text, params) {
+    return getDatabase().query(text, params);
+}
+async function withTransaction(fn) {
+    return getDatabase().withTransaction(fn);
+}
+async function shutdown() {
+    if (_instance) {
+        await _instance.shutdown();
+        _instance = null;
+    }
+}
+//# sourceMappingURL=pool.js.map

package/dist/shared/db/pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pool.js","sourceRoot":"","sources":["../../../src/shared/db/pool.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;AA6Eb,kCAMC;AAED,sBAEC;AAED,0CAEC;AAED,4BAKC;AAhGD,+BAA4B;AAC5B,2BAA6B;AAC7B,sDAA8B;AAkB9B,MAAM,mBAAmB,GAAG,IAAA,WAAI,EAAC,IAAA,YAAO,GAAE,EAAE,YAAY,EAAE,WAAW,CAAC,CAAC;AAEvE,MAAM,cAAc;IACV,EAAE,CAAmB;IAE7B,YAAY,QAAgB;QAC1B,IAAI,CAAC,EAAE,GAAG,IAAI,iBAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACzC,OAAO,CAAC,GAAG,CAAC,wBAAwB,QAAQ,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,KAAK,CAAI,IAAY,EAAE,MAAkB;QAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAEhE,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,QAAQ,EAAE,CAAC;gBACb,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,IAAI,EAAE,EAAE,CAAC,GAAiB,EAAE,IAAe,EAAE,EAAE;oBACrE,IAAI,GAAG;wBAAE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;oBAC5B,OAAO,CAAC,EAAE,IAAI,EAAE,IAAW,EAAE,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;gBACxD,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,IAAI,EAAE,EAAE,UAAmC,GAAiB;oBAClF,IAAI,GAAG;wBAAE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;oBAC5B,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;gBAChD,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAgD;QACpE,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC1B,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,IAAoC,CAAC,CAAC;YAC/C,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;YAC7B,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,GAAiB,EAAE,EAAE;gBAClC,IAAI,GAAG;oBAAE,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;YACZ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,WAAW;QACf,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAC/B,CAAC;CACF;AAED,IAAI,SAAS,GAAoB,IAAI,CAAC;AAEtC,SAAgB,WAAW;IACzB,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,UAAU,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,mBAAmB,CAAC;QAClE,SAAS,GAAG,IAAI,cAAc,CAAC,UAAU,CAAC,CAAC;IAC7C,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAEM,KAAK,UAAU,KAAK,CAAU,IAAY,EAAE,MAAkB;IACnE,OAAO,WAAW,EAAE,CAAC,KAAK,CAAI,IAAI,EAAE,MAAM,CAAC,CAAC;AAC9C,CAAC;AAEM,KAAK,UAAU,eAAe,CAAC,EAAgD;IACpF,OAAO,WAAW,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;AAC3C,CAAC;AAEM,KAAK,UAAU,QAAQ;IAC5B,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC3B,SAAS,GAAG,IAAI,CAAC;IACnB,CAAC;AACH,CAAC"}

package/dist/shared/errors.js

@@ -0,0 +1,59 @@
+'use strict';
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NotFoundError = exports.ProviderError = exports.ValidationError = exports.RateLimitError = exports.AuthError = exports.AppError = void 0;
+class AppError extends Error {
+    statusCode;
+    code;
+    extra;
+    constructor(message, statusCode, code, extra) {
+        super(message);
+        this.name = this.constructor.name;
+        this.statusCode = statusCode;
+        this.code = code;
+        this.extra = extra;
+        Object.setPrototypeOf(this, new.target.prototype);
+    }
+    toJSON() {
+        const body = { error: this.code ?? this.message };
+        if (this.extra)
+            Object.assign(body, this.extra);
+        return body;
+    }
+}
+exports.AppError = AppError;
+/** 401 — Invalid or missing API key */
+class AuthError extends AppError {
+    constructor(message = 'Invalid API key') {
+        super(message, 401, message);
+    }
+}
+exports.AuthError = AuthError;
+/** 429 — Rate limit exceeded */
+class RateLimitError extends AppError {
+    constructor(retryAfterSeconds) {
+        super('Rate limit exceeded', 429, 'rate_limit_exceeded', { retryAfterSeconds });
+    }
+}
+exports.RateLimitError = RateLimitError;
+/** 400 — Bad request / validation failure */
+class ValidationError extends AppError {
+    constructor(message) {
+        super(message, 400, message);
+    }
+}
+exports.ValidationError = ValidationError;
+/** 503 — All providers unavailable */
+class ProviderError extends AppError {
+    constructor() {
+        super('All providers unavailable', 503, 'providers_unavailable');
+    }
+}
+exports.ProviderError = ProviderError;
+/** 404 — Resource not found */
+class NotFoundError extends AppError {
+    constructor(resource = 'Resource') {
+        super(`${resource} not found`, 404, `${resource.toLowerCase()}_not_found`);
+    }
+}
+exports.NotFoundError = NotFoundError;
+//# sourceMappingURL=errors.js.map

package/dist/shared/errors.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../src/shared/errors.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;AAEb,MAAa,QAAS,SAAQ,KAAK;IACxB,UAAU,CAAS;IACnB,IAAI,CAAqB;IACzB,KAAK,CAAsC;IAEpD,YAAY,OAAe,EAAE,UAAkB,EAAE,IAAa,EAAE,KAA+B;QAC7F,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;QAClC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IACpD,CAAC;IAED,MAAM;QACJ,MAAM,IAAI,GAA4B,EAAE,KAAK,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAC3E,IAAI,IAAI,CAAC,KAAK;YAAE,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QAChD,OAAO,IAAI,CAAC;IACd,CAAC;CACF;AAnBD,4BAmBC;AAED,uCAAuC;AACvC,MAAa,SAAU,SAAQ,QAAQ;IACrC,YAAY,OAAO,GAAG,iBAAiB;QACrC,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC/B,CAAC;CACF;AAJD,8BAIC;AAED,gCAAgC;AAChC,MAAa,cAAe,SAAQ,QAAQ;IAC1C,YAAY,iBAAyB;QACnC,KAAK,CAAC,qBAAqB,EAAE,GAAG,EAAE,qBAAqB,EAAE,EAAE,iBAAiB,EAAE,CAAC,CAAC;IAClF,CAAC;CACF;AAJD,wCAIC;AAED,6CAA6C;AAC7C,MAAa,eAAgB,SAAQ,QAAQ;IAC3C,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;IAC/B,CAAC;CACF;AAJD,0CAIC;AAED,sCAAsC;AACtC,MAAa,aAAc,SAAQ,QAAQ;IACzC;QACE,KAAK,CAAC,2BAA2B,EAAE,GAAG,EAAE,uBAAuB,CAAC,CAAC;IACnE,CAAC;CACF;AAJD,sCAIC;AAED,+BAA+B;AAC/B,MAAa,aAAc,SAAQ,QAAQ;IACzC,YAAY,QAAQ,GAAG,UAAU;QAC/B,KAAK,CAAC,GAAG,QAAQ,YAAY,EAAE,GAAG,EAAE,GAAG,QAAQ,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;IAC7E,CAAC;CACF;AAJD,sCAIC"}

package/dist/shared/index.js

@@ -0,0 +1,24 @@
+"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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./constants"), exports);
+__exportStar(require("./errors"), exports);
+__exportStar(require("./crypto"), exports);
+__exportStar(require("./ndjson"), exports);
+__exportStar(require("./types"), exports);
+__exportStar(require("./db/pool"), exports);
+__exportStar(require("./runtime/ollama"), exports);
+//# sourceMappingURL=index.js.map

package/dist/shared/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/shared/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8CAA4B;AAC5B,2CAAyB;AACzB,2CAAyB;AACzB,2CAAyB;AACzB,0CAAwB;AACxB,4CAA0B;AAC1B,mDAAiC"}

package/dist/shared/ndjson.js

@@ -0,0 +1,39 @@
+'use strict';
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createNdjsonParser = createNdjsonParser;
+function createNdjsonParser() {
+    let buffer = '';
+    return {
+        feed(chunk) {
+            buffer += chunk;
+            const results = [];
+            const lines = buffer.split('\n');
+            buffer = lines.pop() ?? '';
+            for (const line of lines) {
+                const trimmed = line.trim();
+                if (!trimmed)
+                    continue;
+                try {
+                    results.push(JSON.parse(trimmed));
+                }
+                catch {
+                    // skip unparseable lines (SSE prefixes, comments, etc.)
+                }
+            }
+            return results;
+        },
+        flush() {
+            const trimmed = buffer.trim();
+            buffer = '';
+            if (!trimmed)
+                return [];
+            try {
+                return [JSON.parse(trimmed)];
+            }
+            catch {
+                return [];
+            }
+        },
+    };
+}
+//# sourceMappingURL=ndjson.js.map

package/dist/shared/ndjson.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ndjson.js","sourceRoot":"","sources":["../../src/shared/ndjson.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAOb,gDAkCC;AAlCD,SAAgB,kBAAkB;IAChC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,OAAO;QACL,IAAI,CAAC,KAAa;YAChB,MAAM,IAAI,KAAK,CAAC;YAChB,MAAM,OAAO,GAAc,EAAE,CAAC;YAC9B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC5B,IAAI,CAAC,OAAO;oBAAE,SAAS;gBACvB,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;gBACpC,CAAC;gBAAC,MAAM,CAAC;oBACP,wDAAwD;gBAC1D,CAAC;YACH,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,KAAK;YACH,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,GAAG,EAAE,CAAC;YACZ,IAAI,CAAC,OAAO;gBAAE,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;YAC/B,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/shared/runtime/ollama/index.js

@@ -0,0 +1,55 @@
+'use strict';
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getLocalModels = getLocalModels;
+exports.streamChat = streamChat;
+exports.checkOllama = checkOllama;
+exports.pullModel = pullModel;
+const constants_1 = require("../../constants");
+const HEALTH_TIMEOUT_MS = 5_000;
+const LIST_TIMEOUT_MS = 8_000;
+const STREAM_TIMEOUT_MS = 30_000;
+const PULL_TIMEOUT_MS = 300_000; // model pulls can take minutes
+async function getLocalModels() {
+    const res = await fetch(`${constants_1.OLLAMA_HOST}/api/tags`, {
+        signal: AbortSignal.timeout(LIST_TIMEOUT_MS),
+    });
+    if (!res.ok)
+        throw new Error(`Ollama unreachable: ${res.status}`);
+    const data = (await res.json());
+    return data.models ?? [];
+}
+async function streamChat(body, signal) {
+    const timeoutSignal = AbortSignal.timeout(STREAM_TIMEOUT_MS);
+    const effectiveSignal = signal
+        ? AbortSignal.any
+            ? AbortSignal.any([timeoutSignal, signal])
+            : timeoutSignal
+        : timeoutSignal;
+    const res = await fetch(`${constants_1.OLLAMA_HOST}/api/chat`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ ...body, stream: true }),
+        signal: effectiveSignal,
+    });
+    if (!res.ok)
+        throw new Error(`Ollama chat error: ${res.status}`);
+    return res;
+}
+async function checkOllama() {
+    const res = await fetch(`${constants_1.OLLAMA_HOST}/`, {
+        signal: AbortSignal.timeout(HEALTH_TIMEOUT_MS),
+    });
+    if (!res.ok)
+        throw new Error(`Ollama health check failed: ${res.status}`);
+}
+async function pullModel(modelName) {
+    const res = await fetch(`${constants_1.OLLAMA_HOST}/api/pull`, {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ name: modelName }),
+        signal: AbortSignal.timeout(PULL_TIMEOUT_MS),
+    });
+    if (!res.ok)
+        throw new Error(`Failed to pull model ${modelName}: ${res.status}`);
+}
+//# sourceMappingURL=index.js.map

package/dist/shared/runtime/ollama/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/shared/runtime/ollama/index.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;AAeb,wCAOC;AAED,gCAgBC;AAED,kCAKC;AAED,8BAQC;AAvDD,+CAA8C;AAE9C,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAChC,MAAM,eAAe,GAAG,KAAK,CAAC;AAC9B,MAAM,iBAAiB,GAAG,MAAM,CAAC;AACjC,MAAM,eAAe,GAAG,OAAO,CAAC,CAAC,+BAA+B;AAQzD,KAAK,UAAU,cAAc;IAClC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,uBAAW,WAAW,EAAE;QACjD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,eAAe,CAAC;KAC7C,CAAC,CAAC;IACH,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;IAClE,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA+B,CAAC;IAC9D,OAAO,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC;AAC3B,CAAC;AAEM,KAAK,UAAU,UAAU,CAAC,IAA6B,EAAE,MAAoB;IAClF,MAAM,aAAa,GAAG,WAAW,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;IAC7D,MAAM,eAAe,GAAG,MAAM;QAC5B,CAAC,CAAC,WAAW,CAAC,GAAG;YACf,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;YAC1C,CAAC,CAAC,aAAa;QACjB,CAAC,CAAC,aAAa,CAAC;IAElB,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,uBAAW,WAAW,EAAE;QACjD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QAC/C,MAAM,EAAE,eAAe;KACxB,CAAC,CAAC;IACH,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;IACjE,OAAO,GAAG,CAAC;AACb,CAAC;AAEM,KAAK,UAAU,WAAW;IAC/B,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,uBAAW,GAAG,EAAE;QACzC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,iBAAiB,CAAC;KAC/C,CAAC,CAAC;IACH,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;AAC5E,CAAC;AAEM,KAAK,UAAU,SAAS,CAAC,SAAiB;IAC/C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,uBAAW,WAAW,EAAE;QACjD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;QAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;QACzC,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,eAAe,CAAC;KAC7C,CAAC,CAAC;IACH,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,SAAS,KAAK,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;AACnF,CAAC"}

package/dist/shared/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/shared/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/shared/types.ts"],"names":[],"mappings":""}

package/dist/training/adapters/axolotl.js

@@ -0,0 +1,83 @@
+'use strict';
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.run = run;
+const child_process_1 = require("child_process");
+const path_1 = __importDefault(require("path"));
+const fs_1 = __importDefault(require("fs"));
+const js_yaml_1 = __importDefault(require("js-yaml"));
+const types_1 = require("../types");
+const validator_1 = require("../validator");
+const RUNNER = path_1.default.resolve(__dirname, 'axolotl_runner.py');
+async function run(config, logEmitter) {
+    const cfg = (0, types_1.applyDefaults)(config);
+    let datasetType = 'alpaca';
+    try {
+        const firstLine = fs_1.default.readFileSync(cfg.datasetPath, 'utf8').split('\n')[0].trim();
+        if (firstLine && (0, validator_1.isSharegptFormat)(JSON.parse(firstLine))) {
+            datasetType = 'sharegpt';
+        }
+    }
+    catch {
+        // dataset already validated upstream; default to alpaca
+    }
+    const axolotlCfg = {
+        base_model: cfg.baseModel,
+        datasets: [{ path: cfg.datasetPath, type: datasetType }],
+        output_dir: cfg.outputDir,
+        num_epochs: cfg.epochs,
+        micro_batch_size: cfg.batchSize,
+        gradient_accumulation_steps: cfg.gradientAccumulation,
+        learning_rate: cfg.learningRate,
+        lora_r: cfg.loraR,
+        lora_alpha: cfg.loraAlpha,
+        sequence_len: cfg.maxSeqLength,
+        load_in_4bit: false, // minimal: no quantisation on Free tier
+        adapter: 'lora',
+    };
+    const cfgPath = `/tmp/qs-axolotl-${Date.now()}.yml`;
+    fs_1.default.writeFileSync(cfgPath, js_yaml_1.default.dump(axolotlCfg));
+    return new Promise((resolve, reject) => {
+        const proc = (0, child_process_1.spawn)('python3', [RUNNER, '--config', cfgPath], {
+            stdio: ['ignore', 'pipe', 'pipe'],
+        });
+        proc.stdout?.on('data', (d) => {
+            for (const line of d.toString().split('\n')) {
+                if (line.trim())
+                    logEmitter.emit('log', line);
+            }
+        });
+        proc.stderr?.on('data', (d) => {
+            for (const line of d.toString().split('\n')) {
+                if (line.trim())
+                    logEmitter.emit('log', line);
+            }
+        });
+        proc.on('close', (code) => {
+            try {
+                fs_1.default.unlinkSync(cfgPath);
+            }
+            catch {
+                /* best effort cleanup */
+            }
+            if (code === 0) {
+                resolve({ outputPath: cfg.outputDir });
+            }
+            else {
+                reject(new Error(`Axolotl runner exited with code ${code}`));
+            }
+        });
+        proc.on('error', (err) => {
+            try {
+                fs_1.default.unlinkSync(cfgPath);
+            }
+            catch {
+                /* best effort cleanup */
+            }
+            reject(err);
+        });
+    });
+}
+//# sourceMappingURL=axolotl.js.map

package/dist/training/adapters/axolotl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"axolotl.js","sourceRoot":"","sources":["../../../src/training/adapters/axolotl.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;AAYb,kBAqEC;AA/ED,iDAAsC;AACtC,gDAAwB;AACxB,4CAAoB;AACpB,sDAA2B;AAE3B,oCAAgE;AAChE,4CAAgD;AAEhD,MAAM,MAAM,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;AAErD,KAAK,UAAU,GAAG,CAAC,MAA6B,EAAE,UAAwB;IAC/E,MAAM,GAAG,GAAG,IAAA,qBAAa,EAAC,MAAM,CAAC,CAAC;IAElC,IAAI,WAAW,GAA0B,QAAQ,CAAC;IAClD,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,YAAE,CAAC,YAAY,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACjF,IAAI,SAAS,IAAI,IAAA,4BAAgB,EAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC;YACzD,WAAW,GAAG,UAAU,CAAC;QAC3B,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,wDAAwD;IAC1D,CAAC;IAED,MAAM,UAAU,GAAG;QACjB,UAAU,EAAE,GAAG,CAAC,SAAS;QACzB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;QACxD,UAAU,EAAE,GAAG,CAAC,SAAS;QACzB,UAAU,EAAE,GAAG,CAAC,MAAM;QACtB,gBAAgB,EAAE,GAAG,CAAC,SAAS;QAC/B,2BAA2B,EAAE,GAAG,CAAC,oBAAoB;QACrD,aAAa,EAAE,GAAG,CAAC,YAAY;QAC/B,MAAM,EAAE,GAAG,CAAC,KAAK;QACjB,UAAU,EAAE,GAAG,CAAC,SAAS;QACzB,YAAY,EAAE,GAAG,CAAC,YAAY;QAC9B,YAAY,EAAE,KAAK,EAAE,wCAAwC;QAC7D,OAAO,EAAE,MAAM;KAChB,CAAC;IAEF,MAAM,OAAO,GAAG,mBAAmB,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC;IACpD,YAAE,CAAC,aAAa,CAAC,OAAO,EAAE,iBAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;IAEjD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,IAAI,GAAG,IAAA,qBAAK,EAAC,SAAS,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE;YAC3D,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;SAClC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE;YACpC,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,IAAI,IAAI,CAAC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE;YACpC,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,IAAI,IAAI,CAAC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAmB,EAAE,EAAE;YACvC,IAAI,CAAC;gBACH,YAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YACzB,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;YACD,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACf,OAAO,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;YACzC,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACvB,IAAI,CAAC;gBACH,YAAE,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;YACzB,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;YACD,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/training/adapters/axolotl_runner.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+"""
+Minimal Axolotl fine-tuning runner (Free Tier).
+No DeepSpeed, no multi-GPU — designed for single-GPU / CPU.
+
+Requirements:
+    pip install axolotl
+"""
+
+import argparse
+import sys
+import os
+
+parser = argparse.ArgumentParser(description='LocoPilot Axolotl training runner')
+parser.add_argument('--config', required=True, help='Path to YAML config file')
+args = parser.parse_args()
+
+if not os.path.exists(args.config):
+    print(f'[axolotl] ERROR: Config file not found: {args.config}', file=sys.stderr)
+    sys.exit(1)
+
+try:
+    from axolotl.cli import train
+except ImportError as e:
+    print(f'[axolotl] ERROR: Missing dependency — {e}', file=sys.stderr)
+    print('[axolotl] Install with: pip install axolotl', file=sys.stderr)
+    sys.exit(1)
+
+print(f'[axolotl] Starting training with config: {args.config}')
+try:
+    # axolotl.cli.train accepts a config path and kicks off training
+    train([args.config])
+    print('[axolotl] Training complete')
+except SystemExit as e:
+    # axolotl may call sys.exit(0) on success
+    if e.code not in (0, None):
+        print(f'[axolotl] ERROR: Training failed with exit code {e.code}', file=sys.stderr)
+        sys.exit(e.code)

package/dist/training/adapters/mlx.js

@@ -0,0 +1,57 @@
+'use strict';
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.run = run;
+const child_process_1 = require("child_process");
+const path_1 = __importDefault(require("path"));
+const fs_1 = __importDefault(require("fs"));
+const types_1 = require("../types");
+const RUNNER = path_1.default.resolve(__dirname, 'mlx_runner.py');
+async function run(config, logEmitter) {
+    const cfg = (0, types_1.applyDefaults)(config);
+    const configPath = `/tmp/qs-mlx-${Date.now()}.json`;
+    fs_1.default.writeFileSync(configPath, JSON.stringify(cfg));
+    return new Promise((resolve, reject) => {
+        const proc = (0, child_process_1.spawn)('python3', [RUNNER, '--config', configPath], {
+            stdio: ['ignore', 'pipe', 'pipe'],
+        });
+        proc.stdout?.on('data', (d) => {
+            for (const line of d.toString().split('\n')) {
+                if (line.trim())
+                    logEmitter.emit('log', line);
+            }
+        });
+        proc.stderr?.on('data', (d) => {
+            for (const line of d.toString().split('\n')) {
+                if (line.trim())
+                    logEmitter.emit('log', line);
+            }
+        });
+        proc.on('close', (code) => {
+            try {
+                fs_1.default.unlinkSync(configPath);
+            }
+            catch {
+                /* best effort cleanup */
+            }
+            if (code === 0) {
+                resolve({ outputPath: cfg.outputDir });
+            }
+            else {
+                reject(new Error(`MLX runner exited with code ${code}`));
+            }
+        });
+        proc.on('error', (err) => {
+            try {
+                fs_1.default.unlinkSync(configPath);
+            }
+            catch {
+                /* best effort cleanup */
+            }
+            reject(err);
+        });
+    });
+}
+//# sourceMappingURL=mlx.js.map

package/dist/training/adapters/mlx.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mlx.js","sourceRoot":"","sources":["../../../src/training/adapters/mlx.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;AAUb,kBA2CC;AAnDD,iDAAsC;AACtC,gDAAwB;AACxB,4CAAoB;AAEpB,oCAAgE;AAEhE,MAAM,MAAM,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;AAEjD,KAAK,UAAU,GAAG,CAAC,MAA6B,EAAE,UAAwB;IAC/E,MAAM,GAAG,GAAG,IAAA,qBAAa,EAAC,MAAM,CAAC,CAAC;IAClC,MAAM,UAAU,GAAG,eAAe,IAAI,CAAC,GAAG,EAAE,OAAO,CAAC;IACpD,YAAE,CAAC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;IAElD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,IAAI,GAAG,IAAA,qBAAK,EAAC,SAAS,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,UAAU,CAAC,EAAE;YAC9D,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;SAClC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE;YACpC,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,IAAI,IAAI,CAAC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE;YACpC,KAAK,MAAM,IAAI,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,IAAI,IAAI,CAAC,IAAI,EAAE;oBAAE,UAAU,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAChD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAmB,EAAE,EAAE;YACvC,IAAI,CAAC;gBACH,YAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAC5B,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;YACD,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACf,OAAO,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,SAAS,EAAE,CAAC,CAAC;YACzC,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,KAAK,CAAC,+BAA+B,IAAI,EAAE,CAAC,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACvB,IAAI,CAAC;gBACH,YAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAC5B,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;YACD,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/training/adapters/mlx_runner.py

@@ -0,0 +1,175 @@
+#!/usr/bin/env python3
+"""
+MLX fine-tuning runner (Free Tier — Apple Silicon).
+Uses mlx-lm for LoRA training on Apple Metal GPU.
+
+Requirements:
+    pip install mlx-lm
+"""
+
+import argparse
+import json
+import os
+import shutil
+import sys
+import tempfile
+
+parser = argparse.ArgumentParser(description='LocoPilot MLX training runner')
+parser.add_argument('--config', required=True, help='Path to JSON config file')
+args = parser.parse_args()
+
+try:
+    with open(args.config) as f:
+        cfg = json.load(f)
+except Exception as e:
+    print(f'[mlx] ERROR: Could not read config: {e}', file=sys.stderr)
+    sys.exit(1)
+
+required = ['baseModel', 'datasetPath', 'outputDir', 'epochs', 'batchSize',
+            'loraR', 'loraAlpha', 'maxSeqLength', 'learningRate']
+for key in required:
+    if key not in cfg:
+        print(f'[mlx] ERROR: Missing config key: {key}', file=sys.stderr)
+        sys.exit(1)
+
+try:
+    import mlx.core  # noqa: F401
+    import mlx_lm    # noqa: F401
+except ImportError as e:
+    print(f'[mlx] ERROR: Missing dependency — {e}', file=sys.stderr)
+    print('[mlx] Install with: pip install mlx-lm', file=sys.stderr)
+    sys.exit(1)
+
+
+def convert_to_mlx_format(dataset_path):
+    """Convert alpaca/sharegpt JSONL to mlx-lm text format."""
+    examples = []
+    with open(dataset_path) as f:
+        for line in f:
+            line = line.strip()
+            if not line:
+                continue
+            row = json.loads(line)
+
+            if 'instruction' in row:
+                instruction = row.get('instruction', '')
+                inp = row.get('input', '')
+                output = row.get('output', '')
+                if inp:
+                    text = (f'### Instruction:\n{instruction}\n\n'
+                            f'### Input:\n{inp}\n\n'
+                            f'### Response:\n{output}')
+                else:
+                    text = f'### Instruction:\n{instruction}\n\n### Response:\n{output}'
+                examples.append({'text': text})
+
+            elif 'conversations' in row:
+                turns = row.get('conversations', [])
+                text = ''
+                for turn in turns:
+                    role = 'Human' if turn.get('from') == 'human' else 'Assistant'
+                    text += f'{role}: {turn.get("value", "")}\n'
+                examples.append({'text': text.strip()})
+
+    return examples
+
+
+print(f'[mlx] Preparing dataset from: {cfg["datasetPath"]}')
+
+# Initialize before try so the finally block can always reference it safely
+data_dir = None
+
+try:
+    data_dir = tempfile.mkdtemp(prefix='qs-mlx-data-')
+
+    examples = convert_to_mlx_format(cfg['datasetPath'])
+    if not examples:
+        print('[mlx] ERROR: Dataset produced no examples after conversion', file=sys.stderr)
+        sys.exit(1)
+
+    # Split 90/10 train/valid
+    split = max(1, int(len(examples) * 0.9))
+    train_examples = examples[:split]
+    valid_examples = examples[split:] or examples[:1]
+
+    with open(os.path.join(data_dir, 'train.jsonl'), 'w') as f:
+        for ex in train_examples:
+            f.write(json.dumps(ex) + '\n')
+    with open(os.path.join(data_dir, 'valid.jsonl'), 'w') as f:
+        for ex in valid_examples:
+            f.write(json.dumps(ex) + '\n')
+
+    print(f'[mlx] Dataset: {len(train_examples)} train, {len(valid_examples)} valid examples')
+
+    os.makedirs(cfg['outputDir'], exist_ok=True)
+    print(f'[mlx] Loading model: {cfg["baseModel"]}')
+    print(f'[mlx] Starting LoRA training — {cfg["epochs"]} epoch(s), batch size {cfg["batchSize"]}')
+
+    _used_python_api = False
+    try:
+        from mlx_lm import load
+        from mlx_lm.tuner.trainer import TrainingArgs, train
+        from mlx_lm.tuner.datasets import load_dataset as mlx_load_dataset
+
+        model, tokenizer = load(cfg['baseModel'])
+
+        training_args = TrainingArgs(
+            batch_size=cfg['batchSize'],
+            iters=cfg['epochs'] * max(len(train_examples) // cfg['batchSize'], 1),
+            val_batches=5,
+            steps_per_report=10,
+            steps_per_eval=50,
+            adapter_path=cfg['outputDir'],
+            max_seq_length=cfg['maxSeqLength'],
+            grad_checkpoint=False,
+            learning_rate=cfg['learningRate'],
+            lora_parameters={
+                'rank': cfg['loraR'],
+                'alpha': cfg['loraAlpha'],
+                'dropout': 0.0,
+                'scale': cfg['loraAlpha'] / cfg['loraR'],
+            },
+        )
+
+        train_set, valid_set, _ = mlx_load_dataset(training_args, tokenizer, data_dir)
+        train(model, tokenizer, train_set, valid_set, training_args)
+        _used_python_api = True
+
+    except (ImportError, AttributeError, TypeError):
+        # Python API unavailable or API changed — fall back to stable mlx_lm.lora CLI
+        pass
+
+    if not _used_python_api:
+        import subprocess
+        print('[mlx] Using mlx-lm CLI (stable fallback)')
+        iters = cfg['epochs'] * max(len(train_examples) // cfg['batchSize'], 1)
+        cmd = [
+            sys.executable, '-m', 'mlx_lm.lora',
+            '--model', cfg['baseModel'],
+            '--train',
+            '--data', data_dir,
+            '--batch-size', str(cfg['batchSize']),
+            '--num-layers', str(cfg['loraR']),
+            '--iters', str(iters),
+            '--adapter-path', cfg['outputDir'],
+            '--learning-rate', str(cfg['learningRate']),
+            '--max-seq-length', str(cfg['maxSeqLength']),
+        ]
+        result = subprocess.run(cmd)
+        if result.returncode != 0:
+            print(f'[mlx] ERROR: Training failed with exit code {result.returncode}', file=sys.stderr)
+            sys.exit(result.returncode)
+
+    print(f'[mlx] Adapter saved to {cfg["outputDir"]}')
+
+except SystemExit:
+    raise
+except Exception as e:
+    print(f'[mlx] ERROR: Training failed: {e}', file=sys.stderr)
+    sys.exit(1)
+finally:
+    if data_dir is not None:
+        try:
+            shutil.rmtree(data_dir)
+        except Exception:
+            pass