feat: add gemini-1.5-pro-latest and gemini-1.5-flash-latest model

coaidev · May 20, 2024 · aa06dab · aa06dab
1 parent 96fad63
commit aa06dab
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/app/src/admin/channel.ts b/app/src/admin/channel.ts
@@ -244,7 +244,13 @@ export const ChannelInfos: Record<string, ChannelInfo> = {
   palm: {
     endpoint: "https://generativelanguage.googleapis.com",
     format: "<api-key>",
-    models: ["chat-bison-001", "gemini-pro", "gemini-pro-vision"],
+    models: [
+      "chat-bison-001",
+      "gemini-pro",
+      "gemini-pro-vision",
+      "gemini-1.5-pro-latest",
+      "gemini-1.5-flash-latest",
+    ],
     description:
       "> Google Gemini / PaLM2 密钥格式为 **api-key**，接入点填写 *https://generativelanguage.googleapis.com* 或其反代地址 \n" +
       "> Google 对请求 IP 地域有限制，可能出现 **User Location Is Not Supported** 的错误，可以看运气通过反代解决。 \n" +

diff --git a/app/src/admin/colors.ts b/app/src/admin/colors.ts
@@ -65,6 +65,8 @@ export const modelColorMapper: Record<string, string> = {
   "chat-bison-001": "red-500",
   "gemini-pro": "red-500",
   "gemini-pro-vision": "red-500",
+  "gemini-1.5-pro-latest": "red-500",
+  "gemini-1.5-flash-latest": "red-500",
 
   "bing-creative": "blue-700",
   "bing-balanced": "blue-700",

diff --git a/app/src/admin/datasets/charge.ts b/app/src/admin/datasets/charge.ts
@@ -214,7 +214,12 @@ export const pricing: PricingDataset = [
     output: 0.001,
   },
   {
-    models: ["gemini-pro", "gemini-pro-vision"],
+    models: [
+      "gemini-pro",
+      "gemini-pro-vision",
+      "gemini-1.5-pro-latest",
+      "gemini-1.5-flash-latest",
+    ],
     input: 0.000125,
     output: 0.000375,
   },