@infrarix/locopilot 1.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +239 -0
- package/dist/api/index.js +79 -0
- package/dist/api/index.js.map +1 -0
- package/dist/api/middleware/rateLimiter.js +27 -0
- package/dist/api/middleware/rateLimiter.js.map +1 -0
- package/dist/api/routes/chat.js +75 -0
- package/dist/api/routes/chat.js.map +1 -0
- package/dist/api/routes/completions.js +72 -0
- package/dist/api/routes/completions.js.map +1 -0
- package/dist/api/routes/health.js +52 -0
- package/dist/api/routes/health.js.map +1 -0
- package/dist/api/routes/models.js +50 -0
- package/dist/api/routes/models.js.map +1 -0
- package/dist/api/routes/training.js +10 -0
- package/dist/api/routes/training.js.map +1 -0
- package/dist/api/services/localRouter.js +201 -0
- package/dist/api/services/localRouter.js.map +1 -0
- package/dist/api/services/localStubs.js +28 -0
- package/dist/api/services/localStubs.js.map +1 -0
- package/dist/api/services/ollama.js +22 -0
- package/dist/api/services/ollama.js.map +1 -0
- package/dist/api/types.js +3 -0
- package/dist/api/types.js.map +1 -0
- package/dist/api/utils/sse.js +78 -0
- package/dist/api/utils/sse.js.map +1 -0
- package/dist/cli/commands/doctor.js +230 -0
- package/dist/cli/commands/doctor.js.map +1 -0
- package/dist/cli/commands/expose.js +98 -0
- package/dist/cli/commands/expose.js.map +1 -0
- package/dist/cli/commands/init.js +340 -0
- package/dist/cli/commands/init.js.map +1 -0
- package/dist/cli/commands/login.js +116 -0
- package/dist/cli/commands/login.js.map +1 -0
- package/dist/cli/commands/logout.js +38 -0
- package/dist/cli/commands/logout.js.map +1 -0
- package/dist/cli/commands/logs.js +95 -0
- package/dist/cli/commands/logs.js.map +1 -0
- package/dist/cli/commands/models.js +106 -0
- package/dist/cli/commands/models.js.map +1 -0
- package/dist/cli/commands/start.js +132 -0
- package/dist/cli/commands/start.js.map +1 -0
- package/dist/cli/commands/train.js +211 -0
- package/dist/cli/commands/train.js.map +1 -0
- package/dist/cli/commands/usage.js +43 -0
- package/dist/cli/commands/usage.js.map +1 -0
- package/dist/cli/commands/whoami.js +54 -0
- package/dist/cli/commands/whoami.js.map +1 -0
- package/dist/cli/index.js +49 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/cli/utils/banner.js +177 -0
- package/dist/cli/utils/banner.js.map +1 -0
- package/dist/cli/utils/paths.js +37 -0
- package/dist/cli/utils/paths.js.map +1 -0
- package/dist/cloud/client.js +157 -0
- package/dist/cloud/client.js.map +1 -0
- package/dist/shared/constants.js +39 -0
- package/dist/shared/constants.js.map +1 -0
- package/dist/shared/crypto.js +26 -0
- package/dist/shared/crypto.js.map +1 -0
- package/dist/shared/db/pool.js +83 -0
- package/dist/shared/db/pool.js.map +1 -0
- package/dist/shared/errors.js +59 -0
- package/dist/shared/errors.js.map +1 -0
- package/dist/shared/index.js +24 -0
- package/dist/shared/index.js.map +1 -0
- package/dist/shared/ndjson.js +39 -0
- package/dist/shared/ndjson.js.map +1 -0
- package/dist/shared/runtime/ollama/index.js +55 -0
- package/dist/shared/runtime/ollama/index.js.map +1 -0
- package/dist/shared/types.js +3 -0
- package/dist/shared/types.js.map +1 -0
- package/dist/training/adapters/axolotl.js +83 -0
- package/dist/training/adapters/axolotl.js.map +1 -0
- package/dist/training/adapters/axolotl_runner.py +38 -0
- package/dist/training/adapters/mlx.js +57 -0
- package/dist/training/adapters/mlx.js.map +1 -0
- package/dist/training/adapters/mlx_runner.py +175 -0
- package/dist/training/adapters/unsloth.js +57 -0
- package/dist/training/adapters/unsloth.js.map +1 -0
- package/dist/training/adapters/unsloth_runner.py +116 -0
- package/dist/training/index.js +47 -0
- package/dist/training/index.js.map +1 -0
- package/dist/training/types.js +18 -0
- package/dist/training/types.js.map +1 -0
- package/dist/training/validator.js +67 -0
- package/dist/training/validator.js.map +1 -0
- package/dist/worker/executor.js +98 -0
- package/dist/worker/executor.js.map +1 -0
- package/dist/worker/handlers.js +197 -0
- package/dist/worker/handlers.js.map +1 -0
- package/dist/worker/index.js +45 -0
- package/dist/worker/index.js.map +1 -0
- package/dist/worker/logStore.js +24 -0
- package/dist/worker/logStore.js.map +1 -0
- package/dist/worker/types.js +3 -0
- package/dist/worker/types.js.map +1 -0
- package/dist/worker/worker.js +12 -0
- package/dist/worker/worker.js.map +1 -0
- package/package.json +81 -0
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const commander_1 = require("commander");
|
|
7
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
8
|
+
const fs_1 = __importDefault(require("fs"));
|
|
9
|
+
const paths_1 = require("../utils/paths");
|
|
10
|
+
function readToken() {
|
|
11
|
+
try {
|
|
12
|
+
if (!fs_1.default.existsSync(paths_1.CONFIG_PATH))
|
|
13
|
+
return null;
|
|
14
|
+
const data = JSON.parse(fs_1.default.readFileSync(paths_1.CONFIG_PATH, 'utf8'));
|
|
15
|
+
const t = data.token;
|
|
16
|
+
return typeof t === 'string' && t.startsWith('qs_') ? t : null;
|
|
17
|
+
}
|
|
18
|
+
catch {
|
|
19
|
+
return null;
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
const cmd = new commander_1.Command('logs').description('Stream logs for a training job');
|
|
23
|
+
cmd.requiredOption('--job <id>', 'Training job ID');
|
|
24
|
+
cmd.action(async (opts) => {
|
|
25
|
+
(0, paths_1.loadEnv)();
|
|
26
|
+
const base = `http://localhost:${process.env.API_PORT ?? '8080'}`;
|
|
27
|
+
const token = readToken();
|
|
28
|
+
const url = `${base}/v1/locopilot/training/jobs/${opts.job}/logs`;
|
|
29
|
+
const headers = {};
|
|
30
|
+
if (token)
|
|
31
|
+
headers['Authorization'] = `Bearer ${token}`;
|
|
32
|
+
console.log(chalk_1.default.gray(`\n Streaming logs for job ${opts.job}...\n`));
|
|
33
|
+
let res;
|
|
34
|
+
try {
|
|
35
|
+
res = await fetch(url, { headers, signal: AbortSignal.timeout(600_000) });
|
|
36
|
+
}
|
|
37
|
+
catch (err) {
|
|
38
|
+
console.error(chalk_1.default.red(` Could not connect to API: ${err.message}`));
|
|
39
|
+
console.error(chalk_1.default.gray(' Is LocoPilot running? Try: locopilot start'));
|
|
40
|
+
process.exit(1);
|
|
41
|
+
}
|
|
42
|
+
if (!res.ok) {
|
|
43
|
+
const body = (await res.json().catch(() => ({})));
|
|
44
|
+
console.error(chalk_1.default.red(` API error ${res.status}:`), body.error ?? 'Unknown error');
|
|
45
|
+
if (res.status === 404) {
|
|
46
|
+
console.error(chalk_1.default.gray(' Check the job ID: locopilot train --config <file>'));
|
|
47
|
+
}
|
|
48
|
+
process.exit(1);
|
|
49
|
+
}
|
|
50
|
+
const reader = res.body?.getReader();
|
|
51
|
+
if (!reader) {
|
|
52
|
+
console.error(chalk_1.default.red(' No response body from server.'));
|
|
53
|
+
process.exit(1);
|
|
54
|
+
}
|
|
55
|
+
const decoder = new TextDecoder();
|
|
56
|
+
let buffer = '';
|
|
57
|
+
try {
|
|
58
|
+
while (true) {
|
|
59
|
+
const { done, value } = await reader.read();
|
|
60
|
+
if (done)
|
|
61
|
+
break;
|
|
62
|
+
buffer += decoder.decode(value, { stream: true });
|
|
63
|
+
const lines = buffer.split('\n');
|
|
64
|
+
buffer = lines.pop() ?? '';
|
|
65
|
+
for (const line of lines) {
|
|
66
|
+
if (!line.startsWith('data: '))
|
|
67
|
+
continue;
|
|
68
|
+
const data = line.slice(6).trim();
|
|
69
|
+
if (data === '[DONE]') {
|
|
70
|
+
console.log(chalk_1.default.green('\n Stream ended.\n'));
|
|
71
|
+
return;
|
|
72
|
+
}
|
|
73
|
+
if (data)
|
|
74
|
+
process.stdout.write(data + '\n');
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
// Flush remaining buffer
|
|
78
|
+
if (buffer.startsWith('data: ')) {
|
|
79
|
+
const data = buffer.slice(6).trim();
|
|
80
|
+
if (data && data !== '[DONE]')
|
|
81
|
+
process.stdout.write(data + '\n');
|
|
82
|
+
}
|
|
83
|
+
console.log(chalk_1.default.gray('\n Log stream closed. Job may still be running.'));
|
|
84
|
+
console.log(chalk_1.default.gray(` Re-run: locopilot logs --job ${opts.job}\n`));
|
|
85
|
+
}
|
|
86
|
+
catch (err) {
|
|
87
|
+
console.error(chalk_1.default.yellow(`\n Stream ended: ${err.message}`));
|
|
88
|
+
console.error(chalk_1.default.gray(` Re-run: locopilot logs --job ${opts.job}\n`));
|
|
89
|
+
}
|
|
90
|
+
finally {
|
|
91
|
+
reader.releaseLock();
|
|
92
|
+
}
|
|
93
|
+
});
|
|
94
|
+
exports.default = cmd;
|
|
95
|
+
//# sourceMappingURL=logs.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"logs.js","sourceRoot":"","sources":["../../../src/cli/commands/logs.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;AAEb,yCAAoC;AACpC,kDAA0B;AAC1B,4CAAoB;AACpB,0CAAsD;AAMtD,SAAS,SAAS;IAChB,IAAI,CAAC;QACH,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,mBAAW,CAAC;YAAE,OAAO,IAAI,CAAC;QAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,YAAE,CAAC,YAAY,CAAC,mBAAW,EAAE,MAAM,CAAC,CAA4B,CAAC;QACzF,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC;QACrB,OAAO,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACjE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,GAAG,GAAG,IAAI,mBAAO,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,gCAAgC,CAAC,CAAC;AAC9E,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,iBAAiB,CAAC,CAAC;AAEpD,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,IAAiB,EAAE,EAAE;IACrC,IAAA,eAAO,GAAE,CAAC;IACV,MAAM,IAAI,GAAG,oBAAoB,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,MAAM,EAAE,CAAC;IAClE,MAAM,KAAK,GAAG,SAAS,EAAE,CAAC;IAE1B,MAAM,GAAG,GAAG,GAAG,IAAI,+BAA+B,IAAI,CAAC,GAAG,OAAO,CAAC;IAClE,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,IAAI,KAAK;QAAE,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAC;IAExD,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,8BAA8B,IAAI,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAEvE,IAAI,GAAa,CAAC;IAClB,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAC5E,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,+BAAgC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAClF,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC,CAAC;QAC1E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAA4B,CAAC,CAAuB,CAAC;QACnG,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,eAAe,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,IAAI,CAAC,KAAK,IAAI,eAAe,CAAC,CAAC;QACtF,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACvB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC,CAAC;QACnF,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;IACrC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC,CAAC;QAC5D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,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,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;oBAAE,SAAS;gBACzC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAClC,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACtB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC,CAAC;oBAChD,OAAO;gBACT,CAAC;gBACD,IAAI,IAAI;oBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,yBAAyB;QACzB,IAAI,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACpC,IAAI,IAAI,IAAI,IAAI,KAAK,QAAQ;gBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;QACnE,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC,CAAC;QAC5E,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,kCAAkC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;IAC1E,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,MAAM,CAAC,qBAAsB,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3E,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,kCAAkC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;IAC5E,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,kBAAe,GAAG,CAAC"}
|
|
@@ -0,0 +1,106 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const commander_1 = require("commander");
|
|
7
|
+
const child_process_1 = require("child_process");
|
|
8
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
9
|
+
const ora_1 = __importDefault(require("ora"));
|
|
10
|
+
const paths_1 = require("../utils/paths");
|
|
11
|
+
// Allowlist: only alphanumeric + safe separators (no shell metacharacters)
|
|
12
|
+
const MODEL_NAME_REGEX = /^[a-zA-Z0-9:._-]+$/;
|
|
13
|
+
const cmd = new commander_1.Command('models').description('Manage models');
|
|
14
|
+
cmd
|
|
15
|
+
.command('pull <model>')
|
|
16
|
+
.description('Pull a model into Ollama')
|
|
17
|
+
.action((model) => {
|
|
18
|
+
if (!MODEL_NAME_REGEX.test(model)) {
|
|
19
|
+
console.error(chalk_1.default.red(' Invalid model name. Use only letters, numbers, colons, hyphens, dots, and underscores.'));
|
|
20
|
+
process.exit(1);
|
|
21
|
+
}
|
|
22
|
+
console.log(chalk_1.default.bold(`\n Pulling ${model}...\n`));
|
|
23
|
+
try {
|
|
24
|
+
// execFileSync with stdio: 'inherit' shows ollama's own progress bar
|
|
25
|
+
(0, child_process_1.execFileSync)('ollama', ['pull', model], { stdio: 'inherit' });
|
|
26
|
+
console.log(chalk_1.default.green(`\n ✔ Model ${model} ready.\n`));
|
|
27
|
+
}
|
|
28
|
+
catch (err) {
|
|
29
|
+
const e = err;
|
|
30
|
+
if (e.code === 'ENOENT') {
|
|
31
|
+
console.error(chalk_1.default.red('\n ollama not found.'));
|
|
32
|
+
const hint = process.platform === 'darwin'
|
|
33
|
+
? 'brew install --cask ollama'
|
|
34
|
+
: process.platform === 'win32'
|
|
35
|
+
? 'winget install Ollama.Ollama'
|
|
36
|
+
: 'curl -fsSL https://ollama.com/install.sh | sh';
|
|
37
|
+
console.error(chalk_1.default.gray(` Install: ${hint}\n`));
|
|
38
|
+
}
|
|
39
|
+
else {
|
|
40
|
+
console.error(chalk_1.default.red(`\n ✖ Failed to pull ${model}: ${e.message}\n`));
|
|
41
|
+
}
|
|
42
|
+
process.exit(1);
|
|
43
|
+
}
|
|
44
|
+
});
|
|
45
|
+
cmd
|
|
46
|
+
.command('list')
|
|
47
|
+
.description('List available models')
|
|
48
|
+
.action(async () => {
|
|
49
|
+
(0, paths_1.loadEnv)();
|
|
50
|
+
const base = `http://localhost:${process.env.API_PORT ?? '8080'}`;
|
|
51
|
+
const spinner = (0, ora_1.default)(' Fetching models...').start();
|
|
52
|
+
let data;
|
|
53
|
+
try {
|
|
54
|
+
const res = await fetch(`${base}/v1/models`, {
|
|
55
|
+
signal: AbortSignal.timeout(5_000),
|
|
56
|
+
});
|
|
57
|
+
if (!res.ok) {
|
|
58
|
+
spinner.fail(chalk_1.default.red(`API returned ${res.status}`));
|
|
59
|
+
console.error(chalk_1.default.gray(' Is LocoPilot running? Try: locopilot start'));
|
|
60
|
+
process.exit(1);
|
|
61
|
+
}
|
|
62
|
+
const body = (await res.json());
|
|
63
|
+
data = body.data;
|
|
64
|
+
spinner.stop();
|
|
65
|
+
}
|
|
66
|
+
catch (err) {
|
|
67
|
+
spinner.fail(chalk_1.default.red(`Could not connect to API: ${err.message}`));
|
|
68
|
+
console.error(chalk_1.default.gray(' Is LocoPilot running? Try: locopilot start'));
|
|
69
|
+
process.exit(1);
|
|
70
|
+
}
|
|
71
|
+
console.log(chalk_1.default.bold('\n Models:\n'));
|
|
72
|
+
if (!data || data.length === 0) {
|
|
73
|
+
console.log(chalk_1.default.gray(' No models found. Run: locopilot models pull <model>\n'));
|
|
74
|
+
return;
|
|
75
|
+
}
|
|
76
|
+
for (const m of data) {
|
|
77
|
+
const tag = m.owned_by === 'local' ? chalk_1.default.green('[local] ') : chalk_1.default.blue('[remote]');
|
|
78
|
+
console.log(` ${tag} ${m.id}`);
|
|
79
|
+
}
|
|
80
|
+
console.log();
|
|
81
|
+
});
|
|
82
|
+
cmd
|
|
83
|
+
.command('rm <model>')
|
|
84
|
+
.description('Remove a local model from Ollama')
|
|
85
|
+
.action((model) => {
|
|
86
|
+
if (!MODEL_NAME_REGEX.test(model)) {
|
|
87
|
+
console.error(chalk_1.default.red(' Invalid model name.'));
|
|
88
|
+
process.exit(1);
|
|
89
|
+
}
|
|
90
|
+
try {
|
|
91
|
+
(0, child_process_1.execFileSync)('ollama', ['rm', model], { stdio: 'inherit' });
|
|
92
|
+
console.log(chalk_1.default.green(`\n ✔ Model ${model} removed.\n`));
|
|
93
|
+
}
|
|
94
|
+
catch (err) {
|
|
95
|
+
const e = err;
|
|
96
|
+
if (e.code === 'ENOENT') {
|
|
97
|
+
console.error(chalk_1.default.red('\n ollama not found.\n'));
|
|
98
|
+
}
|
|
99
|
+
else {
|
|
100
|
+
console.error(chalk_1.default.red(`\n ✖ Failed to remove ${model}: ${e.message}\n`));
|
|
101
|
+
}
|
|
102
|
+
process.exit(1);
|
|
103
|
+
}
|
|
104
|
+
});
|
|
105
|
+
exports.default = cmd;
|
|
106
|
+
//# sourceMappingURL=models.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"models.js","sourceRoot":"","sources":["../../../src/cli/commands/models.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;AAEb,yCAAoC;AACpC,iDAA6C;AAC7C,kDAA0B;AAC1B,8CAAsB;AACtB,0CAAyC;AAEzC,2EAA2E;AAC3E,MAAM,gBAAgB,GAAG,oBAAoB,CAAC;AAE9C,MAAM,GAAG,GAAG,IAAI,mBAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;AAE/D,GAAG;KACA,OAAO,CAAC,cAAc,CAAC;KACvB,WAAW,CAAC,0BAA0B,CAAC;KACvC,MAAM,CAAC,CAAC,KAAa,EAAE,EAAE;IACxB,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CACX,eAAK,CAAC,GAAG,CAAC,0FAA0F,CAAC,CACtG,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,eAAe,KAAK,OAAO,CAAC,CAAC,CAAC;IAErD,IAAI,CAAC;QACH,qEAAqE;QACrE,IAAA,4BAAY,EAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC9D,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,eAAe,KAAK,WAAW,CAAC,CAAC,CAAC;IAC5D,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,GAAG,GAA4B,CAAC;QACvC,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACxB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC,CAAC;YAClD,MAAM,IAAI,GACR,OAAO,CAAC,QAAQ,KAAK,QAAQ;gBAC3B,CAAC,CAAC,4BAA4B;gBAC9B,CAAC,CAAC,OAAO,CAAC,QAAQ,KAAK,OAAO;oBAC5B,CAAC,CAAC,8BAA8B;oBAChC,CAAC,CAAC,+CAA+C,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC;QACpD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,wBAAwB,KAAK,KAAK,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QAC5E,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,GAAG;KACA,OAAO,CAAC,MAAM,CAAC;KACf,WAAW,CAAC,uBAAuB,CAAC;KACpC,MAAM,CAAC,KAAK,IAAI,EAAE;IACjB,IAAA,eAAO,GAAE,CAAC;IACV,MAAM,IAAI,GAAG,oBAAoB,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,MAAM,EAAE,CAAC;IAElE,MAAM,OAAO,GAAG,IAAA,aAAG,EAAC,sBAAsB,CAAC,CAAC,KAAK,EAAE,CAAC;IAEpD,IAAI,IAAyD,CAAC;IAC9D,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,YAAY,EAAE;YAC3C,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,KAAK,CAAC;SACnC,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,GAAG,CAAC,gBAAgB,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YACtD,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC,CAAC;YAC1E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuD,CAAC;QACtF,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;QACjB,OAAO,CAAC,IAAI,EAAE,CAAC;IACjB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,IAAI,CAAC,eAAK,CAAC,GAAG,CAAC,6BAA8B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC/E,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC,CAAC;QAC1E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC;IACzC,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC,CAAC;QACnF,OAAO;IACT,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;QACrB,MAAM,GAAG,GAAG,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,eAAK,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,eAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACtF,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,CAAC,GAAG,EAAE,CAAC;AAChB,CAAC,CAAC,CAAC;AAEL,GAAG;KACA,OAAO,CAAC,YAAY,CAAC;KACrB,WAAW,CAAC,kCAAkC,CAAC;KAC/C,MAAM,CAAC,CAAC,KAAa,EAAE,EAAE;IACxB,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC,CAAC;QAClD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,CAAC;QACH,IAAA,4BAAY,EAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;QAC5D,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,eAAe,KAAK,aAAa,CAAC,CAAC,CAAC;IAC9D,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,GAAG,GAA4B,CAAC;QACvC,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACxB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,CAAC;QACtD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,0BAA0B,KAAK,KAAK,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QAC9E,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEL,kBAAe,GAAG,CAAC"}
|
|
@@ -0,0 +1,132 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
14
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
15
|
+
}) : function(o, v) {
|
|
16
|
+
o["default"] = v;
|
|
17
|
+
});
|
|
18
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
19
|
+
var ownKeys = function(o) {
|
|
20
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
21
|
+
var ar = [];
|
|
22
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
23
|
+
return ar;
|
|
24
|
+
};
|
|
25
|
+
return ownKeys(o);
|
|
26
|
+
};
|
|
27
|
+
return function (mod) {
|
|
28
|
+
if (mod && mod.__esModule) return mod;
|
|
29
|
+
var result = {};
|
|
30
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
31
|
+
__setModuleDefault(result, mod);
|
|
32
|
+
return result;
|
|
33
|
+
};
|
|
34
|
+
})();
|
|
35
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
36
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
37
|
+
};
|
|
38
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
39
|
+
const commander_1 = require("commander");
|
|
40
|
+
const child_process_1 = require("child_process");
|
|
41
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
42
|
+
const fs_1 = __importDefault(require("fs"));
|
|
43
|
+
const path_1 = __importDefault(require("path"));
|
|
44
|
+
const paths = __importStar(require("../utils/paths"));
|
|
45
|
+
const banner_1 = require("../utils/banner");
|
|
46
|
+
const IS_WINDOWS = process.platform === 'win32';
|
|
47
|
+
const cmd = new commander_1.Command('start').description('Start the LocoPilot local API server');
|
|
48
|
+
cmd.option('-p, --port <port>', 'Port to listen on (default: 8080)');
|
|
49
|
+
cmd.action(async (opts) => {
|
|
50
|
+
(0, banner_1.printBanner)();
|
|
51
|
+
paths.ensureHomeDir();
|
|
52
|
+
paths.loadEnv();
|
|
53
|
+
const port = opts.port ?? process.env.API_PORT ?? '8080';
|
|
54
|
+
// ── Pre-flight: check that init has been run ───────────────────────────
|
|
55
|
+
if (!fs_1.default.existsSync(paths.SQLITE_PATH)) {
|
|
56
|
+
console.error(chalk_1.default.red(' ✖ Database not found.'));
|
|
57
|
+
console.error(chalk_1.default.gray(' Run: locopilot init'));
|
|
58
|
+
process.exit(1);
|
|
59
|
+
}
|
|
60
|
+
console.log(chalk_1.default.bold(` Starting LocoPilot on port ${port}...\n`));
|
|
61
|
+
// ── Resolve the API entry point ────────────────────────────────────────
|
|
62
|
+
// __dirname in dist/ = dist/cli/commands/ → 3 up = project root
|
|
63
|
+
// __dirname in src/ = src/cli/commands/ → 3 up = project root
|
|
64
|
+
const projectRoot = path_1.default.resolve(__dirname, '../../..');
|
|
65
|
+
const apiDistPath = path_1.default.join(projectRoot, 'dist', 'api', 'index.js');
|
|
66
|
+
const apiSrcPath = path_1.default.join(projectRoot, 'src', 'api', 'index.ts');
|
|
67
|
+
let runCmd;
|
|
68
|
+
let runArgs;
|
|
69
|
+
if (fs_1.default.existsSync(apiDistPath)) {
|
|
70
|
+
runCmd = process.execPath; // node
|
|
71
|
+
runArgs = [apiDistPath];
|
|
72
|
+
}
|
|
73
|
+
else if (fs_1.default.existsSync(apiSrcPath)) {
|
|
74
|
+
// Development: fall back to ts-node
|
|
75
|
+
const tsNode = IS_WINDOWS ? 'ts-node.cmd' : 'ts-node';
|
|
76
|
+
runCmd = IS_WINDOWS ? 'npx.cmd' : 'npx';
|
|
77
|
+
runArgs = [tsNode, '--project', path_1.default.join(projectRoot, 'tsconfig.json'), apiSrcPath];
|
|
78
|
+
}
|
|
79
|
+
else {
|
|
80
|
+
console.error(chalk_1.default.red(' ✖ Could not locate API entry point.'));
|
|
81
|
+
console.error(chalk_1.default.gray(` Expected: ${apiDistPath}`));
|
|
82
|
+
console.error(chalk_1.default.gray(' Run: npm run build'));
|
|
83
|
+
process.exit(1);
|
|
84
|
+
}
|
|
85
|
+
console.log(chalk_1.default.gray(` Entry: ${runCmd === process.execPath ? apiDistPath : apiSrcPath}`));
|
|
86
|
+
const child = (0, child_process_1.spawn)(runCmd, runArgs, {
|
|
87
|
+
stdio: 'inherit',
|
|
88
|
+
// On Windows, shell:false requires exact binary name; process.execPath is always correct for node.
|
|
89
|
+
shell: false,
|
|
90
|
+
env: {
|
|
91
|
+
...process.env,
|
|
92
|
+
API_PORT: port,
|
|
93
|
+
SQLITE_PATH: paths.SQLITE_PATH,
|
|
94
|
+
OLLAMA_HOST: process.env.OLLAMA_HOST ?? 'http://localhost:11434',
|
|
95
|
+
},
|
|
96
|
+
});
|
|
97
|
+
// Forward signals to the child so Ctrl+C propagates cleanly
|
|
98
|
+
const forward = (sig) => {
|
|
99
|
+
try {
|
|
100
|
+
child.kill(sig);
|
|
101
|
+
}
|
|
102
|
+
catch {
|
|
103
|
+
/* already exited */
|
|
104
|
+
}
|
|
105
|
+
};
|
|
106
|
+
process.on('SIGINT', () => forward('SIGINT'));
|
|
107
|
+
process.on('SIGTERM', () => forward('SIGTERM'));
|
|
108
|
+
child.on('error', (err) => {
|
|
109
|
+
if (err.code === 'ENOENT') {
|
|
110
|
+
console.error(chalk_1.default.red(`\n ✖ Command not found: ${runCmd}`));
|
|
111
|
+
if (!fs_1.default.existsSync(apiDistPath)) {
|
|
112
|
+
console.error(chalk_1.default.gray(' Build first: npm run build'));
|
|
113
|
+
}
|
|
114
|
+
}
|
|
115
|
+
else {
|
|
116
|
+
console.error(chalk_1.default.red(`\n ✖ Failed to start: ${err.message}`));
|
|
117
|
+
}
|
|
118
|
+
process.exit(1);
|
|
119
|
+
});
|
|
120
|
+
child.on('exit', (code, signal) => {
|
|
121
|
+
if (signal) {
|
|
122
|
+
// Graceful signal-based exit — not an error
|
|
123
|
+
process.exit(0);
|
|
124
|
+
}
|
|
125
|
+
if (code !== 0 && code !== null) {
|
|
126
|
+
console.error(chalk_1.default.red(`\n ✖ API exited with code ${code}`));
|
|
127
|
+
process.exit(code);
|
|
128
|
+
}
|
|
129
|
+
});
|
|
130
|
+
});
|
|
131
|
+
exports.default = cmd;
|
|
132
|
+
//# sourceMappingURL=start.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"start.js","sourceRoot":"","sources":["../../../src/cli/commands/start.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEb,yCAAoC;AACpC,iDAAsC;AACtC,kDAA0B;AAC1B,4CAAoB;AACpB,gDAAwB;AACxB,sDAAwC;AACxC,4CAA8C;AAE9C,MAAM,UAAU,GAAG,OAAO,CAAC,QAAQ,KAAK,OAAO,CAAC;AAEhD,MAAM,GAAG,GAAG,IAAI,mBAAO,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,sCAAsC,CAAC,CAAC;AAErF,GAAG,CAAC,MAAM,CAAC,mBAAmB,EAAE,mCAAmC,CAAC,CAAC;AAErE,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,IAAuB,EAAE,EAAE;IAC3C,IAAA,oBAAW,GAAE,CAAC;IACd,KAAK,CAAC,aAAa,EAAE,CAAC;IACtB,KAAK,CAAC,OAAO,EAAE,CAAC;IAEhB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,MAAM,CAAC;IAEzD,0EAA0E;IAC1E,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC;QACtC,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC;QACrD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,gCAAgC,IAAI,OAAO,CAAC,CAAC,CAAC;IAErE,0EAA0E;IAC1E,gEAAgE;IAChE,gEAAgE;IAChE,MAAM,WAAW,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;IACxD,MAAM,WAAW,GAAG,cAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;IACtE,MAAM,UAAU,GAAG,cAAI,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;IAEpE,IAAI,MAAc,CAAC;IACnB,IAAI,OAAiB,CAAC;IAEtB,IAAI,YAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC/B,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO;QAClC,OAAO,GAAG,CAAC,WAAW,CAAC,CAAC;IAC1B,CAAC;SAAM,IAAI,YAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QACrC,oCAAoC;QACpC,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC;QACtD,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC;QACxC,OAAO,GAAG,CAAC,MAAM,EAAE,WAAW,EAAE,cAAI,CAAC,IAAI,CAAC,WAAW,EAAE,eAAe,CAAC,EAAE,UAAU,CAAC,CAAC;IACvF,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC,CAAC;QAClE,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,iBAAiB,WAAW,EAAE,CAAC,CAAC,CAAC;QAC1D,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,YAAY,MAAM,KAAK,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;IAE9F,MAAM,KAAK,GAAG,IAAA,qBAAK,EAAC,MAAM,EAAE,OAAO,EAAE;QACnC,KAAK,EAAE,SAAS;QAChB,mGAAmG;QACnG,KAAK,EAAE,KAAK;QACZ,GAAG,EAAE;YACH,GAAG,OAAO,CAAC,GAAG;YACd,QAAQ,EAAE,IAAI;YACd,WAAW,EAAE,KAAK,CAAC,WAAW;YAC9B,WAAW,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,wBAAwB;SACjE;KACF,CAAC,CAAC;IAEH,4DAA4D;IAC5D,MAAM,OAAO,GAAG,CAAC,GAAmB,EAAQ,EAAE;QAC5C,IAAI,CAAC;YACH,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClB,CAAC;QAAC,MAAM,CAAC;YACP,oBAAoB;QACtB,CAAC;IACH,CAAC,CAAC;IACF,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC9C,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhD,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAA0B,EAAE,EAAE;QAC/C,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,4BAA4B,MAAM,EAAE,CAAC,CAAC,CAAC;YAC/D,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;gBAChC,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,0BAA0B,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACpE,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;QAChC,IAAI,MAAM,EAAE,CAAC;YACX,4CAA4C;YAC5C,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QACD,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAChC,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,8BAA8B,IAAI,EAAE,CAAC,CAAC,CAAC;YAC/D,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACrB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,kBAAe,GAAG,CAAC"}
|
|
@@ -0,0 +1,211 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const commander_1 = require("commander");
|
|
7
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
8
|
+
const fs_1 = __importDefault(require("fs"));
|
|
9
|
+
const path_1 = __importDefault(require("path"));
|
|
10
|
+
const paths_1 = require("../utils/paths");
|
|
11
|
+
const client_1 = require("../../cloud/client");
|
|
12
|
+
function readProToken() {
|
|
13
|
+
try {
|
|
14
|
+
if (!fs_1.default.existsSync(paths_1.CONFIG_PATH))
|
|
15
|
+
return null;
|
|
16
|
+
const cfg = JSON.parse(fs_1.default.readFileSync(paths_1.CONFIG_PATH, 'utf8'));
|
|
17
|
+
const t = cfg.token ?? null;
|
|
18
|
+
return t && t.startsWith('qs_') ? t : null;
|
|
19
|
+
}
|
|
20
|
+
catch {
|
|
21
|
+
return null;
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
const cmd = new commander_1.Command('train').description('Submit a fine-tuning job');
|
|
25
|
+
cmd.requiredOption('--config <file>', 'Path to training config JSON');
|
|
26
|
+
cmd.option('--cloud', 'Route job to LocoPilot Cloud (requires locopilot login)', false);
|
|
27
|
+
cmd.action(async (opts) => {
|
|
28
|
+
(0, paths_1.loadEnv)();
|
|
29
|
+
const configPath = path_1.default.resolve(opts.config);
|
|
30
|
+
if (!fs_1.default.existsSync(configPath)) {
|
|
31
|
+
console.error(chalk_1.default.red(` Config file not found: ${configPath}`));
|
|
32
|
+
process.exit(1);
|
|
33
|
+
}
|
|
34
|
+
let config;
|
|
35
|
+
try {
|
|
36
|
+
config = JSON.parse(fs_1.default.readFileSync(configPath, 'utf8'));
|
|
37
|
+
}
|
|
38
|
+
catch (err) {
|
|
39
|
+
console.error(chalk_1.default.red(` Invalid JSON in config file: ${err.message}`));
|
|
40
|
+
process.exit(1);
|
|
41
|
+
}
|
|
42
|
+
for (const field of ['framework', 'baseModel', 'datasetPath', 'outputDir']) {
|
|
43
|
+
if (!config[field]) {
|
|
44
|
+
console.error(chalk_1.default.red(` Config missing required field: ${field}`));
|
|
45
|
+
process.exit(1);
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
// Resolve relative paths relative to the config file's directory
|
|
49
|
+
const configDir = path_1.default.dirname(configPath);
|
|
50
|
+
if (config.datasetPath && !path_1.default.isAbsolute(config.datasetPath)) {
|
|
51
|
+
config.datasetPath = path_1.default.resolve(configDir, config.datasetPath);
|
|
52
|
+
}
|
|
53
|
+
if (config.outputDir && !path_1.default.isAbsolute(config.outputDir)) {
|
|
54
|
+
config.outputDir = path_1.default.resolve(configDir, config.outputDir);
|
|
55
|
+
}
|
|
56
|
+
// ── Cloud path ─────────────────────────────────────────────────────────
|
|
57
|
+
if (opts.cloud) {
|
|
58
|
+
const token = readProToken();
|
|
59
|
+
if (!token) {
|
|
60
|
+
console.error(chalk_1.default.red(' Not logged in. Run: locopilot login'));
|
|
61
|
+
process.exit(1);
|
|
62
|
+
}
|
|
63
|
+
const authHeader = `Bearer ${token}`;
|
|
64
|
+
console.log(chalk_1.default.bold('\n Submitting cloud training job...\n'));
|
|
65
|
+
let submitRes;
|
|
66
|
+
try {
|
|
67
|
+
submitRes = await (0, client_1.callCloudTrain)(config, authHeader);
|
|
68
|
+
}
|
|
69
|
+
catch (err) {
|
|
70
|
+
if (err instanceof client_1.ProSubscriptionRequiredError) {
|
|
71
|
+
console.error(chalk_1.default.red('\n ✖ Pro subscription not active.'));
|
|
72
|
+
console.error(chalk_1.default.gray(` Manage at: ${err.upgradeUrl}\n`));
|
|
73
|
+
process.exit(1);
|
|
74
|
+
}
|
|
75
|
+
console.error(chalk_1.default.red(` Could not connect to LocoPilot Cloud: ${err.message}`));
|
|
76
|
+
process.exit(1);
|
|
77
|
+
}
|
|
78
|
+
const job = (await submitRes.json());
|
|
79
|
+
if (!submitRes.ok) {
|
|
80
|
+
console.error(chalk_1.default.red(' ✖ Job rejected:'), job.error ?? JSON.stringify(job));
|
|
81
|
+
process.exit(1);
|
|
82
|
+
}
|
|
83
|
+
console.log(chalk_1.default.green(` ✔ Job created: ${job.id}`));
|
|
84
|
+
console.log(chalk_1.default.gray(` Framework: ${config.framework}`));
|
|
85
|
+
console.log(chalk_1.default.gray(` Base model: ${config.baseModel}`));
|
|
86
|
+
console.log(chalk_1.default.bold('\n Tailing logs...\n'));
|
|
87
|
+
try {
|
|
88
|
+
const logsRes = await (0, client_1.callCloudTrainLogs)(job.id, authHeader);
|
|
89
|
+
await tailLogsFromResponse(logsRes);
|
|
90
|
+
}
|
|
91
|
+
catch (err) {
|
|
92
|
+
console.error(chalk_1.default.yellow(`\n Could not connect to log stream: ${err.message}`));
|
|
93
|
+
}
|
|
94
|
+
try {
|
|
95
|
+
const statusRes = await (0, client_1.callCloudTrainStatus)(job.id, authHeader);
|
|
96
|
+
await printFinalStatusFromResponse(statusRes);
|
|
97
|
+
}
|
|
98
|
+
catch {
|
|
99
|
+
// best-effort
|
|
100
|
+
}
|
|
101
|
+
return;
|
|
102
|
+
}
|
|
103
|
+
// ── Free tier path ─────────────────────────────────────────────────────
|
|
104
|
+
const base = `http://localhost:${process.env.API_PORT ?? '8080'}`;
|
|
105
|
+
console.log(chalk_1.default.bold('\n Submitting training job...\n'));
|
|
106
|
+
let res;
|
|
107
|
+
try {
|
|
108
|
+
res = await fetch(`${base}/v1/locopilot/training/jobs`, {
|
|
109
|
+
method: 'POST',
|
|
110
|
+
headers: { 'Content-Type': 'application/json' },
|
|
111
|
+
body: JSON.stringify(config),
|
|
112
|
+
signal: AbortSignal.timeout(30_000),
|
|
113
|
+
});
|
|
114
|
+
}
|
|
115
|
+
catch (err) {
|
|
116
|
+
console.error(chalk_1.default.red(` Could not connect to API: ${err.message}`));
|
|
117
|
+
console.error(chalk_1.default.gray(' Is LocoPilot running? Try: locopilot start'));
|
|
118
|
+
process.exit(1);
|
|
119
|
+
}
|
|
120
|
+
const job = (await res.json());
|
|
121
|
+
if (!res.ok) {
|
|
122
|
+
console.error(chalk_1.default.red(' ✖ Job rejected:'), job.error ?? JSON.stringify(job));
|
|
123
|
+
process.exit(1);
|
|
124
|
+
}
|
|
125
|
+
console.log(chalk_1.default.green(` ✔ Job created: ${job.id}`));
|
|
126
|
+
console.log(chalk_1.default.gray(` Framework: ${config.framework}`));
|
|
127
|
+
console.log(chalk_1.default.gray(` Base model: ${config.baseModel}`));
|
|
128
|
+
console.log(chalk_1.default.bold('\n Tailing logs...\n'));
|
|
129
|
+
try {
|
|
130
|
+
const logsRes = await fetch(`${base}/v1/locopilot/training/jobs/${job.id}/logs`, {
|
|
131
|
+
signal: AbortSignal.timeout(600_000),
|
|
132
|
+
});
|
|
133
|
+
await tailLogsFromResponse(logsRes);
|
|
134
|
+
}
|
|
135
|
+
catch (err) {
|
|
136
|
+
console.error(chalk_1.default.yellow(`\n Could not connect to log stream: ${err.message}`));
|
|
137
|
+
}
|
|
138
|
+
try {
|
|
139
|
+
const statusRes = await fetch(`${base}/v1/locopilot/training/jobs/${job.id}`, {
|
|
140
|
+
signal: AbortSignal.timeout(10_000),
|
|
141
|
+
});
|
|
142
|
+
await printFinalStatusFromResponse(statusRes);
|
|
143
|
+
}
|
|
144
|
+
catch {
|
|
145
|
+
// best-effort
|
|
146
|
+
}
|
|
147
|
+
});
|
|
148
|
+
async function tailLogsFromResponse(res) {
|
|
149
|
+
if (!res.ok) {
|
|
150
|
+
console.error(chalk_1.default.red(` Log stream error ${res.status}`));
|
|
151
|
+
return;
|
|
152
|
+
}
|
|
153
|
+
const reader = res.body?.getReader();
|
|
154
|
+
if (!reader)
|
|
155
|
+
return;
|
|
156
|
+
const decoder = new TextDecoder();
|
|
157
|
+
let buffer = '';
|
|
158
|
+
try {
|
|
159
|
+
while (true) {
|
|
160
|
+
const { done, value } = await reader.read();
|
|
161
|
+
if (done)
|
|
162
|
+
break;
|
|
163
|
+
buffer += decoder.decode(value, { stream: true });
|
|
164
|
+
const lines = buffer.split('\n');
|
|
165
|
+
buffer = lines.pop() ?? '';
|
|
166
|
+
for (const line of lines) {
|
|
167
|
+
if (!line.startsWith('data: '))
|
|
168
|
+
continue;
|
|
169
|
+
const data = line.slice(6).trim();
|
|
170
|
+
if (data === '[DONE]') {
|
|
171
|
+
// Stream ended — actual success/failure is reported by printFinalStatusFromResponse
|
|
172
|
+
return;
|
|
173
|
+
}
|
|
174
|
+
if (data)
|
|
175
|
+
process.stdout.write(data + '\n');
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
// Flush remaining buffer
|
|
179
|
+
if (buffer.startsWith('data: ')) {
|
|
180
|
+
const data = buffer.slice(6).trim();
|
|
181
|
+
if (data && data !== '[DONE]')
|
|
182
|
+
process.stdout.write(data + '\n');
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
catch (err) {
|
|
186
|
+
console.error(chalk_1.default.yellow(`\n Log stream ended: ${err.message}\n`));
|
|
187
|
+
}
|
|
188
|
+
finally {
|
|
189
|
+
reader.releaseLock();
|
|
190
|
+
}
|
|
191
|
+
}
|
|
192
|
+
async function printFinalStatusFromResponse(res) {
|
|
193
|
+
if (!res.ok)
|
|
194
|
+
return;
|
|
195
|
+
const job = (await res.json());
|
|
196
|
+
const status = job.status ?? 'unknown';
|
|
197
|
+
if (status === 'completed') {
|
|
198
|
+
if (job.output_path) {
|
|
199
|
+
console.log(chalk_1.default.green(` Adapter saved to: ${job.output_path}\n`));
|
|
200
|
+
}
|
|
201
|
+
}
|
|
202
|
+
else if (status === 'failed') {
|
|
203
|
+
console.error(chalk_1.default.red(` ✖ Job failed: ${job.error ?? 'unknown error'}\n`));
|
|
204
|
+
process.exitCode = 1;
|
|
205
|
+
}
|
|
206
|
+
else {
|
|
207
|
+
console.log(chalk_1.default.gray(` Job status: ${status}\n`));
|
|
208
|
+
}
|
|
209
|
+
}
|
|
210
|
+
exports.default = cmd;
|
|
211
|
+
//# sourceMappingURL=train.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"train.js","sourceRoot":"","sources":["../../../src/cli/commands/train.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;AAEb,yCAAoC;AACpC,kDAA0B;AAC1B,4CAAoB;AACpB,gDAAwB;AACxB,0CAAsD;AACtD,+CAK4B;AA2B5B,SAAS,YAAY;IACnB,IAAI,CAAC;QACH,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,mBAAW,CAAC;YAAE,OAAO,IAAI,CAAC;QAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAE,CAAC,YAAY,CAAC,mBAAW,EAAE,MAAM,CAAC,CAAoB,CAAC;QAChF,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC;QAC5B,OAAO,CAAC,IAAI,CAAC,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAC7C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,GAAG,GAAG,IAAI,mBAAO,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,0BAA0B,CAAC,CAAC;AACzE,GAAG,CAAC,cAAc,CAAC,iBAAiB,EAAE,8BAA8B,CAAC,CAAC;AACtE,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,yDAAyD,EAAE,KAAK,CAAC,CAAC;AAExF,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,IAAkB,EAAE,EAAE;IACtC,IAAA,eAAO,GAAE,CAAC;IAEV,MAAM,UAAU,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC7C,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC/B,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,4BAA4B,UAAU,EAAE,CAAC,CAAC,CAAC;QACnE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,MAA2B,CAAC;IAChC,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAE,CAAC,YAAY,CAAC,UAAU,EAAE,MAAM,CAAC,CAAwB,CAAC;IAClF,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,kCAAmC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACrF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,CAAC,WAAW,EAAE,WAAW,EAAE,aAAa,EAAE,WAAW,CAAU,EAAE,CAAC;QACpF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,oCAAoC,KAAK,EAAE,CAAC,CAAC,CAAC;YACtE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;IAED,iEAAiE;IACjE,MAAM,SAAS,GAAG,cAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC3C,IAAI,MAAM,CAAC,WAAW,IAAI,CAAC,cAAI,CAAC,UAAU,CAAC,MAAM,CAAC,WAAW,CAAC,EAAE,CAAC;QAC/D,MAAM,CAAC,WAAW,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;IACnE,CAAC;IACD,IAAI,MAAM,CAAC,SAAS,IAAI,CAAC,cAAI,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3D,MAAM,CAAC,SAAS,GAAG,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IAC/D,CAAC;IAED,0EAA0E;IAC1E,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;QACf,MAAM,KAAK,GAAG,YAAY,EAAE,CAAC;QAC7B,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC,CAAC;YAClE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,UAAU,GAAG,UAAU,KAAK,EAAE,CAAC;QACrC,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC,CAAC;QAElE,IAAI,SAAmB,CAAC;QACxB,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,IAAA,uBAAc,EAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,qCAA4B,EAAE,CAAC;gBAChD,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC,CAAC;gBAC/D,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,kBAAkB,GAAG,CAAC,UAAU,IAAI,CAAC,CAAC,CAAC;gBAChE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,CAAC;YACD,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,2CAA4C,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC9F,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,GAAG,GAAG,CAAC,MAAM,SAAS,CAAC,IAAI,EAAE,CAAgB,CAAC;QACpD,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,EAAE,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;YAChF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,oBAAoB,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,mBAAmB,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QAC/D,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,mBAAmB,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;QAC/D,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC;QAEjD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,IAAA,2BAAkB,EAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC7D,MAAM,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,MAAM,CAAC,wCAAyC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAChG,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,IAAA,6BAAoB,EAAC,GAAG,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACjE,MAAM,4BAA4B,CAAC,SAAS,CAAC,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,cAAc;QAChB,CAAC;QACD,OAAO;IACT,CAAC;IAED,0EAA0E;IAC1E,MAAM,IAAI,GAAG,oBAAoB,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,MAAM,EAAE,CAAC;IAElE,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC,CAAC;IAE5D,IAAI,GAAa,CAAC;IAClB,IAAI,CAAC;QACH,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,6BAA6B,EAAE;YACtD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;YAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;YAC5B,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,+BAAgC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAClF,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,IAAI,CAAC,8CAA8C,CAAC,CAAC,CAAC;QAC1E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,MAAM,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAgB,CAAC;IAC9C,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,EAAE,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;QAChF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,oBAAoB,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IACvD,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,mBAAmB,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;IAC/D,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,mBAAmB,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;IAC/D,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,CAAC;IAEjD,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,+BAA+B,GAAG,CAAC,EAAE,OAAO,EAAE;YAC/E,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC;SACrC,CAAC,CAAC;QACH,MAAM,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,MAAM,CAAC,wCAAyC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IAChG,CAAC;IAED,IAAI,CAAC;QACH,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,+BAA+B,GAAG,CAAC,EAAE,EAAE,EAAE;YAC5E,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC;SACpC,CAAC,CAAC;QACH,MAAM,4BAA4B,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;IAAC,MAAM,CAAC;QACP,cAAc;IAChB,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,KAAK,UAAU,oBAAoB,CAAC,GAAa;IAC/C,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,sBAAsB,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC7D,OAAO;IACT,CAAC;IAED,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;IACrC,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAClC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,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,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;oBAAE,SAAS;gBACzC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAClC,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACtB,oFAAoF;oBACpF,OAAO;gBACT,CAAC;gBACD,IAAI,IAAI;oBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,yBAAyB;QACzB,IAAI,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACpC,IAAI,IAAI,IAAI,IAAI,KAAK,QAAQ;gBAAE,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,MAAM,CAAC,yBAA0B,GAAa,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;IACnF,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;AACH,CAAC;AAED,KAAK,UAAU,4BAA4B,CAAC,GAAa;IACvD,IAAI,CAAC,GAAG,CAAC,EAAE;QAAE,OAAO;IACpB,MAAM,GAAG,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAgB,CAAC;IAC9C,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,SAAS,CAAC;IAEvC,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,WAAW,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,KAAK,CAAC,uBAAuB,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;SAAM,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,mBAAmB,GAAG,CAAC,KAAK,IAAI,eAAe,IAAI,CAAC,CAAC,CAAC;QAC9E,OAAO,CAAC,QAAQ,GAAG,CAAC,CAAC;IACvB,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,iBAAiB,MAAM,IAAI,CAAC,CAAC,CAAC;IACvD,CAAC;AACH,CAAC;AAED,kBAAe,GAAG,CAAC"}
|
|
@@ -0,0 +1,43 @@
|
|
|
1
|
+
'use strict';
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const commander_1 = require("commander");
|
|
7
|
+
const chalk_1 = __importDefault(require("chalk"));
|
|
8
|
+
const client_1 = require("../../cloud/client");
|
|
9
|
+
const cmd = new commander_1.Command('usage').description('Show token usage and estimated cost for the current period');
|
|
10
|
+
cmd.action(async () => {
|
|
11
|
+
const token = (0, client_1.getCloudToken)();
|
|
12
|
+
if (!token) {
|
|
13
|
+
console.log(chalk_1.default.gray('\n Not logged in.'));
|
|
14
|
+
console.log(chalk_1.default.gray(' Run: ') + chalk_1.default.white('locopilot login') + '\n');
|
|
15
|
+
process.exit(0);
|
|
16
|
+
}
|
|
17
|
+
try {
|
|
18
|
+
const res = await (0, client_1.callCloudUsage)(`Bearer ${token}`);
|
|
19
|
+
if (res.status === 401) {
|
|
20
|
+
console.error(chalk_1.default.red('\n ✖ Token rejected by cloud.\n'));
|
|
21
|
+
process.exit(1);
|
|
22
|
+
}
|
|
23
|
+
if (!res.ok) {
|
|
24
|
+
console.error(chalk_1.default.red(`\n ✖ /api/usage returned ${res.status}.\n`));
|
|
25
|
+
process.exit(1);
|
|
26
|
+
}
|
|
27
|
+
const u = (await res.json());
|
|
28
|
+
const start = new Date(u.period_start).toLocaleDateString();
|
|
29
|
+
const end = new Date(u.period_end).toLocaleDateString();
|
|
30
|
+
console.log();
|
|
31
|
+
console.log(' ' + chalk_1.default.gray('Period: ') + chalk_1.default.white(`${start} → ${end}`));
|
|
32
|
+
console.log(' ' + chalk_1.default.gray('Tokens: ') + chalk_1.default.white(u.tokens.toLocaleString()));
|
|
33
|
+
console.log(' ' + chalk_1.default.gray('Requests: ') + chalk_1.default.white(u.requests.toLocaleString()));
|
|
34
|
+
console.log(' ' + chalk_1.default.gray('Cost: ') + chalk_1.default.white(`$${u.estimated_cost.toFixed(4)} (estimated)`));
|
|
35
|
+
console.log();
|
|
36
|
+
}
|
|
37
|
+
catch (err) {
|
|
38
|
+
console.error(chalk_1.default.red(`\n ✖ Network error: ${err.message}\n`));
|
|
39
|
+
process.exit(1);
|
|
40
|
+
}
|
|
41
|
+
});
|
|
42
|
+
exports.default = cmd;
|
|
43
|
+
//# sourceMappingURL=usage.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage.js","sourceRoot":"","sources":["../../../src/cli/commands/usage.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;;;;;AAEb,yCAAoC;AACpC,kDAA0B;AAC1B,+CAAmE;AAWnE,MAAM,GAAG,GAAG,IAAI,mBAAO,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,4DAA4D,CAAC,CAAC;AAE3G,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE;IACpB,MAAM,KAAK,GAAG,IAAA,sBAAa,GAAE,CAAC;IAC9B,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,CAAC;QAC9C,OAAO,CAAC,GAAG,CAAC,eAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,eAAK,CAAC,KAAK,CAAC,iBAAiB,CAAC,GAAG,IAAI,CAAC,CAAC;QAC3E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,IAAA,uBAAc,EAAC,UAAU,KAAK,EAAE,CAAC,CAAC;QACpD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACvB,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC,CAAC;YAC7D,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,6BAA6B,GAAG,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC;YACvE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;QACD,MAAM,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAkB,CAAC;QAC9C,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,kBAAkB,EAAE,CAAC;QAC5D,MAAM,GAAG,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,kBAAkB,EAAE,CAAC;QAExD,OAAO,CAAC,GAAG,EAAE,CAAC;QACd,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,eAAK,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,eAAK,CAAC,KAAK,CAAC,GAAG,KAAK,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC;QACjF,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,eAAK,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,eAAK,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACvF,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,eAAK,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,eAAK,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACzF,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,eAAK,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,eAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC;QAC3G,OAAO,CAAC,GAAG,EAAE,CAAC;IAChB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,wBAAyB,GAAa,CAAC,OAAO,IAAI,CAAC,CAAC,CAAC;QAC7E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC,CAAC,CAAC;AAEH,kBAAe,GAAG,CAAC"}
|