tuningengines-cli 0.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/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +75 -0
- package/dist/cli.js.map +1 -0
- package/dist/client.d.ts +68 -0
- package/dist/client.d.ts.map +1 -0
- package/dist/client.js +161 -0
- package/dist/client.js.map +1 -0
- package/dist/commands/account.d.ts +4 -0
- package/dist/commands/account.d.ts.map +1 -0
- package/dist/commands/account.js +73 -0
- package/dist/commands/account.js.map +1 -0
- package/dist/commands/billing.d.ts +4 -0
- package/dist/commands/billing.d.ts.map +1 -0
- package/dist/commands/billing.js +83 -0
- package/dist/commands/billing.js.map +1 -0
- package/dist/commands/config.d.ts +3 -0
- package/dist/commands/config.d.ts.map +1 -0
- package/dist/commands/config.js +42 -0
- package/dist/commands/config.js.map +1 -0
- package/dist/commands/jobs.d.ts +4 -0
- package/dist/commands/jobs.d.ts.map +1 -0
- package/dist/commands/jobs.js +300 -0
- package/dist/commands/jobs.js.map +1 -0
- package/dist/commands/models.d.ts +4 -0
- package/dist/commands/models.d.ts.map +1 -0
- package/dist/commands/models.js +241 -0
- package/dist/commands/models.js.map +1 -0
- package/dist/config.d.ts +9 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +105 -0
- package/dist/config.js.map +1 -0
- package/dist/mcp.d.ts +2 -0
- package/dist/mcp.d.ts.map +1 -0
- package/dist/mcp.js +245 -0
- package/dist/mcp.js.map +1 -0
- package/dist/output.d.ts +8 -0
- package/dist/output.d.ts.map +1 -0
- package/dist/output.js +53 -0
- package/dist/output.js.map +1 -0
- package/package.json +44 -0
|
@@ -0,0 +1,300 @@
|
|
|
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
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
36
|
+
exports.registerJobCommands = registerJobCommands;
|
|
37
|
+
const output = __importStar(require("../output"));
|
|
38
|
+
function registerJobCommands(program, getClient) {
|
|
39
|
+
const jobs = program.command("jobs").description("Manage training jobs");
|
|
40
|
+
jobs
|
|
41
|
+
.command("list")
|
|
42
|
+
.description("List training jobs")
|
|
43
|
+
.option("-s, --status <status>", "Filter by status (queued, running, succeeded, failed, canceled)")
|
|
44
|
+
.option("-l, --limit <n>", "Max results", "20")
|
|
45
|
+
.option("--json", "Output as JSON")
|
|
46
|
+
.action(async (opts) => {
|
|
47
|
+
try {
|
|
48
|
+
const client = getClient();
|
|
49
|
+
const result = await client.listJobs({
|
|
50
|
+
status: opts.status,
|
|
51
|
+
limit: parseInt(opts.limit),
|
|
52
|
+
});
|
|
53
|
+
const jobs = result.data || [];
|
|
54
|
+
if (opts.json) {
|
|
55
|
+
output.json(jobs);
|
|
56
|
+
return;
|
|
57
|
+
}
|
|
58
|
+
if (jobs.length === 0) {
|
|
59
|
+
console.log("No jobs found.");
|
|
60
|
+
return;
|
|
61
|
+
}
|
|
62
|
+
output.table(["ID", "Status", "Base Model", "Output", "GPU Min", "Cost", "Created"], jobs.map((j) => [
|
|
63
|
+
j.id,
|
|
64
|
+
j.status,
|
|
65
|
+
j.base_model || "-",
|
|
66
|
+
j.output_name || "-",
|
|
67
|
+
String(j.gpu_minutes || 0),
|
|
68
|
+
j.estimated_charges ? `$${Number(j.estimated_charges).toFixed(2)}` : "-",
|
|
69
|
+
new Date(j.created_at).toLocaleDateString(),
|
|
70
|
+
]));
|
|
71
|
+
}
|
|
72
|
+
catch (err) {
|
|
73
|
+
console.error(err.message);
|
|
74
|
+
process.exit(1);
|
|
75
|
+
}
|
|
76
|
+
});
|
|
77
|
+
jobs
|
|
78
|
+
.command("show <id>")
|
|
79
|
+
.description("Show job details")
|
|
80
|
+
.option("--json", "Output as JSON")
|
|
81
|
+
.action(async (id, opts) => {
|
|
82
|
+
try {
|
|
83
|
+
const client = getClient();
|
|
84
|
+
const job = await client.getJob(id);
|
|
85
|
+
if (opts.json) {
|
|
86
|
+
output.json(job);
|
|
87
|
+
return;
|
|
88
|
+
}
|
|
89
|
+
output.keyValue([
|
|
90
|
+
["ID", job.id],
|
|
91
|
+
["Status", job.status],
|
|
92
|
+
["Agent", job.agent],
|
|
93
|
+
["Base Model", job.base_model],
|
|
94
|
+
["Output Name", job.output_name],
|
|
95
|
+
["Repo URL", job.repo_url],
|
|
96
|
+
["Branch", job.branch],
|
|
97
|
+
["GPU Minutes", job.gpu_minutes],
|
|
98
|
+
["Est. Charges", job.estimated_charges ? `$${Number(job.estimated_charges).toFixed(2)}` : "-"],
|
|
99
|
+
["Error", job.error_message || "-"],
|
|
100
|
+
["Retryable", String(job.retryable)],
|
|
101
|
+
["Created", job.created_at],
|
|
102
|
+
["Submitted", job.submitted_at || "-"],
|
|
103
|
+
]);
|
|
104
|
+
}
|
|
105
|
+
catch (err) {
|
|
106
|
+
console.error(err.message);
|
|
107
|
+
process.exit(1);
|
|
108
|
+
}
|
|
109
|
+
});
|
|
110
|
+
jobs
|
|
111
|
+
.command("create")
|
|
112
|
+
.description("Submit a new training job")
|
|
113
|
+
.requiredOption("--base-model <model>", "Base model HuggingFace ID")
|
|
114
|
+
.requiredOption("--output-name <name>", "Name for the fine-tuned model")
|
|
115
|
+
.option("--repo-url <url>", "GitHub repository URL")
|
|
116
|
+
.option("--branch <branch>", "Git branch", "main")
|
|
117
|
+
.option("--github-token <token>", "GitHub personal access token")
|
|
118
|
+
.option("--epochs <n>", "Number of training epochs")
|
|
119
|
+
.option("--max-examples <n>", "Maximum training examples")
|
|
120
|
+
.option("--base-user-model-id <id>", "Use a trained model as base")
|
|
121
|
+
.option("--s3-bucket <bucket>", "S3 bucket for export")
|
|
122
|
+
.option("--s3-key <key>", "AWS access key ID")
|
|
123
|
+
.option("--s3-secret <secret>", "AWS secret access key")
|
|
124
|
+
.option("--s3-region <region>", "AWS region")
|
|
125
|
+
.option("--json", "Output as JSON")
|
|
126
|
+
.action(async (opts) => {
|
|
127
|
+
try {
|
|
128
|
+
const client = getClient();
|
|
129
|
+
const params = {
|
|
130
|
+
base_model: opts.baseModel,
|
|
131
|
+
output_name: opts.outputName,
|
|
132
|
+
};
|
|
133
|
+
if (opts.repoUrl)
|
|
134
|
+
params.repo_url = opts.repoUrl;
|
|
135
|
+
if (opts.branch)
|
|
136
|
+
params.branch = opts.branch;
|
|
137
|
+
if (opts.githubToken)
|
|
138
|
+
params.github_token = opts.githubToken;
|
|
139
|
+
if (opts.epochs)
|
|
140
|
+
params.num_epochs = parseInt(opts.epochs);
|
|
141
|
+
if (opts.maxExamples)
|
|
142
|
+
params.max_examples = parseInt(opts.maxExamples);
|
|
143
|
+
if (opts.baseUserModelId)
|
|
144
|
+
params.base_user_model_id = opts.baseUserModelId;
|
|
145
|
+
if (opts.s3Bucket)
|
|
146
|
+
params.s3_output_bucket = opts.s3Bucket;
|
|
147
|
+
if (opts.s3Key)
|
|
148
|
+
params.s3_access_key_id = opts.s3Key;
|
|
149
|
+
if (opts.s3Secret)
|
|
150
|
+
params.s3_secret_access_key = opts.s3Secret;
|
|
151
|
+
if (opts.s3Region)
|
|
152
|
+
params.s3_region = opts.s3Region;
|
|
153
|
+
const job = await client.createJob(params);
|
|
154
|
+
if (opts.json) {
|
|
155
|
+
output.json(job);
|
|
156
|
+
return;
|
|
157
|
+
}
|
|
158
|
+
console.log(`Job created: ${job.id}`);
|
|
159
|
+
console.log(`Status: ${job.status}`);
|
|
160
|
+
console.log(`Base Model: ${job.base_model}`);
|
|
161
|
+
console.log(`Output: ${job.output_name}`);
|
|
162
|
+
}
|
|
163
|
+
catch (err) {
|
|
164
|
+
console.error(err.message);
|
|
165
|
+
process.exit(1);
|
|
166
|
+
}
|
|
167
|
+
});
|
|
168
|
+
jobs
|
|
169
|
+
.command("cancel <id>")
|
|
170
|
+
.description("Cancel a running job")
|
|
171
|
+
.option("--json", "Output as JSON")
|
|
172
|
+
.action(async (id, opts) => {
|
|
173
|
+
try {
|
|
174
|
+
const client = getClient();
|
|
175
|
+
const job = await client.cancelJob(id);
|
|
176
|
+
if (opts.json) {
|
|
177
|
+
output.json(job);
|
|
178
|
+
return;
|
|
179
|
+
}
|
|
180
|
+
console.log(`Job ${job.id} canceled.`);
|
|
181
|
+
}
|
|
182
|
+
catch (err) {
|
|
183
|
+
console.error(err.message);
|
|
184
|
+
process.exit(1);
|
|
185
|
+
}
|
|
186
|
+
});
|
|
187
|
+
jobs
|
|
188
|
+
.command("status <id>")
|
|
189
|
+
.description("Get live job status")
|
|
190
|
+
.option("--watch", "Poll continuously until job completes")
|
|
191
|
+
.option("--interval <seconds>", "Poll interval in seconds", "30")
|
|
192
|
+
.option("--json", "Output as JSON")
|
|
193
|
+
.action(async (id, opts) => {
|
|
194
|
+
try {
|
|
195
|
+
const client = getClient();
|
|
196
|
+
const poll = async () => {
|
|
197
|
+
const status = await client.getJobStatus(id);
|
|
198
|
+
if (opts.json) {
|
|
199
|
+
output.json(status);
|
|
200
|
+
}
|
|
201
|
+
else {
|
|
202
|
+
output.keyValue([
|
|
203
|
+
["Status", status.status],
|
|
204
|
+
["GPU Minutes", status.gpu_minutes],
|
|
205
|
+
["Est. Charges", status.estimated_charges ? `$${Number(status.estimated_charges).toFixed(2)}` : "-"],
|
|
206
|
+
["Balance", output.formatCents(status.balance_cents)],
|
|
207
|
+
["Delivery", status.delivery_status || "-"],
|
|
208
|
+
]);
|
|
209
|
+
}
|
|
210
|
+
return status;
|
|
211
|
+
};
|
|
212
|
+
const status = await poll();
|
|
213
|
+
if (opts.watch) {
|
|
214
|
+
const terminalStatuses = ["succeeded", "failed", "canceled"];
|
|
215
|
+
if (terminalStatuses.includes(status.status)) {
|
|
216
|
+
console.log("\nJob is in terminal state.");
|
|
217
|
+
return;
|
|
218
|
+
}
|
|
219
|
+
const interval = parseInt(opts.interval) * 1000;
|
|
220
|
+
const pollLoop = async () => {
|
|
221
|
+
await new Promise((r) => setTimeout(r, interval));
|
|
222
|
+
if (!opts.json)
|
|
223
|
+
console.log("\n---");
|
|
224
|
+
const s = await poll();
|
|
225
|
+
if (!terminalStatuses.includes(s.status)) {
|
|
226
|
+
await pollLoop();
|
|
227
|
+
}
|
|
228
|
+
else {
|
|
229
|
+
console.log("\nJob completed.");
|
|
230
|
+
}
|
|
231
|
+
};
|
|
232
|
+
await pollLoop();
|
|
233
|
+
}
|
|
234
|
+
}
|
|
235
|
+
catch (err) {
|
|
236
|
+
console.error(err.message);
|
|
237
|
+
process.exit(1);
|
|
238
|
+
}
|
|
239
|
+
});
|
|
240
|
+
jobs
|
|
241
|
+
.command("retry <id>")
|
|
242
|
+
.description("Retry a failed job from checkpoint")
|
|
243
|
+
.option("--json", "Output as JSON")
|
|
244
|
+
.action(async (id, opts) => {
|
|
245
|
+
try {
|
|
246
|
+
const client = getClient();
|
|
247
|
+
const job = await client.retryJob(id);
|
|
248
|
+
if (opts.json) {
|
|
249
|
+
output.json(job);
|
|
250
|
+
return;
|
|
251
|
+
}
|
|
252
|
+
console.log(`Retry job created: ${job.id}`);
|
|
253
|
+
console.log(`Status: ${job.status}`);
|
|
254
|
+
}
|
|
255
|
+
catch (err) {
|
|
256
|
+
console.error(err.message);
|
|
257
|
+
process.exit(1);
|
|
258
|
+
}
|
|
259
|
+
});
|
|
260
|
+
jobs
|
|
261
|
+
.command("estimate")
|
|
262
|
+
.description("Get a cost estimate for a training job")
|
|
263
|
+
.requiredOption("--base-model <model>", "Base model HuggingFace ID")
|
|
264
|
+
.option("--epochs <n>", "Number of epochs", "3")
|
|
265
|
+
.option("--max-examples <n>", "Maximum examples")
|
|
266
|
+
.option("--repo-size-mb <mb>", "Repository size in MB")
|
|
267
|
+
.option("--json", "Output as JSON")
|
|
268
|
+
.action(async (opts) => {
|
|
269
|
+
try {
|
|
270
|
+
const client = getClient();
|
|
271
|
+
const params = {
|
|
272
|
+
base_model: opts.baseModel,
|
|
273
|
+
num_epochs: parseInt(opts.epochs),
|
|
274
|
+
};
|
|
275
|
+
if (opts.maxExamples)
|
|
276
|
+
params.max_examples = parseInt(opts.maxExamples);
|
|
277
|
+
if (opts.repoSizeMb)
|
|
278
|
+
params.repo_size_mb = parseFloat(opts.repoSizeMb);
|
|
279
|
+
const result = await client.estimateJob(params);
|
|
280
|
+
if (opts.json) {
|
|
281
|
+
output.json(result);
|
|
282
|
+
return;
|
|
283
|
+
}
|
|
284
|
+
const est = result.estimate;
|
|
285
|
+
const bal = result.balance;
|
|
286
|
+
console.log(`Estimated Cost: $${Number(est.estimated_cost_usd).toFixed(2)}`);
|
|
287
|
+
if (est.min_cost_usd && est.max_cost_usd) {
|
|
288
|
+
console.log(`Range: $${Number(est.min_cost_usd).toFixed(2)} - $${Number(est.max_cost_usd).toFixed(2)}`);
|
|
289
|
+
}
|
|
290
|
+
console.log(`Current Balance: ${output.formatCents(bal.current_cents)}`);
|
|
291
|
+
console.log(`Required: ${output.formatCents(bal.required_cents)}`);
|
|
292
|
+
console.log(`Sufficient: ${bal.sufficient ? "Yes" : "No"}`);
|
|
293
|
+
}
|
|
294
|
+
catch (err) {
|
|
295
|
+
console.error(err.message);
|
|
296
|
+
process.exit(1);
|
|
297
|
+
}
|
|
298
|
+
});
|
|
299
|
+
}
|
|
300
|
+
//# sourceMappingURL=jobs.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"jobs.js","sourceRoot":"","sources":["../../src/commands/jobs.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,kDAiRC;AAnRD,kDAAoC;AAEpC,SAAgB,mBAAmB,CACjC,OAAgB,EAChB,SAAoC;IAEpC,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC;IAEzE,IAAI;SACD,OAAO,CAAC,MAAM,CAAC;SACf,WAAW,CAAC,oBAAoB,CAAC;SACjC,MAAM,CAAC,uBAAuB,EAAE,iEAAiE,CAAC;SAClG,MAAM,CAAC,iBAAiB,EAAE,aAAa,EAAE,IAAI,CAAC;SAC9C,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;gBACnC,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC;aAC5B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YAE/B,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAClB,OAAO;YACT,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACtB,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;gBAC9B,OAAO;YACT,CAAC;YAED,MAAM,CAAC,KAAK,CACV,CAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,CAAC,EACtE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC;gBACnB,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,MAAM;gBACR,CAAC,CAAC,UAAU,IAAI,GAAG;gBACnB,CAAC,CAAC,WAAW,IAAI,GAAG;gBACpB,MAAM,CAAC,CAAC,CAAC,WAAW,IAAI,CAAC,CAAC;gBAC1B,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG;gBACxE,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,kBAAkB,EAAE;aAC5C,CAAC,CACH,CAAC;QACJ,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI;SACD,OAAO,CAAC,WAAW,CAAC;SACpB,WAAW,CAAC,kBAAkB,CAAC;SAC/B,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,IAAI,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAEpC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,MAAM,CAAC,QAAQ,CAAC;gBACd,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;gBACd,CAAC,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC;gBACtB,CAAC,OAAO,EAAE,GAAG,CAAC,KAAK,CAAC;gBACpB,CAAC,YAAY,EAAE,GAAG,CAAC,UAAU,CAAC;gBAC9B,CAAC,aAAa,EAAE,GAAG,CAAC,WAAW,CAAC;gBAChC,CAAC,UAAU,EAAE,GAAG,CAAC,QAAQ,CAAC;gBAC1B,CAAC,QAAQ,EAAE,GAAG,CAAC,MAAM,CAAC;gBACtB,CAAC,aAAa,EAAE,GAAG,CAAC,WAAW,CAAC;gBAChC,CAAC,cAAc,EAAE,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;gBAC9F,CAAC,OAAO,EAAE,GAAG,CAAC,aAAa,IAAI,GAAG,CAAC;gBACnC,CAAC,WAAW,EAAE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBACpC,CAAC,SAAS,EAAE,GAAG,CAAC,UAAU,CAAC;gBAC3B,CAAC,WAAW,EAAE,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC;aACvC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI;SACD,OAAO,CAAC,QAAQ,CAAC;SACjB,WAAW,CAAC,2BAA2B,CAAC;SACxC,cAAc,CAAC,sBAAsB,EAAE,2BAA2B,CAAC;SACnE,cAAc,CAAC,sBAAsB,EAAE,+BAA+B,CAAC;SACvE,MAAM,CAAC,kBAAkB,EAAE,uBAAuB,CAAC;SACnD,MAAM,CAAC,mBAAmB,EAAE,YAAY,EAAE,MAAM,CAAC;SACjD,MAAM,CAAC,wBAAwB,EAAE,8BAA8B,CAAC;SAChE,MAAM,CAAC,cAAc,EAAE,2BAA2B,CAAC;SACnD,MAAM,CAAC,oBAAoB,EAAE,2BAA2B,CAAC;SACzD,MAAM,CAAC,2BAA2B,EAAE,6BAA6B,CAAC;SAClE,MAAM,CAAC,sBAAsB,EAAE,sBAAsB,CAAC;SACtD,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC;SAC7C,MAAM,CAAC,sBAAsB,EAAE,uBAAuB,CAAC;SACvD,MAAM,CAAC,sBAAsB,EAAE,YAAY,CAAC;SAC5C,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAQ;gBAClB,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,WAAW,EAAE,IAAI,CAAC,UAAU;aAC7B,CAAC;YACF,IAAI,IAAI,CAAC,OAAO;gBAAE,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,OAAO,CAAC;YACjD,IAAI,IAAI,CAAC,MAAM;gBAAE,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAC7C,IAAI,IAAI,CAAC,WAAW;gBAAE,MAAM,CAAC,YAAY,GAAG,IAAI,CAAC,WAAW,CAAC;YAC7D,IAAI,IAAI,CAAC,MAAM;gBAAE,MAAM,CAAC,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3D,IAAI,IAAI,CAAC,WAAW;gBAAE,MAAM,CAAC,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACvE,IAAI,IAAI,CAAC,eAAe;gBAAE,MAAM,CAAC,kBAAkB,GAAG,IAAI,CAAC,eAAe,CAAC;YAC3E,IAAI,IAAI,CAAC,QAAQ;gBAAE,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC3D,IAAI,IAAI,CAAC,KAAK;gBAAE,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC;YACrD,IAAI,IAAI,CAAC,QAAQ;gBAAE,MAAM,CAAC,oBAAoB,GAAG,IAAI,CAAC,QAAQ,CAAC;YAC/D,IAAI,IAAI,CAAC,QAAQ;gBAAE,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC;YAEpD,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAE3C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;YACrC,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;YAC7C,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI;SACD,OAAO,CAAC,aAAa,CAAC;SACtB,WAAW,CAAC,sBAAsB,CAAC;SACnC,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,IAAI,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;YAEvC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,OAAO,GAAG,CAAC,EAAE,YAAY,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI;SACD,OAAO,CAAC,aAAa,CAAC;SACtB,WAAW,CAAC,qBAAqB,CAAC;SAClC,MAAM,CAAC,SAAS,EAAE,uCAAuC,CAAC;SAC1D,MAAM,CAAC,sBAAsB,EAAE,0BAA0B,EAAE,IAAI,CAAC;SAChE,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,IAAI,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAE3B,MAAM,IAAI,GAAG,KAAK,IAAkB,EAAE;gBACpC,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;gBAE7C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBACd,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACtB,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,QAAQ,CAAC;wBACd,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC;wBACzB,CAAC,aAAa,EAAE,MAAM,CAAC,WAAW,CAAC;wBACnC,CAAC,cAAc,EAAE,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;wBACpG,CAAC,SAAS,EAAE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;wBACrD,CAAC,UAAU,EAAE,MAAM,CAAC,eAAe,IAAI,GAAG,CAAC;qBAC5C,CAAC,CAAC;gBACL,CAAC;gBAED,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;YAE5B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;gBACf,MAAM,gBAAgB,GAAG,CAAC,WAAW,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;gBAC7D,IAAI,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC7C,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;oBAC3C,OAAO;gBACT,CAAC;gBAED,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;gBAChD,MAAM,QAAQ,GAAG,KAAK,IAAmB,EAAE;oBACzC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,CAAC;oBAClD,IAAI,CAAC,IAAI,CAAC,IAAI;wBAAE,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;oBACrC,MAAM,CAAC,GAAG,MAAM,IAAI,EAAE,CAAC;oBACvB,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;wBACzC,MAAM,QAAQ,EAAE,CAAC;oBACnB,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;oBAClC,CAAC;gBACH,CAAC,CAAC;gBACF,MAAM,QAAQ,EAAE,CAAC;YACnB,CAAC;QACH,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI;SACD,OAAO,CAAC,YAAY,CAAC;SACrB,WAAW,CAAC,oCAAoC,CAAC;SACjD,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,IAAI,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAEtC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACjB,OAAO;YACT,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,sBAAsB,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,WAAW,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;QACvC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI;SACD,OAAO,CAAC,UAAU,CAAC;SACnB,WAAW,CAAC,wCAAwC,CAAC;SACrD,cAAc,CAAC,sBAAsB,EAAE,2BAA2B,CAAC;SACnE,MAAM,CAAC,cAAc,EAAE,kBAAkB,EAAE,GAAG,CAAC;SAC/C,MAAM,CAAC,oBAAoB,EAAE,kBAAkB,CAAC;SAChD,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC;SACtD,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAQ;gBAClB,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC;aAClC,CAAC;YACF,IAAI,IAAI,CAAC,WAAW;gBAAE,MAAM,CAAC,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACvE,IAAI,IAAI,CAAC,UAAU;gBAAE,MAAM,CAAC,YAAY,GAAG,UAAU,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEvE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAEhD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpB,OAAO;YACT,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ,CAAC;YAC5B,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC;YAC3B,OAAO,CAAC,GAAG,CAAC,oBAAoB,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC7E,IAAI,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;gBACzC,OAAO,CAAC,GAAG,CAAC,WAAW,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC1G,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,oBAAoB,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;YACzE,OAAO,CAAC,GAAG,CAAC,aAAa,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;YACnE,OAAO,CAAC,GAAG,CAAC,eAAe,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9D,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../../src/commands/models.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,mBAAmB,EAAE,MAAM,WAAW,CAAC;AAGhD,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,mBAAmB,GACnC,IAAI,CAsNN"}
|
|
@@ -0,0 +1,241 @@
|
|
|
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
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
36
|
+
exports.registerModelCommands = registerModelCommands;
|
|
37
|
+
const output = __importStar(require("../output"));
|
|
38
|
+
function registerModelCommands(program, getClient) {
|
|
39
|
+
const models = program.command("models").description("Manage your trained models");
|
|
40
|
+
models
|
|
41
|
+
.command("list")
|
|
42
|
+
.description("List your trained and imported models")
|
|
43
|
+
.option("--json", "Output as JSON")
|
|
44
|
+
.action(async (opts) => {
|
|
45
|
+
try {
|
|
46
|
+
const client = getClient();
|
|
47
|
+
const result = await client.listUserModels();
|
|
48
|
+
const models = result.data || [];
|
|
49
|
+
if (opts.json) {
|
|
50
|
+
output.json(models);
|
|
51
|
+
return;
|
|
52
|
+
}
|
|
53
|
+
if (models.length === 0) {
|
|
54
|
+
console.log("No models found.");
|
|
55
|
+
return;
|
|
56
|
+
}
|
|
57
|
+
output.table(["ID", "Name", "Status", "Source", "Base Model", "Size", "Created"], models.map((m) => [
|
|
58
|
+
m.id,
|
|
59
|
+
m.name || "-",
|
|
60
|
+
m.status,
|
|
61
|
+
m.source_type || "-",
|
|
62
|
+
m.base_model || "-",
|
|
63
|
+
m.size_bytes ? formatBytes(m.size_bytes) : "-",
|
|
64
|
+
new Date(m.created_at).toLocaleDateString(),
|
|
65
|
+
]));
|
|
66
|
+
}
|
|
67
|
+
catch (err) {
|
|
68
|
+
console.error(err.message);
|
|
69
|
+
process.exit(1);
|
|
70
|
+
}
|
|
71
|
+
});
|
|
72
|
+
models
|
|
73
|
+
.command("show <id>")
|
|
74
|
+
.description("Show model details")
|
|
75
|
+
.option("--json", "Output as JSON")
|
|
76
|
+
.action(async (id, opts) => {
|
|
77
|
+
try {
|
|
78
|
+
const client = getClient();
|
|
79
|
+
const model = await client.getUserModel(id);
|
|
80
|
+
if (opts.json) {
|
|
81
|
+
output.json(model);
|
|
82
|
+
return;
|
|
83
|
+
}
|
|
84
|
+
output.keyValue([
|
|
85
|
+
["ID", model.id],
|
|
86
|
+
["Name", model.name],
|
|
87
|
+
["Status", model.status],
|
|
88
|
+
["Source", model.source_type],
|
|
89
|
+
["Base Model", model.base_model],
|
|
90
|
+
["Size", model.size_bytes ? formatBytes(model.size_bytes) : "-"],
|
|
91
|
+
["Volume Path", model.modal_volume_path || "-"],
|
|
92
|
+
["Training Job", model.training_job_id || "-"],
|
|
93
|
+
["Created", model.created_at],
|
|
94
|
+
["Updated", model.updated_at],
|
|
95
|
+
]);
|
|
96
|
+
}
|
|
97
|
+
catch (err) {
|
|
98
|
+
console.error(err.message);
|
|
99
|
+
process.exit(1);
|
|
100
|
+
}
|
|
101
|
+
});
|
|
102
|
+
models
|
|
103
|
+
.command("delete <id>")
|
|
104
|
+
.description("Delete a model")
|
|
105
|
+
.action(async (id) => {
|
|
106
|
+
try {
|
|
107
|
+
const client = getClient();
|
|
108
|
+
const result = await client.deleteUserModel(id);
|
|
109
|
+
console.log(result.message || "Model deleted.");
|
|
110
|
+
}
|
|
111
|
+
catch (err) {
|
|
112
|
+
console.error(err.message);
|
|
113
|
+
process.exit(1);
|
|
114
|
+
}
|
|
115
|
+
});
|
|
116
|
+
models
|
|
117
|
+
.command("status <id>")
|
|
118
|
+
.description("Check import/export progress")
|
|
119
|
+
.option("--json", "Output as JSON")
|
|
120
|
+
.action(async (id, opts) => {
|
|
121
|
+
try {
|
|
122
|
+
const client = getClient();
|
|
123
|
+
const status = await client.getUserModelStatus(id);
|
|
124
|
+
if (opts.json) {
|
|
125
|
+
output.json(status);
|
|
126
|
+
return;
|
|
127
|
+
}
|
|
128
|
+
output.keyValue([
|
|
129
|
+
["Status", status.status],
|
|
130
|
+
["Display", status.display_status || "-"],
|
|
131
|
+
["Size", status.size || "-"],
|
|
132
|
+
["Error", status.error || "-"],
|
|
133
|
+
]);
|
|
134
|
+
}
|
|
135
|
+
catch (err) {
|
|
136
|
+
console.error(err.message);
|
|
137
|
+
process.exit(1);
|
|
138
|
+
}
|
|
139
|
+
});
|
|
140
|
+
models
|
|
141
|
+
.command("import")
|
|
142
|
+
.description("Import a model from S3")
|
|
143
|
+
.requiredOption("--name <name>", "Model name")
|
|
144
|
+
.requiredOption("--s3-url <url>", "S3 URL (s3://bucket/path)")
|
|
145
|
+
.requiredOption("--base-model <model>", "Base model HuggingFace ID")
|
|
146
|
+
.requiredOption("--s3-key <key>", "AWS access key ID")
|
|
147
|
+
.requiredOption("--s3-secret <secret>", "AWS secret access key")
|
|
148
|
+
.requiredOption("--s3-region <region>", "AWS region")
|
|
149
|
+
.option("--json", "Output as JSON")
|
|
150
|
+
.action(async (opts) => {
|
|
151
|
+
try {
|
|
152
|
+
const client = getClient();
|
|
153
|
+
const model = await client.importModel({
|
|
154
|
+
name: opts.name,
|
|
155
|
+
source_s3_url: opts.s3Url,
|
|
156
|
+
base_model: opts.baseModel,
|
|
157
|
+
s3_access_key_id: opts.s3Key,
|
|
158
|
+
s3_secret_access_key: opts.s3Secret,
|
|
159
|
+
s3_region: opts.s3Region,
|
|
160
|
+
});
|
|
161
|
+
if (opts.json) {
|
|
162
|
+
output.json(model);
|
|
163
|
+
return;
|
|
164
|
+
}
|
|
165
|
+
console.log(`Import started: ${model.id}`);
|
|
166
|
+
console.log(`Name: ${model.name}`);
|
|
167
|
+
console.log("Use 'te models status <id>' to check progress.");
|
|
168
|
+
}
|
|
169
|
+
catch (err) {
|
|
170
|
+
console.error(err.message);
|
|
171
|
+
process.exit(1);
|
|
172
|
+
}
|
|
173
|
+
});
|
|
174
|
+
models
|
|
175
|
+
.command("export <id>")
|
|
176
|
+
.description("Export a model to S3")
|
|
177
|
+
.requiredOption("--s3-bucket <bucket>", "S3 bucket name")
|
|
178
|
+
.option("--s3-prefix <prefix>", "S3 key prefix")
|
|
179
|
+
.requiredOption("--s3-key <key>", "AWS access key ID")
|
|
180
|
+
.requiredOption("--s3-secret <secret>", "AWS secret access key")
|
|
181
|
+
.requiredOption("--s3-region <region>", "AWS region")
|
|
182
|
+
.option("--delete-after", "Delete model from cloud after export")
|
|
183
|
+
.option("--json", "Output as JSON")
|
|
184
|
+
.action(async (id, opts) => {
|
|
185
|
+
try {
|
|
186
|
+
const client = getClient();
|
|
187
|
+
const model = await client.exportModel(id, {
|
|
188
|
+
s3_bucket: opts.s3Bucket,
|
|
189
|
+
s3_prefix: opts.s3Prefix,
|
|
190
|
+
s3_access_key_id: opts.s3Key,
|
|
191
|
+
s3_secret_access_key: opts.s3Secret,
|
|
192
|
+
s3_region: opts.s3Region,
|
|
193
|
+
delete_after: opts.deleteAfter || false,
|
|
194
|
+
});
|
|
195
|
+
if (opts.json) {
|
|
196
|
+
output.json(model);
|
|
197
|
+
return;
|
|
198
|
+
}
|
|
199
|
+
console.log(`Export started for model ${model.id}`);
|
|
200
|
+
console.log("Use 'te models status <id>' to check progress.");
|
|
201
|
+
}
|
|
202
|
+
catch (err) {
|
|
203
|
+
console.error(err.message);
|
|
204
|
+
process.exit(1);
|
|
205
|
+
}
|
|
206
|
+
});
|
|
207
|
+
// Also register "base-models" subcommand for listing supported base models
|
|
208
|
+
models
|
|
209
|
+
.command("base")
|
|
210
|
+
.description("List supported base models for fine-tuning")
|
|
211
|
+
.option("--json", "Output as JSON")
|
|
212
|
+
.action(async (opts) => {
|
|
213
|
+
try {
|
|
214
|
+
const client = getClient();
|
|
215
|
+
const result = await client.listModels();
|
|
216
|
+
const models = result.models || [];
|
|
217
|
+
if (opts.json) {
|
|
218
|
+
output.json(models);
|
|
219
|
+
return;
|
|
220
|
+
}
|
|
221
|
+
output.table(["ID", "Name", "Size", "GPU hr/epoch"], models.map((m) => [
|
|
222
|
+
m.id,
|
|
223
|
+
m.name,
|
|
224
|
+
m.size || "-",
|
|
225
|
+
String(m.gpu_hours_per_epoch || "-"),
|
|
226
|
+
]));
|
|
227
|
+
}
|
|
228
|
+
catch (err) {
|
|
229
|
+
console.error(err.message);
|
|
230
|
+
process.exit(1);
|
|
231
|
+
}
|
|
232
|
+
});
|
|
233
|
+
}
|
|
234
|
+
function formatBytes(bytes) {
|
|
235
|
+
if (bytes === 0)
|
|
236
|
+
return "0 B";
|
|
237
|
+
const units = ["B", "KB", "MB", "GB", "TB"];
|
|
238
|
+
const i = Math.floor(Math.log(bytes) / Math.log(1024));
|
|
239
|
+
return `${(bytes / Math.pow(1024, i)).toFixed(1)} ${units[i]}`;
|
|
240
|
+
}
|
|
241
|
+
//# sourceMappingURL=models.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"models.js","sourceRoot":"","sources":["../../src/commands/models.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAIA,sDAyNC;AA3ND,kDAAoC;AAEpC,SAAgB,qBAAqB,CACnC,OAAgB,EAChB,SAAoC;IAEpC,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,CAAC,4BAA4B,CAAC,CAAC;IAEnF,MAAM;SACH,OAAO,CAAC,MAAM,CAAC;SACf,WAAW,CAAC,uCAAuC,CAAC;SACpD,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,cAAc,EAAE,CAAC;YAC7C,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YAEjC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpB,OAAO;YACT,CAAC;YAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxB,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;gBAChC,OAAO;YACT,CAAC;YAED,MAAM,CAAC,KAAK,CACV,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,EAAE,SAAS,CAAC,EACnE,MAAM,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC;gBACrB,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,IAAI,IAAI,GAAG;gBACb,CAAC,CAAC,MAAM;gBACR,CAAC,CAAC,WAAW,IAAI,GAAG;gBACpB,CAAC,CAAC,UAAU,IAAI,GAAG;gBACnB,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG;gBAC9C,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,kBAAkB,EAAE;aAC5C,CAAC,CACH,CAAC;QACJ,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,MAAM;SACH,OAAO,CAAC,WAAW,CAAC;SACpB,WAAW,CAAC,oBAAoB,CAAC;SACjC,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,IAAI,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YAE5C,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,MAAM,CAAC,QAAQ,CAAC;gBACd,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC;gBAChB,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC;gBACpB,CAAC,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC;gBACxB,CAAC,QAAQ,EAAE,KAAK,CAAC,WAAW,CAAC;gBAC7B,CAAC,YAAY,EAAE,KAAK,CAAC,UAAU,CAAC;gBAChC,CAAC,MAAM,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;gBAChE,CAAC,aAAa,EAAE,KAAK,CAAC,iBAAiB,IAAI,GAAG,CAAC;gBAC/C,CAAC,cAAc,EAAE,KAAK,CAAC,eAAe,IAAI,GAAG,CAAC;gBAC9C,CAAC,SAAS,EAAE,KAAK,CAAC,UAAU,CAAC;gBAC7B,CAAC,SAAS,EAAE,KAAK,CAAC,UAAU,CAAC;aAC9B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,MAAM;SACH,OAAO,CAAC,aAAa,CAAC;SACtB,WAAW,CAAC,gBAAgB,CAAC;SAC7B,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,EAAE;QAC3B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,IAAI,gBAAgB,CAAC,CAAC;QAClD,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,MAAM;SACH,OAAO,CAAC,aAAa,CAAC;SACtB,WAAW,CAAC,8BAA8B,CAAC;SAC3C,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,IAAI,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAEnD,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpB,OAAO;YACT,CAAC;YAED,MAAM,CAAC,QAAQ,CAAC;gBACd,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC;gBACzB,CAAC,SAAS,EAAE,MAAM,CAAC,cAAc,IAAI,GAAG,CAAC;gBACzC,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,IAAI,GAAG,CAAC;gBAC5B,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,IAAI,GAAG,CAAC;aAC/B,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,MAAM;SACH,OAAO,CAAC,QAAQ,CAAC;SACjB,WAAW,CAAC,wBAAwB,CAAC;SACrC,cAAc,CAAC,eAAe,EAAE,YAAY,CAAC;SAC7C,cAAc,CAAC,gBAAgB,EAAE,2BAA2B,CAAC;SAC7D,cAAc,CAAC,sBAAsB,EAAE,2BAA2B,CAAC;SACnE,cAAc,CAAC,gBAAgB,EAAE,mBAAmB,CAAC;SACrD,cAAc,CAAC,sBAAsB,EAAE,uBAAuB,CAAC;SAC/D,cAAc,CAAC,sBAAsB,EAAE,YAAY,CAAC;SACpD,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC;gBACrC,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,aAAa,EAAE,IAAI,CAAC,KAAK;gBACzB,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,gBAAgB,EAAE,IAAI,CAAC,KAAK;gBAC5B,oBAAoB,EAAE,IAAI,CAAC,QAAQ;gBACnC,SAAS,EAAE,IAAI,CAAC,QAAQ;aACzB,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,mBAAmB,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YAC3C,OAAO,CAAC,GAAG,CAAC,SAAS,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;QAChE,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,MAAM;SACH,OAAO,CAAC,aAAa,CAAC;SACtB,WAAW,CAAC,sBAAsB,CAAC;SACnC,cAAc,CAAC,sBAAsB,EAAE,gBAAgB,CAAC;SACxD,MAAM,CAAC,sBAAsB,EAAE,eAAe,CAAC;SAC/C,cAAc,CAAC,gBAAgB,EAAE,mBAAmB,CAAC;SACrD,cAAc,CAAC,sBAAsB,EAAE,uBAAuB,CAAC;SAC/D,cAAc,CAAC,sBAAsB,EAAE,YAAY,CAAC;SACpD,MAAM,CAAC,gBAAgB,EAAE,sCAAsC,CAAC;SAChE,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,EAAU,EAAE,IAAI,EAAE,EAAE;QACjC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,EAAE,EAAE;gBACzC,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,gBAAgB,EAAE,IAAI,CAAC,KAAK;gBAC5B,oBAAoB,EAAE,IAAI,CAAC,QAAQ;gBACnC,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,YAAY,EAAE,IAAI,CAAC,WAAW,IAAI,KAAK;aACxC,CAAC,CAAC;YAEH,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,4BAA4B,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YACpD,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;QAChE,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,2EAA2E;IAC3E,MAAM;SACH,OAAO,CAAC,MAAM,CAAC;SACf,WAAW,CAAC,4CAA4C,CAAC;SACzD,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC;SAClC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC;YACzC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;YAEnC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACpB,OAAO;YACT,CAAC;YAED,MAAM,CAAC,KAAK,CACV,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,cAAc,CAAC,EACtC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC;gBACrB,CAAC,CAAC,EAAE;gBACJ,CAAC,CAAC,IAAI;gBACN,CAAC,CAAC,IAAI,IAAI,GAAG;gBACb,MAAM,CAAC,CAAC,CAAC,mBAAmB,IAAI,GAAG,CAAC;aACrC,CAAC,CACH,CAAC;QACJ,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,WAAW,CAAC,KAAa;IAChC,IAAI,KAAK,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC9B,MAAM,KAAK,GAAG,CAAC,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IAC5C,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IACvD,OAAO,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AACjE,CAAC"}
|
package/dist/config.d.ts
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
export interface Config {
|
|
2
|
+
api_key?: string;
|
|
3
|
+
api_url?: string;
|
|
4
|
+
}
|
|
5
|
+
export declare function loadConfig(): Config;
|
|
6
|
+
export declare function getApiKey(): string;
|
|
7
|
+
export declare function getApiUrl(): string;
|
|
8
|
+
export declare function saveConfig(updates: Partial<Config>): void;
|
|
9
|
+
//# sourceMappingURL=config.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAQA,MAAM,WAAW,MAAM;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,wBAAgB,UAAU,IAAI,MAAM,CAmBnC;AAED,wBAAgB,SAAS,IAAI,MAAM,CAYlC;AAED,wBAAgB,SAAS,IAAI,MAAM,CAGlC;AAED,wBAAgB,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,GAAG,IAAI,CAuBzD"}
|