dsh-all-usage 1.1.2 → 1.1.3

package/lib/plugin.js

@@ -0,0 +1,276 @@
+import { randomBytes } from 'node:crypto'
+import { createAggregation } from './aggregation.js'
+import { createBalance } from './balance.js'
+import { registerRoutes } from './http.js'
+import { createLedger } from './ledger.js'
+import { createPricingRuntime } from './pricing-runtime.js'
+import { createSessionSync } from './session-sync.js'
+import { createCostAccumulator, createEmptyPricingState } from './pricing.js'
+
+const name = 'dsh-all-usage'
+const inject = ['sessionQuery', 'workspaceRegistry', 'timer', 'sessionPersistence', 'storage', 'webServer']
+
+function apply(ctx) {
+  const services = {
+    credentials: ctx.get('credentials'),
+    settings: ctx.get('settings'),
+    storage: ctx.get('storage'),
+    webServer: ctx.get('webServer'),
+    sessionPersistence: ctx.get('sessionPersistence'),
+  }
+  const state = {
+    wsMeta: new Map(),
+    pathIndex: new Map(),
+    memberOf: new Map(),
+    byDay: new Map(),
+    byDayUtc: new Map(),
+    perWorkspace: new Map(),
+    perModel: new Map(),
+    usageByStep: new Map(),
+    turnRecords: new Map(),
+    usageByLocalDate: new Map(),
+    usageByUtcDate: new Map(),
+    sessionModel: new Map(),
+    totals: { turns: 0, input: 0, output: 0, cacheRead: 0, cacheWrite: 0, reasoning: 0, cost: createCostAccumulator() },
+    sessionCount: new Set(),
+    sessionSeq: new Map(),
+    chains: new Map(),
+    liveResyncPending: new Set(),
+    liveResyncTimers: new Map(),
+    liveResyncAttempts: new Map(),
+    scan: { started: false, done: false, scanned: 0, total: 0, failed: 0 },
+    aliases: {},
+    kvUnit: null,
+    aliasWriteChain: Promise.resolve(),
+    aliasesReady: Promise.resolve(),
+    balanceCache: { fetchedAt: 0, payload: null },
+    requestToken: randomBytes(32).toString('base64url'),
+    instanceId: randomBytes(12).toString('base64url'),
+    statsRevision: 0,
+    dataRevision: 0,
+    metadataRevision: 0,
+    scanRevision: 0,
+    pricingRevision: 0,
+    statsUpdatedAt: Date.now(),
+    statsDirtyScheduled: false,
+    statsDirtyKinds: new Set(),
+    sync: {
+      lastStartedAt: 0,
+      lastCompletedAt: 0,
+      lastErrorAt: 0,
+      lastErrorCode: null,
+      persistenceSnapshotsAvailable: false,
+      sessionsTotal: 0,
+      sessionsRead: 0,
+      sessionsSkippedByRevision: 0,
+      sessionsRestoredFromLedger: 0,
+      sessionsFailed: 0,
+    },
+    ledgerRecords: new Map(),
+    queryCache: new Map(),
+    recordsQueryCache: new Map(),
+    snapshotCache: null,
+    ledgerRevision: Date.now(),
+    ledgerUnit: null,
+    ledgerReady: Promise.resolve(),
+    ledgerWriteChain: Promise.resolve(),
+    ledgerPending: new Map(),
+    ledgerWriteWaiters: new Map(),
+    ledgerWriteRunning: false,
+    ledgerWriteScheduled: false,
+    ledgerWriteTimer: null,
+    ledgerDirtySessions: new Set(),
+    ledgerDirtyEpochs: new Map(),
+    ledgerWriteFailedSessions: new Set(),
+    pricingState: createEmptyPricingState(),
+    pricingResolutionCache: new Map(),
+    pricingUnit: null,
+    pricingReady: Promise.resolve(),
+    pricingWriteChain: Promise.resolve(),
+    pricingSyncInFlight: false,
+    pricingSyncTimer: null,
+    aggregationGeneration: 0,
+    disposed: false,
+    baselineRetryDelay: 1000,
+    baselineRetryScheduled: false,
+    baselineFallbackTimer: null,
+    knownSessionIds: new Set(),
+    reconcileHintScheduled: false,
+    reconcilePending: false,
+    reconcileInFlight: false,
+    reconcileTimer: null,
+  }
+
+  const host = { ctx, services, state, webServer: services.webServer, aggregation: null, ledger: null, pricing: null, balance: null, sessionSync: null, aliases: null }
+  function commitPendingStats() {
+    if (!state.statsDirtyScheduled) return
+    state.statsDirtyScheduled = false
+    const kinds = state.statsDirtyKinds
+    state.statsDirtyKinds = new Set()
+    if (state.disposed) return
+    state.statsRevision += 1
+    if (kinds.has('data')) state.dataRevision += 1
+    if (kinds.has('metadata')) state.metadataRevision += 1
+    if (kinds.has('scan')) state.scanRevision += 1
+    if (kinds.has('pricing')) state.pricingRevision += 1
+    state.statsUpdatedAt = Date.now()
+  }
+  function markStatsChanged(kind = 'data') {
+    const kinds = Array.isArray(kind) ? kind : [kind]
+    let recordsChanged = false
+    for (const value of kinds) {
+      if (value !== 'data' && value !== 'metadata' && value !== 'scan' && value !== 'pricing') continue
+      state.statsDirtyKinds.add(value)
+      if (value === 'data' || value === 'pricing') recordsChanged = true
+    }
+    state.snapshotCache = null
+    if (recordsChanged) state.recordsQueryCache.clear()
+    if (state.disposed || state.statsDirtyScheduled) return
+    state.statsDirtyScheduled = true
+    if (typeof queueMicrotask === 'function') queueMicrotask(commitPendingStats)
+    else Promise.resolve().then(commitPendingStats)
+  }
+  function resetSyncState() {
+    state.sync.lastStartedAt = 0
+    state.sync.lastCompletedAt = 0
+    state.sync.lastErrorAt = 0
+    state.sync.lastErrorCode = null
+    state.sync.persistenceSnapshotsAvailable = false
+    state.sync.sessionsTotal = 0
+    state.sync.sessionsRead = 0
+    state.sync.sessionsSkippedByRevision = 0
+    state.sync.sessionsRestoredFromLedger = 0
+    state.sync.sessionsFailed = 0
+  }
+  function beginSync() {
+    resetSyncState()
+    state.sync.lastStartedAt = Date.now()
+    markStatsChanged('scan')
+  }
+  function noteSyncError(code) {
+    state.sync.lastErrorAt = Date.now()
+    state.sync.lastErrorCode = code
+    markStatsChanged('scan')
+  }
+  function syncSnapshot() {
+    return {
+      lastStartedAt: state.sync.lastStartedAt,
+      lastCompletedAt: state.sync.lastCompletedAt,
+      lastErrorAt: state.sync.lastErrorAt,
+      lastErrorCode: state.sync.lastErrorCode,
+      persistenceSnapshotsAvailable: state.sync.persistenceSnapshotsAvailable,
+      sessionsTotal: state.sync.sessionsTotal,
+      sessionsRead: state.sync.sessionsRead,
+      sessionsSkippedByRevision: state.sync.sessionsSkippedByRevision,
+      sessionsRestoredFromLedger: state.sync.sessionsRestoredFromLedger,
+      sessionsFailed: state.sync.sessionsFailed,
+    }
+  }
+  Object.assign(host, { commitPendingStats, markStatsChanged, resetSyncState, beginSync, noteSyncError, syncSnapshot })
+
+  host.pricingSnapshot = (...args) => host.pricing.pricingSnapshot(...args)
+  host.syncSnapshot = (...args) => syncSnapshot(...args)
+  host.aggregation = createAggregation(host)
+  host.ledger = createLedger(host)
+  host.pricing = createPricingRuntime(host)
+  host.balance = createBalance(host)
+  host.sessionSync = createSessionSync(host)
+
+  async function loadAliases() {
+    const storage = services.storage
+    if (storage === undefined || storage === null) return
+    try {
+      const backend = storage.backend.get('json')
+      if (backend === undefined || backend === null || backend.kv === undefined) return
+      const unit = await backend.kv.open({ name: 'all_usage_aliases', version: 0, tables: [], hasGlobal: true })
+      if (state.disposed) { await unit.close().catch(() => {}); return }
+      state.kvUnit = unit
+      const snap = await unit.loadAll()
+      const global = snap && snap.global
+      if (global !== null && global !== undefined && typeof global === 'object') {
+        let changed = false
+        for (const key of Object.keys(global)) {
+          const value = global[key]
+          const next = typeof value === 'string' ? value.trim() : ''
+          if (next !== '' && state.aliases[key] !== next) {
+            state.aliases[key] = next
+            changed = true
+          }
+        }
+        if (changed) markStatsChanged('metadata')
+      }
+    } catch (err) {
+      console.error('[all-usage] alias storage unavailable:', err)
+    }
+  }
+  function persistAliases() {
+    if (state.disposed) return state.aliasWriteChain
+    const snapshotAliases = {}
+    for (const key of Object.keys(state.aliases)) snapshotAliases[key] = state.aliases[key]
+    state.aliasWriteChain = state.aliasWriteChain.then(() => {
+      if (state.kvUnit === null || state.kvUnit === undefined) return undefined
+      return state.kvUnit.setGlobal(snapshotAliases).catch((err) => {
+        console.error('[all-usage] alias persist failed:', err)
+      })
+    })
+    return state.aliasWriteChain
+  }
+  function setAlias(wsId, raw) {
+    if (typeof wsId !== 'string' || wsId.length === 0 || wsId.length > 256) return { ok: false, message: 'invalid-workspace', aliases: Object.assign({}, state.aliases) }
+    if (typeof raw !== 'string') return { ok: false, message: 'invalid-alias', aliases: Object.assign({}, state.aliases) }
+    const alias = raw.trim().slice(0, 80)
+    if (!state.wsMeta.has(wsId)) return { ok: false, message: 'unknown-workspace', aliases: Object.assign({}, state.aliases) }
+    const previous = typeof state.aliases[wsId] === 'string' ? state.aliases[wsId] : ''
+    if (alias === previous) return { ok: true, aliases: Object.assign({}, state.aliases) }
+    if (alias === '') delete state.aliases[wsId]
+    else state.aliases[wsId] = alias
+    persistAliases()
+    markStatsChanged('metadata')
+    return { ok: true, aliases: Object.assign({}, state.aliases) }
+  }
+  host.aliases = { loadAliases, persistAliases, setAlias }
+
+  ctx.effect(() => async () => {
+    state.disposed = true
+    if (state.reconcileTimer !== null) {
+      clearTimeout(state.reconcileTimer)
+      state.reconcileTimer = null
+    }
+    if (state.baselineFallbackTimer !== null) {
+      clearTimeout(state.baselineFallbackTimer)
+      state.baselineFallbackTimer = null
+    }
+    if (state.pricingSyncTimer !== null) {
+      clearTimeout(state.pricingSyncTimer)
+      state.pricingSyncTimer = null
+    }
+    for (const timer of state.liveResyncTimers.values()) clearTimeout(timer)
+    state.liveResyncTimers.clear()
+    state.liveResyncAttempts.clear()
+    state.liveResyncPending.clear()
+    state.baselineRetryScheduled = false
+    if (state.ledgerWriteTimer !== null) {
+      clearTimeout(state.ledgerWriteTimer)
+      state.ledgerWriteTimer = null
+    }
+    state.ledgerWriteScheduled = false
+    state.chains.clear()
+    await Promise.all([state.aliasesReady, state.ledgerReady, state.pricingReady, state.aliasWriteChain, state.pricingWriteChain])
+    await host.ledger.drainLedgerWrites()
+    const units = [state.kvUnit, state.ledgerUnit, state.pricingUnit]
+    state.kvUnit = null
+    state.ledgerUnit = null
+    state.pricingUnit = null
+    await Promise.all(units.map((unit) => unit === null || unit === undefined ? undefined : unit.close().catch(() => {})))
+  })
+
+  registerRoutes(host)
+  state.ledgerReady = host.ledger.loadLedger()
+  state.pricingReady = host.pricing.loadPricing().then(() => { host.pricing.schedulePricingSync() })
+  void host.sessionSync.runBaseline()
+  host.sessionSync.scheduleReconcileTimer()
+  state.aliasesReady = loadAliases()
+}
+
+export { name, inject, apply }
+export default { name, inject, apply }

package/lib/pricing-runtime.js

@@ -0,0 +1,282 @@
+import { COST_SCHEMA_VERSION, calculateCost, fetchModelsDevCatalog, normalizeCostSnapshot, normalizePricingState, officialProviderIds, serializeCostAggregate, serializePricingState } from './pricing.js'
+
+const LEDGER_VERSION = 3
+
+export function createPricingRuntime(host) {
+  const { state, markStatsChanged } = host
+  const { storage } = host.services
+  const {
+    coerceIdentity,
+    dateKeys,
+    ensureDay,
+    ensureDayModel,
+    ensureDayWs,
+    ensureModel,
+    ensureWs,
+    addCostAggregateDirection,
+    adjustQueryCost,
+    resolveCurrentPricing,
+  } = host.aggregation
+  const persistLedgerRecord = (...args) => host.ledger.persistLedgerRecord(...args)
+  const drainLedgerWrites = (...args) => host.ledger.drainLedgerWrites(...args)
+  const nextLedgerRevision = (...args) => host.ledger.nextLedgerRevision(...args)
+
+  function adjustCostOnly(item, cost, direction) {
+    const identity = coerceIdentity(item.identity || item.modelId)
+    const dates = item && typeof item.date === 'string' && typeof item.dateUtc === 'string' ? { local: item.date, utc: item.dateUtc } : dateKeys(item.time)
+    const targets = []
+    const seen = new Set()
+    for (const target of [
+      ensureModel(identity).cost,
+      state.totals.cost,
+      ensureWs(item.wsId).cost,
+      ensureDay(state.byDay, dates.local).cost,
+      ensureDay(state.byDayUtc, dates.utc).cost,
+      ensureDayWs(ensureDay(state.byDay, dates.local), item.wsId).cost,
+      ensureDayWs(ensureDay(state.byDayUtc, dates.utc), item.wsId).cost,
+      ensureDayModel(ensureDay(state.byDay, dates.local), identity).cost,
+      ensureDayModel(ensureDay(state.byDayUtc, dates.utc), identity).cost,
+    ]) {
+      if (seen.has(target)) continue
+      seen.add(target)
+      targets.push(target)
+    }
+    for (const target of targets) addCostAggregateDirection(target, cost, direction)
+  }
+  function usedPricingModels(tierSchedules = null) {
+    const rows = []
+    const seen = new Set()
+    const scheduleIds = new Map()
+    const scheduleIdFor = (tiers) => {
+      if (!Array.isArray(tierSchedules) || tiers.length === 0) return null
+      const key = JSON.stringify(tiers)
+      const existing = scheduleIds.get(key)
+      if (existing !== undefined) return existing
+      const id = 'tier-' + tierSchedules.length
+      scheduleIds.set(key, id)
+      tierSchedules.push({ id, tiers: tiers.map((tier) => ({ ...tier })) })
+      return id
+    }
+    for (const item of state.usageByStep.values()) {
+      const identity = coerceIdentity(item.identity || item.modelId)
+      if (seen.has(identity.identityKey)) continue
+      seen.add(identity.identityKey)
+      const resolved = resolveCurrentPricing(identity)
+      const tiers = Array.isArray(resolved.tiers) ? resolved.tiers : []
+      const tierScheduleId = resolved.tiered === true ? scheduleIdFor(tiers) : null
+      rows.push({
+        identityKey: identity.identityKey,
+        provider: identity.provider,
+        requestedModel: identity.requestedModel,
+        actualModel: identity.actualModel,
+        model: identity.label,
+        status: resolved.status,
+        reason: resolved.reason || '',
+        pricingModel: resolved.pricingModel || null,
+        providerId: resolved.providerId || null,
+        source: resolved.source || 'none',
+        currency: resolved.currency || 'USD',
+        rates: resolved.rates || null,
+        tiered: resolved.tiered === true,
+        tierCount: tiers.length,
+        ...(tierScheduleId === null ? {} : { tierScheduleId }),
+        tieredInvalid: resolved.tieredInvalid === true,
+        inputTokenSemantics: resolved.inputTokenSemantics || 'fresh',
+        multiplier: resolved.multiplier || '1',
+      })
+      if (rows.length >= 500) break
+    }
+    rows.sort((a, b) => String(a.model).localeCompare(String(b.model)))
+    return rows
+  }
+  function pricingModelSearch(query, limit = 20) {
+    const raw = typeof query === 'string' ? query.trim().toLowerCase() : ''
+    if (raw === '') return []
+    const normalized = raw.replace(/\s+/g, ' ')
+    const selected = new Map()
+    for (const entry of state.pricingState.catalogEntries) {
+      const official = officialProviderIds(entry.modelId)
+      if (!official.has(String(entry.providerId || '').toLowerCase())) continue
+      const modelId = entry.modelId.toLowerCase()
+      const displayName = String(entry.displayName || '').toLowerCase()
+      if (!modelId.includes(normalized) && !displayName.includes(normalized)) continue
+      const score = modelId === normalized ? 0 : modelId.startsWith(normalized) ? 1 : displayName.startsWith(normalized) ? 2 : 3
+      const previous = selected.get(entry.modelId)
+      if (previous === undefined || score < previous.score) selected.set(entry.modelId, { value: entry.modelId, label: entry.displayName, providerId: entry.providerId, tiered: entry.tiered === true, tierCount: Array.isArray(entry.tiers) ? entry.tiers.length : 0, score })
+    }
+    return Array.from(selected.values()).sort((a, b) => a.score - b.score || a.value.localeCompare(b.value)).slice(0, Math.max(1, Math.min(50, Number.isInteger(limit) ? limit : 20))).map(({ score, ...entry }) => entry)
+  }
+  function pricingSnapshot(options = {}) {
+    const pricingStateSnapshot = state.pricingState
+    const detailed = options.detailed !== false
+    const tierSchedules = detailed ? [] : null
+    const snapshot = {
+      schemaVersion: pricingStateSnapshot.schemaVersion,
+      source: { ...pricingStateSnapshot.source },
+      sync: { ...pricingStateSnapshot.sync },
+      catalogModelCount: pricingStateSnapshot.catalogEntries.length,
+      overrideCount: pricingStateSnapshot.overrides.length,
+      mappingCount: pricingStateSnapshot.mappings.length,
+      configured: pricingStateSnapshot.catalogEntries.length > 0 || pricingStateSnapshot.overrides.length > 0 || pricingStateSnapshot.mappings.length > 0,
+      usedModels: usedPricingModels(tierSchedules),
+      cost: serializeCostAggregate(state.totals.cost),
+    }
+    if (detailed) {
+      snapshot.config = {
+        sync: { ...pricingStateSnapshot.sync },
+        mappings: pricingStateSnapshot.mappings.map((mapping) => ({ ...mapping })),
+        overrides: pricingStateSnapshot.overrides.map(({ providerId, ...entry }) => ({ ...entry, ...(Array.isArray(entry.tiers) ? { tiers: entry.tiers.map((tier) => ({ ...tier })) } : {}) })),
+      }
+      snapshot.tierSchedules = tierSchedules
+    }
+    return snapshot
+  }
+  async function loadPricing() {
+    if (storage === undefined || storage.backend === undefined || typeof storage.backend.get !== 'function') return
+    try {
+      const backend = storage.backend.get('json')
+      if (backend === undefined || backend === null || backend.kv === undefined) return
+      const unit = await backend.kv.open({ name: 'all_usage_pricing', version: 0, tables: [], hasGlobal: true })
+      if (state.disposed) { await unit.close().catch(() => {}); return }
+      state.pricingUnit = unit
+      const snapshot = await unit.loadAll()
+      const global = snapshot && snapshot.global
+      const raw = global && typeof global === 'object' && global.pricing !== undefined ? global.pricing : global
+      state.pricingState = normalizePricingState(raw)
+      state.pricingResolutionCache.clear()
+    } catch (err) {
+      console.error('[all-usage] pricing catalog unavailable:', err)
+    }
+  }
+  function persistPricing() {
+    if (state.disposed) return state.pricingWriteChain
+    const payload = { pricing: serializePricingState(state.pricingState) }
+    state.pricingWriteChain = state.pricingWriteChain.then(async () => {
+      if (state.pricingUnit === null || state.pricingUnit === undefined) return
+      await state.pricingUnit.setGlobal(payload)
+    })
+    state.pricingWriteChain = state.pricingWriteChain.catch((err) => {
+      console.error('[all-usage] pricing persist failed:', err)
+    })
+    return state.pricingWriteChain
+  }
+  async function syncPricing(force = false) {
+    await Promise.all([state.pricingReady, state.ledgerReady])
+    if (state.pricingSyncInFlight) return { ok: false, message: 'pricing-sync-in-progress', pricing: pricingSnapshot() }
+    const now = Date.now()
+    if (!force && state.pricingState.sync.lastSuccessAt > 0 && now - state.pricingState.sync.lastSuccessAt < state.pricingState.sync.intervalMs) return { ok: true, skipped: true, pricing: pricingSnapshot() }
+    state.pricingSyncInFlight = true
+    state.pricingState.sync.lastAttemptAt = now
+    try {
+      const result = await fetchModelsDevCatalog()
+      if (!result.ok) {
+        state.pricingState.source.lastError = result.error
+        state.pricingState.sync.lastError = result.error
+        // A failed attempt changes only sync health, not any configured price,
+        // so it must not invalidate the query snapshot/records caches.
+        markStatsChanged('sync-health')
+        await persistPricing()
+        return { ok: false, message: result.error, pricing: pricingSnapshot() }
+      }
+      const previousHash = state.pricingState.source && state.pricingState.source.catalogHash !== undefined ? String(state.pricingState.source.catalogHash) : ''
+      state.pricingState = normalizePricingState({
+        ...serializePricingState(state.pricingState),
+        source: { url: result.catalog.sourceUrl, fetchedAt: result.catalog.fetchedAt, catalogHash: result.catalog.catalogHash, lastError: '' },
+        sync: { ...state.pricingState.sync, lastSuccessAt: result.catalog.fetchedAt, lastError: '' },
+        catalogEntries: result.catalog.entries,
+      })
+      state.pricingResolutionCache.clear()
+      const backfill = backfillUnpricedCosts()
+      // Only catalog changes or newly priced usage can alter computed costs; a
+      // successful sync with an unchanged catalog must not invalidate the query
+      // snapshot, scoped caches, or records cursors. The sync-health bump still
+      // rebuilds the snapshot cache so lastSuccessAt is not stale on the next
+      // full response.
+      if (String(result.catalog.catalogHash || '') !== previousHash || backfill.priced > 0) markStatsChanged('pricing')
+      else markStatsChanged('sync-health')
+      await persistPricing()
+      await drainLedgerWrites()
+      return { ok: true, skipped: false, backfill, pricing: pricingSnapshot() }
+    } finally {
+      state.pricingSyncInFlight = false
+      schedulePricingSync()
+    }
+  }
+  function backfillUnpricedCosts() {
+    let considered = 0
+    let priced = 0
+    const touched = new Set()
+    for (const item of state.usageByStep.values()) {
+      const oldCost = normalizeCostSnapshot(item.cost)
+      if (oldCost !== null && oldCost.pricingMode === 'official-model' && oldCost.status === 'priced') continue
+      considered += 1
+      const next = calculateCost(item.values, resolveCurrentPricing(item.identity || item.modelId))
+      if (next.status !== 'priced') continue
+      if (oldCost !== null) {
+        adjustCostOnly(item, oldCost, -1)
+        adjustQueryCost(item, oldCost, -1)
+      }
+      item.cost = next
+      adjustCostOnly(item, next, 1)
+      adjustQueryCost(item, next, 1)
+      const record = state.ledgerRecords.get(item.sid)
+      if (record !== undefined) {
+        const stored = record.usage.find((candidate) => candidate.key === item.key)
+        if (stored !== undefined) { stored.cost = next; touched.add(record) }
+      }
+      priced += 1
+    }
+    for (const record of touched) {
+      record.version = LEDGER_VERSION
+      record.updatedAt = nextLedgerRevision()
+      // Backfilling prices must not clear a mixed-workspace upgrade flag: stay
+      // unfoldable until a rebuild normalizes every historical item.
+      const stillMixed = record.turns.some((turn) => turn && turn.workspaceId !== record.workspaceId) || record.usage.some((item) => item && item.workspaceId !== record.workspaceId)
+      record.needsUpgrade = stillMixed
+      void persistLedgerRecord(record)
+    }
+    let remaining = 0
+    for (const item of state.usageByStep.values()) {
+      const cost = normalizeCostSnapshot(item.cost)
+      if (cost === null || cost.status !== 'priced') remaining += 1
+    }
+    return { considered, priced, remaining }
+  }
+  function updatePricingState(raw, backfill) {
+    const input = raw && typeof raw === 'object' && raw.pricing && typeof raw.pricing === 'object' ? raw.pricing : raw
+    const current = serializePricingState(state.pricingState)
+    const merged = { ...current, ...(input && typeof input === 'object' ? input : {}) }
+    if (input && typeof input === 'object' && input.sync && typeof input.sync === 'object' && !Array.isArray(input.sync)) merged.sync = { ...current.sync, ...input.sync }
+    state.pricingState = normalizePricingState(merged)
+    state.pricingResolutionCache.clear()
+    const result = backfill === true ? backfillUnpricedCosts() : { considered: 0, priced: 0, remaining: state.totals.cost.unpricedCalls + state.totals.cost.ambiguousCalls + state.totals.cost.unsupportedCalls }
+    markStatsChanged('pricing')
+    schedulePricingSync()
+    return result
+  }
+  function schedulePricingSync() {
+    if (state.pricingSyncTimer !== null) { clearTimeout(state.pricingSyncTimer); state.pricingSyncTimer = null }
+    if (state.disposed || state.pricingState.sync.autoEnabled !== true) return
+    const elapsed = state.pricingState.sync.lastSuccessAt > 0 ? Date.now() - state.pricingState.sync.lastSuccessAt : state.pricingState.sync.intervalMs
+    const delay = Math.max(0, state.pricingState.sync.intervalMs - elapsed)
+    state.pricingSyncTimer = setTimeout(() => {
+      state.pricingSyncTimer = null
+      void syncPricing(true)
+    }, delay)
+    if (state.pricingSyncTimer && typeof state.pricingSyncTimer.unref === 'function') state.pricingSyncTimer.unref()
+  }
+
+
+  return {
+    adjustCostOnly,
+    usedPricingModels,
+    pricingModelSearch,
+    pricingSnapshot,
+    loadPricing,
+    persistPricing,
+    syncPricing,
+    backfillUnpricedCosts,
+    updatePricingState,
+    schedulePricingSync
+  }
+}