feat: add additional fallback

PostHog · Feb 1, 2025 · df7222d · df7222d
1 parent f3d3210
commit df7222d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/plugin-server/src/worker/ingestion/event-pipeline/processAiEvent.ts b/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
 }