@simulatte/doppler 0.1.6 → 0.1.7

package/src/tooling/command-runner-shared.js

@@ -1,5 +1,5 @@
 import { buildRuntimeContractPatch } from './command-api.js';
-import { mergeRuntimeValues } from '../config/runtime-merge.js';
+import { applyOrderedRuntimeInputs } from './runtime-input-composition.js';
 
 function cloneRuntimeConfig(runtimeConfig) {
   if (runtimeConfig == null) return runtimeConfig;
@@ -9,6 +9,46 @@ function cloneRuntimeConfig(runtimeConfig) {
   return JSON.parse(JSON.stringify(runtimeConfig));
 }
 
+function assertCalibrateRuntimeCompatibility(request, runtimeConfig) {
+  if (request?.intent !== 'calibrate') {
+    return;
+  }
+
+  const shared = runtimeConfig?.shared ?? {};
+  const debug = shared.debug ?? {};
+  const benchmarkRun = shared.benchmark?.run ?? {};
+  const violations = [];
+
+  if (debug.trace?.enabled === true) {
+    violations.push('runtime.shared.debug.trace.enabled');
+  }
+  if (debug.pipeline?.enabled === true) {
+    violations.push('runtime.shared.debug.pipeline.enabled');
+  }
+  if (Array.isArray(debug.probes) && debug.probes.length > 0) {
+    violations.push('runtime.shared.debug.probes');
+  }
+  if (debug.profiler?.enabled === true) {
+    violations.push('runtime.shared.debug.profiler.enabled');
+  }
+  if (benchmarkRun.debug === true) {
+    violations.push('runtime.shared.benchmark.run.debug');
+  }
+  if (benchmarkRun.profile === true) {
+    violations.push('runtime.shared.benchmark.run.profile');
+  }
+  if (benchmarkRun.captureMemoryTimeSeries === true) {
+    violations.push('runtime.shared.benchmark.run.captureMemoryTimeSeries');
+  }
+
+  if (violations.length > 0) {
+    throw new Error(
+      `tooling command: calibrate intent forbids investigation instrumentation (${violations.join(', ')}). ` +
+      'Disable those runtime config fields or use the debug command instead.'
+    );
+  }
+}
+
 function resetRuntimeState(runtimeBridge) {
   if (!runtimeBridge?.setRuntimeConfig) {
     throw new Error('runtime bridge must provide setRuntimeConfig().');
@@ -22,12 +62,6 @@ function resetRuntimeState(runtimeBridge) {
   runtimeBridge.setRuntimeConfig(null);
 }
 
-function mergeRuntimePatch(runtimeBridge, patch) {
-  if (!patch) return;
-  const mergedRuntime = mergeRuntimeValues(runtimeBridge.getRuntimeConfig(), patch);
-  runtimeBridge.setRuntimeConfig(mergedRuntime);
-}
-
 function snapshotRuntimeState(runtimeBridge) {
   return {
     runtimeConfig: cloneRuntimeConfig(runtimeBridge.getRuntimeConfig()),
@@ -73,17 +107,17 @@ function restoreRuntimeState(runtimeBridge, snapshot) {
 
 export async function applyRuntimeInputs(request, runtimeBridge, options = {}) {
   resetRuntimeState(runtimeBridge);
-
-  if (request.runtimePreset) {
-    await runtimeBridge.applyRuntimePreset(request.runtimePreset, options);
-  }
-
-  if (request.runtimeConfigUrl) {
-    await runtimeBridge.applyRuntimeConfigFromUrl(request.runtimeConfigUrl, options);
-  }
-
-  mergeRuntimePatch(runtimeBridge, request.runtimeConfig);
-  mergeRuntimePatch(runtimeBridge, buildRuntimeContractPatch(request));
+  await applyOrderedRuntimeInputs(runtimeBridge, {
+    configChain: request.configChain ?? null,
+    runtimePreset: request.runtimePreset ?? null,
+    runtimeConfigUrl: request.runtimeConfigUrl ?? null,
+    runtimeConfig: request.runtimeConfig ?? null,
+    runtimeContractPatch: buildRuntimeContractPatch(request),
+  }, {
+    applyRuntimePreset: runtimeBridge.applyRuntimePreset?.bind(runtimeBridge),
+    applyRuntimeConfigFromUrl: runtimeBridge.applyRuntimeConfigFromUrl?.bind(runtimeBridge),
+  }, options);
+  assertCalibrateRuntimeCompatibility(request, runtimeBridge.getRuntimeConfig());
 }
 
 export async function runWithRuntimeIsolation(runtimeBridge, run) {

package/src/tooling/lean-execution-contract.js

@@ -1,12 +1,159 @@
 import {
-  extractExecutionContractFacts,
   sanitizeLeanModuleName,
 } from '../config/execution-contract-check.js';
+import { DEFAULT_BATCHING_DEFAULTS, DEFAULT_GENERATION_CONFIG } from '../config/schema/inference-defaults.schema.js';
+import { DEFAULT_KVCACHE_CONFIG } from '../config/schema/kvcache.schema.js';
 
 export {
-  extractExecutionContractFacts,
   sanitizeLeanModuleName,
-} from '../config/execution-contract-check.js';
+};
+
+const KV_LAYOUTS = new Set(['contiguous', 'paged', 'tiered', 'bdpa', 'bdpa_paged']);
+const PHASES = new Set(['prefill', 'decode', 'both']);
+const COLD_QUANT_MODES = new Set(['none', 'int8', 'int4']);
+const ATTENTION_OPS = new Set(['attention']);
+const EMBED_OPS = new Set(['embed', 'gather']);
+const SAMPLE_OPS = new Set(['sample']);
+
+function isPlainObject(value) {
+  return value != null && typeof value === 'object' && !Array.isArray(value);
+}
+
+function assertManifestObject(manifest) {
+  if (!isPlainObject(manifest)) {
+    throw new Error('lean execution contract: manifest must be an object.');
+  }
+  return manifest;
+}
+
+function assertPositiveInteger(value, label) {
+  if (!Number.isInteger(value) || value < 0) {
+    throw new Error(`lean execution contract: ${label} must be a non-negative integer.`);
+  }
+  return value;
+}
+
+function assertPositiveIntegerOrDefault(value, fallback, label) {
+  if (value == null) {
+    return assertPositiveInteger(fallback, `${label} fallback`);
+  }
+  return assertPositiveInteger(value, label);
+}
+
+function normalizeKVLayout(value) {
+  const normalized = String(value ?? DEFAULT_KVCACHE_CONFIG.layout).trim().toLowerCase();
+  if (!KV_LAYOUTS.has(normalized)) {
+    throw new Error(
+      `lean execution contract: unsupported KV layout "${value}". ` +
+      `Expected one of ${[...KV_LAYOUTS].join(', ')}.`
+    );
+  }
+  return normalized;
+}
+
+function normalizePhase(value, label) {
+  const normalized = String(value ?? '').trim().toLowerCase();
+  if (!PHASES.has(normalized)) {
+    throw new Error(
+      `lean execution contract: ${label} must be one of ${[...PHASES].join(', ')}.`
+    );
+  }
+  return normalized;
+}
+
+function normalizeColdQuantMode(kvcache) {
+  const tieringMode = String(
+    kvcache?.tiering?.mode
+      ?? DEFAULT_KVCACHE_CONFIG.tiering.mode
+  ).trim().toLowerCase();
+  if (tieringMode === 'off' || tieringMode === 'fp16') {
+    return 'none';
+  }
+  if (!COLD_QUANT_MODES.has(tieringMode)) {
+    throw new Error(
+      `lean execution contract: unsupported tiered cold quant mode "${tieringMode}".`
+    );
+  }
+  return tieringMode;
+}
+
+function classifyOp(op) {
+  const normalized = String(op ?? '').trim().toLowerCase();
+  if (!normalized) {
+    throw new Error('lean execution contract: execution step op is required.');
+  }
+  if (ATTENTION_OPS.has(normalized)) return 'attention';
+  if (EMBED_OPS.has(normalized)) return 'embed';
+  if (SAMPLE_OPS.has(normalized)) return 'sample';
+  if (normalized.includes('norm')) return 'norm';
+  if (normalized.includes('residual')) return 'residual';
+  if (normalized.endsWith('_proj') || normalized.startsWith('rope_') || normalized === 'activation') {
+    return 'projection';
+  }
+  return 'other';
+}
+
+export function extractExecutionContractFacts(manifest) {
+  const normalizedManifest = assertManifestObject(manifest);
+  const modelId = String(normalizedManifest.modelId ?? 'model').trim() || 'model';
+  const architecture = isPlainObject(normalizedManifest.architecture)
+    ? normalizedManifest.architecture
+    : {};
+  const inference = isPlainObject(normalizedManifest.inference)
+    ? normalizedManifest.inference
+    : {};
+  const sessionDefaults = isPlainObject(inference.sessionDefaults)
+    ? inference.sessionDefaults
+    : {};
+  const execution = isPlainObject(inference.execution)
+    ? inference.execution
+    : {};
+  const kvcache = isPlainObject(sessionDefaults.kvcache)
+    ? sessionDefaults.kvcache
+    : {};
+  const decodeLoop = isPlainObject(sessionDefaults.decodeLoop)
+    ? sessionDefaults.decodeLoop
+    : {};
+
+  const steps = Array.isArray(execution.steps)
+    ? execution.steps.map((step, index) => {
+      if (!isPlainObject(step)) {
+        throw new Error(`lean execution contract: execution.steps[${index}] must be an object.`);
+      }
+      const id = String(step.id ?? '').trim();
+      if (!id) {
+        throw new Error(`lean execution contract: execution.steps[${index}].id is required.`);
+      }
+      return {
+        id,
+        phase: normalizePhase(step.phase, `execution.steps[${index}].phase`),
+        opClass: classifyOp(step.op),
+      };
+    })
+    : [];
+
+  return {
+    modelId,
+    session: {
+      layout: normalizeKVLayout(kvcache.layout),
+      disableCommandBatching: decodeLoop.disableCommandBatching
+        ?? DEFAULT_GENERATION_CONFIG.disableCommandBatching,
+      decodeBatchSize: assertPositiveIntegerOrDefault(
+        decodeLoop.batchSize,
+        DEFAULT_BATCHING_DEFAULTS.batchSize,
+        'sessionDefaults.decodeLoop.batchSize'
+      ),
+      headDim: assertPositiveInteger(architecture.headDim, 'architecture.headDim'),
+      kvLen: assertPositiveIntegerOrDefault(
+        architecture.maxSeqLen ?? kvcache.maxSeqLen,
+        DEFAULT_KVCACHE_CONFIG.maxSeqLen,
+        'architecture.maxSeqLen'
+      ),
+      coldQuantMode: normalizeColdQuantMode(kvcache),
+    },
+    steps,
+  };
+}
 
 function asLeanString(value) {
   return JSON.stringify(String(value));