fix: image

deanxv · Jul 10, 2024 · c01393c · c01393c
1 parent dbfd0bd
commit c01393c
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1 deletion.
diff --git a/common/constants.go b/common/constants.go
@@ -1,6 +1,6 @@
 package common
 
-var Version = "v4.5.9" // this hard coding will be replaced automatically when building, no need to manually change
+var Version = "v4.5.10" // this hard coding will be replaced automatically when building, no need to manually change
 
 const (
 	RequestIdKey = "X-Request-Id"

diff --git a/discord/processmessage.go b/discord/processmessage.go
@@ -38,6 +38,17 @@ func processMessageUpdateForOpenAI(m *discordgo.MessageUpdate) model.OpenAIChatC
 		}
 	}
 
+	if len(m.Attachments) != 0 {
+		for _, attachment := range m.Attachments {
+			if attachment.ProxyURL != "" && !strings.Contains(m.Content, attachment.ProxyURL) {
+				if m.Content != "" {
+					m.Content += "\n"
+				}
+				m.Content += fmt.Sprintf("%s\n![Image](%s)", attachment.ProxyURL, attachment.ProxyURL)
+			}
+		}
+	}
+
 	promptTokens := common.CountTokens(m.ReferencedMessage.Content)
 	completionTokens := common.CountTokens(m.Content)
 
@@ -158,6 +169,17 @@ func processMessageCreateForOpenAI(m *discordgo.MessageCreate) model.OpenAIChatC
 		}
 	}
 
+	if len(m.Attachments) != 0 {
+		for _, attachment := range m.Attachments {
+			if attachment.ProxyURL != "" && !strings.Contains(m.Content, attachment.ProxyURL) {
+				if m.Content != "" {
+					m.Content += "\n"
+				}
+				m.Content += fmt.Sprintf("%s\n![Image](%s)", attachment.ProxyURL, attachment.ProxyURL)
+			}
+		}
+	}
+
 	promptTokens := common.CountTokens(m.ReferencedMessage.Content)
 	completionTokens := common.CountTokens(m.Content)