From b189faca5200ead8061c0b7d348accfdf4876b31 Mon Sep 17 00:00:00 2001
From: Weaxs <459312872@qq.com>
Date: Wed, 29 May 2024 14:57:23 +0800
Subject: [PATCH] feat: update ernie model (#4756)

---
 .../wenxin/llm/ernie-3.5-128k.yaml            | 37 +++++++++++++++++++
 .../wenxin/llm/ernie-3.5-4k-0205.yaml         |  1 +
 .../wenxin/llm/ernie-3.5-8k-1222.yaml         |  1 +
 .../model_providers/wenxin/llm/ernie_bot.py   |  5 ++-
 4 files changed, 43 insertions(+), 1 deletion(-)
 create mode 100644 api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-128k.yaml

diff --git a/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-128k.yaml b/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-128k.yaml
new file mode 100644
index 0000000000..b1b1ba1f69
--- /dev/null
+++ b/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-128k.yaml
@@ -0,0 +1,37 @@
+model: ernie-3.5-128k
+label:
+  en_US: Ernie-3.5-128K
+model_type: llm
+features:
+  - agent-thought
+model_properties:
+  mode: chat
+  context_size: 128000
+parameter_rules:
+  - name: temperature
+    use_template: temperature
+    min: 0.1
+    max: 1.0
+    default: 0.8
+  - name: top_p
+    use_template: top_p
+  - name: max_tokens
+    use_template: max_tokens
+    default: 4096
+    min: 2
+    max: 4096
+  - name: presence_penalty
+    use_template: presence_penalty
+  - name: frequency_penalty
+    use_template: frequency_penalty
+  - name: response_format
+    use_template: response_format
+  - name: disable_search
+    label:
+      zh_Hans: 禁用搜索
+      en_US: Disable Search
+    type: boolean
+    help:
+      zh_Hans: 禁用模型自行进行外部搜索。
+      en_US: Disable the model to perform external search.
+    required: false
diff --git a/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-4k-0205.yaml b/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-4k-0205.yaml
index 9487342a1d..1e8cf96440 100644
--- a/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-4k-0205.yaml
+++ b/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-4k-0205.yaml
@@ -35,3 +35,4 @@ parameter_rules:
       zh_Hans: 禁用模型自行进行外部搜索。
       en_US: Disable the model to perform external search.
     required: false
+deprecated: true
diff --git a/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-8k-1222.yaml b/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-8k-1222.yaml
index 5dfcd5825b..c43588cfe1 100644
--- a/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-8k-1222.yaml
+++ b/api/core/model_runtime/model_providers/wenxin/llm/ernie-3.5-8k-1222.yaml
@@ -35,3 +35,4 @@ parameter_rules:
       zh_Hans: 禁用模型自行进行外部搜索。
       en_US: Disable the model to perform external search.
     required: false
+deprecated: true
diff --git a/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py b/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py
index a6d0bbb396..4646ba384a 100644
--- a/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py
+++ b/api/core/model_runtime/model_providers/wenxin/llm/ernie_bot.py
@@ -129,6 +129,7 @@ class ErnieBotModel:
         'ernie-3.5-8k-0205': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-3.5-8k-0205',
         'ernie-3.5-8k-1222': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-3.5-8k-1222',
         'ernie-3.5-4k-0205': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-3.5-4k-0205',
+        'ernie-3.5-128k': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-3.5-128k',
         'ernie-4.0-8k': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro',
         'ernie-speed-8k': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie_speed',
         'ernie-speed-128k': 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/ernie-speed-128k',
@@ -144,7 +145,9 @@ class ErnieBotModel:
         'ernie-3.5-8k',
         'ernie-3.5-8k-0205',
         'ernie-3.5-8k-1222',
-        'ernie-3.5-4k-0205'
+        'ernie-3.5-4k-0205',
+        'ernie-3.5-128k',
+        'ernie-4.0-8k'
     ]
 
     api_key: str = ''