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

添加doubao 图片功能日志输出

This commit is contained in:
junleea 2025-04-01 14:28:12 +08:00
parent 4d974e4eab
commit 13bcffa72d
2 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
proto/im.go
View File

@ -68,13 +68,14 @@ type SessionResponse struct {
}
type IMParamContext struct {
UserID int `json:"user_id"` //用户id
SessionID int `json:"session_id"` //会话id
FunctionID int `json:"function_id"` //功能id
ModelID int `json:"model_id"` //模型id
ModelType string `json:"model_type"` //模型类型
Question string `json:"question"` //问题
Channel string `json:"channel"` //消息队列
UserID int `json:"user_id"` //用户id
SessionID int `json:"session_id"` //会话id
FunctionID int `json:"function_id"` //功能id
ModelID int `json:"model_id"` //模型id
ModelType string `json:"model_type"` //模型类型
Question string `json:"question"` //问题
Channel string `json:"channel"` //消息队列
IsHasImage bool `json:"is_has_image"` //是否有图片
}
// 用户向模型发送图片的消息结构

8
service/doubao.go
View File

@ -90,8 +90,8 @@ func DouBaoV2(modelParam proto.ModelParam, imCtx *proto.IMParamContext) {
if err != nil {
log.Println("get doubao session history message error:", err)
}
//messagesStr, _ := json.Marshal(messages)
//log.Println("doubao messages:", string(messagesStr))
messagesStr, _ := json.Marshal(messages)
log.Println("doubao messages:", string(messagesStr))
req := model.CreateChatCompletionRequest{
Model: modelParam.Model,
@ -137,7 +137,9 @@ func DouBaoV2(modelParam proto.ModelParam, imCtx *proto.IMParamContext) {
choices := recv.Choices[0]
if choices.FinishReason == proto.FinishReasonStop {
answer += choices.Delta.Content
tokens += recv.Usage.TotalTokens
if recv.Usage != nil {
tokens += recv.Usage.TotalTokens
}
break
} else {
answer += choices.Delta.Content