feat: add Mistral AI cost table and vendor integration (#357)

* feat: add Mistral AI cost table and vendor integration

* feat: add Mistral AI cost table

* fix: address PR comments for Mistral AI integration

* feat: add Mistral handling in litellm condition and remove duplicate XAI logo code

* Update model check to 'tral'

* Update vendor detection to use 'tral' for Mistral models for consistency

---------

Co-authored-by: devin-ai-integration[bot] <158243242+devin-ai-integration[bot]@users.noreply.github.com>

components/shared/vendor-metadata.tsx

@@ -367,6 +367,8 @@ export function VendorLogo({
     );
   }
 
+
+
   if (vendor.includes("xai")) {
     const color = vendorColor("vercel");
     return (

lib/constants.ts

@@ -146,6 +146,49 @@ export const OPENAI_PRICING: Record<string, CostTableEntry> = {
 
 };
 
+export const MISTRAL_PRICING: Record<string, CostTableEntry> = {
+  "mistral-large-latest": {
+    input: 0.002,
+    output: 0.006,
+  },
+  "pixtral-large-latest": {
+    input: 0.002,
+    output: 0.006,
+  },
+  "mistral-small-latest": {
+    input: 0.0002,
+    output: 0.0006,
+  },
+  "codestral-latest": {
+    input: 0.0002,
+    output: 0.0006,
+  },
+  "ministral-8b-latest": {
+    input: 0.0001,
+    output: 0.0001,
+  },
+  "ministral-3b-latest": {
+    input: 0.00004,
+    output: 0.00004,
+  },
+  "mistral-embed": {
+    input: 0.0001,
+    output: 0,
+  },
+  "open-mistral-7b": {
+    input: 0.00025,
+    output: 0.00025,
+  },
+  "open-mixtral-8x7b": {
+    input: 0.0007,
+    output: 0.0007,
+  },
+  "open-mixtral-8x22b": {
+    input: 0.002,
+    output: 0.006,
+  },
+};
+
 export const XAI_PRICING: Record<string, CostTableEntry> = {
   "grok-beta": {
     input: 0.005,

lib/utils.ts

@@ -16,6 +16,7 @@ import {
   CostTableEntry,
   GROQ_PRICING,
   LangTraceAttributes,
+  MISTRAL_PRICING,
   OPENAI_PRICING,
   PERPLEXITY_PRICING,
   SpanStatusCode,
@@ -526,8 +527,8 @@ export function calculatePriceFromUsage(
       vendor = "openai";
     } else if (model.includes("claude")) {
       vendor = "anthropic";
-    } else if (model.includes("mistral")) {
-      vendor = "mistral"; // Assuming there is a MISTRAL_PRICING object
+    } else if (model.includes("tral")) {
+      vendor = "mistral";
     } else if (model.includes("grok")) {
       vendor = "xai";
     } else if (model.includes("gemini")) {
@@ -581,6 +582,8 @@ export function calculatePriceFromUsage(
       costTable = GEMINI_PRICING[model];
     } else if (model.includes("grok")) {
       costTable = XAI_PRICING[model];
+    } else if (model.includes("tral")) {
+      costTable = MISTRAL_PRICING[model];
     }
   } else if (vendor === "openai") {
     // check if model is present as key in OPENAI_PRICING
@@ -642,6 +645,8 @@ export function calculatePriceFromUsage(
     costTable = GEMINI_PRICING[model];
   } else if (vendor === "xai") {
     costTable = XAI_PRICING[model];
+  } else if (vendor === "mistral") {
+    costTable = MISTRAL_PRICING[model];
   }
   if (costTable) {
     const total =