added x cost support (#3151)

Helicone · Jan 21, 2025 · 1f094fd · 1f094fd
1 parent 7ba8e3f
commit 1f094fd
Show file tree

Hide file tree

Showing 5 changed files with 205 additions and 0 deletions.
diff --git a/bifrost/packages/cost/providers/x/index.ts b/bifrost/packages/cost/providers/x/index.ts
@@ -0,0 +1,49 @@
+/**
+ *
+ * DO NOT EDIT THIS FILE UNLESS IT IS IN /costs
+ */
+
+import { ModelRow } from "../../interfaces/Cost";
+
+export const costs: ModelRow[] = [
+  {
+    model: {
+      operator: "equals",
+      value: "grok-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-vision-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-vision-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+];
diff --git a/costs/src/providers/mappings.ts b/costs/src/providers/mappings.ts
@@ -16,6 +16,7 @@ import { openAIProvider } from "./openai";
 import { anthropicProvider } from "./anthropic";
 import { costs as awsBedrockCosts } from "./awsBedrock";
 import { costs as deepseekCosts } from "./deepseek";
+import { costs as xCosts } from "./x";
 
 const openAiPattern = /^https:\/\/api\.openai\.com/;
 const anthropicPattern = /^https:\/\/api\.anthropic\.com/;
@@ -52,6 +53,8 @@ const firecrawl = /^https:\/\/api\.firecrawl\.dev/;
 const awsBedrock = /^https:\/\/bedrock-runtime\.[a-z0-9-]+\.amazonaws\.com\/.*/;
 // https://api.deepseek.com
 const deepseek = /^https:\/\/api\.deepseek\.com/;
+// https://api.x.ai
+const x = /^https:\/\/api\.x\.ai/;
 
 export const providersNames = [
   "OPENAI",
@@ -78,6 +81,7 @@ export const providersNames = [
   "FIRECRAWL",
   "AWS",
   "DEEPSEEK",
+  "X",
 ] as const;
 
 export type ProviderName = (typeof providersNames)[number];
@@ -127,6 +131,11 @@ export const providers: {
     pattern: cloudflareAiGatewayPattern,
     provider: "CLOUDFLARE",
   },
+  {
+    pattern: x,
+    provider: "X",
+    costs: xCosts,
+  },
   {
     pattern: twoYFV,
     provider: "2YFV",

diff --git a/valhalla/jawn/src/packages/cost/providers/x/index.ts b/valhalla/jawn/src/packages/cost/providers/x/index.ts
@@ -0,0 +1,49 @@
+/**
+ *
+ * DO NOT EDIT THIS FILE UNLESS IT IS IN /costs
+ */
+
+import { ModelRow } from "../../interfaces/Cost";
+
+export const costs: ModelRow[] = [
+  {
+    model: {
+      operator: "equals",
+      value: "grok-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-vision-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-vision-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+];
diff --git a/web/packages/cost/providers/x/index.ts b/web/packages/cost/providers/x/index.ts
@@ -0,0 +1,49 @@
+/**
+ *
+ * DO NOT EDIT THIS FILE UNLESS IT IS IN /costs
+ */
+
+import { ModelRow } from "../../interfaces/Cost";
+
+export const costs: ModelRow[] = [
+  {
+    model: {
+      operator: "equals",
+      value: "grok-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-vision-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-vision-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+];
diff --git a/worker/src/packages/cost/providers/x/index.ts b/worker/src/packages/cost/providers/x/index.ts
@@ -0,0 +1,49 @@
+/**
+ *
+ * DO NOT EDIT THIS FILE UNLESS IT IS IN /costs
+ */
+
+import { ModelRow } from "../../interfaces/Cost";
+
+export const costs: ModelRow[] = [
+  {
+    model: {
+      operator: "equals",
+      value: "grok-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-vision-beta",
+    },
+    cost: {
+      prompt_token: 0.000005,
+      completion_token: 0.000015,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+  {
+    model: {
+      operator: "equals",
+      value: "grok-2-vision-1212",
+    },
+    cost: {
+      prompt_token: 0.000002,
+      completion_token: 0.00001,
+    },
+  },
+];