@fraziym/axoniz 0.1.0

package/dist/web/static/js/compare/models.js

@@ -0,0 +1,104 @@
+// compare/models.js — model classification, fetching, display names, persistence
+import Storage from '../storage.js';
+import state from './state.js';
+import uiModule from '../ui.js';
+import { sortModelObjects } from '../modelSort.js';
+
+var escapeHtml = uiModule.esc;
+
+// ── Model classification constants ──
+const NON_CHAT_PREFIXES = ['tts-', 'whisper-', 'text-embedding-', 'text-moderation-', 'moderation-', 'embedding'];
+const NON_CHAT_SUFFIXES = ['deep-research', '-online'];
+const IMAGE_PREFIXES = ['dall-e-3', 'gpt-image', 'chatgpt-image'];
+const DEPRECATED_IMAGE = ['dall-e-2'];
+
+function classifyModel(id) {
+  const lower = id.toLowerCase();
+  if (DEPRECATED_IMAGE.some(p => lower.startsWith(p))) return 'other';
+  if (IMAGE_PREFIXES.some(p => lower.startsWith(p))) return 'image';
+  if (NON_CHAT_PREFIXES.some(p => lower.startsWith(p))) return 'other';
+  if (NON_CHAT_SUFFIXES.some(p => lower.endsWith(p) || lower.includes(p))) return 'other';
+  return 'chat';
+}
+
+/** Build display names for selected models, adding endpoint name when the same model appears from multiple providers. */
+function _modelDisplayNames(models) {
+  const nameCount = {};
+  for (const m of models) {
+    const short = m.name || m.model.split('/').pop();
+    nameCount[short] = (nameCount[short] || 0) + 1;
+  }
+  return models.map(m => {
+    const short = m.name || m.model.split('/').pop();
+    if (nameCount[short] > 1 && m.endpointName) return short + ' (' + escapeHtml(m.endpointName) + ')';
+    return short;
+  });
+}
+
+/** Save selected models and synth models to localStorage, keyed by compare mode. */
+function _persistSelections() {
+  if (state._selectedModels.length > 0) {
+    Storage.setJSON('zero-compare-selections-' + (state._compareMode || 'chat'), state._selectedModels);
+  }
+  if ((state._compareMode === 'search' || state._compareMode === 'research') && state._searchSynthModels) {
+    Storage.setJSON('zero-compare-synth-' + state._compareMode, state._searchSynthModels);
+  }
+}
+
+// ── Model fetching with cache ──
+const MODELS_CACHE_TTL = 30000; // 30 seconds
+
+/** Fetch available models from API. */
+async function fetchModels() {
+  const now = Date.now();
+  if (state._fetchModelsCache && (now - state._fetchModelsCacheTime) < MODELS_CACHE_TTL) {
+    return state._fetchModelsCache;
+  }
+  const res = await fetch(`${state.API_BASE}/api/models`);
+  const data = await res.json();
+  const models = [];
+  if (data.items && data.items.length > 0) {
+    data.items.forEach(item => {
+      const displayNames = item.models_display || item.models || [];
+      const extraDisplay = item.models_extra_display || item.models_extra || [];
+      // Curated list (item.models) takes priority; non-curated extras come
+      // after so newer/uncatalogued models (e.g. deepseek-v4-pro) still show.
+      (item.models || []).forEach((mid, i) => {
+        models.push({
+          id: mid,
+          url: item.url,
+          name: (displayNames[i] || mid).split('/').pop(),
+          endpointId: item.endpoint_id || null,
+          endpointName: item.endpoint_name || '',
+          type: classifyModel(mid),
+        });
+      });
+      (item.models_extra || []).forEach((mid, i) => {
+        models.push({
+          id: mid,
+          url: item.url,
+          name: (extraDisplay[i] || mid).split('/').pop(),
+          endpointId: item.endpoint_id || null,
+          endpointName: item.endpoint_name || '',
+          type: classifyModel(mid),
+        });
+      });
+    });
+  }
+  state._fetchModelsCache = sortModelObjects(models);
+  state._fetchModelsCacheTime = now;
+  return state._fetchModelsCache;
+}
+
+// ── Shuffle pool persistence ──
+const POOL_STORAGE_KEY = 'zero-shuffle-pool-excluded';
+
+function getExcludedModels() {
+  return Storage.getJSON(POOL_STORAGE_KEY, []);
+}
+
+function setExcludedModels(arr) {
+  Storage.setJSON(POOL_STORAGE_KEY, arr);
+}
+
+export { classifyModel, _modelDisplayNames, fetchModels, _persistSelections, getExcludedModels, setExcludedModels };