create-walle 0.9.25 → 0.9.26

package/template/wall-e/llm/provider-health-state.js

@@ -0,0 +1,176 @@
+'use strict';
+
+const DEFAULT_PROVIDER_STATUS_MAX_AGE_MS = 10 * 60 * 1000;
+
+function nowIso() {
+  return new Date().toISOString();
+}
+
+function canonicalProvider(value) {
+  return String(value || '').trim().toLowerCase();
+}
+
+function canonicalModel(value) {
+  return String(value || '').trim().toLowerCase();
+}
+
+function providerModelKey(provider, model) {
+  const p = canonicalProvider(provider || 'default');
+  const m = canonicalModel(model || 'default');
+  return `${p}:${m}`;
+}
+
+function providerStatusIssueId(provider, model) {
+  const p = encodeURIComponent(canonicalProvider(provider || 'default'));
+  const m = encodeURIComponent(canonicalModel(model || 'default'));
+  return `provider_status:${p}:${m}`;
+}
+
+function parseProviderStatusIssueId(value) {
+  const id = String(value || '');
+  if (!id.startsWith('provider_status:')) return null;
+  const rest = id.slice('provider_status:'.length);
+  const firstColon = rest.indexOf(':');
+  if (firstColon < 0) return null;
+  try {
+    return {
+      provider: decodeURIComponent(rest.slice(0, firstColon)),
+      model: decodeURIComponent(rest.slice(firstColon + 1)),
+    };
+  } catch {
+    return null;
+  }
+}
+
+function parseStatusMap(brain) {
+  if (!brain || typeof brain.getKv !== 'function') return {};
+  try {
+    const raw = brain.getKv('provider_health_status');
+    const parsed = raw ? JSON.parse(raw) : {};
+    return parsed && typeof parsed === 'object' && !Array.isArray(parsed) ? parsed : {};
+  } catch {
+    return {};
+  }
+}
+
+function writeStatusMap(brain, statuses) {
+  if (!brain || typeof brain.setKv !== 'function') return false;
+  brain.setKv('provider_health_status', JSON.stringify(statuses || {}));
+  return true;
+}
+
+function publicProviderStatus(status, nowMs = Date.now()) {
+  if (!status || typeof status !== 'object') return null;
+  const checkedAt = status.checked_at || status.checkedAt || '';
+  const checkedMs = checkedAt ? Date.parse(checkedAt) : 0;
+  return {
+    provider: status.provider || '',
+    model: status.model || '',
+    issue_id: providerStatusIssueId(status.provider, status.model),
+    ok: status.ok === true,
+    status: status.ok === true ? 'ok' : 'error',
+    source: status.source || '',
+    error_type: status.error_type || '',
+    error: status.error || '',
+    latency_ms: Number.isFinite(Number(status.latency_ms)) ? Number(status.latency_ms) : null,
+    checked_at: checkedAt,
+    age_ms: checkedMs ? Math.max(0, nowMs - checkedMs) : null,
+    diagnostics: status.diagnostics && typeof status.diagnostics === 'object' ? status.diagnostics : null,
+  };
+}
+
+function recordProviderHealthStatus(status, opts = {}) {
+  if (!status || !status.provider) return { recorded: false };
+  let brain = opts.brain;
+  if (!brain) {
+    try { brain = require('../brain'); } catch { brain = null; }
+  }
+  if (!brain) return { recorded: false };
+
+  const provider = canonicalProvider(status.provider);
+  const model = String(status.model || '').trim();
+  const key = providerModelKey(provider, model);
+  const statuses = parseStatusMap(brain);
+  const entry = {
+    provider,
+    model,
+    ok: status.ok === true,
+    source: String(status.source || 'provider').slice(0, 80),
+    error_type: String(status.error_type || status.type || '').slice(0, 80),
+    error: String(status.error || status.message || '').slice(0, 240),
+    latency_ms: Number.isFinite(Number(status.latency_ms)) ? Number(status.latency_ms) : null,
+    checked_at: status.checked_at || status.checkedAt || nowIso(),
+    diagnostics: status.diagnostics && typeof status.diagnostics === 'object' ? status.diagnostics : null,
+  };
+  statuses[key] = entry;
+  writeStatusMap(brain, statuses);
+
+  let cleared = 0;
+  if (entry.ok && opts.clearTransientAlerts !== false) {
+    try {
+      const planner = require('../skills/skill-planner');
+      cleared = planner.clearRecoveredProviderAlerts({ provider, model }).cleared || 0;
+    } catch {}
+  }
+
+  return { recorded: true, key, cleared };
+}
+
+function getProviderHealthStatus(provider, model, opts = {}) {
+  let brain = opts.brain;
+  if (!brain) {
+    try { brain = require('../brain'); } catch { brain = null; }
+  }
+  if (!brain) return null;
+  const statuses = parseStatusMap(brain);
+  const providerKey = canonicalProvider(provider);
+  const modelKey = canonicalModel(model);
+  const exact = statuses[providerModelKey(providerKey, modelKey)];
+  const providerOnly = Object.values(statuses)
+    .filter((status) => canonicalProvider(status.provider) === providerKey)
+    .sort((a, b) => Date.parse(b.checked_at || '') - Date.parse(a.checked_at || ''))[0];
+  const status = exact || providerOnly || null;
+  if (!status) return null;
+  const maxAgeMs = opts.maxAgeMs ?? DEFAULT_PROVIDER_STATUS_MAX_AGE_MS;
+  const checkedMs = Date.parse(status.checked_at || '');
+  const referenceMs = opts.nowMs || Date.now();
+  if (Number.isFinite(maxAgeMs) && maxAgeMs > 0 && checkedMs && referenceMs - checkedMs > maxAgeMs) {
+    return null;
+  }
+  return publicProviderStatus(status, referenceMs);
+}
+
+function clearProviderHealthStatus(target = {}, opts = {}) {
+  const provider = canonicalProvider(target.provider);
+  const model = canonicalModel(target.model);
+  if (!provider) return { cleared: 0 };
+  let brain = opts.brain;
+  if (!brain) {
+    try { brain = require('../brain'); } catch { brain = null; }
+  }
+  if (!brain) return { cleared: 0 };
+
+  const statuses = parseStatusMap(brain);
+  let cleared = 0;
+  for (const [key, status] of Object.entries(statuses)) {
+    const sameProvider = canonicalProvider(status?.provider) === provider || key.startsWith(`${provider}:`);
+    if (!sameProvider) continue;
+    const sameModel = !model || canonicalModel(status?.model) === model || key === providerModelKey(provider, model);
+    if (!sameModel) continue;
+    delete statuses[key];
+    cleared += 1;
+  }
+  if (cleared) writeStatusMap(brain, statuses);
+  return { cleared };
+}
+
+module.exports = {
+  clearProviderHealthStatus,
+  DEFAULT_PROVIDER_STATUS_MAX_AGE_MS,
+  getProviderHealthStatus,
+  parseProviderStatusIssueId,
+  providerModelKey,
+  providerStatusIssueId,
+  publicProviderStatus,
+  recordProviderHealthStatus,
+};