mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-18 00:16:37 +08:00
fix: GLM-4V 的 Vision 兼容问题 (close #136)
This commit is contained in:
parent
a70ca53449
commit
d0acecb2ab
@ -74,6 +74,25 @@ func getZhipuToken(apikey string) string {
|
|||||||
func requestOpenAI2Zhipu(request dto.GeneralOpenAIRequest) *dto.GeneralOpenAIRequest {
|
func requestOpenAI2Zhipu(request dto.GeneralOpenAIRequest) *dto.GeneralOpenAIRequest {
|
||||||
messages := make([]dto.Message, 0, len(request.Messages))
|
messages := make([]dto.Message, 0, len(request.Messages))
|
||||||
for _, message := range request.Messages {
|
for _, message := range request.Messages {
|
||||||
|
if !message.IsStringContent() {
|
||||||
|
mediaMessages := message.ParseContent()
|
||||||
|
for j, mediaMessage := range mediaMessages {
|
||||||
|
if mediaMessage.Type == dto.ContentTypeImageURL {
|
||||||
|
imageUrl := mediaMessage.ImageUrl.(dto.MessageImageUrl)
|
||||||
|
// check if base64
|
||||||
|
if strings.HasPrefix(imageUrl.Url, "data:image/") {
|
||||||
|
// 去除base64数据的URL前缀(如果有)
|
||||||
|
if idx := strings.Index(imageUrl.Url, ","); idx != -1 {
|
||||||
|
imageUrl.Url = imageUrl.Url[idx+1:]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
mediaMessage.ImageUrl = imageUrl
|
||||||
|
mediaMessages[j] = mediaMessage
|
||||||
|
}
|
||||||
|
}
|
||||||
|
messageRaw, _ := json.Marshal(mediaMessages)
|
||||||
|
message.Content = messageRaw
|
||||||
|
}
|
||||||
messages = append(messages, dto.Message{
|
messages = append(messages, dto.Message{
|
||||||
Role: message.Role,
|
Role: message.Role,
|
||||||
Content: message.Content,
|
Content: message.Content,
|
||||||
@ -138,7 +157,7 @@ func streamResponseZhipu2OpenAI(zhipuResponse *ZhipuV4StreamResponse) *dto.ChatC
|
|||||||
Id: zhipuResponse.Id,
|
Id: zhipuResponse.Id,
|
||||||
Object: "chat.completion.chunk",
|
Object: "chat.completion.chunk",
|
||||||
Created: zhipuResponse.Created,
|
Created: zhipuResponse.Created,
|
||||||
Model: "glm-4",
|
Model: "glm-4v",
|
||||||
Choices: []dto.ChatCompletionsStreamResponseChoice{choice},
|
Choices: []dto.ChatCompletionsStreamResponseChoice{choice},
|
||||||
}
|
}
|
||||||
return &response
|
return &response
|
||||||
|
Loading…
Reference in New Issue
Block a user