feat: support deepseek

feat: support deepseek Co-Authored-By: Minghan Zhang <[email protected]>
yangboz · Jun 29, 2024 · bad4689 · bad4689
1 parent afb13dd
commit bad4689
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/app/src/admin/colors.ts b/app/src/admin/colors.ts
@@ -68,6 +68,9 @@ export const modelColorMapper: Record<string, string> = {
   "gemini-1.5-pro-latest": "red-500",
   "gemini-1.5-flash-latest": "red-500",
 
+  "deepseek-coder": "blue-700",
+  "deepseek-chat": "blue-700",
+
   "bing-creative": "blue-700",
   "bing-balanced": "blue-700",
   "bing-precise": "blue-700",

diff --git a/app/src/admin/datasets/charge.ts b/app/src/admin/datasets/charge.ts
@@ -229,6 +229,12 @@ export const pricing: PricingDataset = [
     output: 0.1,
     currency: Currency.CNY,
   },
+  {
+    models: ["deepseek-chat", "deepseek-coder"],
+    input: 0.001,
+    output: 0.002,
+    currency: Currency.CNY,
+  },
   {
     models: ["360-gpt-v9"],
     input: 0.12,