feat: add additional fallback for anthropic (PostHog#28179)

plugin-server/src/utils/ai-costs/anthropic_overrides.ts

@@ -0,0 +1,32 @@
+import type { ModelRow } from './types'
+
+export const costs: ModelRow[] = [
+    {
+        model: 'claude-3-5-haiku',
+        cost: {
+            prompt_token: 8e-7,
+            completion_token: 0.000004,
+        },
+    },
+    {
+        model: 'claude-3-5-sonnet',
+        cost: {
+            prompt_token: 0.000003,
+            completion_token: 0.000015,
+        },
+    },
+    {
+        model: 'claude-3-opus',
+        cost: {
+            prompt_token: 0.000015,
+            completion_token: 0.000075,
+        },
+    },
+    {
+        model: 'claude-2',
+        cost: {
+            prompt_token: 0.000008,
+            completion_token: 0.000024,
+        },
+    },
+]

plugin-server/src/utils/ai-costs/index.ts

@@ -1,4 +1,5 @@
 import { costs as anthropicCosts } from './anthropic'
+import { costs as anthropicOverrides } from './anthropic_overrides'
 import { costs as cohereCosts } from './cohere'
 import { costs as deepseekCosts } from './deepseek'
 import { costs as googleCosts } from './google'
@@ -11,6 +12,7 @@ import type { ModelRow } from './types'
 export const costs: ModelRow[] = [
     ...openaiCosts,
     ...anthropicCosts,
+    ...anthropicOverrides,
     ...googleCosts,
     ...deepseekCosts,
     ...perplexityCosts,

plugin-server/src/worker/ingestion/event-pipeline/processAiEvent.ts

@@ -86,9 +86,18 @@ export const extractCoreModelParams = (event: PluginEvent): PluginEvent => {
 }
 
 const findCostFromModel = (aiModel: string): ModelRow | undefined => {
+    // Check if the model is an exact match
     let cost = costs.find((cost) => cost.model.toLowerCase() === aiModel.toLowerCase())
+    // Check if the model is a variant of a known model
     if (!cost) {
         cost = costs.find((cost) => aiModel.toLowerCase().includes(cost.model.toLowerCase()))
     }
+    // Check if the model is a variant of a known model
+    if (!cost) {
+        cost = costs.find((cost) => cost.model.toLowerCase().includes(aiModel.toLowerCase()))
+    }
+    if (!cost) {
+        console.warn(`No cost found for model: ${aiModel}`)
+    }
     return cost
 }