spark添加模型参数domain
This commit is contained in:
parent
b0b14fef3c
commit
b17df76df7
|
|
@ -25,6 +25,7 @@ type WSMessageReq struct {
|
|||
|
||||
type ModelParam struct {
|
||||
Model string `json:"model"` //模型名称
|
||||
Domain string `json:"domain"` //domain参数,spark需要
|
||||
Url string `json:"url"` //模型地址
|
||||
APPID string `json:"appid"` //应用id
|
||||
APISecret string `json:"apiSecret"` //应用密钥
|
||||
|
|
|
|||
|
|
@ -123,7 +123,7 @@ func Spark(modelParam proto.ModelParam, question, channel string, SessionID, Use
|
|||
|
||||
//发送消息
|
||||
go func() {
|
||||
data := genSparkParams(question, modelParam.APPID)
|
||||
data := genSparkParams(question, modelParam.APPID, modelParam.Domain)
|
||||
//将数据转换为json
|
||||
//dataByte, err := json.Marshal(data)
|
||||
if err != nil {
|
||||
|
|
@ -200,7 +200,7 @@ func SparkToGeneralMassageAndSendMsgQueue(data *SparkResponse, channel string, S
|
|||
}
|
||||
|
||||
// 生成参数
|
||||
func genSparkParams(question, appid string) SparkRequest {
|
||||
func genSparkParams(question, appid string, domain string) SparkRequest {
|
||||
messages := []MessageContent{
|
||||
{Role: "user", Content: question},
|
||||
}
|
||||
|
|
@ -211,7 +211,7 @@ func genSparkParams(question, appid string) SparkRequest {
|
|||
},
|
||||
Parameter: Parameter{
|
||||
Chat: ChatParameter{
|
||||
Domain: "4.0Ultra",
|
||||
Domain: domain,
|
||||
Temperature: 0.8,
|
||||
MaxTokens: 2048,
|
||||
},
|
||||
|
|
|
|||
Loading…
Reference in New Issue