feat: support OPENAI json_schema (#7258)

2024-08-15 11:29:19 +08:00 · 2024-08-15 11:29:19 +08:00 · 6ff7fd80a1
commit 6ff7fd80a1
parent 5aa373dc04
6 changed files with 44 additions and 10 deletions
--- a/api/core/model_runtime/entities/defaults.py
+++ b/api/core/model_runtime/entities/defaults.py
@ -1,4 +1,3 @@
-
 from core.model_runtime.entities.model_entities import DefaultParameterName

 PARAMETER_RULE_TEMPLATE: dict[DefaultParameterName, dict] = {
@ -94,5 +93,16 @@ PARAMETER_RULE_TEMPLATE: dict[DefaultParameterName, dict] = {
        },
        'required': False,
        'options': ['JSON', 'XML'],
-    }
-}
+    },
+    DefaultParameterName.JSON_SCHEMA: {
+        'label': {
+            'en_US': 'JSON Schema',
+        },
+        'type': 'text',
+        'help': {
+            'en_US': 'Set a response json schema will ensure LLM to adhere it.',
+            'zh_Hans': '设置返回的json schema，llm将按照它返回',
+        },
+        'required': False,
+    },
+}