CodeCow/coze-discord-proxy
1
0
Fork 0
mirror of https://github.com/deanxv/coze-discord-proxy.git synced 2026-06-04 21:02:47 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: /images/generations RevisedPrompt

Browse Source
This commit is contained in:
dean 2024-06-07 16:33:45 +08:00
parent 7bcd4255b3
commit 932bceed6e
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
discord/processmessage.go
View File

@ -85,18 +85,20 @@ func processMessageUpdateForOpenAIImage(m *discordgo.MessageUpdate) model.OpenAI
for _, match := range subMatches {
response.Data = append(response.Data, &model.OpenAIImagesGenerationDataResponse{
URL: match[1],
URL: match[1],
RevisedPrompt: m.Content,
})
}
if len(m.Embeds) != 0 {
for _, embed := range m.Embeds {
if embed.Image != nil && !strings.Contains(m.Content, embed.Image.URL) {
if m.Content != "" {
m.Content += "\n"
}
// if m.Content != "" {
// m.Content += "\n"
// }
response.Data = append(response.Data, &model.OpenAIImagesGenerationDataResponse{
URL: embed.Image.URL,
URL: embed.Image.URL,
RevisedPrompt: m.Content,
})
}
}
@ -183,18 +185,20 @@ func processMessageCreateForOpenAIImage(m *discordgo.MessageCreate) model.OpenAI
for i, match := range subMatches {
response.Data = append(response.Data, &model.OpenAIImagesGenerationDataResponse{
URL: match[i],
URL: match[i],
RevisedPrompt: m.Content,
})
}
if len(m.Embeds) != 0 {
for _, embed := range m.Embeds {
if embed.Image != nil && !strings.Contains(m.Content, embed.Image.URL) {
if m.Content != "" {
m.Content += "\n"
}
//if m.Content != "" {
// m.Content += "\n"
//}
response.Data = append(response.Data, &model.OpenAIImagesGenerationDataResponse{
URL: embed.Image.URL,
URL: embed.Image.URL,
RevisedPrompt: m.Content,
})
}
}