@aiyou-dev/cli 0.1.0 → 1.0.2

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.
Files changed (89) hide show
  1. package/README.md +438 -0
  2. package/dist/commands/index.d.ts.map +1 -1
  3. package/dist/commands/index.js +360 -4
  4. package/dist/commands/index.js.map +1 -1
  5. package/dist/config.js +3 -0
  6. package/dist/config.js.map +1 -1
  7. package/dist/init/interactive.d.ts +2 -0
  8. package/dist/init/interactive.d.ts.map +1 -1
  9. package/dist/init/interactive.js +2 -2
  10. package/dist/init/interactive.js.map +1 -1
  11. package/dist/init/opencode-config.d.ts +46 -0
  12. package/dist/init/opencode-config.d.ts.map +1 -0
  13. package/dist/init/opencode-config.js +82 -0
  14. package/dist/init/opencode-config.js.map +1 -0
  15. package/dist/init/settings-generator.d.ts +21 -10
  16. package/dist/init/settings-generator.d.ts.map +1 -1
  17. package/dist/init/settings-generator.js +121 -35
  18. package/dist/init/settings-generator.js.map +1 -1
  19. package/dist/init/team-setup.d.ts +43 -0
  20. package/dist/init/team-setup.d.ts.map +1 -0
  21. package/dist/init/team-setup.js +237 -0
  22. package/dist/init/team-setup.js.map +1 -0
  23. package/dist/mcp/tools/ast-tools.d.ts +6 -0
  24. package/dist/mcp/tools/ast-tools.d.ts.map +1 -0
  25. package/dist/mcp/tools/ast-tools.js +101 -0
  26. package/dist/mcp/tools/ast-tools.js.map +1 -0
  27. package/dist/mcp/tools/config-tools.js +1 -1
  28. package/dist/mcp/tools/config-tools.js.map +1 -1
  29. package/dist/mcp/tools/hooks-tools.d.ts.map +1 -1
  30. package/dist/mcp/tools/hooks-tools.js +99 -1
  31. package/dist/mcp/tools/hooks-tools.js.map +1 -1
  32. package/dist/mcp/tools/index.d.ts +1 -1
  33. package/dist/mcp/tools/index.d.ts.map +1 -1
  34. package/dist/mcp/tools/index.js +13 -1
  35. package/dist/mcp/tools/index.js.map +1 -1
  36. package/dist/mcp/tools/model-engine-tools.d.ts +3 -0
  37. package/dist/mcp/tools/model-engine-tools.d.ts.map +1 -0
  38. package/dist/mcp/tools/model-engine-tools.js +52 -0
  39. package/dist/mcp/tools/model-engine-tools.js.map +1 -0
  40. package/dist/mcp/tools/models-tools.d.ts +6 -0
  41. package/dist/mcp/tools/models-tools.d.ts.map +1 -0
  42. package/dist/mcp/tools/models-tools.js +162 -0
  43. package/dist/mcp/tools/models-tools.js.map +1 -0
  44. package/dist/mcp/tools/proxy-tools.d.ts +6 -0
  45. package/dist/mcp/tools/proxy-tools.d.ts.map +1 -0
  46. package/dist/mcp/tools/proxy-tools.js +243 -0
  47. package/dist/mcp/tools/proxy-tools.js.map +1 -0
  48. package/dist/mcp/tools/rd-tools.d.ts +6 -0
  49. package/dist/mcp/tools/rd-tools.d.ts.map +1 -0
  50. package/dist/mcp/tools/rd-tools.js +205 -0
  51. package/dist/mcp/tools/rd-tools.js.map +1 -0
  52. package/dist/mcp/tools/semantic-tools.d.ts +6 -0
  53. package/dist/mcp/tools/semantic-tools.d.ts.map +1 -0
  54. package/dist/mcp/tools/semantic-tools.js +129 -0
  55. package/dist/mcp/tools/semantic-tools.js.map +1 -0
  56. package/dist/models/index.d.ts +6 -0
  57. package/dist/models/index.d.ts.map +1 -0
  58. package/dist/models/index.js +6 -0
  59. package/dist/models/index.js.map +1 -0
  60. package/dist/models/launcher.d.ts +15 -0
  61. package/dist/models/launcher.d.ts.map +1 -0
  62. package/dist/models/launcher.js +187 -0
  63. package/dist/models/launcher.js.map +1 -0
  64. package/dist/models/manager.d.ts +13 -0
  65. package/dist/models/manager.d.ts.map +1 -0
  66. package/dist/models/manager.js +247 -0
  67. package/dist/models/manager.js.map +1 -0
  68. package/dist/models/minio.d.ts +17 -0
  69. package/dist/models/minio.d.ts.map +1 -0
  70. package/dist/models/minio.js +93 -0
  71. package/dist/models/minio.js.map +1 -0
  72. package/dist/models/types.d.ts +48 -0
  73. package/dist/models/types.d.ts.map +1 -0
  74. package/dist/models/types.js +2 -0
  75. package/dist/models/types.js.map +1 -0
  76. package/dist/models/vram.d.ts +9 -0
  77. package/dist/models/vram.d.ts.map +1 -0
  78. package/dist/models/vram.js +139 -0
  79. package/dist/models/vram.js.map +1 -0
  80. package/dist/napi/proxy.d.ts +214 -0
  81. package/dist/napi/proxy.d.ts.map +1 -0
  82. package/dist/napi/proxy.js +39 -0
  83. package/dist/napi/proxy.js.map +1 -0
  84. package/dist/production/error-handler.js +1 -1
  85. package/dist/production/error-handler.js.map +1 -1
  86. package/dist/types.d.ts +35 -0
  87. package/dist/types.d.ts.map +1 -1
  88. package/dist/types.js.map +1 -1
  89. package/package.json +9 -5
@@ -0,0 +1,247 @@
1
+ import { existsSync, statSync, writeFileSync, readFileSync, unlinkSync } from "node:fs";
2
+ import { join } from "node:path";
3
+ import { homedir } from "node:os";
4
+ import { color } from "../output.js";
5
+ import { ask, confirm } from "../init/interactive.js";
6
+ import { detectGpuInfo, getModelVram, validateCombination, listLocalModels } from "./vram.js";
7
+ import { checkMinioHealth, listModelsInBucket, downloadModel, getModelsDir } from "./minio.js";
8
+ import { startServer, stopAll, isPortInUse } from "./launcher.js";
9
+ import { updateOpenCodeConfig, setDefaultModel } from "../init/opencode-config.js";
10
+ const STATE_FILE = join(homedir(), ".aiyoucli", "models-state.json");
11
+ const MODEL_ROLES = ["central", "executor", "auditor"];
12
+ const ROLE_PORTS = { central: 8000, executor: 8001, auditor: 8002 };
13
+ function getRoleCount(mode) {
14
+ switch (mode) {
15
+ case "uni-model": return 1;
16
+ case "dual-model": return 2;
17
+ case "tree-model": return 3;
18
+ }
19
+ }
20
+ function getRolesForMode(mode) {
21
+ return MODEL_ROLES.slice(0, getRoleCount(mode));
22
+ }
23
+ function saveState(state) {
24
+ writeFileSync(STATE_FILE, JSON.stringify(state, null, 2));
25
+ }
26
+ function cleanModelName(file) {
27
+ return file
28
+ .replace(/\.gguf$/i, "")
29
+ .replace(/-(Q[0-9]_[A-Z_]+|q[0-9]_[a-z_0-9]+|IQ[0-9]_[A-Z]+)$/i, "")
30
+ .replace(/[._]/g, "-")
31
+ .toLowerCase();
32
+ }
33
+ function loadState() {
34
+ try {
35
+ const data = JSON.parse(readFileSync(STATE_FILE, "utf-8"));
36
+ return data;
37
+ }
38
+ catch {
39
+ return null;
40
+ }
41
+ }
42
+ async function selectModel(prompt, available, defaultName) {
43
+ console.log(`\n ${color.bold(prompt)}`);
44
+ const filtered = available.filter((m) => m.name !== "embed-server.py" && !m.name.startsWith("all-MiniLM"));
45
+ for (let i = 0; i < filtered.length; i++) {
46
+ const m = filtered[i];
47
+ const check = m.name === defaultName ? color.green(" ⟵ default") : "";
48
+ const exists = m.exists ? color.dim(" [local]") : "";
49
+ console.log(` ${i + 1}. ${m.name} ${color.dim(`(${m.estimatedVramGB.toFixed(1)}GB VRAM)`)}${exists}${check}`);
50
+ }
51
+ const answer = await ask(` > Selecciona un modelo (1-${filtered.length})${defaultName ? ` [default: ${defaultName}]` : ""}: `);
52
+ if (answer === "" && defaultName) {
53
+ const found = filtered.find((m) => m.name === defaultName) ?? filtered[0];
54
+ return found;
55
+ }
56
+ const idx = parseInt(answer, 10) - 1;
57
+ if (idx >= 0 && idx < filtered.length) {
58
+ return filtered[idx];
59
+ }
60
+ return filtered[0];
61
+ }
62
+ async function selectWorkMode() {
63
+ console.log(`\n ${color.bold("Work mode:")}`);
64
+ const modes = [
65
+ { label: "uni-model", value: "uni-model", desc: "1 modelo central" },
66
+ { label: "dual-model", value: "dual-model", desc: "central + executor" },
67
+ { label: "tree-model", value: "tree-model", desc: "central + executor + auditor" },
68
+ ];
69
+ for (let i = 0; i < modes.length; i++) {
70
+ console.log(` ${i + 1}. ${color.cyan(modes[i].label)} ${color.dim(modes[i].desc)}`);
71
+ }
72
+ const answer = await ask(" > Selecciona modo de trabajo (1-3) [default: uni-model]: ");
73
+ const idx = parseInt(answer, 10) - 1;
74
+ if (idx >= 0 && idx < modes.length) {
75
+ return modes[idx].value;
76
+ }
77
+ return "uni-model";
78
+ }
79
+ export async function startInteractive(defaultModel) {
80
+ console.log(`\n${color.bold(color.cyan("■ aiyoucli models start"))}\n`);
81
+ const modelsDir = getModelsDir();
82
+ const minio = checkMinioHealth();
83
+ if (minio.ok) {
84
+ console.log(` ${color.green("✓")} MinIO: ${minio.message}`);
85
+ }
86
+ else {
87
+ console.log(` ${color.yellow("⚠")} ${minio.message}`);
88
+ }
89
+ const gpu = detectGpuInfo();
90
+ const freeVramGB = gpu.freeMemMiB / 1024;
91
+ if (gpu.available) {
92
+ console.log(` ${color.green("✓")} GPU: ${gpu.name} (${freeVramGB.toFixed(1)} GB libres)`);
93
+ }
94
+ else {
95
+ console.log(` ${color.yellow("⚠")} GPU: no detectada, usando default ${freeVramGB.toFixed(1)} GB`);
96
+ }
97
+ const localModels = listLocalModels(modelsDir);
98
+ const bucketModels = minio.ok ? listModelsInBucket() : [];
99
+ const localFiles = new Set(localModels.map((m) => m.name));
100
+ const availableModels = [
101
+ ...localModels,
102
+ ...bucketModels
103
+ .filter((f) => !localFiles.has(f))
104
+ .map((f) => ({
105
+ name: f,
106
+ path: join(modelsDir, f),
107
+ sizeGB: 0,
108
+ quant: "unknown",
109
+ estimatedVramGB: getModelVram(f),
110
+ exists: false,
111
+ })),
112
+ ];
113
+ if (availableModels.length === 0) {
114
+ return { ok: false, running: [], message: "No hay modelos disponibles en MinIO ni localmente." };
115
+ }
116
+ console.log(` ${color.green("✓")} ${availableModels.length} modelos disponibles (${localModels.length} locales)`);
117
+ const workMode = await selectWorkMode();
118
+ const roles = getRolesForMode(workMode);
119
+ const selected = [];
120
+ for (let i = 0; i < roles.length; i++) {
121
+ const role = roles[i];
122
+ const port = ROLE_PORTS[role];
123
+ if (isPortInUse(port) && port !== ROLE_PORTS.central) {
124
+ console.log(` ${color.yellow(`⚠ Puerto ${port} en uso, seleccionando puerto alternativo`)}`);
125
+ }
126
+ const model = await selectModel(i === 0 ? `Modelo ${color.cyan(role)} (puerto ${port}):` : `Modelo ${color.cyan(role)} (puerto ${port}):`, availableModels, i === 0 ? (defaultModel ?? "gemma-4-E2B-it-qat-UD-Q4_K_XL.gguf") : undefined);
127
+ if (!model) {
128
+ return { ok: false, running: [], message: "Seleccion cancelada." };
129
+ }
130
+ selected.push({ file: model.name, path: model.path, role, port });
131
+ }
132
+ const validation = validateCombination(selected.map((s) => ({ file: s.file })), freeVramGB);
133
+ console.log(`\n ${color.bold("Resumen:")}`);
134
+ console.log(` Modo: ${color.cyan(workMode)}`);
135
+ for (const s of selected) {
136
+ const vram = getModelVram(s.file);
137
+ console.log(` ${color.dim(s.role.padEnd(10))} ${s.file} ${color.dim(`(${vram.toFixed(1)}GB, :${s.port})`)}`);
138
+ }
139
+ console.log(` ${validation.message}`);
140
+ if (!validation.valid) {
141
+ console.log("");
142
+ for (const s of validation.suggestions) {
143
+ console.log(` ${color.yellow("→")} ${s}`);
144
+ }
145
+ return { ok: false, running: [], message: "Combinacion invalida por VRAM." };
146
+ }
147
+ const ok = await confirm("\n Confirmar y lanzar?");
148
+ if (!ok) {
149
+ return { ok: false, running: [], message: "Cancelado por el usuario." };
150
+ }
151
+ for (const s of selected) {
152
+ if (!existsSync(s.path) || !statSync(s.path).isFile()) {
153
+ console.log(` ${color.dim("Descargando")} ${s.file} desde MinIO...`);
154
+ const dl = downloadModel(s.file);
155
+ if (!dl.ok) {
156
+ return { ok: false, running: [], message: dl.message };
157
+ }
158
+ s.path = dl.path;
159
+ }
160
+ else {
161
+ console.log(` ${color.green("✓")} ${s.file} ya esta localmente`);
162
+ }
163
+ }
164
+ const running = [];
165
+ for (const s of selected) {
166
+ console.log(`\n ${color.dim("Lanzando")} ${s.role} en :${s.port}...`);
167
+ const result = await startServer({ role: s.role, file: s.file, path: s.path, port: s.port });
168
+ if ("error" in result) {
169
+ console.log(` ${color.red("✗")} ${result.error}`);
170
+ continue;
171
+ }
172
+ running.push(result);
173
+ console.log(` ${color.green("✓")} ${s.role} listo (PID ${result.pid}) — ${s.file}`);
174
+ }
175
+ const state = {
176
+ running,
177
+ startedAt: new Date().toISOString(),
178
+ workMode,
179
+ };
180
+ saveState(state);
181
+ if (running.length === 0) {
182
+ return { ok: false, running: [], message: "No se pudo iniciar ningun modelo." };
183
+ }
184
+ for (const r of running) {
185
+ const modelName = cleanModelName(r.assignment.file);
186
+ const opencodeResult = updateOpenCodeConfig(r.assignment.file, modelName, r.assignment.port);
187
+ if (opencodeResult.ok) {
188
+ console.log(` ${color.green("✓")} ${opencodeResult.message}`);
189
+ }
190
+ else {
191
+ console.log(` ${color.yellow("⚠")} ${opencodeResult.message}`);
192
+ }
193
+ }
194
+ const firstRunning = running[0];
195
+ if (firstRunning) {
196
+ const modelName = cleanModelName(firstRunning.assignment.file);
197
+ const defaultResult = setDefaultModel(modelName);
198
+ if (defaultResult.ok) {
199
+ console.log(` ${color.green("✓")} ${defaultResult.message}`);
200
+ }
201
+ }
202
+ displayStatus(running);
203
+ const total = selected.length;
204
+ const started = running.length;
205
+ const failed = total - started;
206
+ const msg = failed > 0
207
+ ? `${started}/${total} modelos iniciados. ${failed} fallaron.`
208
+ : `${total} modelos iniciados correctamente.`;
209
+ return { ok: started > 0, running, message: msg };
210
+ }
211
+ export function displayStatus(running) {
212
+ console.log(`\n ${color.bold("🟢 Modelos activos")}`);
213
+ for (const r of running) {
214
+ console.log(` ${color.cyan(r.assignment.role.padEnd(10))} → ${r.assignment.file} :${r.assignment.port} PID ${r.pid}`);
215
+ }
216
+ }
217
+ export function showStatus() {
218
+ const state = loadState();
219
+ if (!state || state.running.length === 0) {
220
+ return "No hay modelos activos.";
221
+ }
222
+ let out = `Modo: ${state.workMode}\n`;
223
+ for (const r of state.running) {
224
+ out += ` ${r.assignment.role} → ${r.assignment.file} :${r.assignment.port} (PID ${r.pid})\n`;
225
+ }
226
+ return out;
227
+ }
228
+ export function stopInteractive() {
229
+ const state = loadState();
230
+ if (!state) {
231
+ return { ok: false, message: "No hay sesion activa de modelos." };
232
+ }
233
+ const { stopped, errors } = stopAll();
234
+ const statePath = STATE_FILE;
235
+ try {
236
+ unlinkSync(statePath);
237
+ }
238
+ catch { }
239
+ const msg = stopped.length > 0
240
+ ? `Detenidos: ${stopped.join(", ")}`
241
+ : "No habia modelos activos";
242
+ if (errors.length > 0) {
243
+ return { ok: true, message: `${msg}. Errores: ${errors.join(", ")}` };
244
+ }
245
+ return { ok: true, message: msg };
246
+ }
247
+ //# sourceMappingURL=manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"manager.js","sourceRoot":"","sources":["../../src/models/manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACxF,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,KAAK,EAAE,MAAM,cAAc,CAAC;AACrC,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AAEtD,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAC9F,OAAO,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC/F,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAClE,OAAO,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAEnF,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,mBAAmB,CAAC,CAAC;AAErE,MAAM,WAAW,GAAgB,CAAC,SAAS,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;AACpE,MAAM,UAAU,GAA8B,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAE/F,SAAS,YAAY,CAAC,IAAc;IAClC,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,WAAW,CAAC,CAAC,OAAO,CAAC,CAAC;QAC3B,KAAK,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC;QAC5B,KAAK,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,IAAc;IACrC,OAAO,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;AAClD,CAAC;AAED,SAAS,SAAS,CAAC,KAAuB;IACxC,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI;SACR,OAAO,CAAC,UAAU,EAAE,EAAE,CAAC;SACvB,OAAO,CAAC,sDAAsD,EAAE,EAAE,CAAC;SACnE,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;SACrB,WAAW,EAAE,CAAC;AACnB,CAAC;AAED,SAAS,SAAS;IAChB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;QAC3D,OAAO,IAAwB,CAAC;IAClC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,WAAW,CAAC,MAAc,EAAE,SAAuB,EAAE,WAAoB;IACtF,OAAO,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,iBAAiB,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC;IAE3G,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACrD,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,MAAM,GAAG,KAAK,EAAE,CAAC,CAAC;IACjH,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,+BAA+B,QAAQ,CAAC,MAAM,IAAI,WAAW,CAAC,CAAC,CAAC,cAAc,WAAW,GAAG,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAChI,IAAI,MAAM,KAAK,EAAE,IAAI,WAAW,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC1E,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC;QACtC,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC;IACvB,CAAC;IACD,OAAO,QAAQ,CAAC,CAAC,CAAC,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,cAAc;IAC3B,OAAO,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAuD;QAChE,EAAE,KAAK,EAAE,WAAW,EAAG,KAAK,EAAE,WAAW,EAAG,IAAI,EAAE,kBAAkB,EAAE;QACtE,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,oBAAoB,EAAE;QACxE,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,8BAA8B,EAAE;KACnF,CAAC;IAEF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxF,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,6DAA6D,CAAC,CAAC;IACxF,MAAM,GAAG,GAAG,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACnC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;IAC1B,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,YAAqB;IAErB,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,IAAI,CAAC,CAAC;IAExE,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IAEjC,MAAM,KAAK,GAAG,gBAAgB,EAAE,CAAC;IACjC,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;QACb,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAC/D,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,MAAM,GAAG,GAAY,aAAa,EAAE,CAAC;IACrC,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,GAAG,IAAI,CAAC;IACzC,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,GAAG,CAAC,IAAI,KAAK,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;IAC7F,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,sCAAsC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC;IACtG,CAAC;IAED,MAAM,WAAW,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;IAC/C,MAAM,YAAY,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1D,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3D,MAAM,eAAe,GAAiB;QACpC,GAAG,WAAW;QACd,GAAG,YAAY;aACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;aACjC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACX,IAAI,EAAE,CAAC;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;YACxB,MAAM,EAAE,CAAC;YACT,KAAK,EAAE,SAAS;YAChB,eAAe,EAAE,YAAY,CAAC,CAAC,CAAC;YAChC,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;KACN,CAAC;IAEF,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,oDAAoD,EAAE,CAAC;IACnG,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,eAAe,CAAC,MAAM,yBAAyB,WAAW,CAAC,MAAM,WAAW,CAAC,CAAC;IAEnH,MAAM,QAAQ,GAAG,MAAM,cAAc,EAAE,CAAC;IACxC,MAAM,KAAK,GAAG,eAAe,CAAC,QAAQ,CAAC,CAAC;IAExC,MAAM,QAAQ,GAAoE,EAAE,CAAC;IAErF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;QAE9B,IAAI,WAAW,CAAC,IAAI,CAAC,IAAI,IAAI,KAAK,UAAU,CAAC,OAAO,EAAE,CAAC;YACrD,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,MAAM,CAAC,YAAY,IAAI,2CAA2C,CAAC,EAAE,CAAC,CAAC;QAChG,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,WAAW,CAC7B,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,UAAU,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,IAAI,IAAI,EACzG,eAAe,EACf,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,IAAI,oCAAoC,CAAC,CAAC,CAAC,CAAC,SAAS,CAC7E,CAAC;QAEF,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,sBAAsB,EAAE,CAAC;QACrE,CAAC;QAED,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,MAAM,UAAU,GAAG,mBAAmB,CACpC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,EACvC,UAAU,CACX,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IAC7C,OAAO,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IACnD,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC,CAAC;IACjH,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,KAAK,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;IAEvC,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACtB,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAChB,KAAK,MAAM,CAAC,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;YACvC,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,gCAAgC,EAAE,CAAC;IAC/E,CAAC;IAED,MAAM,EAAE,GAAG,MAAM,OAAO,CAAC,yBAAyB,CAAC,CAAC;IACpD,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,2BAA2B,EAAE,CAAC;IAC1E,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;YACtD,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,IAAI,iBAAiB,CAAC,CAAC;YACtE,MAAM,EAAE,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;gBACX,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,CAAC,OAAO,EAAE,CAAC;YACzD,CAAC;YACD,CAAC,CAAC,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC;QACnB,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,qBAAqB,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED,MAAM,OAAO,GAAmB,EAAE,CAAC;IACnC,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzB,OAAO,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,QAAQ,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC;QACvE,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7F,IAAI,OAAO,IAAI,MAAM,EAAE,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;YACnD,SAAS;QACX,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrB,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,eAAe,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IACvF,CAAC;IAED,MAAM,KAAK,GAAqB;QAC9B,OAAO;QACP,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,QAAQ;KACT,CAAC;IACF,SAAS,CAAC,KAAK,CAAC,CAAC;IAEjB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,mCAAmC,EAAE,CAAC;IAClF,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,cAAc,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QACpD,MAAM,cAAc,GAAG,oBAAoB,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC7F,IAAI,cAAc,CAAC,EAAE,EAAE,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC;QACjE,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,cAAc,CAAC,OAAO,EAAE,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAChC,IAAI,YAAY,EAAE,CAAC;QACjB,MAAM,SAAS,GAAG,cAAc,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,aAAa,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,aAAa,CAAC,EAAE,EAAE,CAAC;YACrB,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED,aAAa,CAAC,OAAO,CAAC,CAAC;IACvB,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC;IAC9B,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC;IAC/B,MAAM,MAAM,GAAG,KAAK,GAAG,OAAO,CAAC;IAC/B,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC;QACpB,CAAC,CAAC,GAAG,OAAO,IAAI,KAAK,uBAAuB,MAAM,YAAY;QAC9D,CAAC,CAAC,GAAG,KAAK,mCAAmC,CAAC;IAChD,OAAO,EAAE,EAAE,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,OAAuB;IACnD,OAAO,CAAC,GAAG,CAAC,OAAO,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC;IACvD,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,UAAU,CAAC,IAAI,MAAM,CAAC,CAAC,UAAU,CAAC,IAAI,SAAS,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;IAC3H,CAAC;AACH,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,MAAM,KAAK,GAAG,SAAS,EAAE,CAAC;IAC1B,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzC,OAAO,yBAAyB,CAAC;IACnC,CAAC;IAED,IAAI,GAAG,GAAG,SAAS,KAAK,CAAC,QAAQ,IAAI,CAAC;IACtC,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;QAC9B,GAAG,IAAI,KAAK,CAAC,CAAC,UAAU,CAAC,IAAI,MAAM,CAAC,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,CAAC,UAAU,CAAC,IAAI,SAAS,CAAC,CAAC,GAAG,KAAK,CAAC;IAChG,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,MAAM,UAAU,eAAe;IAC7B,MAAM,KAAK,GAAG,SAAS,EAAE,CAAC;IAC1B,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,kCAAkC,EAAE,CAAC;IACpE,CAAC;IAED,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,CAAC;IAEtC,MAAM,SAAS,GAAG,UAAU,CAAC;IAC7B,IAAI,CAAC;QACH,UAAU,CAAC,SAAS,CAAC,CAAC;IACxB,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;IAEV,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC;QAC5B,CAAC,CAAC,cAAc,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;QACpC,CAAC,CAAC,0BAA0B,CAAC;IAC/B,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,GAAG,cAAc,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;IACxE,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;AACpC,CAAC"}
@@ -0,0 +1,17 @@
1
+ export declare function getModelsDir(): string;
2
+ export declare function checkMinioHealth(): {
3
+ ok: boolean;
4
+ message: string;
5
+ };
6
+ export declare function listModelsInBucket(): string[];
7
+ export declare function getFileSizeInBucket(modelFile: string): number;
8
+ export declare function downloadModel(modelFile: string): {
9
+ ok: boolean;
10
+ path: string;
11
+ message: string;
12
+ };
13
+ export declare function batchDownload(models: string[]): {
14
+ ok: boolean;
15
+ results: string[];
16
+ };
17
+ //# sourceMappingURL=minio.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"minio.d.ts","sourceRoot":"","sources":["../../src/models/minio.ts"],"names":[],"mappings":"AAYA,wBAAgB,YAAY,IAAI,MAAM,CAKrC;AAED,wBAAgB,gBAAgB,IAAI;IAAE,EAAE,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAgBnE;AAED,wBAAgB,kBAAkB,IAAI,MAAM,EAAE,CAU7C;AAED,wBAAgB,mBAAmB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAU7D;AAED,wBAAgB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG;IAAE,EAAE,EAAE,OAAO,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAgC/F;AAED,wBAAgB,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG;IAAE,EAAE,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,EAAE,CAAA;CAAE,CAQlF"}
@@ -0,0 +1,93 @@
1
+ import { execSync } from "node:child_process";
2
+ import { existsSync, mkdirSync } from "node:fs";
3
+ import { join } from "node:path";
4
+ import { homedir } from "node:os";
5
+ const MINIO_URL = "http://127.0.0.1:9000";
6
+ const MINIO_CONTAINER = "bgust-minio";
7
+ const MINIO_BUCKET = "llm-models";
8
+ const MINIO_ACCESS_KEY = "minioadmin";
9
+ const MINIO_SECRET_KEY = "minioadmin";
10
+ const MODELS_DIR = join(homedir(), ".aiyoucli", "models");
11
+ export function getModelsDir() {
12
+ if (!existsSync(MODELS_DIR)) {
13
+ mkdirSync(MODELS_DIR, { recursive: true });
14
+ }
15
+ return MODELS_DIR;
16
+ }
17
+ export function checkMinioHealth() {
18
+ try {
19
+ const code = execSync(`curl -s -o /dev/null -w "%{http_code}" ${MINIO_URL}/minio/health/live`, { encoding: "utf-8", timeout: 5000 }).trim();
20
+ if (code === "200") {
21
+ return { ok: true, message: `MinIO activo en ${MINIO_URL}` };
22
+ }
23
+ return { ok: false, message: `MinIO respondio con codigo ${code}` };
24
+ }
25
+ catch {
26
+ return {
27
+ ok: false,
28
+ message: `No se pudo conectar a MinIO en ${MINIO_URL}. Asegurate de que el contenedor ${MINIO_CONTAINER} este corriendo.`,
29
+ };
30
+ }
31
+ }
32
+ export function listModelsInBucket() {
33
+ try {
34
+ const out = execSync(`docker exec ${MINIO_CONTAINER} ls /data/${MINIO_BUCKET}/`, { encoding: "utf-8", timeout: 10000 }).trim();
35
+ return out.split("\n").filter((f) => f.endsWith(".gguf")).sort();
36
+ }
37
+ catch {
38
+ return [];
39
+ }
40
+ }
41
+ export function getFileSizeInBucket(modelFile) {
42
+ try {
43
+ const out = execSync(`docker exec ${MINIO_CONTAINER} du -sb /data/${MINIO_BUCKET}/${modelFile} 2>/dev/null | cut -f1`, { encoding: "utf-8", timeout: 10000 }).trim();
44
+ return parseInt(out, 10) || 0;
45
+ }
46
+ catch {
47
+ return 0;
48
+ }
49
+ }
50
+ export function downloadModel(modelFile) {
51
+ const destDir = getModelsDir();
52
+ const destPath = join(destDir, modelFile);
53
+ if (existsSync(destPath)) {
54
+ const isFile = execSync(`test -f "${destPath}" && echo "file" || echo "not"`, {
55
+ encoding: "utf-8", timeout: 3000
56
+ }).trim();
57
+ if (isFile === "file") {
58
+ return { ok: true, path: destPath, message: `Ya existe localmente: ${modelFile}` };
59
+ }
60
+ // Remove stale directory artifact from previous docker cp
61
+ try {
62
+ execSync(`rm -rf "${destPath}"`, { timeout: 5000 });
63
+ }
64
+ catch {
65
+ return { ok: false, path: "", message: `No se pudo limpiar el directorio corrupto: ${modelFile}` };
66
+ }
67
+ }
68
+ const tmpPath = `${destPath}.tmp`;
69
+ try {
70
+ execSync(`curl --aws-sigv4 "aws:amz:us-east-1:s3" --user "${MINIO_ACCESS_KEY}:${MINIO_SECRET_KEY}" "http://127.0.0.1:9000/${MINIO_BUCKET}/${modelFile}" -o "${tmpPath}" 2>/dev/null`, { timeout: 600000, stdio: "pipe" });
71
+ execSync(`mv "${tmpPath}" "${destPath}"`, { timeout: 5000 });
72
+ return { ok: true, path: destPath, message: `Descargado: ${modelFile}` };
73
+ }
74
+ catch (err) {
75
+ try {
76
+ execSync(`rm -f "${tmpPath}"`, { timeout: 3000 });
77
+ }
78
+ catch { /* cleanup */ }
79
+ const msg = err instanceof Error ? err.message : String(err);
80
+ return { ok: false, path: "", message: `Error descargando ${modelFile}: ${msg}` };
81
+ }
82
+ }
83
+ export function batchDownload(models) {
84
+ const results = [];
85
+ for (const m of models) {
86
+ const r = downloadModel(m);
87
+ results.push(r.message);
88
+ if (!r.ok)
89
+ return { ok: false, results };
90
+ }
91
+ return { ok: true, results };
92
+ }
93
+ //# sourceMappingURL=minio.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"minio.js","sourceRoot":"","sources":["../../src/models/minio.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAChD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAElC,MAAM,SAAS,GAAG,uBAAuB,CAAC;AAC1C,MAAM,eAAe,GAAG,aAAa,CAAC;AACtC,MAAM,YAAY,GAAG,YAAY,CAAC;AAClC,MAAM,gBAAgB,GAAG,YAAY,CAAC;AACtC,MAAM,gBAAgB,GAAG,YAAY,CAAC;AACtC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;AAE1D,MAAM,UAAU,YAAY;IAC1B,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC5B,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC7C,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,QAAQ,CACnB,0CAA0C,SAAS,oBAAoB,EACvE,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CACrC,CAAC,IAAI,EAAE,CAAC;QACT,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;YACnB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,mBAAmB,SAAS,EAAE,EAAE,CAAC;QAC/D,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,8BAA8B,IAAI,EAAE,EAAE,CAAC;IACtE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,EAAE,EAAE,KAAK;YACT,OAAO,EAAE,kCAAkC,SAAS,oCAAoC,eAAe,kBAAkB;SAC1H,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,kBAAkB;IAChC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,QAAQ,CAClB,eAAe,eAAe,aAAa,YAAY,GAAG,EAC1D,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CACtC,CAAC,IAAI,EAAE,CAAC;QACT,OAAO,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACnE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,SAAiB;IACnD,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,QAAQ,CAClB,eAAe,eAAe,iBAAiB,YAAY,IAAI,SAAS,wBAAwB,EAChG,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CACtC,CAAC,IAAI,EAAE,CAAC;QACT,OAAO,QAAQ,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,SAAiB;IAC7C,MAAM,OAAO,GAAG,YAAY,EAAE,CAAC;IAC/B,MAAM,QAAQ,GAAG,IAAI,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;IAE1C,IAAI,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,QAAQ,CAAC,YAAY,QAAQ,gCAAgC,EAAE;YAC5E,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI;SACjC,CAAC,CAAC,IAAI,EAAE,CAAC;QACV,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YACtB,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,yBAAyB,SAAS,EAAE,EAAE,CAAC;QACrF,CAAC;QACD,0DAA0D;QAC1D,IAAI,CAAC;YACH,QAAQ,CAAC,WAAW,QAAQ,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QACtD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,8CAA8C,SAAS,EAAE,EAAE,CAAC;QACrG,CAAC;IACH,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,QAAQ,MAAM,CAAC;IAClC,IAAI,CAAC;QACH,QAAQ,CACN,mDAAmD,gBAAgB,IAAI,gBAAgB,4BAA4B,YAAY,IAAI,SAAS,SAAS,OAAO,eAAe,EAC3K,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CACnC,CAAC;QACF,QAAQ,CAAC,OAAO,OAAO,MAAM,QAAQ,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7D,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,eAAe,SAAS,EAAE,EAAE,CAAC;IAC3E,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,CAAC;YAAC,QAAQ,CAAC,UAAU,OAAO,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC;QAClF,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7D,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,qBAAqB,SAAS,KAAK,GAAG,EAAE,EAAE,CAAC;IACpF,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,MAAgB;IAC5C,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;QACvB,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;QAC3B,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QACxB,IAAI,CAAC,CAAC,CAAC,EAAE;YAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAC3C,CAAC;IACD,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AAC/B,CAAC"}
@@ -0,0 +1,48 @@
1
+ export type WorkMode = "uni-model" | "dual-model" | "tree-model";
2
+ export type ModelRole = "central" | "executor" | "auditor";
3
+ export interface ModelAssignment {
4
+ role: ModelRole;
5
+ file: string;
6
+ path: string;
7
+ port: number;
8
+ }
9
+ export interface RunningModel {
10
+ pid: number;
11
+ assignment: ModelAssignment;
12
+ startedAt: string;
13
+ }
14
+ export interface GpuInfo {
15
+ name: string;
16
+ totalMemMiB: number;
17
+ freeMemMiB: number;
18
+ available: boolean;
19
+ }
20
+ export interface ModelEntry {
21
+ name: string;
22
+ path: string;
23
+ sizeGB: number;
24
+ quant: string;
25
+ estimatedVramGB: number;
26
+ exists: boolean;
27
+ }
28
+ export interface VramTableEntry {
29
+ minMemGB: number;
30
+ recommended: boolean;
31
+ }
32
+ export interface ValidationResult {
33
+ valid: boolean;
34
+ totalVramGB: number;
35
+ freeVramGB: number;
36
+ models: {
37
+ name: string;
38
+ vramGB: number;
39
+ }[];
40
+ message: string;
41
+ suggestions: string[];
42
+ }
43
+ export interface ModelEngineState {
44
+ running: RunningModel[];
45
+ startedAt: string;
46
+ workMode: WorkMode;
47
+ }
48
+ //# sourceMappingURL=types.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/models/types.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,QAAQ,GAAG,WAAW,GAAG,YAAY,GAAG,YAAY,CAAC;AAEjE,MAAM,MAAM,SAAS,GAAG,SAAS,GAAG,UAAU,GAAG,SAAS,CAAC;AAE3D,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,SAAS,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,YAAY;IAC3B,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,eAAe,CAAC;IAC5B,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE,OAAO,CAAC;CACjB;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,OAAO,CAAC;CACtB;AAED,MAAM,WAAW,gBAAgB;IAC/B,KAAK,EAAE,OAAO,CAAC;IACf,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC3C,OAAO,EAAE,MAAM,CAAC;IAChB,WAAW,EAAE,MAAM,EAAE,CAAC;CACvB;AAED,MAAM,WAAW,gBAAgB;IAC/B,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,QAAQ,CAAC;CACpB"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/models/types.ts"],"names":[],"mappings":""}
@@ -0,0 +1,9 @@
1
+ import type { GpuInfo, ModelEntry, ValidationResult } from "./types.js";
2
+ export declare function getModelVram(modelName: string): number;
3
+ export declare function detectGpuInfo(): GpuInfo;
4
+ export declare function estimateCpuMode(totalRamMiB: number): GpuInfo;
5
+ export declare function validateCombination(models: {
6
+ file: string;
7
+ }[], freeVramGB: number): ValidationResult;
8
+ export declare function listLocalModels(modelsPath: string): ModelEntry[];
9
+ //# sourceMappingURL=vram.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"vram.d.ts","sourceRoot":"","sources":["../../src/models/vram.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAiCxE,wBAAgB,YAAY,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAEtD;AAED,wBAAgB,aAAa,IAAI,OAAO,CA4CvC;AAED,wBAAgB,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAO5D;AAED,wBAAgB,mBAAmB,CACjC,MAAM,EAAE;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,EAAE,EAC1B,UAAU,EAAE,MAAM,GACjB,gBAAgB,CA0BlB;AAED,wBAAgB,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,UAAU,EAAE,CAuBhE"}
@@ -0,0 +1,139 @@
1
+ import { execSync } from "node:child_process";
2
+ import { readdirSync, statSync, existsSync } from "node:fs";
3
+ import { join } from "node:path";
4
+ const VRAM_TABLE = {
5
+ "DeepSeek-R1-Distill-Qwen-1.5B-Q4_K_M.gguf": 1.4,
6
+ "deepseek-coder-1.3b-instruct.Q4_K_M.gguf": 1.3,
7
+ "DeepSeek-R1-Distill-Qwen-7B-IQ2_M.gguf": 2.5,
8
+ "gemma-2-2b-it-Q4_K_M.gguf": 2.0,
9
+ "gemma-2-27b-it-IQ2_M.gguf": 9.0,
10
+ "gemma-4-E2B-it-qat-UD-Q4_K_XL.gguf": 2.8,
11
+ "gemma-4-E2B_q4_0-it.gguf": 3.5,
12
+ "granite-3.0-2b-instruct-Q4_K_M.gguf": 1.8,
13
+ "granite-3.0-3b-a800m-instruct-Q4_K_M.gguf": 2.2,
14
+ "granite-4.0-1b-Q4_K_M.gguf": 1.0,
15
+ "granite-4.0-h-micro-Q4_K_M.gguf": 0.8,
16
+ "granite-4.0-h-tiny-Q4_K_M.gguf": 0.6,
17
+ "granite-4.1-3b-Q4_K_M.gguf": 2.5,
18
+ "granite-8b-code-instruct-128k.Q4_K_M.gguf": 5.2,
19
+ "granite-guardian-3.0-2b.Q4_K_M.gguf": 1.8,
20
+ "Llama-3.2-1B-Instruct-Q4_K_M.gguf": 1.0,
21
+ "Llama-3.2-3B-Instruct-Q4_K_M.gguf": 2.4,
22
+ "Ministral-3-3B-Instruct-2512-Q4_K_M.gguf": 2.5,
23
+ "Nemotron-Mini-4B-Instruct-Q4_K_M.gguf": 3.0,
24
+ "Phi-3.5-mini-instruct-Q4_K_M.gguf": 2.5,
25
+ "Phi-4-mini-reasoning-Q4_K_M.gguf": 3.2,
26
+ "qwen2.5-3b-instruct-q4_k_m.gguf": 2.4,
27
+ "Qwen2.5-7B-Instruct-IQ2_M.gguf": 3.0,
28
+ "Qwen2.5-7B-Instruct-Q4_K_M.gguf": 4.9,
29
+ "qwen2.5-coder-1.5b-instruct-q4_k_m.gguf": 1.2,
30
+ "qwen2.5-coder-3b-instruct-q4_k_m.gguf": 2.4,
31
+ "qwen2.5-coder-7b-instruct-q4_k_m.gguf": 4.8,
32
+ "SmolLM2-1.7B-Instruct-Q4_K_M.gguf": 1.4,
33
+ };
34
+ export function getModelVram(modelName) {
35
+ return VRAM_TABLE[modelName] ?? 2.0;
36
+ }
37
+ export function detectGpuInfo() {
38
+ try {
39
+ const out = execSync("nvidia-smi --query-gpu=name,memory.total,memory.free --format=csv,noheader", { encoding: "utf-8", timeout: 5000 }).trim();
40
+ const parts = out.split(", ");
41
+ if (parts.length >= 3) {
42
+ const name = parts[0];
43
+ const total = parseInt(parts[1], 10);
44
+ const free = parseInt(parts[2], 10);
45
+ return { name, totalMemMiB: total, freeMemMiB: free, available: true };
46
+ }
47
+ }
48
+ catch {
49
+ // nvidia-smi not available, try other methods
50
+ }
51
+ try {
52
+ const out = execSync("rocm-smi --json", {
53
+ encoding: "utf-8",
54
+ timeout: 5000,
55
+ });
56
+ const data = JSON.parse(out);
57
+ const first = Object.values(data)[0];
58
+ if (first) {
59
+ const total = parseInt(first["VRAM Total Memory (MB)"] ?? "0", 10);
60
+ const free = parseInt(first["VRAM Free Memory (MB)"] ?? "0", 10);
61
+ return {
62
+ name: first["Card series"] ?? "AMD GPU",
63
+ totalMemMiB: total,
64
+ freeMemMiB: free,
65
+ available: total > 0,
66
+ };
67
+ }
68
+ }
69
+ catch {
70
+ // rocm-smi not available
71
+ }
72
+ return {
73
+ name: "unknown",
74
+ totalMemMiB: 6144,
75
+ freeMemMiB: 4096,
76
+ available: false,
77
+ };
78
+ }
79
+ export function estimateCpuMode(totalRamMiB) {
80
+ return {
81
+ name: "CPU (RAM)",
82
+ totalMemMiB: totalRamMiB,
83
+ freeMemMiB: Math.round(totalRamMiB * 0.7),
84
+ available: false,
85
+ };
86
+ }
87
+ export function validateCombination(models, freeVramGB) {
88
+ const modelVrams = models.map((m) => ({
89
+ name: m.file,
90
+ vramGB: getModelVram(m.file),
91
+ }));
92
+ const totalVramGB = modelVrams.reduce((sum, m) => sum + m.vramGB, 0);
93
+ const suggestions = [];
94
+ if (totalVramGB > freeVramGB) {
95
+ const sorted = [...modelVrams].sort((a, b) => a.vramGB - b.vramGB);
96
+ suggestions.push(`Prueba con modelos mas pequeños: ${sorted.slice(0, 2).map((m) => m.name).join(", ")}`);
97
+ suggestions.push(`O usa modo uni-model con un solo modelo de hasta ${freeVramGB.toFixed(1)}GB`);
98
+ }
99
+ return {
100
+ valid: totalVramGB <= freeVramGB,
101
+ totalVramGB,
102
+ freeVramGB,
103
+ models: modelVrams,
104
+ message: totalVramGB <= freeVramGB
105
+ ? `✅ ${totalVramGB.toFixed(1)} GB de ${freeVramGB.toFixed(1)} GB libres`
106
+ : `✗ ${totalVramGB.toFixed(1)} GB estimados, solo ${freeVramGB.toFixed(1)} GB libres`,
107
+ suggestions,
108
+ };
109
+ }
110
+ export function listLocalModels(modelsPath) {
111
+ if (!existsSync(modelsPath)) {
112
+ return [];
113
+ }
114
+ const entries = [];
115
+ for (const file of readdirSync(modelsPath)) {
116
+ if (!file.endsWith(".gguf"))
117
+ continue;
118
+ const fullPath = join(modelsPath, file);
119
+ const st = statSync(fullPath);
120
+ if (!st.isFile())
121
+ continue;
122
+ const sizeGB = st.size / (1024 * 1024 * 1024);
123
+ const quant = detectQuant(file);
124
+ entries.push({
125
+ name: file,
126
+ path: join(modelsPath, file),
127
+ sizeGB: Math.round(sizeGB * 100) / 100,
128
+ quant,
129
+ estimatedVramGB: getModelVram(file),
130
+ exists: true,
131
+ });
132
+ }
133
+ return entries;
134
+ }
135
+ function detectQuant(fileName) {
136
+ const match = fileName.match(/-(Q[0-9]_[A-Z_]+|q[0-9]_[a-z_0-9]+|IQ[0-9]_[A-Z]+)\.gguf$/);
137
+ return match?.[1] ?? "unknown";
138
+ }
139
+ //# sourceMappingURL=vram.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"vram.js","sourceRoot":"","sources":["../../src/models/vram.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAC5D,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAGjC,MAAM,UAAU,GAA2B;IACzC,2CAA2C,EAAE,GAAG;IAChD,0CAA0C,EAAE,GAAG;IAC/C,wCAAwC,EAAE,GAAG;IAC7C,2BAA2B,EAAE,GAAG;IAChC,2BAA2B,EAAE,GAAG;IAChC,oCAAoC,EAAE,GAAG;IACzC,0BAA0B,EAAE,GAAG;IAC/B,qCAAqC,EAAE,GAAG;IAC1C,2CAA2C,EAAE,GAAG;IAChD,4BAA4B,EAAE,GAAG;IACjC,iCAAiC,EAAE,GAAG;IACtC,gCAAgC,EAAE,GAAG;IACrC,4BAA4B,EAAE,GAAG;IACjC,2CAA2C,EAAE,GAAG;IAChD,qCAAqC,EAAE,GAAG;IAC1C,mCAAmC,EAAE,GAAG;IACxC,mCAAmC,EAAE,GAAG;IACxC,0CAA0C,EAAE,GAAG;IAC/C,uCAAuC,EAAE,GAAG;IAC5C,mCAAmC,EAAE,GAAG;IACxC,kCAAkC,EAAE,GAAG;IACvC,iCAAiC,EAAE,GAAG;IACtC,gCAAgC,EAAE,GAAG;IACrC,iCAAiC,EAAE,GAAG;IACtC,yCAAyC,EAAE,GAAG;IAC9C,uCAAuC,EAAE,GAAG;IAC5C,uCAAuC,EAAE,GAAG;IAC5C,mCAAmC,EAAE,GAAG;CACzC,CAAC;AAEF,MAAM,UAAU,YAAY,CAAC,SAAiB;IAC5C,OAAO,UAAU,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC;AACtC,CAAC;AAED,MAAM,UAAU,aAAa;IAC3B,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,QAAQ,CAClB,4EAA4E,EAC5E,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CACrC,CAAC,IAAI,EAAE,CAAC;QACT,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACtB,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACtB,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACpC,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;QACzE,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,8CAA8C;IAChD,CAAC;IAED,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,QAAQ,CAAC,iBAAiB,EAAE;YACtC,QAAQ,EAAE,OAAO;YACjB,OAAO,EAAE,IAAI;SACd,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAA2B,CAAC;QAC/D,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,wBAAwB,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;YACnE,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,uBAAuB,CAAC,IAAI,GAAG,EAAE,EAAE,CAAC,CAAC;YACjE,OAAO;gBACL,IAAI,EAAE,KAAK,CAAC,aAAa,CAAC,IAAI,SAAS;gBACvC,WAAW,EAAE,KAAK;gBAClB,UAAU,EAAE,IAAI;gBAChB,SAAS,EAAE,KAAK,GAAG,CAAC;aACrB,CAAC;QACJ,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,yBAAyB;IAC3B,CAAC;IAED,OAAO;QACL,IAAI,EAAE,SAAS;QACf,WAAW,EAAE,IAAI;QACjB,UAAU,EAAE,IAAI;QAChB,SAAS,EAAE,KAAK;KACjB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,WAAmB;IACjD,OAAO;QACL,IAAI,EAAE,WAAW;QACjB,WAAW,EAAE,WAAW;QACxB,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,GAAG,CAAC;QACzC,SAAS,EAAE,KAAK;KACjB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,MAA0B,EAC1B,UAAkB;IAElB,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACpC,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;KAC7B,CAAC,CAAC,CAAC;IACJ,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAErE,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,IAAI,WAAW,GAAG,UAAU,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC;QACnE,WAAW,CAAC,IAAI,CACd,oCAAoC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACvF,CAAC;QACF,WAAW,CAAC,IAAI,CAAC,oDAAoD,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAClG,CAAC;IAED,OAAO;QACL,KAAK,EAAE,WAAW,IAAI,UAAU;QAChC,WAAW;QACX,UAAU;QACV,MAAM,EAAE,UAAU;QAClB,OAAO,EAAE,WAAW,IAAI,UAAU;YAChC,CAAC,CAAC,KAAK,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY;YACxE,CAAC,CAAC,KAAK,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY;QACvF,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,UAAkB;IAChD,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC5B,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAiB,EAAE,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC;QAC3C,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,SAAS;QACtC,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;QACxC,MAAM,EAAE,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC9B,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE;YAAE,SAAS;QAC3B,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,GAAG,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;QAChC,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,IAAI;YACV,IAAI,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC;YAC5B,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,GAAG,GAAG;YACtC,KAAK;YACL,eAAe,EAAE,YAAY,CAAC,IAAI,CAAC;YACnC,MAAM,EAAE,IAAI;SACb,CAAC,CAAC;IACL,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,WAAW,CAAC,QAAgB;IACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,2DAA2D,CAAC,CAAC;IAC1F,OAAO,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC;AACjC,CAAC"}