@simulatte/doppler 0.1.4 → 0.1.6

package/src/index-browser.js

@@ -1,5 +1,5 @@
-export const DOPPLER_VERSION = '0.1.0';
-export { doppler } from './client/doppler-api.js';
+export { DOPPLER_VERSION } from './version.js';
+export { doppler } from './client/doppler-api.browser.js';
 
 // Core loaders
 export {

package/src/index.js

@@ -1,4 +1,4 @@
-export const DOPPLER_VERSION = '0.1.0';
+export { DOPPLER_VERSION } from './version.js';
 export { doppler } from './client/doppler-api.js';
 
 // Core loaders

package/src/inference/browser-harness.js

@@ -15,10 +15,16 @@ import {
   getActiveKernelPathSource,
   getActiveKernelPathPolicy,
 } from '../config/kernel-path-loader.js';
-import { selectRuleValue } from '../rules/rule-registry.js';
+import {
+  getInferenceLayerPatternContractArtifact,
+  selectRuleValue,
+} from '../rules/rule-registry.js';
 import { mergeRuntimeValues } from '../config/runtime-merge.js';
 import { isPlainObject } from '../utils/plain-object.js';
+import { validateBrowserSuiteMetrics } from '../config/schema/browser-suite-metrics.schema.js';
 import { validateTrainingMetricsReport } from '../config/schema/training-metrics.schema.js';
+import { buildExecutionContractArtifact } from '../config/execution-contract-check.js';
+import { buildManifestRequiredInferenceFieldsArtifact } from '../config/required-inference-fields-contract-check.js';
 
 const TRAINING_SUITE_MODULE_PATH = '../training/suite.js';
 const NODE_SOURCE_RUNTIME_MODULE_PATH = '../tooling/node-source-runtime.js';
@@ -41,6 +47,29 @@ async function runTrainingBenchSuite(options = {}) {
   return module.runTrainingBenchSuite(options);
 }
 
+function buildSuiteContractMetrics(suite, baseMetrics, manifest) {
+  const executionContractArtifact = buildExecutionContractArtifact(manifest);
+  const executionV0GraphContractArtifact = executionContractArtifact?.executionV0?.graph ?? null;
+  const layerPatternContractArtifact = getInferenceLayerPatternContractArtifact();
+  const requiredInferenceFieldsArtifact = manifest?.modelType === 'transformer'
+    && isPlainObject(manifest?.inference?.attention)
+    ? buildManifestRequiredInferenceFieldsArtifact(
+      manifest?.inference ?? null,
+      `${manifest?.modelId ?? 'unknown'}.inference`
+    )
+    : null;
+  return validateBrowserSuiteMetrics({
+    ...baseMetrics,
+    schemaVersion: 1,
+    source: 'doppler',
+    suite,
+    ...(executionContractArtifact ? { executionContractArtifact } : {}),
+    executionV0GraphContractArtifact,
+    layerPatternContractArtifact,
+    requiredInferenceFieldsArtifact,
+  });
+}
+
 function parseReportTimestamp(rawTimestamp, label = 'timestamp') {
   if (rawTimestamp == null) {
     return null;
@@ -900,6 +929,9 @@ async function resolveHarnessOverride(options = {}) {
 
 async function initializeSuiteModel(options = {}) {
   if (options.harnessOverride) {
+    if (options.runtime?.runtimeConfig) {
+      setRuntimeConfig(options.runtime.runtimeConfig);
+    }
     return resolveHarnessOverride(options);
   }
   const loadStart = performance.now();
@@ -959,6 +991,14 @@ async function runKernelSuite(options = {}) {
 
 const DEFAULT_HARNESS_PROMPT = 'Summarize this input in one sentence.';
 const DEFAULT_RUNTIME_PLACEHOLDER_PROMPT = 'Hello from Doppler.';
+const DEFAULT_QWEN_PROMPT = Object.freeze({
+  messages: Object.freeze([
+    Object.freeze({
+      role: 'user',
+      content: 'Answer in one short sentence: What color is the sky on a clear day?',
+    }),
+  ]),
+});
 const DEFAULT_TRANSLATEGEMMA_PROMPT = Object.freeze({
   messages: Object.freeze([
     Object.freeze({
@@ -1244,6 +1284,9 @@ function resolvePromptTemplateType(source) {
 }
 
 function buildDefaultGenerationPrompt(templateType) {
+  if (templateType === 'qwen') {
+    return clonePromptInput(DEFAULT_QWEN_PROMPT);
+  }
   if (templateType === 'translategemma') {
     return clonePromptInput(DEFAULT_TRANSLATEGEMMA_PROMPT);
   }
@@ -1251,7 +1294,7 @@ function buildDefaultGenerationPrompt(templateType) {
 }
 
 function shouldPreferModelDefaultPrompt(runtimePrompt, templateType) {
-  if (templateType !== 'translategemma') {
+  if (templateType !== 'translategemma' && templateType !== 'qwen') {
     return false;
   }
   if (typeof runtimePrompt !== 'string') {
@@ -1260,6 +1303,31 @@ function shouldPreferModelDefaultPrompt(runtimePrompt, templateType) {
   return runtimePrompt.trim() === DEFAULT_RUNTIME_PLACEHOLDER_PROMPT;
 }
 
+function assertPromptContract(runtimePrompt, templateType, source = 'runtime.inference.prompt') {
+  if (templateType !== 'translategemma') {
+    return;
+  }
+  if (runtimePrompt === undefined || runtimePrompt === null) {
+    return;
+  }
+  if (typeof runtimePrompt === 'string') {
+    const trimmed = runtimePrompt.trim();
+    if (!trimmed || trimmed === DEFAULT_RUNTIME_PLACEHOLDER_PROMPT) {
+      return;
+    }
+    throw new Error(
+      `TranslateGemma harness prompt contract violation: ${source} must be ` +
+      '{ messages: [...] } with source_lang_code/target_lang_code blocks, not a plain string.'
+    );
+  }
+  if (!isStructuredPromptInput(runtimePrompt)) {
+    throw new Error(
+      `TranslateGemma harness prompt contract violation: ${source} must be ` +
+      '{ messages: [...] } with source_lang_code/target_lang_code blocks.'
+    );
+  }
+}
+
 function describePromptInput(promptInput) {
   if (typeof promptInput === 'string') {
     return promptInput.trim() || DEFAULT_HARNESS_PROMPT;
@@ -1276,6 +1344,11 @@ function describePromptInput(promptInput) {
   if (sourceLang && targetLang) {
     return `${sourceLang} -> ${targetLang}: ${text || '[non-text request]'}`;
   }
+  const stringContent = asText(firstMessage?.content);
+  if (stringContent) {
+    const role = asText(firstMessage?.role) || 'user';
+    return `${role}: ${stringContent}`;
+  }
   try {
     return JSON.stringify(promptInput);
   } catch {
@@ -1286,6 +1359,7 @@ function describePromptInput(promptInput) {
 function resolveGenerationPromptInput(runtimeConfig, runOverrides = null, source = null) {
   const templateType = resolvePromptTemplateType(source);
   const overridePrompt = runOverrides?.prompt;
+  assertPromptContract(overridePrompt, templateType, 'runOverrides.prompt');
   if (typeof overridePrompt === 'string' && overridePrompt.trim()) {
     return overridePrompt.trim();
   }
@@ -1294,6 +1368,7 @@ function resolveGenerationPromptInput(runtimeConfig, runOverrides = null, source
   }
 
   const runtimePrompt = runtimeConfig?.inference?.prompt;
+  assertPromptContract(runtimePrompt, templateType, 'runtimeConfig.inference.prompt');
   if (shouldPreferModelDefaultPrompt(runtimePrompt, templateType)) {
     return buildDefaultGenerationPrompt(templateType);
   }
@@ -1824,6 +1899,11 @@ async function runInferenceSuite(options = {}) {
     source: 'doppler',
     prefillSemantics: 'internal_prefill_phase',
   });
+  const metricsWithContracts = buildSuiteContractMetrics(
+    options.suiteName || 'inference',
+    metrics,
+    harness.manifest
+  );
   return {
     ...summary,
     modelId: options.modelId || harness.manifest?.modelId || 'unknown',
@@ -1841,7 +1921,7 @@ async function runInferenceSuite(options = {}) {
     timing,
     timingDiagnostics,
     output,
-    metrics,
+    metrics: metricsWithContracts,
     memoryStats,
     deviceInfo: resolveDeviceInfo(),
     pipeline: options.keepPipeline ? harness.pipeline : null,
@@ -2218,6 +2298,7 @@ async function runBenchSuite(options = {}) {
     source: 'doppler',
     prefillSemantics: 'internal_prefill_phase',
   });
+  const metricsWithContracts = buildSuiteContractMetrics('bench', metrics, harness.manifest);
   return {
     ...summary,
     modelId: options.modelId || harness.manifest?.modelId || 'unknown',
@@ -2235,7 +2316,7 @@ async function runBenchSuite(options = {}) {
     timing,
     timingDiagnostics,
     output,
-    metrics,
+    metrics: metricsWithContracts,
     memoryStats,
     deviceInfo: resolveDeviceInfo(),
     pipeline: options.keepPipeline ? harness.pipeline : null,
@@ -2396,25 +2477,9 @@ async function runDiffusionSuite(options = {}) {
     source: 'doppler',
     prefillSemantics: 'internal_prefill_phase',
   });
-
-  return {
-    ...summary,
-    modelId: options.modelId || harness.manifest?.modelId || 'unknown',
-    cacheMode,
-    loadMode,
-    env: {
-      library: 'doppler',
-      runtime: 'browser',
-      device: 'webgpu',
-      browserUserAgent: typeof navigator !== 'undefined' ? (navigator.userAgent || null) : null,
-      browserPlatform: typeof navigator !== 'undefined' ? (navigator.platform || null) : null,
-      browserLanguage: typeof navigator !== 'undefined' ? (navigator.language || null) : null,
-      browserVendor: typeof navigator !== 'undefined' ? (navigator.vendor || null) : null,
-    },
-    timing,
-    timingDiagnostics,
-    output,
-    metrics: {
+  const metricsWithContracts = buildSuiteContractMetrics(
+    'diffusion',
+    {
       warmupRuns,
       timedRuns,
       width,
@@ -2439,6 +2504,27 @@ async function runDiffusionSuite(options = {}) {
       gpu: gpuStats,
       performanceArtifact: diffusionPerformanceArtifact,
     },
+    harness.manifest
+  );
+
+  return {
+    ...summary,
+    modelId: options.modelId || harness.manifest?.modelId || 'unknown',
+    cacheMode,
+    loadMode,
+    env: {
+      library: 'doppler',
+      runtime: 'browser',
+      device: 'webgpu',
+      browserUserAgent: typeof navigator !== 'undefined' ? (navigator.userAgent || null) : null,
+      browserPlatform: typeof navigator !== 'undefined' ? (navigator.platform || null) : null,
+      browserLanguage: typeof navigator !== 'undefined' ? (navigator.language || null) : null,
+      browserVendor: typeof navigator !== 'undefined' ? (navigator.vendor || null) : null,
+    },
+    timing,
+    timingDiagnostics,
+    output,
+    metrics: metricsWithContracts,
     memoryStats,
     deviceInfo: resolveDeviceInfo(),
     pipeline: options.keepPipeline ? harness.pipeline : null,

package/src/inference/pipelines/diffusion/init.js

@@ -1,5 +1,7 @@
 import { DEFAULT_DIFFUSION_CONFIG } from '../../../config/schema/index.js';
 
+const SUPPORTED_DIFFUSION_RUNTIME_LAYOUTS = new Set(['sd3', 'flux', 'sana']);
+
 function mergeSection(base, override) {
   if (!override) return { ...base };
   return { ...base, ...override };
@@ -38,6 +40,9 @@ function resolveSchedulerType(modelScheduler, runtimeScheduler) {
   if (modelClass === 'FlowMatchEulerDiscreteScheduler') {
     return 'flowmatch_euler';
   }
+  if (modelClass === 'SCMScheduler') {
+    return 'scm';
+  }
   if (modelClass === 'EulerDiscreteScheduler') {
     return 'euler';
   }
@@ -58,6 +63,8 @@ function mergeSchedulerConfig(modelConfig, runtimeScheduler) {
     type,
     numTrainTimesteps: modelScheduler.num_train_timesteps ?? runtimeScheduler.numTrainTimesteps,
     shift: modelScheduler.shift ?? runtimeScheduler.shift,
+    predictionType: modelScheduler.prediction_type ?? runtimeScheduler.predictionType,
+    sigmaData: modelScheduler.sigma_data ?? runtimeScheduler.sigmaData,
   };
 }
 
@@ -95,6 +102,13 @@ export function initializeDiffusion(manifest, runtimeConfig) {
     }
     throw new Error('Diffusion manifest missing config.diffusion model contract.');
   }
+  const layout = modelConfig.layout;
+  if (layout && !SUPPORTED_DIFFUSION_RUNTIME_LAYOUTS.has(layout)) {
+    throw new Error(
+      `Diffusion layout "${layout}" is recognized in the manifest, but the GPU runtime is not implemented yet. ` +
+      'Supported runtime layouts: sd3, flux, sana.'
+    );
+  }
 
   const runtimeBase = mergeDiffusionConfig(DEFAULT_DIFFUSION_CONFIG, runtimeConfig?.inference?.diffusion);
   const runtime = {