From 76dd2ec16da77554a6f37e2ddf8ad62ee1aca63a Mon Sep 17 00:00:00 2001
From: LovelyGuYiMeng <76251800+LovelyGuYiMeng@users.noreply.github.com>
Date: Fri, 13 Sep 2024 18:12:37 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=92=84=20style:=20update=20zhipu=20model?=
 =?UTF-8?q?=20info=20(#3949)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/config/modelProviders/zhipu.ts | 83 ++++++++++++++++++++++++++----
 1 file changed, 74 insertions(+), 9 deletions(-)

diff --git a/src/config/modelProviders/zhipu.ts b/src/config/modelProviders/zhipu.ts
index 2b83630b5f4d4..1abe5539d700e 100644
--- a/src/config/modelProviders/zhipu.ts
+++ b/src/config/modelProviders/zhipu.ts
@@ -12,6 +12,11 @@ const ZhiPu: ModelProviderCard = {
       enabled: true,
       functionCall: true,
       id: 'glm-4-alltools',
+      pricing: {
+        currency: 'CNY',
+        input: 100,
+        output: 100,
+      },
       tokens: 128_000,
     },
     {
@@ -21,14 +26,23 @@ const ZhiPu: ModelProviderCard = {
       enabled: true,
       functionCall: true,
       id: 'glm-4-plus',
+      pricing: {
+        currency: 'CNY',
+        input: 50,
+        output: 50,
+      },
       tokens: 128_000,
     },
     {
       description: 'GLM-4-0520 是最新模型版本，专为高度复杂和多样化任务设计，表现卓越。',
       displayName: 'GLM-4-0520',
-      enabled: true,
       functionCall: true,
       id: 'glm-4-0520',
+      pricing: {
+        currency: 'CNY',
+        input: 100,
+        output: 100,
+      },
       tokens: 128_000,
     },
     {
@@ -36,29 +50,50 @@ const ZhiPu: ModelProviderCard = {
       displayName: 'GLM-4',
       functionCall: true,
       id: 'glm-4',
+      pricing: {
+        currency: 'CNY',
+        input: 100,
+        output: 100,
+      },
       tokens: 128_000,
     },
+    {
+      description: 'GLM-4-AirX 提供 GLM-4-Air 的高效版本，推理速度可达其2.6倍。',
+      displayName: 'GLM-4-AirX',
+      enabled: true,
+      functionCall: true,
+      id: 'glm-4-airx',
+      pricing: {
+        currency: 'CNY',
+        input: 10,
+        output: 10,
+      },
+      tokens: 8192,
+    },
     {
       description: 'GLM-4-Air 是性价比高的版本，性能接近GLM-4，提供快速度和实惠的价格。',
       displayName: 'GLM-4-Air',
       enabled: true,
       functionCall: true,
       id: 'glm-4-air',
+      pricing: {
+        currency: 'CNY',
+        input: 1,
+        output: 1,
+      },
       tokens: 128_000,
     },
-    {
-      description: 'GLM-4-AirX 提供 GLM-4-Air 的高效版本，推理速度可达其2.6倍。',
-      displayName: 'GLM-4-AirX',
-      functionCall: true,
-      id: 'glm-4-airx',
-      tokens: 8192,
-    },
     {
       description: 'GLM-4-Long 支持超长文本输入，适合记忆型任务与大规模文档处理。',
       displayName: 'GLM-4-Long',
       enabled: true,
       functionCall: true,
       id: 'glm-4-long',
+      pricing: {
+        currency: 'CNY',
+        input: 1,
+        output: 1,
+      },
       tokens: 1_024_000,
     },
     {
@@ -67,6 +102,11 @@ const ZhiPu: ModelProviderCard = {
       enabled: true,
       functionCall: true,
       id: 'glm-4-flash',
+      pricing: {
+        currency: 'CNY',
+        input: 0,
+        output: 0,
+      },
       tokens: 128_000,
     },
     {
@@ -74,14 +114,23 @@ const ZhiPu: ModelProviderCard = {
       displayName: 'GLM-4V-Plus',
       enabled: true,
       id: 'glm-4v-plus',
+      pricing: {
+        currency: 'CNY',
+        input: 10,
+        output: 10,
+      },
       tokens: 8192,
       vision: true,
     },
     {
       description: 'GLM-4V 提供强大的图像理解与推理能力，支持多种视觉任务。',
       displayName: 'GLM-4V',
-      enabled: true,
       id: 'glm-4v',
+      pricing: {
+        currency: 'CNY',
+        input: 50,
+        output: 50,
+      },
       tokens: 2048,
       vision: true,
     },
@@ -89,19 +138,35 @@ const ZhiPu: ModelProviderCard = {
       description:
         'CodeGeeX-4 是强大的AI编程助手，支持多种编程语言的智能问答与代码补全，提升开发效率。',
       displayName: 'CodeGeeX-4',
+      enabled: true,
       id: 'codegeex-4',
+      pricing: {
+        currency: 'CNY',
+        input: 0.1,
+        output: 0.1,
+      },
       tokens: 128_000,
     },
     {
       description: 'CharGLM-3 专为角色扮演与情感陪伴设计，支持超长多轮记忆与个性化对话，应用广泛。',
       displayName: 'CharGLM-3',
       id: 'charglm-3',
+      pricing: {
+        currency: 'CNY',
+        input: 15,
+        output: 15,
+      },
       tokens: 4096,
     },
     {
       description: 'Emohaa 是心理模型，具备专业咨询能力，帮助用户理解情感问题。',
       displayName: 'Emohaa',
       id: 'emohaa',
+      pricing: {
+        currency: 'CNY',
+        input: 15,
+        output: 15,
+      },
       tokens: 8192,
     },
   ],