feat: 新增返回支持的格式

Author: wsshow

reader.go

@@ -3,9 +3,13 @@ package docreader
 import (
 	"os"
 	"path/filepath"
+	"slices"
 	"strings"
 )
 
+// 支持的文档格式列表
+var supportedFormats = []string{".docx", ".pdf", ".xlsx", ".pptx", ".txt", ".csv", ".md", ".markdown", ".rtf"}
+
 // DocumentReader 定义了文档读取器的通用接口
 type DocumentReader interface {
 	// ReadText 读取文档的文本内容
@@ -22,6 +26,22 @@ type Document struct {
 	Metadata map[string]string
 }
 
+// GetSupportedFormats 返回当前支持的文档格式列表
+func GetSupportedFormats() []string {
+	formats := make([]string, len(supportedFormats))
+	copy(formats, supportedFormats)
+	return formats
+}
+
+// IsFormatSupported 检查指定的文件格式是否被支持
+func IsFormatSupported(ext string) bool {
+	ext = strings.ToLower(ext)
+	if !strings.HasPrefix(ext, ".") {
+		ext = "." + ext
+	}
+	return slices.Contains(supportedFormats, ext)
+}
+
 // ReadDocument 根据文件扩展名自动选择合适的读取器
 func ReadDocument(filePath string) (*Document, error) {
 	// 检查文件是否存在