openai: add #include-attachments option to load all text based attachments

Author: brainexe

client/slack.go

@@ -4,6 +4,7 @@ package client
 
 import (
 	"fmt"
+	"io"
 	"strings"
 	"sync"
 
@@ -125,6 +126,9 @@ type SlackClient interface {
 
 	// PinMessage will pin a message to the channel
 	PinMessage(channel string, timestamp string) error
+
+	// GetFile downloads a file from Slack by its private URL
+	GetFile(downloadURL string, writer io.Writer) error
 }
 
 // Slack is wrapper to the slack.Client which also holds the the socketmode.Client and all needed config
@@ -336,6 +340,11 @@ func (s *Slack) UploadFile(params slack.UploadFileV2Parameters) (*slack.FileSumm
 	return s.UploadFileV2(params)
 }
 
+// GetFile downloads a file from Slack by its private URL
+func (s *Slack) GetFile(downloadURL string, writer io.Writer) error {
+	return s.Client.GetFile(downloadURL, writer)
+}
+
 // GetUserIDAndName returns the user-id and user-name based on a identifier. If can get a user-id or name
 func GetUserIDAndName(identifier string) (id string, name string) {
 	identifier = strings.TrimPrefix(identifier, "@")

command/openai/command.go

@@ -1,6 +1,7 @@
 package openai
 
 import (
+	"bytes"
 	"fmt"
 	"regexp"
 	"strings"
@@ -87,7 +88,7 @@ func (c *openaiCommand) startConversation(message msg.Ref, text string) bool {
 
 	// Handle #message-history hashtag
 	if hashtagOptions.MessageHistory > 0 {
-		channelMessages, err := c.getChannelHistory(message.GetChannel(), hashtagOptions.MessageHistory)
+		channelMessages, err := c.getChannelHistory(message.GetChannel(), hashtagOptions.MessageHistory, hashtagOptions.IncludeAttachments)
 		if err != nil {
 			c.ReplyError(message, fmt.Errorf("can't load channel history: %w", err))
 			return true
@@ -410,8 +411,32 @@ func (c *openaiCommand) GetTemplateFunction() template.FuncMap {
 	}
 }
 
+// loadTextAttachments downloads text file attachments and returns them formatted
+func (c *openaiCommand) loadTextAttachments(files []slack.File) string {
+	log.Warn("loadTextAttachments") // todo temporary
+
+	var result strings.Builder
+	for _, file := range files {
+		if !strings.HasPrefix(file.Mimetype, "text/") {
+			log.Infof("Skipping attachment %s: mimetype is %s", file.Name, file.Mimetype)
+			continue
+		}
+
+		var buf bytes.Buffer
+		log.Infof("Downloading attachment %s", file.Name)
+
+		if err := c.GetFile(file.URLPrivate, &buf); err != nil {
+			log.Warnf("Failed to download attachment %s: %v", file.Name, err)
+			continue
+		}
+
+		result.WriteString(fmt.Sprintf(" <Attachment filename=\"%s\">%s</Attachment>", file.Name, buf.String()))
+	}
+	return result.String()
+}
+
 // getChannelHistory fetches the last N messages from a channel (including thread messages)
-func (c *openaiCommand) getChannelHistory(channel string, count int) ([]slack.Message, error) {
+func (c *openaiCommand) getChannelHistory(channel string, count int, includeAttachments bool) ([]slack.Message, error) {
 	params := &slack.GetConversationHistoryParameters{
 		ChannelID: channel,
 		Limit:     count,
@@ -425,6 +450,10 @@ func (c *openaiCommand) getChannelHistory(channel string, count int) ([]slack.Me
 	// Collect main channel messages and their thread replies
 	allMessages := make([]slack.Message, 0)
 	for _, message := range response.Messages {
+		// Load text attachments if requested
+		if includeAttachments && len(message.Files) > 0 {
+			message.Text += c.loadTextAttachments(message.Files)
+		}
 		allMessages = append(allMessages, message)
 
 		// If this message has replies, fetch the thread messages
@@ -441,7 +470,13 @@ func (c *openaiCommand) getChannelHistory(channel string, count int) ([]slack.Me
 			}
 
 			// Skip the first message as it's the parent (already added)
+			// Also load attachments for thread messages if requested
 			if len(threadMessages) > 1 {
+				for i := 1; i < len(threadMessages); i++ {
+					if includeAttachments && len(threadMessages[i].Files) > 0 {
+						threadMessages[i].Text += c.loadTextAttachments(threadMessages[i].Files)
+					}
+				}
 				allMessages = append(allMessages, threadMessages[1:]...)
 			}
 		}
@@ -460,7 +495,7 @@ func (c *openaiCommand) GetHelp() []bot.Help {
 	return []bot.Help{
 		{
 			Command:     "openai <question>",
-			Description: "Starts a chatgpt/openai conversation in a new thread. Supports hashtags for advanced options: \n- #model-<name> (e.g., #model-gpt-4o), \n- #high-thinking/#medium-thinking/#low-thinking/#no-thinking for reasoning control\n- #message-history or #message-history-<N> to include recent channel messages as context\n- #no-streaming to disable streaming and get the full response at once\n- #no-thread to reply directly without creating a thread\n- #debug to show debug information about the request",
+			Description: "Starts a chatgpt/openai conversation in a new thread. Supports hashtags for advanced options: \n- #model-<name> (e.g., #model-gpt-4o), \n- #high-thinking/#medium-thinking/#low-thinking/#no-thinking for reasoning control\n- #message-history or #message-history-<N> to include recent channel messages as context\n- #include-attachments to include text file attachments when using #message-history\n- #no-streaming to disable streaming and get the full response at once\n- #no-thread to reply directly without creating a thread\n- #debug to show debug information about the request",
 			Category:    category,
 			Examples: []string{
 				"openai why is the sky blue?",
@@ -470,6 +505,7 @@ func (c *openaiCommand) GetHelp() []bot.Help {
 				"openai #no-streaming give me a detailed explanation",
 				"openai #no-thread quick question without a thread",
 				"openai #debug analyze this code for performance issues",
+				"openai #message-history-10 #include-attachments summarize the shared files",
 			},
 		},
 		{

command/openai/config.go

@@ -46,7 +46,7 @@ func (c *Config) IsEnabled() bool {
 
 var defaultConfig = Config{
 	APIHost:              apiHost,
-	Model:                "gpt-5", // aka model behind ChatGPT
+	Model:                "gpt-5.2", // aka model behind ChatGPT
 	UpdateInterval:       time.Second * 1,
 	APITimeout:           time.Second * 120,
 	HistorySize:          25,

command/openai/hashtag_parser.go

@@ -4,6 +4,14 @@ import (
 	"regexp"
 	"strconv"
 	"strings"
+
+	log "github.com/sirupsen/logrus"
+)
+
+// Pre-compiled regexes for hashtag parsing
+var (
+	messageHistoryWithCountRe = regexp.MustCompile(`#message-history-(\d+)`)
+	modelRe                   = regexp.MustCompile(`#model-([\w.-]+)`)
 )
 
 // removeHashtag removes a hashtag from the text if present and returns whether it was found
@@ -17,12 +25,13 @@ func removeHashtag(text *string, hashtag string) bool {
 
 // HashtagOptions contains parsed hashtag options from user input
 type HashtagOptions struct {
-	ReasoningEffort string // "minimal", "medium", "high", or ""
-	Model           string // override model, empty means use config default
-	MessageHistory  int    // number of channel messages to include, 0 means disabled
-	NoStreaming     bool   // disable streaming responses, get full response at once
-	NoThread        bool   // disable thread replies, reply directly to the message instead
-	Debug           bool   // show debug information at the end of the response
+	ReasoningEffort    string // "minimal", "medium", "high", or ""
+	Model              string // override model, empty means use config default
+	MessageHistory     int    // number of channel messages to include, 0 means disabled
+	NoStreaming        bool   // disable streaming responses, get full response at once
+	NoThread           bool   // disable thread replies, reply directly to the message instead
+	Debug              bool   // show debug information at the end of the response
+	IncludeAttachments bool   // include text file attachments in message history
 }
 
 // ParseHashtags extracts hashtag options from the input text and returns
@@ -36,7 +45,6 @@ func ParseHashtags(text string) (cleanText string, options HashtagOptions) {
 	// 4. #high-thinking, #medium-thinking, #minimal-thinking, #no-thinking
 
 	// Parse message-history with number: #message-history-20
-	messageHistoryWithCountRe := regexp.MustCompile(`#message-history-(\d+)`)
 	if matches := messageHistoryWithCountRe.FindStringSubmatch(text); len(matches) > 1 {
 		if count, err := strconv.Atoi(matches[1]); err == nil {
 			options.MessageHistory = count
@@ -48,7 +56,6 @@ func ParseHashtags(text string) (cleanText string, options HashtagOptions) {
 	}
 
 	// Parse model: #model-gpt-4o
-	modelRe := regexp.MustCompile(`#model-([\w.-]+)`)
 	if matches := modelRe.FindStringSubmatch(text); len(matches) > 1 {
 		options.Model = matches[1]
 		text = modelRe.ReplaceAllString(text, "")
@@ -78,8 +85,14 @@ func ParseHashtags(text string) (cleanText string, options HashtagOptions) {
 	// Parse debug option
 	options.Debug = removeHashtag(&text, "#debug")
 
+	// Parse include-attachments option
+	options.IncludeAttachments = removeHashtag(&text, "#include-attachments")
+
 	// Clean up extra whitespace
 	cleanText = strings.Join(strings.Fields(text), " ")
 
+	// just temporary
+	log.Warn("Parsed Hashtag Options: ", options)
+
 	return cleanText, options
 }

command/openai/hashtag_parser_test.go

@@ -6,15 +6,16 @@ import (
 
 func TestParseHashtags(t *testing.T) {
 	tests := []struct {
-		name              string
-		input             string
-		expectedText      string
-		expectedModel     string
-		expectedReason    string
-		expectedHistory   int
-		expectedStreaming bool
-		expectedNoThread  bool
-		expectedDebug     bool
+		name                string
+		input               string
+		expectedText        string
+		expectedModel       string
+		expectedReason      string
+		expectedHistory     int
+		expectedStreaming   bool
+		expectedNoThread    bool
+		expectedDebug       bool
+		expectedAttachments bool
 	}{
 		{
 			name:          "No hashtags",
@@ -132,6 +133,31 @@ func TestParseHashtags(t *testing.T) {
 			expectedNoThread:  true,
 			expectedDebug:     true,
 		},
+		{
+			name:                "Include attachments only",
+			input:               "#include-attachments Show me the files",
+			expectedText:        "Show me the files",
+			expectedAttachments: true,
+		},
+		{
+			name:                "Include attachments with message history",
+			input:               "#message-history-10 #include-attachments Summarize the shared files",
+			expectedText:        "Summarize the shared files",
+			expectedHistory:     10,
+			expectedAttachments: true,
+		},
+		{
+			name:                "All hashtags including attachments",
+			input:               "#model-o1 #high-thinking #message-history-20 #include-attachments #no-streaming #no-thread #debug Complete request with files",
+			expectedText:        "Complete request with files",
+			expectedModel:       "o1",
+			expectedReason:      "high",
+			expectedHistory:     20,
+			expectedStreaming:   true,
+			expectedNoThread:    true,
+			expectedDebug:       true,
+			expectedAttachments: true,
+		},
 	}
 
 	for _, tt := range tests {
@@ -159,6 +185,9 @@ func TestParseHashtags(t *testing.T) {
 			if options.Debug != tt.expectedDebug {
 				t.Errorf("Expected Debug %v, got %v", tt.expectedDebug, options.Debug)
 			}
+			if options.IncludeAttachments != tt.expectedAttachments {
+				t.Errorf("Expected IncludeAttachments %v, got %v", tt.expectedAttachments, options.IncludeAttachments)
+			}
 		})
 	}
 }