90 lines
2.9 KiB
Go
90 lines
2.9 KiB
Go
/*
|
||
* Copyright 2025 coze-dev Authors
|
||
*
|
||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||
* you may not use this file except in compliance with the License.
|
||
* You may obtain a copy of the License at
|
||
*
|
||
* http://www.apache.org/licenses/LICENSE-2.0
|
||
*
|
||
* Unless required by applicable law or agreed to in writing, software
|
||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
* See the License for the specific language governing permissions and
|
||
* limitations under the License.
|
||
*/
|
||
|
||
package entity
|
||
|
||
import (
|
||
"encoding/json"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
|
||
"github.com/coze-dev/coze-studio/backend/api/model/crossdomain/modelmgr"
|
||
)
|
||
|
||
func TestDefaultParameter(t *testing.T) {
|
||
dps := []*modelmgr.Parameter{
|
||
{
|
||
Name: "temperature",
|
||
Label: &modelmgr.MultilingualText{
|
||
ZH: "生成随机性",
|
||
EN: "Temperature",
|
||
},
|
||
Desc: &modelmgr.MultilingualText{
|
||
ZH: "- **temperature**: 调高温度会使得模型的输出更多样性和创新性，反之，降低温度会使输出内容更加遵循指令要求但减少多样性。建议不要与“Top p”同时调整。",
|
||
EN: "**Temperature**:\\n\\n- When you increase this value, the model outputs more diverse and innovative content; when you decrease it, the model outputs less diverse content that strictly follows the given instructions.\\n- It is recommended not to adjust this value with \\\"Top p\\\" at the same time.",
|
||
},
|
||
Type: modelmgr.ValueTypeFloat,
|
||
Min: "0",
|
||
Max: "1",
|
||
Precision: 1,
|
||
DefaultVal: modelmgr.DefaultValue{
|
||
modelmgr.DefaultTypeDefault: "1.0",
|
||
modelmgr.DefaultTypeCreative: "1",
|
||
modelmgr.DefaultTypeBalance: "0.8",
|
||
modelmgr.DefaultTypePrecise: "0.3",
|
||
},
|
||
Style: modelmgr.DisplayStyle{
|
||
Widget: modelmgr.WidgetSlider,
|
||
Label: &modelmgr.MultilingualText{
|
||
ZH: "生成多样性",
|
||
EN: "Generation diversity",
|
||
},
|
||
},
|
||
},
|
||
{
|
||
Name: "max_tokens",
|
||
Label: &modelmgr.MultilingualText{
|
||
ZH: "最大回复长度",
|
||
EN: "Response max length",
|
||
},
|
||
Desc: &modelmgr.MultilingualText{
|
||
ZH: "控制模型输出的Tokens 长度上限。通常 100 Tokens 约等于 150 个中文汉字。",
|
||
EN: "You can specify the maximum length of the tokens output through this value. Typically, 100 tokens are approximately equal to 150 Chinese characters.",
|
||
},
|
||
Type: modelmgr.ValueTypeInt,
|
||
Min: "1",
|
||
Max: "12288",
|
||
Precision: 0,
|
||
DefaultVal: modelmgr.DefaultValue{
|
||
modelmgr.DefaultTypeDefault: "4096",
|
||
},
|
||
Style: modelmgr.DisplayStyle{
|
||
Widget: modelmgr.WidgetSlider,
|
||
Label: &modelmgr.MultilingualText{
|
||
ZH: "输入及输出设置",
|
||
EN: "Input and output settings",
|
||
},
|
||
},
|
||
},
|
||
}
|
||
|
||
data, err := json.Marshal(dps)
|
||
assert.NoError(t, err)
|
||
|
||
t.Logf("default parameters: %s", string(data))
|
||
}
|