@tuned-tensor/local 0.2.8 → 0.3.0

package/dist/model-registry.js

@@ -5,9 +5,7 @@ import { verifyTarGzipArchive } from "./artifacts.js";
 export const TRAINING_MODELS = [
     {
         id: "Qwen/Qwen3.5-2B",
-        aliases: ["qwen/qwen3.5-2b", "Qwen/Qwen3.5-2B-Base", "qwen/qwen3.5-2b-base"],
         family: "qwen3_5",
-        loader: "causal_lm",
         defaultLearningRate: 0.00001,
         defaultPerDeviceBatchSize: 1,
         defaultGradientAccumulationSteps: 8,
@@ -15,156 +13,52 @@ export const TRAINING_MODELS = [
         defaultLoraAlpha: 32,
         defaultLoraDropout: 0.05,
         defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: false,
-    },
-    {
-        id: "Qwen/Qwen3.5-4B",
-        aliases: ["qwen/qwen3.5-4b", "Qwen/Qwen3.5-4B-Base", "qwen/qwen3.5-4b-base"],
-        family: "qwen3_5",
-        loader: "causal_lm",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 16,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: false,
-    },
-    {
-        id: "Qwen/Qwen3-VL-2B-Instruct",
-        aliases: ["qwen/qwen3-vl-2b-instruct", "Qwen/Qwen3-VL-2B", "qwen/qwen3-vl-2b"],
-        family: "qwen3_vl",
-        loader: "image_text_to_text",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 8,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: false,
-    },
-    {
-        id: "google/gemma-4-E2B-it",
-        aliases: ["google/gemma-4-E2B", "google/gemma-4-e2b", "google/gemma-4-e2b-it"],
-        family: "gemma4",
-        loader: "image_text_to_text",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 8,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: false,
-    },
-    {
-        id: "google/gemma-4-E4B-it",
-        aliases: ["google/gemma-4-E4B", "google/gemma-4-e4b", "google/gemma-4-e4b-it"],
-        family: "gemma4",
-        loader: "image_text_to_text",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 16,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: false,
-    },
-    {
-        id: "meta-llama/Llama-3.2-3B-Instruct",
-        aliases: ["meta-llama/llama-3.2-3b-instruct", "meta-llama/Llama-3.2-3B"],
-        family: "llama3_2",
-        loader: "causal_lm",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 16,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: true,
-    },
-    {
-        id: "microsoft/Phi-4-mini-instruct",
-        aliases: ["microsoft/phi-4-mini-instruct", "phi-4-mini-instruct"],
-        family: "phi4",
-        loader: "causal_lm",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 16,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: false,
-        requiresHfToken: false,
-    },
-    {
-        id: "ibm-granite/granite-3.3-2b-instruct",
-        aliases: ["granite-3.3-2b-instruct"],
-        family: "granite3_3",
-        loader: "causal_lm",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 8,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: false,
-    },
-    {
-        id: "bigcode/starcoder2-3b",
-        aliases: ["starcoder2-3b"],
-        family: "starcoder2",
-        loader: "causal_lm",
-        defaultLearningRate: 0.00001,
-        defaultPerDeviceBatchSize: 1,
-        defaultGradientAccumulationSteps: 8,
-        defaultLoraRank: 16,
-        defaultLoraAlpha: 32,
-        defaultLoraDropout: 0.05,
-        defaultMaxSeqLength: 2048,
-        trustRemoteCode: true,
-        requiresHfToken: false,
     },
 ];
-const EXTERNAL_MODEL_PREFIXES = ["external:", "command:"];
-export function isExternalTrainingModel(modelId) {
-    const normalized = modelId.trim().toLowerCase();
-    return EXTERNAL_MODEL_PREFIXES.some((prefix) => normalized.startsWith(prefix));
-}
 export function resolveTrainingModel(modelId) {
     const normalized = modelId.trim().toLowerCase();
-    const model = TRAINING_MODELS.find((candidate) => candidate.id.toLowerCase() === normalized
-        || candidate.aliases.some((alias) => alias.toLowerCase() === normalized));
+    const model = TRAINING_MODELS.find((candidate) => candidate.id.toLowerCase() === normalized);
     if (!model) {
-        const supported = TRAINING_MODELS.map((candidate) => candidate.id).join(", ");
-        throw new Error(`Unsupported base model "${modelId}". Supported models: ${supported}`);
+        throw new Error(`Unsupported base model "${modelId}". Supported model: ${TRAINING_MODELS[0].id}`);
     }
     return model;
 }
 export function canonicalizeTrainingModel(modelId) {
-    if (isExternalTrainingModel(modelId)) {
-        const trimmed = modelId.trim();
-        const [prefix, ...rest] = trimmed.split(":");
-        const id = rest.join(":").trim();
-        if (!id)
-            throw new Error(`External model "${modelId}" must include an id after "${prefix}:".`);
-        return `${prefix.toLowerCase()}:${id}`;
-    }
     return resolveTrainingModel(modelId).id;
 }
+const CERTIFIED_TEXT_CONFIG = {
+    model_type: "qwen3_5_text",
+    hidden_size: 2048,
+    num_hidden_layers: 24,
+    num_attention_heads: 8,
+    num_key_value_heads: 2,
+    intermediate_size: 6144,
+    vocab_size: 248320,
+};
+/** Reject a same-family snapshot that is not the certified 2B architecture. */
+export function assertCertifiedBaseModelConfig(value, label = "base-model config.json") {
+    if (!value || typeof value !== "object" || Array.isArray(value)) {
+        throw new Error(`${label} must contain a JSON object.`);
+    }
+    const config = value;
+    const architectures = config.architectures;
+    const textConfig = config.text_config;
+    if (config.model_type !== "qwen3_5"
+        || !Array.isArray(architectures)
+        || !architectures.includes("Qwen3_5ForConditionalGeneration")
+        || !textConfig
+        || typeof textConfig !== "object"
+        || Array.isArray(textConfig)) {
+        throw new Error(`${label} is not the certified Qwen/Qwen3.5-2B architecture.`);
+    }
+    const text = textConfig;
+    for (const [key, expected] of Object.entries(CERTIFIED_TEXT_CONFIG)) {
+        if (text[key] !== expected) {
+            throw new Error(`${label} is not the certified Qwen/Qwen3.5-2B architecture: `
+                + `text_config.${key} must be ${JSON.stringify(expected)}.`);
+        }
+    }
+}
 export function localModelArtifactPath(uri) {
     if (uri.startsWith("file://"))
         return fileURLToPath(new URL(uri));
@@ -173,189 +67,79 @@ export function localModelArtifactPath(uri) {
     }
     return resolve(uri);
 }
-function isModelPayloadFile(path) {
-    const name = path.split(/[\\/]/).at(-1)?.toLowerCase() ?? "";
-    if (name === "training-metrics.json"
-        || name === "trainer_state.json"
-        || name === "training_args.bin"
-        || name === "config.json"
-        || name === "generation_config.json"
-        || name === "adapter_config.json"
-        || name === "readme.md"
-        || name.startsWith("tokenizer")
-        || name.startsWith("special_tokens")
-        || name.startsWith("added_tokens")
-        || name === "vocab.json"
-        || name === "merges.txt")
-        return false;
-    return true;
-}
-function isRecognizedModelPayloadFile(path) {
-    const name = path.split(/[\\/]/).at(-1)?.toLowerCase() ?? "";
-    if (name === "training_args.bin")
-        return false;
-    return name.endsWith(".safetensors")
-        || name.endsWith(".bin")
-        || name.endsWith(".pt")
-        || name.endsWith(".pth")
-        || name.endsWith(".gguf")
-        || name.endsWith(".onnx")
-        || name.endsWith(".ckpt")
-        || name.endsWith(".npz");
+function adapterWeightName(name) {
+    const lower = name.toLowerCase();
+    return lower === "adapter_model.safetensors" || lower === "adapter_model.bin";
 }
-function isAdapterWeightFile(path) {
-    const name = path.split(/[\\/]/).at(-1)?.toLowerCase() ?? "";
-    return name === "adapter_model.safetensors" || name === "adapter_model.bin";
-}
-function isFullModelWeightFile(path) {
-    const name = path.split(/[\\/]/).at(-1)?.toLowerCase() ?? "";
-    return (name.endsWith(".safetensors") && !name.startsWith("adapter_"))
-        || /^pytorch_model.*\.bin$/.test(name);
-}
-async function inspectDirectory(path) {
-    let fileCount = 0;
-    let totalBytes = 0;
-    let payloadFileCount = 0;
-    let payloadBytes = 0;
-    let recognizedPayloadFileCount = 0;
-    let recognizedPayloadBytes = 0;
-    let adapterWeightFileCount = 0;
-    let adapterWeightBytes = 0;
-    let fullModelWeightFileCount = 0;
-    let fullModelWeightBytes = 0;
-    let hasAdapterConfig = false;
+async function inspectAdapterDirectory(path) {
+    let count = 0;
+    let bytes = 0;
+    let hasConfig = false;
     for (const entry of await readdir(path, { withFileTypes: true })) {
         const child = resolve(path, entry.name);
         if (entry.isSymbolicLink()) {
             throw new Error(`Model artifact must not contain symbolic links: ${child}`);
         }
         if (entry.isDirectory()) {
-            const nested = await inspectDirectory(child);
-            fileCount += nested.fileCount;
-            totalBytes += nested.totalBytes;
-            payloadFileCount += nested.payloadFileCount;
-            payloadBytes += nested.payloadBytes;
-            recognizedPayloadFileCount += nested.recognizedPayloadFileCount;
-            recognizedPayloadBytes += nested.recognizedPayloadBytes;
-            adapterWeightFileCount += nested.adapterWeightFileCount;
-            adapterWeightBytes += nested.adapterWeightBytes;
-            fullModelWeightFileCount += nested.fullModelWeightFileCount;
-            fullModelWeightBytes += nested.fullModelWeightBytes;
-            hasAdapterConfig ||= nested.hasAdapterConfig;
+            const nested = await inspectAdapterDirectory(child);
+            count += nested.count;
+            bytes += nested.bytes;
+            hasConfig ||= nested.hasConfig;
             continue;
         }
         if (!entry.isFile())
             continue;
         const metadata = await lstat(child);
-        if (entry.name.toLowerCase() === "adapter_config.json" && metadata.size > 0)
-            hasAdapterConfig = true;
-        fileCount += 1;
-        totalBytes += metadata.size;
-        if (isModelPayloadFile(child)) {
-            payloadFileCount += 1;
-            payloadBytes += metadata.size;
-        }
-        if (isRecognizedModelPayloadFile(child)) {
-            recognizedPayloadFileCount += 1;
-            recognizedPayloadBytes += metadata.size;
+        if (adapterWeightName(entry.name) && metadata.size > 0) {
+            count += 1;
+            bytes += metadata.size;
         }
-        if (isAdapterWeightFile(child)) {
-            adapterWeightFileCount += 1;
-            adapterWeightBytes += metadata.size;
-        }
-        if (isFullModelWeightFile(child)) {
-            fullModelWeightFileCount += 1;
-            fullModelWeightBytes += metadata.size;
+        if (entry.name.toLowerCase() === "adapter_config.json" && metadata.size > 0) {
+            hasConfig = true;
         }
     }
-    return {
-        fileCount,
-        totalBytes,
-        payloadFileCount,
-        payloadBytes,
-        recognizedPayloadFileCount,
-        recognizedPayloadBytes,
-        adapterWeightFileCount,
-        adapterWeightBytes,
-        fullModelWeightFileCount,
-        fullModelWeightBytes,
-        hasAdapterConfig,
-    };
+    return { count, bytes, hasConfig };
 }
 export async function inspectModelArtifact(uri) {
     const path = localModelArtifactPath(uri);
     const metadata = await lstat(path).catch(() => null);
     if (!metadata)
         throw new Error(`Model artifact does not exist: ${path}`);
-    if (metadata.isSymbolicLink())
+    if (metadata.isSymbolicLink()) {
         throw new Error(`Model artifact must not be a symbolic link: ${path}`);
-    if (metadata.isFile()) {
-        return {
-            uri,
-            path,
-            kind: "file",
-            file_count: 1,
-            total_bytes: metadata.size,
-            payload_file_count: 1,
-            payload_bytes: metadata.size,
-            recognized_payload_file_count: isRecognizedModelPayloadFile(path) ? 1 : 0,
-            recognized_payload_bytes: isRecognizedModelPayloadFile(path) ? metadata.size : 0,
-            adapter_weight_file_count: isAdapterWeightFile(path) ? 1 : 0,
-            adapter_weight_bytes: isAdapterWeightFile(path) ? metadata.size : 0,
-            full_model_weight_file_count: isFullModelWeightFile(path) ? 1 : 0,
-            full_model_weight_bytes: isFullModelWeightFile(path) ? metadata.size : 0,
-            has_adapter_config: false,
-        };
     }
     if (metadata.isDirectory()) {
-        const contents = await inspectDirectory(path);
+        const adapter = await inspectAdapterDirectory(path);
         return {
             uri,
             path,
             kind: "directory",
-            file_count: contents.fileCount,
-            total_bytes: contents.totalBytes,
-            payload_file_count: contents.payloadFileCount,
-            payload_bytes: contents.payloadBytes,
-            recognized_payload_file_count: contents.recognizedPayloadFileCount,
-            recognized_payload_bytes: contents.recognizedPayloadBytes,
-            adapter_weight_file_count: contents.adapterWeightFileCount,
-            adapter_weight_bytes: contents.adapterWeightBytes,
-            full_model_weight_file_count: contents.fullModelWeightFileCount,
-            full_model_weight_bytes: contents.fullModelWeightBytes,
-            has_adapter_config: contents.hasAdapterConfig,
+            adapter_weight_file_count: adapter.count,
+            adapter_weight_bytes: adapter.bytes,
+            has_adapter_config: adapter.hasConfig,
         };
     }
-    throw new Error(`Model artifact is not a regular file or directory: ${path}`);
-}
-export async function assertUsableModelArtifact(uri, options = {}) {
-    let inspection = await inspectModelArtifact(uri);
-    if (inspection.kind === "file" && inspection.path.toLowerCase().endsWith(".tar.gz")) {
-        const archive = await verifyTarGzipArchive(inspection.path);
-        inspection = {
-            ...inspection,
-            recognized_payload_file_count: archive.recognized_payload_entries,
-            recognized_payload_bytes: archive.recognized_payload_bytes,
-            adapter_weight_file_count: archive.adapter_weight_entries,
-            adapter_weight_bytes: archive.adapter_weight_bytes,
-            full_model_weight_file_count: archive.full_model_weight_entries,
-            full_model_weight_bytes: archive.full_model_weight_bytes,
-            has_adapter_config: archive.adapter_config_entries > 0,
-        };
+    if (!metadata.isFile() || !path.toLowerCase().endsWith(".tar.gz")) {
+        throw new Error(`Model artifact must be a PEFT adapter directory or .tar.gz archive: ${path}`);
     }
-    if (inspection.payload_file_count === 0 || inspection.payload_bytes === 0) {
-        throw new Error(`Training completed without a usable model artifact: ${inspection.path} has no model payload files.`);
-    }
-    if (!options.allowUnrecognizedPayload
-        && inspection.adapter_weight_file_count === 0
-        && inspection.full_model_weight_file_count === 0) {
-        throw new Error(`Model artifact ${inspection.path} has no loadable adapter or Transformers full-model weights.`);
-    }
-    if (!options.allowUnrecognizedPayload && inspection.adapter_weight_file_count > 0 && !inspection.has_adapter_config) {
-        throw new Error(`Model artifact ${inspection.path} has adapter weights but no non-empty adapter_config.json.`);
+    const archive = await verifyTarGzipArchive(path);
+    return {
+        uri,
+        path,
+        kind: "file",
+        adapter_weight_file_count: archive.adapter_weight_entries,
+        adapter_weight_bytes: archive.adapter_weight_bytes,
+        has_adapter_config: archive.adapter_config_entries > 0,
+    };
+}
+export async function assertUsableModelArtifact(uri) {
+    const inspection = await inspectModelArtifact(uri);
+    if (inspection.adapter_weight_file_count === 0
+        || inspection.adapter_weight_bytes === 0) {
+        throw new Error(`PEFT model artifact ${inspection.path} contains no non-empty adapter_model weights.`);
     }
-    if (!options.allowUnrecognizedPayload && inspection.has_adapter_config && inspection.adapter_weight_file_count === 0) {
-        throw new Error(`Model artifact ${inspection.path} has adapter_config.json but no adapter_model weights.`);
+    if (!inspection.has_adapter_config) {
+        throw new Error(`PEFT model artifact ${inspection.path} contains adapter weights but no non-empty adapter_config.json.`);
     }
     return inspection;
 }

package/dist/model-registry.js.map

@@ -1 +1 @@
-{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../src/model-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AAkBtD,MAAM,CAAC,MAAM,eAAe,GAAoB;IAC9C;QACE,EAAE,EAAE,iBAAiB;QACrB,OAAO,EAAE,CAAC,iBAAiB,EAAE,sBAAsB,EAAE,sBAAsB,CAAC;QAC5E,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,WAAW;QACnB,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,CAAC;QACnC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,KAAK;KACvB;IACD;QACE,EAAE,EAAE,iBAAiB;QACrB,OAAO,EAAE,CAAC,iBAAiB,EAAE,sBAAsB,EAAE,sBAAsB,CAAC;QAC5E,MAAM,EAAE,SAAS;QACjB,MAAM,EAAE,WAAW;QACnB,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,EAAE;QACpC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,KAAK;KACvB;IACD;QACE,EAAE,EAAE,2BAA2B;QAC/B,OAAO,EAAE,CAAC,2BAA2B,EAAE,kBAAkB,EAAE,kBAAkB,CAAC;QAC9E,MAAM,EAAE,UAAU;QAClB,MAAM,EAAE,oBAAoB;QAC5B,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,CAAC;QACnC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,KAAK;KACvB;IACD;QACE,EAAE,EAAE,uBAAuB;QAC3B,OAAO,EAAE,CAAC,oBAAoB,EAAE,oBAAoB,EAAE,uBAAuB,CAAC;QAC9E,MAAM,EAAE,QAAQ;QAChB,MAAM,EAAE,oBAAoB;QAC5B,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,CAAC;QACnC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,KAAK;KACvB;IACD;QACE,EAAE,EAAE,uBAAuB;QAC3B,OAAO,EAAE,CAAC,oBAAoB,EAAE,oBAAoB,EAAE,uBAAuB,CAAC;QAC9E,MAAM,EAAE,QAAQ;QAChB,MAAM,EAAE,oBAAoB;QAC5B,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,EAAE;QACpC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,KAAK;KACvB;IACD;QACE,EAAE,EAAE,kCAAkC;QACtC,OAAO,EAAE,CAAC,kCAAkC,EAAE,yBAAyB,CAAC;QACxE,MAAM,EAAE,UAAU;QAClB,MAAM,EAAE,WAAW;QACnB,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,EAAE;QACpC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,IAAI;KACtB;IACD;QACE,EAAE,EAAE,+BAA+B;QACnC,OAAO,EAAE,CAAC,+BAA+B,EAAE,qBAAqB,CAAC;QACjE,MAAM,EAAE,MAAM;QACd,MAAM,EAAE,WAAW;QACnB,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,EAAE;QACpC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,KAAK;QACtB,eAAe,EAAE,KAAK;KACvB;IACD;QACE,EAAE,EAAE,qCAAqC;QACzC,OAAO,EAAE,CAAC,yBAAyB,CAAC;QACpC,MAAM,EAAE,YAAY;QACpB,MAAM,EAAE,WAAW;QACnB,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,CAAC;QACnC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,KAAK;KACvB;IACD;QACE,EAAE,EAAE,uBAAuB;QAC3B,OAAO,EAAE,CAAC,eAAe,CAAC;QAC1B,MAAM,EAAE,YAAY;QACpB,MAAM,EAAE,WAAW;QACnB,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,CAAC;QACnC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;QACzB,eAAe,EAAE,IAAI;QACrB,eAAe,EAAE,KAAK;KACvB;CACF,CAAC;AAEF,MAAM,uBAAuB,GAAG,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;AAE1D,MAAM,UAAU,uBAAuB,CAAC,OAAe;IACrD,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAChD,OAAO,uBAAuB,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;AACjF,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,OAAe;IAClD,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAChD,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAC/C,SAAS,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,UAAU;WACtC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,UAAU,CAAC,CACzE,CAAC;IACF,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,SAAS,GAAG,eAAe,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9E,MAAM,IAAI,KAAK,CAAC,2BAA2B,OAAO,wBAAwB,SAAS,EAAE,CAAC,CAAC;IACzF,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,OAAe;IACvD,IAAI,uBAAuB,CAAC,OAAO,CAAC,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,CAAC,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC7C,MAAM,EAAE,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACjC,IAAI,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,OAAO,+BAA+B,MAAM,KAAK,CAAC,CAAC;QAC/F,OAAO,GAAG,MAAM,CAAC,WAAW,EAAE,IAAI,EAAE,EAAE,CAAC;IACzC,CAAC;IACD,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;AAC1C,CAAC;AAmBD,MAAM,UAAU,sBAAsB,CAAC,GAAW;IAChD,IAAI,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,aAAa,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAClE,IAAI,sBAAsB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,oDAAoD,GAAG,EAAE,CAAC,CAAC;IAC7E,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAC7D,IACE,IAAI,KAAK,uBAAuB;WAC7B,IAAI,KAAK,oBAAoB;WAC7B,IAAI,KAAK,mBAAmB;WAC5B,IAAI,KAAK,aAAa;WACtB,IAAI,KAAK,wBAAwB;WACjC,IAAI,KAAK,qBAAqB;WAC9B,IAAI,KAAK,WAAW;WACpB,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC;WAC5B,IAAI,CAAC,UAAU,CAAC,gBAAgB,CAAC;WACjC,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC;WAC/B,IAAI,KAAK,YAAY;WACrB,IAAI,KAAK,YAAY;QACxB,OAAO,KAAK,CAAC;IACf,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,4BAA4B,CAAC,IAAY;IAChD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAC7D,IAAI,IAAI,KAAK,mBAAmB;QAAE,OAAO,KAAK,CAAC;IAC/C,OAAO,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC;WAC/B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;WACrB,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;WACpB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;WACrB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;WACtB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;WACtB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;WACtB,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;AAC7B,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY;IACvC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAC7D,OAAO,IAAI,KAAK,2BAA2B,IAAI,IAAI,KAAK,mBAAmB,CAAC;AAC9E,CAAC;AAED,SAAS,qBAAqB,CAAC,IAAY;IACzC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;IAC7D,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;WACjE,wBAAwB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC3C,CAAC;AAED,KAAK,UAAU,gBAAgB,CAAC,IAAY;IAa1C,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,gBAAgB,GAAG,CAAC,CAAC;IACzB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,0BAA0B,GAAG,CAAC,CAAC;IACnC,IAAI,sBAAsB,GAAG,CAAC,CAAC;IAC/B,IAAI,sBAAsB,GAAG,CAAC,CAAC;IAC/B,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,wBAAwB,GAAG,CAAC,CAAC;IACjC,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAC7B,IAAI,gBAAgB,GAAG,KAAK,CAAC;IAC7B,KAAK,MAAM,KAAK,IAAI,MAAM,OAAO,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QACjE,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,KAAK,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,mDAAmD,KAAK,EAAE,CAAC,CAAC;QAC9E,CAAC;QACD,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAC7C,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC;YAC9B,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC;YAChC,gBAAgB,IAAI,MAAM,CAAC,gBAAgB,CAAC;YAC5C,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC;YACpC,0BAA0B,IAAI,MAAM,CAAC,0BAA0B,CAAC;YAChE,sBAAsB,IAAI,MAAM,CAAC,sBAAsB,CAAC;YACxD,sBAAsB,IAAI,MAAM,CAAC,sBAAsB,CAAC;YACxD,kBAAkB,IAAI,MAAM,CAAC,kBAAkB,CAAC;YAChD,wBAAwB,IAAI,MAAM,CAAC,wBAAwB,CAAC;YAC5D,oBAAoB,IAAI,MAAM,CAAC,oBAAoB,CAAC;YACpD,gBAAgB,KAAK,MAAM,CAAC,gBAAgB,CAAC;YAC7C,SAAS;QACX,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE;YAAE,SAAS;QAC9B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,qBAAqB,IAAI,QAAQ,CAAC,IAAI,GAAG,CAAC;YAAE,gBAAgB,GAAG,IAAI,CAAC;QACrG,SAAS,IAAI,CAAC,CAAC;QACf,UAAU,IAAI,QAAQ,CAAC,IAAI,CAAC;QAC5B,IAAI,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,gBAAgB,IAAI,CAAC,CAAC;YACtB,YAAY,IAAI,QAAQ,CAAC,IAAI,CAAC;QAChC,CAAC;QACD,IAAI,4BAA4B,CAAC,KAAK,CAAC,EAAE,CAAC;YACxC,0BAA0B,IAAI,CAAC,CAAC;YAChC,sBAAsB,IAAI,QAAQ,CAAC,IAAI,CAAC;QAC1C,CAAC;QACD,IAAI,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,sBAAsB,IAAI,CAAC,CAAC;YAC5B,kBAAkB,IAAI,QAAQ,CAAC,IAAI,CAAC;QACtC,CAAC;QACD,IAAI,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAAC;YACjC,wBAAwB,IAAI,CAAC,CAAC;YAC9B,oBAAoB,IAAI,QAAQ,CAAC,IAAI,CAAC;QACxC,CAAC;IACH,CAAC;IACD,OAAO;QACL,SAAS;QACT,UAAU;QACV,gBAAgB;QAChB,YAAY;QACZ,0BAA0B;QAC1B,sBAAsB;QACtB,sBAAsB;QACtB,kBAAkB;QAClB,wBAAwB;QACxB,oBAAoB;QACpB,gBAAgB;KACjB,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,GAAW;IACpD,MAAM,IAAI,GAAG,sBAAsB,CAAC,GAAG,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IACrD,IAAI,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,IAAI,EAAE,CAAC,CAAC;IACzE,IAAI,QAAQ,CAAC,cAAc,EAAE;QAAE,MAAM,IAAI,KAAK,CAAC,+CAA+C,IAAI,EAAE,CAAC,CAAC;IACtG,IAAI,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC;QACtB,OAAO;YACL,GAAG;YACH,IAAI;YACJ,IAAI,EAAE,MAAM;YACZ,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,QAAQ,CAAC,IAAI;YAC1B,kBAAkB,EAAE,CAAC;YACrB,aAAa,EAAE,QAAQ,CAAC,IAAI;YAC5B,6BAA6B,EAAE,4BAA4B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACzE,wBAAwB,EAAE,4BAA4B,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAChF,yBAAyB,EAAE,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5D,oBAAoB,EAAE,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACnE,4BAA4B,EAAE,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjE,uBAAuB,EAAE,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACxE,kBAAkB,EAAE,KAAK;SAC1B,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC9C,OAAO;YACL,GAAG;YACH,IAAI;YACJ,IAAI,EAAE,WAAW;YACjB,UAAU,EAAE,QAAQ,CAAC,SAAS;YAC9B,WAAW,EAAE,QAAQ,CAAC,UAAU;YAChC,kBAAkB,EAAE,QAAQ,CAAC,gBAAgB;YAC7C,aAAa,EAAE,QAAQ,CAAC,YAAY;YACpC,6BAA6B,EAAE,QAAQ,CAAC,0BAA0B;YAClE,wBAAwB,EAAE,QAAQ,CAAC,sBAAsB;YACzD,yBAAyB,EAAE,QAAQ,CAAC,sBAAsB;YAC1D,oBAAoB,EAAE,QAAQ,CAAC,kBAAkB;YACjD,4BAA4B,EAAE,QAAQ,CAAC,wBAAwB;YAC/D,uBAAuB,EAAE,QAAQ,CAAC,oBAAoB;YACtD,kBAAkB,EAAE,QAAQ,CAAC,gBAAgB;SAC9C,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,sDAAsD,IAAI,EAAE,CAAC,CAAC;AAChF,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,GAAW,EACX,UAAkD,EAAE;IAEpD,IAAI,UAAU,GAAG,MAAM,oBAAoB,CAAC,GAAG,CAAC,CAAC;IACjD,IAAI,UAAU,CAAC,IAAI,KAAK,MAAM,IAAI,UAAU,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QACpF,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5D,UAAU,GAAG;YACX,GAAG,UAAU;YACb,6BAA6B,EAAE,OAAO,CAAC,0BAA0B;YACjE,wBAAwB,EAAE,OAAO,CAAC,wBAAwB;YAC1D,yBAAyB,EAAE,OAAO,CAAC,sBAAsB;YACzD,oBAAoB,EAAE,OAAO,CAAC,oBAAoB;YAClD,4BAA4B,EAAE,OAAO,CAAC,yBAAyB;YAC/D,uBAAuB,EAAE,OAAO,CAAC,uBAAuB;YACxD,kBAAkB,EAAE,OAAO,CAAC,sBAAsB,GAAG,CAAC;SACvD,CAAC;IACJ,CAAC;IACD,IAAI,UAAU,CAAC,kBAAkB,KAAK,CAAC,IAAI,UAAU,CAAC,aAAa,KAAK,CAAC,EAAE,CAAC;QAC1E,MAAM,IAAI,KAAK,CAAC,uDAAuD,UAAU,CAAC,IAAI,8BAA8B,CAAC,CAAC;IACxH,CAAC;IACD,IACE,CAAC,OAAO,CAAC,wBAAwB;WAC9B,UAAU,CAAC,yBAAyB,KAAK,CAAC;WAC1C,UAAU,CAAC,4BAA4B,KAAK,CAAC,EAChD,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,kBAAkB,UAAU,CAAC,IAAI,8DAA8D,CAAC,CAAC;IACnH,CAAC;IACD,IAAI,CAAC,OAAO,CAAC,wBAAwB,IAAI,UAAU,CAAC,yBAAyB,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,kBAAkB,EAAE,CAAC;QACpH,MAAM,IAAI,KAAK,CAAC,kBAAkB,UAAU,CAAC,IAAI,4DAA4D,CAAC,CAAC;IACjH,CAAC;IACD,IAAI,CAAC,OAAO,CAAC,wBAAwB,IAAI,UAAU,CAAC,kBAAkB,IAAI,UAAU,CAAC,yBAAyB,KAAK,CAAC,EAAE,CAAC;QACrH,MAAM,IAAI,KAAK,CAAC,kBAAkB,UAAU,CAAC,IAAI,wDAAwD,CAAC,CAAC;IAC7G,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC"}
+{"version":3,"file":"model-registry.js","sourceRoot":"","sources":["../src/model-registry.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzC,OAAO,EAAE,oBAAoB,EAAE,MAAM,gBAAgB,CAAC;AActD,MAAM,CAAC,MAAM,eAAe,GAAoB;IAC9C;QACE,EAAE,EAAE,iBAAiB;QACrB,MAAM,EAAE,SAAS;QACjB,mBAAmB,EAAE,OAAO;QAC5B,yBAAyB,EAAE,CAAC;QAC5B,gCAAgC,EAAE,CAAC;QACnC,eAAe,EAAE,EAAE;QACnB,gBAAgB,EAAE,EAAE;QACpB,kBAAkB,EAAE,IAAI;QACxB,mBAAmB,EAAE,IAAI;KAC1B;CACF,CAAC;AAEF,MAAM,UAAU,oBAAoB,CAAC,OAAe;IAClD,MAAM,UAAU,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;IAChD,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAC/C,SAAS,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,UAAU,CAC1C,CAAC;IACF,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CACb,2BAA2B,OAAO,uBAAuB,eAAe,CAAC,CAAC,CAAE,CAAC,EAAE,EAAE,CAClF,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,OAAe;IACvD,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;AAC1C,CAAC;AAED,MAAM,qBAAqB,GAAG;IAC5B,UAAU,EAAE,cAAc;IAC1B,WAAW,EAAE,IAAI;IACjB,iBAAiB,EAAE,EAAE;IACrB,mBAAmB,EAAE,CAAC;IACtB,mBAAmB,EAAE,CAAC;IACtB,iBAAiB,EAAE,IAAI;IACvB,UAAU,EAAE,MAAM;CACV,CAAC;AAEX,+EAA+E;AAC/E,MAAM,UAAU,8BAA8B,CAC5C,KAAc,EACd,KAAK,GAAG,wBAAwB;IAEhC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CAAC,GAAG,KAAK,8BAA8B,CAAC,CAAC;IAC1D,CAAC;IACD,MAAM,MAAM,GAAG,KAAgC,CAAC;IAChD,MAAM,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;IAC3C,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,CAAC;IACtC,IACE,MAAM,CAAC,UAAU,KAAK,SAAS;WAC5B,CAAC,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC;WAC7B,CAAC,aAAa,CAAC,QAAQ,CAAC,iCAAiC,CAAC;WAC1D,CAAC,UAAU;WACX,OAAO,UAAU,KAAK,QAAQ;WAC9B,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAC5B,CAAC;QACD,MAAM,IAAI,KAAK,CACb,GAAG,KAAK,qDAAqD,CAC9D,CAAC;IACJ,CAAC;IACD,MAAM,IAAI,GAAG,UAAqC,CAAC;IACnD,KAAK,MAAM,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAAE,CAAC;QACpE,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,QAAQ,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CACb,GAAG,KAAK,sDAAsD;kBAC5D,eAAe,GAAG,YAAY,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,GAAG,CAC5D,CAAC;QACJ,CAAC;IACH,CAAC;AACH,CAAC;AAWD,MAAM,UAAU,sBAAsB,CAAC,GAAW;IAChD,IAAI,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,aAAa,CAAC,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAClE,IAAI,sBAAsB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,oDAAoD,GAAG,EAAE,CAAC,CAAC;IAC7E,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAY;IACrC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACjC,OAAO,KAAK,KAAK,2BAA2B,IAAI,KAAK,KAAK,mBAAmB,CAAC;AAChF,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,IAAY;IAKjD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,SAAS,GAAG,KAAK,CAAC;IACtB,KAAK,MAAM,KAAK,IAAI,MAAM,OAAO,CAAC,IAAI,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;QACjE,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,KAAK,CAAC,cAAc,EAAE,EAAE,CAAC;YAC3B,MAAM,IAAI,KAAK,CAAC,mDAAmD,KAAK,EAAE,CAAC,CAAC;QAC9E,CAAC;QACD,IAAI,KAAK,CAAC,WAAW,EAAE,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,KAAK,CAAC,CAAC;YACpD,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;YACtB,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;YACtB,SAAS,KAAK,MAAM,CAAC,SAAS,CAAC;YAC/B,SAAS;QACX,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE;YAAE,SAAS;QAC9B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,iBAAiB,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,QAAQ,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACvD,KAAK,IAAI,CAAC,CAAC;YACX,KAAK,IAAI,QAAQ,CAAC,IAAI,CAAC;QACzB,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,KAAK,qBAAqB,IAAI,QAAQ,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC5E,SAAS,GAAG,IAAI,CAAC;QACnB,CAAC;IACH,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;AACrC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CAAC,GAAW;IACpD,MAAM,IAAI,GAAG,sBAAsB,CAAC,GAAG,CAAC,CAAC;IACzC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;IACrD,IAAI,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,IAAI,EAAE,CAAC,CAAC;IACzE,IAAI,QAAQ,CAAC,cAAc,EAAE,EAAE,CAAC;QAC9B,MAAM,IAAI,KAAK,CAAC,+CAA+C,IAAI,EAAE,CAAC,CAAC;IACzE,CAAC;IACD,IAAI,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;QAC3B,MAAM,OAAO,GAAG,MAAM,uBAAuB,CAAC,IAAI,CAAC,CAAC;QACpD,OAAO;YACL,GAAG;YACH,IAAI;YACJ,IAAI,EAAE,WAAW;YACjB,yBAAyB,EAAE,OAAO,CAAC,KAAK;YACxC,oBAAoB,EAAE,OAAO,CAAC,KAAK;YACnC,kBAAkB,EAAE,OAAO,CAAC,SAAS;SACtC,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,KAAK,CACb,uEAAuE,IAAI,EAAE,CAC9E,CAAC;IACJ,CAAC;IACD,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,IAAI,CAAC,CAAC;IACjD,OAAO;QACL,GAAG;QACH,IAAI;QACJ,IAAI,EAAE,MAAM;QACZ,yBAAyB,EAAE,OAAO,CAAC,sBAAsB;QACzD,oBAAoB,EAAE,OAAO,CAAC,oBAAoB;QAClD,kBAAkB,EAAE,OAAO,CAAC,sBAAsB,GAAG,CAAC;KACvD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,GAAW;IAEX,MAAM,UAAU,GAAG,MAAM,oBAAoB,CAAC,GAAG,CAAC,CAAC;IACnD,IACE,UAAU,CAAC,yBAAyB,KAAK,CAAC;WACvC,UAAU,CAAC,oBAAoB,KAAK,CAAC,EACxC,CAAC;QACD,MAAM,IAAI,KAAK,CACb,uBAAuB,UAAU,CAAC,IAAI,+CAA+C,CACtF,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,UAAU,CAAC,kBAAkB,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CACb,uBAAuB,UAAU,CAAC,IAAI,iEAAiE,CACxG,CAAC;IACJ,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC"}

package/dist/model-server.d.ts

@@ -3,7 +3,7 @@ import type { LocalModelRecord } from "./store.js";
 export interface LocalModelServeOptions {
     host?: string;
     port?: number;
-    device?: "auto" | "cpu" | "cuda" | "mps";
+    device?: "cpu" | "cuda";
     maxTokens?: number;
     temperature?: number;
     topP?: number;
@@ -18,7 +18,6 @@ export interface LocalModelServerLaunch {
     command: string;
     commandArgs: string[];
     displayCommand: string[];
-    cwd?: string;
     env: NodeJS.ProcessEnv;
     url: string;
     modelName: string;

package/dist/model-server.js

@@ -1,9 +1,9 @@
 import { spawn } from "node:child_process";
 import { resolve } from "node:path";
 import { fileURLToPath } from "node:url";
-import { minimalMachineLearningEnvironment, withHuggingFaceCacheEnvironment } from "./huggingface-cache.js";
+import { minimalMachineLearningEnvironment, withOfflineHuggingFaceCacheEnvironment, } from "./huggingface-cache.js";
 import { resolveTrainingModel } from "./model-registry.js";
-import { buildEntrypointCommand } from "./process-runner.js";
+import { buildBundledPythonCommand, withBundledPythonEnvironment, } from "./process-runner.js";
 function localArtifactPath(uri) {
     if (uri.startsWith("file://"))
         return fileURLToPath(uri);
@@ -56,41 +56,33 @@ export function buildLocalModelServerLaunch(args) {
             + recordedBaseModelPath);
     }
     const localBaseModelPath = recordedBaseModelPath ?? configuredBaseModelPath;
-    const resolvedModel = resolveTrainingModel(args.model.base_model);
-    const entrypoint = buildEntrypointCommand({
-        backend: "uv",
-        project: args.config.evaluation.inference.project ?? "training/local-runner",
-        cwd: args.config.evaluation.inference.cwd,
-        script: "training/local-runner/src/serve.py",
-    });
+    resolveTrainingModel(args.model.base_model);
+    const entrypoint = buildBundledPythonCommand("serve.py");
     const modelName = args.model.id;
-    const env = withHuggingFaceCacheEnvironment({
-        ...minimalMachineLearningEnvironment(process.env, { includeHfToken: resolvedModel.requiresHfToken }),
-        ...args.config.evaluation.inference.env,
+    const env = withBundledPythonEnvironment(withOfflineHuggingFaceCacheEnvironment({
+        ...minimalMachineLearningEnvironment(process.env),
         TT_MODEL_ARTIFACT: artifactPath,
         TT_BASE_MODEL: localBaseModelPath ?? args.model.base_model,
         ...(options.baseModelRevision && !localBaseModelPath
             ? { TT_BASE_MODEL_REVISION: options.baseModelRevision }
             : {}),
         TT_MODEL_NAME: modelName,
-        TT_MODEL_LOADER: resolvedModel.loader,
+        TT_MODEL_LOADER: "causal_lm",
         TT_HOST: host,
         TT_PORT: String(port),
         TT_DEVICE: options.device ?? args.config.evaluation.inference.device,
         TT_MAX_TOKENS: String(maxTokens),
         TT_TEMPERATURE: String(temperature),
         TT_TOP_P: String(topP),
-        TT_TRUST_REMOTE_CODE: String(resolvedModel.trustRemoteCode && args.config.evaluation.inference.trustRemoteCode),
+        TT_TRUST_REMOTE_CODE: "false",
         TT_MAX_CONCURRENT_REQUESTS: String(maxConcurrentRequests),
-        TT_CHAT_TEMPLATE_KWARGS: JSON.stringify(args.config.evaluation.inference.chatTemplateKwargs ?? {}),
         ...(options.systemPrompt ? { TT_SYSTEM_PROMPT: options.systemPrompt } : {}),
         ...(apiKey ? { TT_API_KEY: apiKey } : {}),
-    }, args.config.paths.modelCache);
+    }, args.config.paths.modelCache));
     return {
         command: entrypoint.command,
         commandArgs: entrypoint.commandArgs,
         displayCommand: entrypoint.displayCommand,
-        cwd: entrypoint.kind === "uv" ? args.config.evaluation.inference.cwd : undefined,
         env,
         url: `http://${httpHost(host)}:${port}`,
         modelName,
@@ -100,7 +92,6 @@ export function buildLocalModelServerLaunch(args) {
 export async function serveLocalModel(launch) {
     await new Promise((resolveServer, reject) => {
         const child = spawn(launch.command, launch.commandArgs, {
-            cwd: launch.cwd,
             env: launch.env,
             stdio: "inherit",
             detached: process.platform !== "win32",

package/dist/model-server.js.map

@@ -1 +1 @@
-{"version":3,"file":"model-server.js","sourceRoot":"","sources":["../src/model-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAEzC,OAAO,EAAE,iCAAiC,EAAE,+BAA+B,EAAE,MAAM,wBAAwB,CAAC;AAC5G,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AA6B7D,SAAS,iBAAiB,CAAC,GAAW;IACpC,IAAI,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;IACzD,IAAI,sBAAsB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,yDAAyD,GAAG,EAAE,CAAC,CAAC;IAClF,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,aAAa,CAAC,IAAY,EAAE,KAAa,EAAE,OAAe,EAAE,OAAe,EAAE,OAAO,GAAG,KAAK;IACnG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,OAAO,IAAI,KAAK,GAAG,OAAO,IAAI,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;QAC3G,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,oBAAoB,OAAO,QAAQ,OAAO,EAAE,CAAC,CAAC;IACvE,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,OAAO,CAAC;AAC5F,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,IAI3C;IACC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;IACnC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,WAAW,CAAC;IACzC,MAAM,MAAM,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACrC,IAAI,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;IACvF,CAAC;IACD,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACtF,IAAI,OAAO,CAAC,SAAS,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,GAAG,OAAO,CAAC,SAAS,cAAc,CAAC,CAAC;IACtF,IAAI,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,oFAAoF,CAAC,CAAC;IACxG,CAAC;IACD,MAAM,IAAI,GAAG,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IAC1E,MAAM,SAAS,GAAG,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC,SAAS,IAAI,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACtF,MAAM,WAAW,GAAG,aAAa,CAAC,aAAa,EAAE,OAAO,CAAC,WAAW,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACjF,MAAM,IAAI,GAAG,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5D,MAAM,qBAAqB,GAAG,aAAa,CACzC,uBAAuB,EACvB,OAAO,CAAC,qBAAqB,IAAI,CAAC,EAClC,CAAC,EACD,CAAC,EACD,IAAI,CACL,CAAC;IACF,MAAM,YAAY,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAChE,MAAM,qBAAqB,GAAG,OAAO,CAAC,oBAAoB;QACxD,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,oBAAoB,CAAC;QACjD,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,uBAAuB,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS;QACzD,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC;QACtC,CAAC,CAAC,SAAS,CAAC;IACd,IACE,qBAAqB;WAClB,uBAAuB;WACvB,OAAO,CAAC,qBAAqB,CAAC,KAAK,uBAAuB,EAC7D,CAAC;QACD,MAAM,IAAI,KAAK,CACb,8BAA8B,uBAAuB,6DAA6D;cAChH,qBAAqB,CACxB,CAAC;IACJ,CAAC;IACD,MAAM,kBAAkB,GAAG,qBAAqB,IAAI,uBAAuB,CAAC;IAC5E,MAAM,aAAa,GAAG,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAClE,MAAM,UAAU,GAAG,sBAAsB,CAAC;QACxC,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,IAAI,uBAAuB;QAC5E,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG;QACzC,MAAM,EAAE,oCAAoC;KAC7C,CAAC,CAAC;IACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;IAChC,MAAM,GAAG,GAAG,+BAA+B,CAAC;QAC1C,GAAG,iCAAiC,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,aAAa,CAAC,eAAe,EAAE,CAAC;QACpG,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG;QACvC,iBAAiB,EAAE,YAAY;QAC/B,aAAa,EAAE,kBAAkB,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU;QAC1D,GAAG,CAAC,OAAO,CAAC,iBAAiB,IAAI,CAAC,kBAAkB;YAClD,CAAC,CAAC,EAAE,sBAAsB,EAAE,OAAO,CAAC,iBAAiB,EAAE;YACvD,CAAC,CAAC,EAAE,CAAC;QACP,aAAa,EAAE,SAAS;QACxB,eAAe,EAAE,aAAa,CAAC,MAAM;QACrC,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC;QACrB,SAAS,EAAE,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,MAAM;QACpE,aAAa,EAAE,MAAM,CAAC,SAAS,CAAC;QAChC,cAAc,EAAE,MAAM,CAAC,WAAW,CAAC;QACnC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC;QACtB,oBAAoB,EAAE,MAAM,CAAC,aAAa,CAAC,eAAe,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,eAAe,CAAC;QAC/G,0BAA0B,EAAE,MAAM,CAAC,qBAAqB,CAAC;QACzD,uBAAuB,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,kBAAkB,IAAI,EAAE,CAAC;QAClG,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3E,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1C,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IACjC,OAAO;QACL,OAAO,EAAE,UAAU,CAAC,OAAO;QAC3B,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,cAAc,EAAE,UAAU,CAAC,cAAc;QACzC,GAAG,EAAE,UAAU,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS;QAChF,GAAG;QACH,GAAG,EAAE,UAAU,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE;QACvC,SAAS;QACT,YAAY;KACb,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,MAA8B;IAClE,MAAM,IAAI,OAAO,CAAO,CAAC,aAAa,EAAE,MAAM,EAAE,EAAE;QAChD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,WAAW,EAAE;YACtD,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,KAAK,EAAE,SAAS;YAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO;SACvC,CAAC,CAAC;QACH,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,cAA0C,CAAC;QAC/C,MAAM,gBAAgB,GAAG,CAAC,MAAsB,EAAE,EAAE;YAClD,IAAI,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;gBAC9C,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;oBACjC,OAAO;gBACT,CAAC;gBAAC,MAAM,CAAC;oBACP,0DAA0D;gBAC5D,CAAC;YACH,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrB,CAAC,CAAC;QACF,MAAM,IAAI,GAAG,CAAC,MAAsB,EAAE,EAAE;YACtC,IAAI,QAAQ;gBAAE,OAAO;YACrB,QAAQ,GAAG,IAAI,CAAC;YAChB,gBAAgB,CAAC,MAAM,CAAC,CAAC;YACzB,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;YACtE,cAAc,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC,CAAC;QACF,MAAM,QAAQ,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YAClC,IAAI,cAAc;gBAAE,YAAY,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,OAAO,EAAE,CAAC;YACV,MAAM,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YACnC,OAAO,EAAE,CAAC;YACV,IAAI,QAAQ,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC5D,aAAa,EAAE,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,aAAa,EAAE,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,KAAK,CAAC,uCAAuC,IAAI,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;YACjF,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}
+{"version":3,"file":"model-server.js","sourceRoot":"","sources":["../src/model-server.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAEzC,OAAO,EACL,iCAAiC,EACjC,sCAAsC,GACvC,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,OAAO,EACL,yBAAyB,EACzB,4BAA4B,GAC7B,MAAM,qBAAqB,CAAC;AA4B7B,SAAS,iBAAiB,CAAC,GAAW;IACpC,IAAI,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC;QAAE,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;IACzD,IAAI,sBAAsB,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,yDAAyD,GAAG,EAAE,CAAC,CAAC;IAClF,CAAC;IACD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,aAAa,CAAC,IAAY,EAAE,KAAa,EAAE,OAAe,EAAE,OAAe,EAAE,OAAO,GAAG,KAAK;IACnG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,OAAO,IAAI,KAAK,GAAG,OAAO,IAAI,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;QAC3G,MAAM,IAAI,KAAK,CAAC,GAAG,IAAI,oBAAoB,OAAO,QAAQ,OAAO,EAAE,CAAC,CAAC;IACvE,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,cAAc,CAAC,IAAY;IAClC,OAAO,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,WAAW,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,OAAO,CAAC;AAC5F,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;AAC1E,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,IAI3C;IACC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;IACnC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,WAAW,CAAC;IACzC,MAAM,MAAM,GAAG,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACrC,IAAI,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,mEAAmE,CAAC,CAAC;IACvF,CAAC;IACD,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACtF,IAAI,OAAO,CAAC,SAAS,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,GAAG,OAAO,CAAC,SAAS,cAAc,CAAC,CAAC;IACtF,IAAI,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,oFAAoF,CAAC,CAAC;IACxG,CAAC;IACD,MAAM,IAAI,GAAG,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IAC1E,MAAM,SAAS,GAAG,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC,SAAS,IAAI,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACtF,MAAM,WAAW,GAAG,aAAa,CAAC,aAAa,EAAE,OAAO,CAAC,WAAW,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACjF,MAAM,IAAI,GAAG,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5D,MAAM,qBAAqB,GAAG,aAAa,CACzC,uBAAuB,EACvB,OAAO,CAAC,qBAAqB,IAAI,CAAC,EAClC,CAAC,EACD,CAAC,EACD,IAAI,CACL,CAAC;IACF,MAAM,YAAY,GAAG,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;IAChE,MAAM,qBAAqB,GAAG,OAAO,CAAC,oBAAoB;QACxD,CAAC,CAAC,iBAAiB,CAAC,OAAO,CAAC,oBAAoB,CAAC;QACjD,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,uBAAuB,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS;QACzD,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC;QACtC,CAAC,CAAC,SAAS,CAAC;IACd,IACE,qBAAqB;WAClB,uBAAuB;WACvB,OAAO,CAAC,qBAAqB,CAAC,KAAK,uBAAuB,EAC7D,CAAC;QACD,MAAM,IAAI,KAAK,CACb,8BAA8B,uBAAuB,6DAA6D;cAChH,qBAAqB,CACxB,CAAC;IACJ,CAAC;IACD,MAAM,kBAAkB,GAAG,qBAAqB,IAAI,uBAAuB,CAAC;IAC5E,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAC5C,MAAM,UAAU,GAAG,yBAAyB,CAAC,UAAU,CAAC,CAAC;IACzD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;IAChC,MAAM,GAAG,GAAG,4BAA4B,CACtC,sCAAsC,CAAC;QACrC,GAAG,iCAAiC,CAAC,OAAO,CAAC,GAAG,CAAC;QACjD,iBAAiB,EAAE,YAAY;QAC/B,aAAa,EAAE,kBAAkB,IAAI,IAAI,CAAC,KAAK,CAAC,UAAU;QAC1D,GAAG,CAAC,OAAO,CAAC,iBAAiB,IAAI,CAAC,kBAAkB;YAClD,CAAC,CAAC,EAAE,sBAAsB,EAAE,OAAO,CAAC,iBAAiB,EAAE;YACvD,CAAC,CAAC,EAAE,CAAC;QACP,aAAa,EAAE,SAAS;QACxB,eAAe,EAAE,WAAW;QAC5B,OAAO,EAAE,IAAI;QACb,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC;QACrB,SAAS,EAAE,OAAO,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,MAAM;QACpE,aAAa,EAAE,MAAM,CAAC,SAAS,CAAC;QAChC,cAAc,EAAE,MAAM,CAAC,WAAW,CAAC;QACnC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC;QACtB,oBAAoB,EAAE,OAAO;QAC7B,0BAA0B,EAAE,MAAM,CAAC,qBAAqB,CAAC;QACzD,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3E,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC1C,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CACjC,CAAC;IACF,OAAO;QACL,OAAO,EAAE,UAAU,CAAC,OAAO;QAC3B,WAAW,EAAE,UAAU,CAAC,WAAW;QACnC,cAAc,EAAE,UAAU,CAAC,cAAc;QACzC,GAAG;QACH,GAAG,EAAE,UAAU,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE;QACvC,SAAS;QACT,YAAY;KACb,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,MAA8B;IAClE,MAAM,IAAI,OAAO,CAAO,CAAC,aAAa,EAAE,MAAM,EAAE,EAAE;QAChD,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,WAAW,EAAE;YACtD,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,KAAK,EAAE,SAAS;YAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ,KAAK,OAAO;SACvC,CAAC,CAAC;QACH,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,cAA0C,CAAC;QAC/C,MAAM,gBAAgB,GAAG,CAAC,MAAsB,EAAE,EAAE;YAClD,IAAI,KAAK,CAAC,GAAG,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;gBAC9C,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;oBACjC,OAAO;gBACT,CAAC;gBAAC,MAAM,CAAC;oBACP,0DAA0D;gBAC5D,CAAC;YACH,CAAC;YACD,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrB,CAAC,CAAC;QACF,MAAM,IAAI,GAAG,CAAC,MAAsB,EAAE,EAAE;YACtC,IAAI,QAAQ;gBAAE,OAAO;YACrB,QAAQ,GAAG,IAAI,CAAC;YAChB,gBAAgB,CAAC,MAAM,CAAC,CAAC;YACzB,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,CAAC;YACtE,cAAc,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC,CAAC;QACF,MAAM,QAAQ,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAChC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;YAClC,IAAI,cAAc;gBAAE,YAAY,CAAC,cAAc,CAAC,CAAC;QACnD,CAAC,CAAC;QACF,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACjC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;QACnC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;YAC5B,OAAO,EAAE,CAAC;YACV,MAAM,CAAC,KAAK,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YACnC,OAAO,EAAE,CAAC;YACV,IAAI,QAAQ,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC5D,aAAa,EAAE,CAAC;YAClB,CAAC;iBAAM,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,aAAa,EAAE,CAAC;YAClB,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,KAAK,CAAC,uCAAuC,IAAI,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;YACjF,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC"}

package/dist/orchestrator.d.ts

@@ -1,41 +1,27 @@
 import { type FineTuneRunRequest, type LocalRunnerConfig, type RunReport } from "./contracts.js";
 import type { LocalRunReporter } from "./run-reporter.js";
-export type LocalRunStage = "prepare" | "baseline" | "train" | "candidate" | "score" | "report" | "all";
 export interface LocalRunResult {
     request: FineTuneRunRequest;
     report: RunReport;
     reportPath: string;
     artifactDir: string;
 }
-export interface LocalStageRunResult {
-    request: FineTuneRunRequest;
-    stage: LocalRunStage;
-    report?: RunReport;
-    reportPath?: string;
-    artifactDir: string;
-    artifacts: {
-        training_jsonl: string;
-        stage_metadata: string;
-        training_report: string;
-        baseline_eval: string;
-        candidate_eval: string;
-        report: string;
-        artifact_manifest: string;
-    };
-}
 export declare function loadJsonFile<T>(path: string): Promise<T>;
 export declare function loadRunRequest(path: string): Promise<FineTuneRunRequest>;
 export declare function loadLocalRunnerConfig(path?: string): Promise<LocalRunnerConfig>;
-/** Hash a local base model while permitting Hugging Face snapshot file links. */
-export declare function fingerprintLocalBaseModel(uri: string): Promise<string>;
-export declare function runLocalFineTuneStage(input: {
+/**
+ * Parses the public contracts and fully validates every dataset before a run
+ * lock, store record, or artifact-directory claim is created.
+ */
+export declare function validateLocalFineTuneInput(input: {
+    request: unknown;
+    config: unknown;
+}): Promise<{
     request: FineTuneRunRequest;
     config: LocalRunnerConfig;
-    reporter?: LocalRunReporter;
-    stage?: LocalRunStage;
-    force?: boolean;
-    modelArtifact?: string;
-}): Promise<LocalStageRunResult>;
+}>;
+/** Hash a local base model while permitting Hugging Face snapshot file links. */
+export declare function fingerprintLocalBaseModel(uri: string): Promise<string>;
 export declare function runLocalFineTune(input: {
     request: FineTuneRunRequest;
     config: LocalRunnerConfig;