fix: image

common/constants.go

@@ -12,7 +12,7 @@ var StreamRequestOutTime = os.Getenv("STREAM_REQUEST_OUT_TIME")
 
 var DebugEnabled = os.Getenv("DEBUG") == "true"
 
-var Version = "v3.2.1" // this hard coding will be replaced automatically when building, no need to manually change
+var Version = "v3.2.2" // this hard coding will be replaced automatically when building, no need to manually change
 
 const (
 	RequestIdKey = "X-Request-Id"

discord/discord.go

@@ -18,6 +18,7 @@ import (
 	"net/url"
 	"os"
 	"os/signal"
+	"regexp"
 	"strconv"
 	"strings"
 	"syscall"
@@ -287,6 +288,15 @@ func processMessageForOpenAI(m *discordgo.MessageUpdate) model.OpenAIChatComplet
 func processMessageForOpenAIImage(m *discordgo.MessageUpdate) model.OpenAIImagesGenerationResponse {
 	var response model.OpenAIImagesGenerationResponse
 
+	re := regexp.MustCompile(`]\((https?://\S+)\)`)
+	submatches := re.FindAllStringSubmatch(m.Content, -1)
+
+	for _, match := range submatches {
+		response.Data = append(response.Data, struct {
+			URL string `json:"url"`
+		}{URL: match[1]})
+	}
+
 	if len(m.Embeds) != 0 {
 		for _, embed := range m.Embeds {
 			if embed.Image != nil && !strings.Contains(m.Content, embed.Image.URL) {