From 3fc0ebdd51251faf3c96cdd4163293a0d97c315f Mon Sep 17 00:00:00 2001
From: zhuhao <37029601+hwzhuhao@users.noreply.github.com>
Date: Fri, 18 Oct 2024 08:19:58 +0800
Subject: [PATCH] feat: add yi-lightning llm model for yi (#9458)

---
 .../model_providers/yi/llm/_position.yaml     |  1 +
 .../model_providers/yi/llm/yi-lightning.yaml  | 43 +++++++++++++++++++
 2 files changed, 44 insertions(+)
 create mode 100644 api/core/model_runtime/model_providers/yi/llm/yi-lightning.yaml

diff --git a/api/core/model_runtime/model_providers/yi/llm/_position.yaml b/api/core/model_runtime/model_providers/yi/llm/_position.yaml
index e876893b41..5fa098beda 100644
--- a/api/core/model_runtime/model_providers/yi/llm/_position.yaml
+++ b/api/core/model_runtime/model_providers/yi/llm/_position.yaml
@@ -7,3 +7,4 @@
 - yi-medium-200k
 - yi-spark
 - yi-large-turbo
+- yi-lightning
diff --git a/api/core/model_runtime/model_providers/yi/llm/yi-lightning.yaml b/api/core/model_runtime/model_providers/yi/llm/yi-lightning.yaml
new file mode 100644
index 0000000000..fccf1b3a26
--- /dev/null
+++ b/api/core/model_runtime/model_providers/yi/llm/yi-lightning.yaml
@@ -0,0 +1,43 @@
+model: yi-lightning
+label:
+  zh_Hans: yi-lightning
+  en_US: yi-lightning
+model_type: llm
+features:
+  - agent-thought
+model_properties:
+  mode: chat
+  context_size: 16384
+parameter_rules:
+  - name: temperature
+    use_template: temperature
+    type: float
+    default: 0.3
+    min: 0.0
+    max: 2.0
+    help:
+      zh_Hans: 控制生成结果的多样性和随机性。数值越小，越严谨；数值越大，越发散。
+      en_US: Control the diversity and randomness of generated results. The smaller the value, the more rigorous it is; the larger the value, the more divergent it is.
+  - name: max_tokens
+    use_template: max_tokens
+    type: int
+    default: 1024
+    min: 1
+    max: 4000
+    help:
+      zh_Hans: 指定生成结果长度的上限。如果生成结果截断，可以调大该参数。
+      en_US: Specifies the upper limit on the length of generated results. If the generated results are truncated, you can increase this parameter.
+  - name: top_p
+    use_template: top_p
+    type: float
+    default: 0.9
+    min: 0.01
+    max: 1.00
+    help:
+      zh_Hans: 控制生成结果的随机性。数值越小，随机性越弱；数值越大，随机性越强。一般而言，top_p 和 temperature 两个参数选择一个进行调整即可。
+      en_US: Control the randomness of generated results. The smaller the value, the weaker the randomness; the larger the value, the stronger the randomness. Generally speaking, you can adjust one of the two parameters top_p and temperature.
+pricing:
+  input: '0.99'
+  output: '0.99'
+  unit: '0.000001'
+  currency: RMB