ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加千帆大模型支持

This commit is contained in:
junleea 2025-05-10 14:12:12 +08:00
parent 602657e094
commit 7c90312197
2 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
proto/im.go
View File

@ -37,13 +37,15 @@ 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"` //应用密钥
APIKey string `json:"apiKey"` //应用key
System string `json:"system"` //系统参数
Model string `json:"model"` //模型名称
Domain string `json:"domain"` //domain参数，spark需要
Url string `json:"url"` //模型地址
APPID string `json:"appid"` //应用id
APISecret string `json:"apiSecret"` //应用密钥
APIKey string `json:"apiKey"` //应用key
System string `json:"system"` //系统参数
AccessKey string `json:"access_key"` //accessKey参数
SecretKey string `json:"Secret_key"` //secretKey参数
//其他参数
Temperature float32 `json:"temperature"` //温度,默认0.5,核采样阈值。取值越高随机性越强，即相同的问题得到的不同答案的可能性越大
MaxTokens int `json:"maxTokens"` //最大生成长度,默认4096

10
service/qianfan.go
View File

@ -14,8 +14,14 @@ import (
)
func Qianfan(modelParam proto.ModelParam, imCtx *proto.IMParamContext) {
qianfan.GetConfig().AccessKey = proto.QianfanAccessKey
qianfan.GetConfig().SecretKey = proto.QianfanSecretKey
log.Println("qianfan model param:", modelParam)
log.Println("qianfan question:", imCtx.Question)
if modelParam.AccessKey == "" || modelParam.APISecret == "" {
log.Println("qianfan AccessKey or APISecret is empty")
return
}
qianfan.GetConfig().AccessKey = modelParam.AccessKey
qianfan.GetConfig().SecretKey = modelParam.APISecret
chat := qianfan.NewChatCompletion(qianfan.WithModel(modelParam.Model))
var messages []qianfan.ChatCompletionMessage