From cd078a62640bf10158392c092c5303989387cf04 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=91=86=E8=90=8C=E9=97=B7=E6=B2=B9=E7=93=B6?=
 <253605712@qq.com>
Date: Fri, 2 Feb 2024 12:33:11 +0800
Subject: [PATCH] feat:add gpt-3.5-turbo-0125 (#2347)

---
 .../model_providers/openai/llm/_position.yaml |  1 +
 .../openai/llm/gpt-3.5-turbo-0125.yaml        | 31 +++++++++++++++++++
 .../openai/llm/gpt-3.5-turbo-1106.yaml        |  2 +-
 3 files changed, 33 insertions(+), 1 deletion(-)
 create mode 100644 api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-0125.yaml

diff --git a/api/core/model_runtime/model_providers/openai/llm/_position.yaml b/api/core/model_runtime/model_providers/openai/llm/_position.yaml
index 3c32708417..ceb7d7cd77 100644
--- a/api/core/model_runtime/model_providers/openai/llm/_position.yaml
+++ b/api/core/model_runtime/model_providers/openai/llm/_position.yaml
@@ -4,6 +4,7 @@
 - gpt-4-1106-preview
 - gpt-4-0125-preview
 - gpt-4-vision-preview
+- gpt-3.5-turbo-0125
 - gpt-3.5-turbo
 - gpt-3.5-turbo-16k
 - gpt-3.5-turbo-16k-0613
diff --git a/api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-0125.yaml b/api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-0125.yaml
new file mode 100644
index 0000000000..3e40db01f9
--- /dev/null
+++ b/api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-0125.yaml
@@ -0,0 +1,31 @@
+model: gpt-3.5-turbo-0125
+label:
+  zh_Hans: gpt-3.5-turbo-0125
+  en_US: gpt-3.5-turbo-0125
+model_type: llm
+features:
+  - multi-tool-call
+  - agent-thought
+  - stream-tool-call
+model_properties:
+  mode: chat
+  context_size: 16385
+parameter_rules:
+  - name: temperature
+    use_template: temperature
+  - name: top_p
+    use_template: top_p
+  - name: presence_penalty
+    use_template: presence_penalty
+  - name: frequency_penalty
+    use_template: frequency_penalty
+  - name: max_tokens
+    use_template: max_tokens
+    default: 512
+    min: 1
+    max: 4096
+pricing:
+  input: '0.0005'
+  output: '0.0015'
+  unit: '0.001'
+  currency: USD
diff --git a/api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-1106.yaml b/api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-1106.yaml
index 4b5d31c774..499792e39d 100644
--- a/api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-1106.yaml
+++ b/api/core/model_runtime/model_providers/openai/llm/gpt-3.5-turbo-1106.yaml
@@ -23,7 +23,7 @@ parameter_rules:
     use_template: max_tokens
     default: 512
     min: 1
-    max: 16385
+    max: 4096
 pricing:
   input: '0.001'
   output: '0.002'