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 15:27:56 +08:00
parent 6ee679f29e
commit 2e02bd22a1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
service/qianfan.go
View File

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