From beeb4ce78e3e77995a5976d847be22d9f528b1db Mon Sep 17 00:00:00 2001
From: sijinhui <sijinhui@qq.com>
Date: Sat, 7 Sep 2024 23:36:48 +0800
Subject: [PATCH] add model

---
 app/components/ui-lib.tsx |  7 +++++++
 app/constant.ts           | 18 +++++++++++++++++-
 app/icons/Moonshot.svg    |  1 +
 3 files changed, 25 insertions(+), 1 deletion(-)
 create mode 100644 app/icons/Moonshot.svg

diff --git a/app/components/ui-lib.tsx b/app/components/ui-lib.tsx
index eb7b70629..bac206535 100644
--- a/app/components/ui-lib.tsx
+++ b/app/components/ui-lib.tsx
@@ -34,6 +34,10 @@ import {
   Button,
 } from "antd";
 import { OpenAIOutlined, GoogleOutlined } from "@ant-design/icons";
+// 自定义图标
+import Icon from "@ant-design/icons";
+import MoonShot from "@/app/icons/Moonshot.svg";
+
 const { Meta } = AntCard;
 import { CheckCard } from "@ant-design/pro-components";
 import { CheckGroupValueType } from "@ant-design/pro-card/es/components/CheckCard/Group";
@@ -613,6 +617,9 @@ export function ModalSelector<T extends CheckGroupValueType>(props: {
     if (value.startsWith("claude")) {
       return <ClaudeInstantIcon />;
     }
+    if (value.startsWith("moon")) {
+      return <Icon component={MoonShot} />;
+    }
     return <></>;
   };
 
diff --git a/app/constant.ts b/app/constant.ts
index 058bdd267..0c482b569 100644
--- a/app/constant.ts
+++ b/app/constant.ts
@@ -338,7 +338,11 @@ const tencentModels = [
   "hunyuan-vision",
 ];
 
-const moonshotModes = ["moonshot-v1-8k", "moonshot-v1-32k", "moonshot-v1-128k"];
+const moonshotModes = [
+  "moonshot-v1-8k",
+  // "moonshot-v1-32k",
+  // "moonshot-v1-128k"
+];
 
 const iflytekModels = [
   "general",
@@ -446,6 +450,18 @@ export const DEFAULT_MODELS = [
       sorted: 3,
     },
   },
+  ...moonshotModes.map((name) => ({
+    name,
+    available: true,
+    sorted: seq++,
+    describe: "月之暗面,可以试试",
+    provider: {
+      id: "moonshot",
+      providerName: "Moonshot",
+      providerType: "moonshot",
+      sorted: 9,
+    },
+  })),
   {
     name: "midjourney",
     describe: "绘图用,不用选",
diff --git a/app/icons/Moonshot.svg b/app/icons/Moonshot.svg
new file mode 100644
index 000000000..59865a011
--- /dev/null
+++ b/app/icons/Moonshot.svg
@@ -0,0 +1 @@
+<?xml version="1.0" standalone="no"?><!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd"><svg t="1725722339262" class="icon" viewBox="0 0 1024 1024" version="1.1" xmlns="http://www.w3.org/2000/svg" p-id="5170" xmlns:xlink="http://www.w3.org/1999/xlink" width="200" height="200"><path d="M572.3 404.2c-14 24.3-27.6 48-42.3 73.5 141.5 37.9 281.3 75.4 423.8 113.6-8.1 29-16.1 57.4-24.5 87.1-288-77.1-573.8-153.6-860.4-230.4 5.3-37.3 14.7-72.5 29.2-106.3 7.3-17 16.1-33.3 24-50 2.7-5.6 5.9-7.4 12.6-5.6 142.4 38.5 285 76.6 427.5 114.9 3.2 1 6.5 2.1 10.1 3.2zM956.1 476.9c-270.8-72.5-541.6-145-816.1-218.4 21.2-23.9 40-46.7 60.5-67.7 25.8-26.3 55.4-48.2 87.7-66 3.9-2.1 9.6-3.2 13.8-2.1 136 36.2 271.9 72.8 407.9 109.3 0.6 0.2 1.1 0.7 2.9 1.9-25.4 18.2-48.2 38.6-67.3 64.9 28.5 7.7 55.5 15.1 82.5 22.4 67.3 18 134.6 36.1 201.9 53.9 6.2 1.6 10 4.2 11.3 10.7 6.1 29.5 12.4 58.9 18.7 88.4-1.2 0.9-2.5 1.8-3.8 2.7zM873.3 780.6c-22.3 22.9-44.5 45.9-67.1 68.5-2.2 2.2-7.8 2.7-11.2 1.8-90.3-23.9-180.6-48.2-270.9-72.3-123.6-33.1-247.2-66.1-370.8-99.1-18.9-5-37.8-10-56.5-15.4-3.2-0.9-7.3-3.8-8.3-6.7-20-56.9-27.1-115.5-23.6-175.6 0.1-1 0.7-2 1.6-4.8 140.9 37.8 281.7 75.6 422.8 113.4-6.2 26.4-12.3 52.3-18.7 79.2 134.2 35.9 267.2 71.6 400.2 107.2 0.8 1.3 1.7 2.6 2.5 3.8zM672.3 929.4c-214.4 85.7-475.1-16.2-567.3-233.2 20.9 5.5 41.1 10.8 61.1 16.2 93.7 25.2 187.3 50.6 281.1 75.5 7.1 1.9 8.8 4.5 9.1 11.3 0.9 21.3 3 42.6 3.8 63.9 0.3 8.4 4.6 10.6 11.5 12.4 62.5 16.5 125 33.3 187.5 50 3.5 1 7.1 2.2 13.2 3.9zM868.3 244.7c-176.9-47.3-353.8-94.7-530.7-142 73.5-34 150.4-45.2 230.1-34.5 124.5 16.7 224.8 75.6 300.6 176.5z" p-id="5171"></path></svg>
\ No newline at end of file