@ifc-lite/viewer 1.14.2 → 1.14.4

package/src/lib/llm/models.ts

@@ -0,0 +1,197 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+/**
+ * LLM model registry.
+ * Model IDs are sourced from environment variables only.
+ * Pro model cost metadata comes from cost-bucket env vars:
+ * - *_PRO_MODELS_LOW => $
+ * - *_PRO_MODELS_MEDIUM => $$
+ * - *_PRO_MODELS_HIGH => $$$
+ */
+
+import type { LLMModel } from './types.js';
+
+function readEnv(key: string): string | undefined {
+  const importMetaEnv = (import.meta as unknown as { env?: Record<string, unknown> }).env;
+  const viteVal = importMetaEnv?.[key];
+  const nodeVal = typeof process !== 'undefined' ? process.env[key] : undefined;
+  const val = typeof viteVal === 'string' ? viteVal : nodeVal;
+  if (typeof val !== 'string') return undefined;
+  const trimmed = val.trim();
+  return trimmed.length > 0 ? trimmed : undefined;
+}
+
+function parseCsvEnv(key: string): string[] {
+  const raw = readEnv(key);
+  if (!raw) return [];
+  return raw
+    .split(',')
+    .map((part) => part.trim())
+    .filter(Boolean);
+}
+
+function parseCsvFromFirstDefined(keys: string[]): string[] {
+  for (const key of keys) {
+    const values = parseCsvEnv(key);
+    if (values.length > 0) return values;
+  }
+  return [];
+}
+
+function uniqueInOrder(ids: string[]): string[] {
+  const out: string[] = [];
+  const seen = new Set<string>();
+  for (const id of ids) {
+    if (!seen.has(id)) {
+      seen.add(id);
+      out.push(id);
+    }
+  }
+  return out;
+}
+
+function titleCaseProvider(rawProvider: string): string {
+  const overrides: Record<string, string> = {
+    openai: 'OpenAI',
+    anthropic: 'Anthropic',
+    google: 'Google',
+    meta: 'Meta',
+    'meta-llama': 'Meta',
+    xai: 'xAI',
+    'x-ai': 'xAI',
+    mistralai: 'Mistral',
+    qwen: 'Alibaba',
+    deepseek: 'DeepSeek',
+    minimax: 'MiniMax',
+    'z-ai': 'Zhipu',
+  };
+
+  const normalized = rawProvider.toLowerCase();
+  if (overrides[normalized]) return overrides[normalized];
+  return rawProvider
+    .split(/[-_]/)
+    .filter(Boolean)
+    .map((part) => part.charAt(0).toUpperCase() + part.slice(1))
+    .join(' ');
+}
+
+function humanizeModelSlug(slug: string): string {
+  const withoutTier = slug.split(':')[0] ?? slug;
+  return withoutTier
+    .replace(/[._-]+/g, ' ')
+    .split(' ')
+    .filter(Boolean)
+    .map((word) => {
+      if (/^[0-9.]+$/.test(word)) return word;
+      const upper = word.toUpperCase();
+      if (upper === 'GPT' || upper === 'OSS' || upper === 'R1') return upper;
+      if (word.length <= 2) return upper;
+      return word.charAt(0).toUpperCase() + word.slice(1);
+    })
+    .join(' ');
+}
+
+function buildModel(id: string, tier: 'free' | 'pro', cost?: LLMModel['cost']): LLMModel {
+  const [providerRaw, modelRaw = id] = id.split('/');
+  return {
+    id,
+    tier,
+    name: humanizeModelSlug(modelRaw),
+    provider: titleCaseProvider(providerRaw ?? 'Unknown'),
+    contextWindow: 128_000,
+    supportsImages: false,
+    supportsFileAttachments: true,
+    cost: tier === 'pro' ? cost : undefined,
+  };
+}
+
+const freeModelIds = uniqueInOrder(parseCsvFromFirstDefined(['VITE_LLM_FREE_MODELS', 'LLM_FREE_MODELS']));
+const proLowCostIds = uniqueInOrder(parseCsvFromFirstDefined(['VITE_LLM_PRO_MODELS_LOW', 'LLM_PRO_MODELS_LOW']));
+const proMediumCostIds = uniqueInOrder(parseCsvFromFirstDefined(['VITE_LLM_PRO_MODELS_MEDIUM', 'LLM_PRO_MODELS_MEDIUM']));
+const proHighCostIds = uniqueInOrder(parseCsvFromFirstDefined(['VITE_LLM_PRO_MODELS_HIGH', 'LLM_PRO_MODELS_HIGH']));
+
+// Backward-compatible fallback for older env shape with one pro list.
+const legacyProIds = uniqueInOrder(parseCsvFromFirstDefined(['VITE_LLM_PRO_MODELS', 'LLM_PRO_MODELS']));
+const useLegacyProList = proLowCostIds.length === 0 && proMediumCostIds.length === 0 && proHighCostIds.length === 0;
+
+const rawFreeModels: LLMModel[] = freeModelIds.map((id) => buildModel(id, 'free'));
+
+const proCostBuckets: Array<{ ids: string[]; cost: LLMModel['cost'] }> = [
+  { ids: proLowCostIds, cost: '$' },
+  { ids: useLegacyProList ? legacyProIds : proMediumCostIds, cost: '$$' },
+  { ids: proHighCostIds, cost: '$$$' },
+];
+
+const seenProModelIds = new Set<string>();
+const rawProModels: LLMModel[] = proCostBuckets.flatMap(({ ids, cost }) =>
+  ids.flatMap((id) => {
+    if (seenProModelIds.has(id)) return [];
+    seenProModelIds.add(id);
+    return [buildModel(id, 'pro', cost)];
+  }),
+);
+
+const imageCapableModelIds = new Set(
+  uniqueInOrder(parseCsvFromFirstDefined(['VITE_LLM_IMAGE_MODELS', 'LLM_IMAGE_MODELS'])),
+);
+const fileCapableModelIds = new Set(
+  uniqueInOrder(parseCsvFromFirstDefined(['VITE_LLM_FILE_ATTACHMENT_MODELS', 'LLM_FILE_ATTACHMENT_MODELS'])),
+);
+const hasImageOverrideList = imageCapableModelIds.size > 0;
+const hasFileOverrideList = fileCapableModelIds.size > 0;
+
+function applyCapabilities(model: LLMModel): LLMModel {
+  const supportsImages = hasImageOverrideList ? imageCapableModelIds.has(model.id) : model.supportsImages;
+  const supportsFileAttachments = hasFileOverrideList
+    ? fileCapableModelIds.has(model.id)
+    : model.supportsFileAttachments;
+  return {
+    ...model,
+    supportsImages,
+    supportsFileAttachments,
+  };
+}
+
+export const FREE_MODELS: LLMModel[] = rawFreeModels.map(applyCapabilities);
+export const PRO_MODELS: LLMModel[] = rawProModels.map(applyCapabilities);
+export const ALL_MODELS = [...FREE_MODELS, ...PRO_MODELS];
+
+const FALLBACK_MODEL: LLMModel = {
+  id: 'llm-model-missing',
+  name: 'No model configured',
+  provider: 'Unknown',
+  tier: 'free',
+  contextWindow: 128_000,
+  supportsImages: false,
+  supportsFileAttachments: true,
+  notes: 'Set VITE_LLM_FREE_MODELS and VITE_LLM_PRO_MODELS_LOW/MEDIUM/HIGH in environment.',
+};
+
+export const DEFAULT_FREE_MODEL = FREE_MODELS[0] ?? PRO_MODELS[0] ?? FALLBACK_MODEL;
+export const DEFAULT_PRO_MODEL = PRO_MODELS[0] ?? DEFAULT_FREE_MODEL;
+
+export function getModelById(id: string): LLMModel | undefined {
+  return ALL_MODELS.find((m) => m.id === id);
+}
+
+/** Check whether a model ID requires a pro subscription */
+export function requiresPro(modelId: string): boolean {
+  const model = getModelById(modelId);
+  return model?.tier === 'pro';
+}
+
+export function getDefaultModelForEntitlement(hasPro: boolean): LLMModel {
+  return hasPro ? DEFAULT_PRO_MODEL : DEFAULT_FREE_MODEL;
+}
+
+export function coerceModelForEntitlement(modelId: string | null | undefined, hasPro: boolean): string {
+  if (modelId) {
+    const model = getModelById(modelId);
+    if (model && (!requiresPro(modelId) || hasPro)) {
+      return modelId;
+    }
+  }
+  return getDefaultModelForEntitlement(hasPro).id;
+}

package/src/lib/llm/repair-loop.test.ts

@@ -0,0 +1,91 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+import test from 'node:test';
+import assert from 'node:assert/strict';
+import type { ChatMessage } from './types.js';
+import { buildRepairAttemptKey, buildRepairSessionKey, getEscalatedRepairScope, pruneMessagesForRepair } from './repair-loop.js';
+import { createPreflightDiagnostic } from './script-diagnostics.js';
+
+test('pruneMessagesForRepair removes prior repair prompt and patch reply context', () => {
+  const messages: ChatMessage[] = [
+    { id: '1', role: 'user', content: 'Create a house', createdAt: 1 },
+    { id: '2', role: 'assistant', content: '```js\nconst h = 1;\n```', createdAt: 2 },
+    { id: '3', role: 'user', content: 'The script needs a root-cause repair.\n\nFailure type: preflight', createdAt: 3 },
+    { id: '4', role: 'assistant', content: '```ifc-script-edits\n{"scriptEdits":[]}\n```', createdAt: 4 },
+    { id: '5', role: 'user', content: 'Add a roof', createdAt: 5 },
+  ];
+
+  const pruned = pruneMessagesForRepair(messages);
+  assert.deepEqual(pruned.map((message) => message.id), ['1', '2', '5']);
+});
+
+test('buildRepairAttemptKey stays stable for same targeted failure', () => {
+  const diagnostics = [
+    createPreflightDiagnostic(
+      'wall_hosted_opening_pattern',
+      'Suspicious door call.',
+      'error',
+      {
+        methodName: 'addIfcDoor',
+        range: { from: 120, to: 180 },
+        snippet: 'bim.create.addIfcDoor(h, ground, { Name: "Front Door" });',
+      },
+    ),
+  ];
+
+  const first = buildRepairAttemptKey({
+    reason: 'preflight',
+    diagnostics,
+    currentCode: 'const h = bim.create.project({ Name: "House" });',
+  });
+  const second = buildRepairAttemptKey({
+    reason: 'preflight',
+    diagnostics,
+    currentCode: 'const h = bim.create.project({ Name: "House" });',
+  });
+
+  assert.equal(first, second);
+});
+
+test('pruneMessagesForRepair also removes malformed assistant repair replies', () => {
+  const messages: ChatMessage[] = [
+    { id: '1', role: 'user', content: 'Create a tower', createdAt: 1 },
+    { id: '2', role: 'assistant', content: '```js\nconst h = 1;\n```', createdAt: 2 },
+    { id: '3', role: 'user', content: 'The script needs a root-cause repair.\n\nFailure type: patch-apply', createdAt: 3 },
+    { id: '4', role: 'assistant', content: 'Here is a corrected snippet:\n```js\nconst width = 30;\n```', createdAt: 4 },
+    { id: '5', role: 'user', content: 'Try again', createdAt: 5 },
+  ];
+
+  const pruned = pruneMessagesForRepair(messages);
+  assert.deepEqual(pruned.map((message) => message.id), ['1', '2', '5']);
+});
+
+test('buildRepairSessionKey tracks the root cause across script revisions', () => {
+  const diagnostics = [
+    createPreflightDiagnostic(
+      'detached_snippet_scope',
+      'Detached snippet risk.',
+      'error',
+      { symbol: 'width' },
+    ),
+  ];
+
+  const first = buildRepairSessionKey({
+    diagnostics,
+    currentCode: 'const width = 30;',
+  });
+  const second = buildRepairSessionKey({
+    diagnostics,
+    currentCode: 'const width = 30;\nconst depth = 40;',
+  });
+
+  assert.equal(first, second);
+});
+
+test('getEscalatedRepairScope widens repair sessions once', () => {
+  assert.equal(getEscalatedRepairScope('local'), 'block');
+  assert.equal(getEscalatedRepairScope('block'), 'structural');
+  assert.equal(getEscalatedRepairScope('structural'), null);
+});

package/src/lib/llm/repair-loop.ts

@@ -0,0 +1,76 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
+
+import type { ChatMessage } from './types.js';
+import { getPrimaryRootCause, type ScriptDiagnostic } from './script-diagnostics.js';
+
+const REPAIR_PROMPT_PREFIX = 'The script needs a root-cause repair.';
+
+export function isRepairPromptMessage(content: string): boolean {
+  return content.startsWith(REPAIR_PROMPT_PREFIX);
+}
+
+export function pruneMessagesForRepair(messages: ChatMessage[]): ChatMessage[] {
+  const result: ChatMessage[] = [];
+  let skippingRepairReply = false;
+
+  for (const message of messages) {
+    if (message.role === 'user' && isRepairPromptMessage(message.content)) {
+      skippingRepairReply = true;
+      continue;
+    }
+
+    if (skippingRepairReply && message.role === 'assistant') {
+      continue;
+    }
+
+    if (message.role === 'user') skippingRepairReply = false;
+    result.push(message);
+  }
+
+  return result;
+}
+
+export function buildRepairAttemptKey(params: {
+  reason: string;
+  currentCode: string;
+  diagnostics?: ScriptDiagnostic[];
+}): string {
+  const primary = getPrimaryRootCause(params.diagnostics ?? []);
+
+  return [
+    params.reason,
+    primary?.rootCauseKey ?? '',
+    primary?.repairScope ?? '',
+    hashText(params.currentCode),
+  ].join('|');
+}
+
+export function buildRepairSessionKey(params: {
+  currentCode: string;
+  diagnostics?: ScriptDiagnostic[];
+}): string {
+  const primary = getPrimaryRootCause(params.diagnostics ?? []);
+  return primary?.rootCauseKey ?? hashText(params.currentCode);
+}
+
+export function getEscalatedRepairScope(scope: ScriptDiagnostic['repairScope']): ScriptDiagnostic['repairScope'] | null {
+  switch (scope) {
+    case 'local':
+      return 'block';
+    case 'block':
+      return 'structural';
+    default:
+      return null;
+  }
+}
+
+export function hashText(text: string): string {
+  let hash = 2166136261;
+  for (let i = 0; i < text.length; i++) {
+    hash ^= text.charCodeAt(i);
+    hash = Math.imul(hash, 16777619);
+  }
+  return (hash >>> 0).toString(16);
+}