feat: add anthropic overrides

PostHog · Feb 1, 2025 · e7f9df0 · e7f9df0
1 parent df7222d
commit e7f9df0
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/plugin-server/src/utils/ai-costs/anthropic_overrides.ts b/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,
+        },
+    },
+]
diff --git a/plugin-server/src/utils/ai-costs/index.ts b/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,