From f288d367ac79151e9be7d238633cf4b7e82e337d Mon Sep 17 00:00:00 2001
From: orangeclk <orangeclk@users.noreply.github.com>
Date: Thu, 8 Aug 2024 14:17:05 +0800
Subject: [PATCH] Add price info for zhipu models (#7084)

---
 .../model_providers/zhipuai/llm/glm-4-0520.yaml              | 5 +++++
 .../model_runtime/model_providers/zhipuai/llm/glm-4-air.yaml | 5 +++++
 .../model_providers/zhipuai/llm/glm-4-airx.yaml              | 5 +++++
 .../model_providers/zhipuai/llm/glm-4-flash.yaml             | 5 +++++
 .../model_providers/zhipuai/text_embedding/embedding-2.yaml  | 4 ++++
 5 files changed, 24 insertions(+)

diff --git a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-0520.yaml b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-0520.yaml
index 3968e8f268..8391278e4f 100644
--- a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-0520.yaml
+++ b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-0520.yaml
@@ -37,3 +37,8 @@ parameter_rules:
     default: 1024
     min: 1
     max: 8192
+pricing:
+  input: '0.1'
+  output: '0.1'
+  unit: '0.001'
+  currency: RMB
diff --git a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-air.yaml b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-air.yaml
index ae2d5e5d53..7caebd3e4b 100644
--- a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-air.yaml
+++ b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-air.yaml
@@ -37,3 +37,8 @@ parameter_rules:
     default: 1024
     min: 1
     max: 8192
+pricing:
+  input: '0.001'
+  output: '0.001'
+  unit: '0.001'
+  currency: RMB
diff --git a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-airx.yaml b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-airx.yaml
index c0038a1ab2..dc123913de 100644
--- a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-airx.yaml
+++ b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-airx.yaml
@@ -37,3 +37,8 @@ parameter_rules:
     default: 1024
     min: 1
     max: 8192
+pricing:
+  input: '0.01'
+  output: '0.01'
+  unit: '0.001'
+  currency: RMB
diff --git a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-flash.yaml b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-flash.yaml
index 650f9faee6..1b1d499ba7 100644
--- a/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-flash.yaml
+++ b/api/core/model_runtime/model_providers/zhipuai/llm/glm-4-flash.yaml
@@ -37,3 +37,8 @@ parameter_rules:
     default: 1024
     min: 1
     max: 8192
+pricing:
+  input: '0.0001'
+  output: '0.0001'
+  unit: '0.001'
+  currency: RMB
diff --git a/api/core/model_runtime/model_providers/zhipuai/text_embedding/embedding-2.yaml b/api/core/model_runtime/model_providers/zhipuai/text_embedding/embedding-2.yaml
index faf0f818c4..72188e3424 100644
--- a/api/core/model_runtime/model_providers/zhipuai/text_embedding/embedding-2.yaml
+++ b/api/core/model_runtime/model_providers/zhipuai/text_embedding/embedding-2.yaml
@@ -2,3 +2,7 @@ model: embedding-2
 model_type: text-embedding
 model_properties:
   context_size: 512
+pricing:
+  input: '0.0005'
+  unit: '0.001'
+  currency: RMB