From d14ba2ed9eaaa87bb038e12825a980211b7e5c20 Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Fri, 12 Jul 2024 16:30:33 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E5=A2=9E=E5=8A=A0=E3=80=91ChatGlm=20?=
 =?UTF-8?q?=E5=AF=B9=E6=8E=A5=20Image?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/views/ai/image/index/components/other/index.vue |  9 +++++++--
 src/views/ai/utils/constants.ts                     | 12 ++++++++++++
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/views/ai/image/index/components/other/index.vue b/src/views/ai/image/index/components/other/index.vue
index 861a1563..352333db 100644
--- a/src/views/ai/image/index/components/other/index.vue
+++ b/src/views/ai/image/index/components/other/index.vue
@@ -61,7 +61,7 @@
       </el-select>
     </el-space>
   </div>
-  <div class="group-item">
+  <div class="group-item" v-if="otherPlatform !== AiPlatformEnum.CHATGLM">
     <div>
       <el-text tag="b">图片尺寸</el-text>
     </div>
@@ -79,7 +79,7 @@
 <script setup lang="ts">
 import {ImageApi, ImageDrawReqVO, ImageVO} from '@/api/ai/image'
 import {
-  AiPlatformEnum,
+  AiPlatformEnum, ChatGlmModels,
   ImageHotWords,
   ImageModelVO,
   OtherPlatformEnum,
@@ -159,6 +159,11 @@ const handlerPlatformChange = async (platform) => {
     models.value = TongYiWanXiangModels
   } else if (AiPlatformEnum.YI_YAN === platform) {
     models.value = QianFanModels
+  } else if (AiPlatformEnum.CHATGLM === platform) {
+    models.value = ChatGlmModels
+    // ChatGlm 模型没有 width、height 随便默认一个值过后台必填检测
+    height.value = 512
+    width.value = 512
   } else {
     models.value = []
   }
diff --git a/src/views/ai/utils/constants.ts b/src/views/ai/utils/constants.ts
index d33493e2..3c886389 100644
--- a/src/views/ai/utils/constants.ts
+++ b/src/views/ai/utils/constants.ts
@@ -21,6 +21,7 @@ export const AiPlatformEnum = {
   STABLE_DIFFUSION: 'StableDiffusion', // Stability AI
   MIDJOURNEY: 'Midjourney', // Midjourney
   SUNO: 'Suno', // Suno AI
+  CHATGLM: 'ChatGlm', // ChatGlm
 }
 
 export const OtherPlatformEnum:ImageModelVO [] = [
@@ -31,6 +32,10 @@ export const OtherPlatformEnum:ImageModelVO [] = [
   {
     key: AiPlatformEnum.YI_YAN,
     name: '百度千帆'
+  },
+  {
+    key: AiPlatformEnum.CHATGLM,
+    name: '智铺 ChatGlm'
   }
 ]
 
@@ -266,6 +271,13 @@ export const QianFanModels: ImageModelVO[] = [
   }
 ]
 
+export const ChatGlmModels: ImageModelVO[] = [
+  {
+    key: 'cogview-3',
+    name: 'cogview-3'
+  }
+]
+
 
 export const StableDiffusionClipGuidancePresets: ImageModelVO[] = [
   {