From ad130110430b4a96a5a63f508fe98b0213e4a200 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=99=A2=E5=93=8E=E5=93=9F=E5=96=82?= <i@oiuv.cn>
Date: Fri, 23 Aug 2024 16:24:45 +0800
Subject: [PATCH] add JSON Mode support for moonshot models (#7568)

---
 .../moonshot/llm/moonshot-v1-128k.yaml               | 12 ++++++++++++
 .../moonshot/llm/moonshot-v1-32k.yaml                | 12 ++++++++++++
 .../model_providers/moonshot/llm/moonshot-v1-8k.yaml | 12 ++++++++++++
 3 files changed, 36 insertions(+)

diff --git a/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-128k.yaml b/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-128k.yaml
index 0d2e51c47f..1078e84c59 100644
--- a/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-128k.yaml
+++ b/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-128k.yaml
@@ -21,6 +21,18 @@ parameter_rules:
     default: 1024
     min: 1
     max: 128000
+  - name: response_format
+    label:
+      zh_Hans: 回复格式
+      en_US: response_format
+    type: string
+    help:
+      zh_Hans: 指定模型必须输出的格式
+      en_US: specifying the format that the model must output
+    required: false
+    options:
+      - text
+      - json_object
 pricing:
   input: '0.06'
   output: '0.06'
diff --git a/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-32k.yaml b/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-32k.yaml
index 9ff537014a..9c739d0501 100644
--- a/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-32k.yaml
+++ b/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-32k.yaml
@@ -21,6 +21,18 @@ parameter_rules:
     default: 1024
     min: 1
     max: 32000
+  - name: response_format
+    label:
+      zh_Hans: 回复格式
+      en_US: response_format
+    type: string
+    help:
+      zh_Hans: 指定模型必须输出的格式
+      en_US: specifying the format that the model must output
+    required: false
+    options:
+      - text
+      - json_object
 pricing:
   input: '0.024'
   output: '0.024'
diff --git a/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-8k.yaml b/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-8k.yaml
index 0f308d3676..187a86999e 100644
--- a/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-8k.yaml
+++ b/api/core/model_runtime/model_providers/moonshot/llm/moonshot-v1-8k.yaml
@@ -21,6 +21,18 @@ parameter_rules:
     default: 512
     min: 1
     max: 8192
+  - name: response_format
+    label:
+      zh_Hans: 回复格式
+      en_US: response_format
+    type: string
+    help:
+      zh_Hans: 指定模型必须输出的格式
+      en_US: specifying the format that the model must output
+    required: false
+    options:
+      - text
+      - json_object
 pricing:
   input: '0.012'
   output: '0.012'