@goodandready/dsh-cost-meter 0.8.28 → 0.8.30

package/lib/analytics.js

@@ -90,4 +90,59 @@ export function calculateBurnRateAndProjection(turns, totalCostUsd, options = {}
     projectedCostUsd,
     confidence,
   }
-}
+}
+
+/**
+ * Returns estimated context window token limit for model identifier.
+ */
+export function getFallbackContextLength(model) {
+  const m = String(model || '').toLowerCase()
+  if (m.includes('claude-3-5') || m.includes('claude-3-7') || m.includes('claude-3-opus')) return 200000
+  if (m.includes('claude')) return 200000
+  if (m.includes('gemini-1.5') || m.includes('gemini-2.0') || m.includes('gemini-exp')) return 1000000
+  if (m.includes('gpt-4.1') || m.includes('gpt-4o') || m.includes('o1') || m.includes('o3')) return 128000
+  if (m.includes('qwen-2.5') || m.includes('qwq')) return 128000
+  if (m.includes('deepseek')) return 128000
+  return 128000
+}
+
+/**
+ * Calculates context window saturation metrics and compaction risk.
+ *
+ * @param {Array<{ turn: number, uncachedInputTokens?: number, cacheReadTokens?: number }>} turns
+ * @param {object} [tariff]
+ * @param {object} [route]
+ * @returns {{ currentTokens: number, contextLength: number, saturationPercent: number, riskLevel: 'low' | 'moderate' | 'high', nextTurnEstUsd: number }}
+ */
+export function calculateContextMetrics(turns, tariff, route) {
+  const lastTurn = (Array.isArray(turns) && turns.length > 0) ? turns[turns.length - 1] : null
+  const currentTokens = lastTurn
+    ? (Number(lastTurn.uncachedInputTokens || 0) + Number(lastTurn.cacheReadTokens || 0))
+    : 0
+
+  const contextLength = (tariff && Number(tariff.contextLength)) || getFallbackContextLength(route?.model)
+  const saturationPercent = contextLength > 0
+    ? Math.min(100, Math.round((currentTokens / contextLength) * 100))
+    : 0
+
+  let riskLevel = 'low'
+  if (saturationPercent >= 80) {
+    riskLevel = 'high'
+  } else if (saturationPercent >= 50) {
+    riskLevel = 'moderate'
+  }
+
+  const rates = tariff?.schedule?.base || {}
+  const inputRate = Number(rates.input) || 0.14
+  const nextTurnEstUsd = currentTokens > 0
+    ? Number(((currentTokens * inputRate) / 1e6).toFixed(6))
+    : 0
+
+  return {
+    currentTokens,
+    contextLength,
+    saturationPercent,
+    riskLevel,
+    nextTurnEstUsd,
+  }
+}