1

kb/controller/document_controller.go

@@ -2,12 +2,9 @@ package controller
 
 import (
 	"context"
-	"io"
 
 	"rag-local/kb/model/dto"
 	"rag-local/kb/service"
-
-	"github.com/gogf/gf/v2/errors/gerror"
 )
 
 type document struct{}
@@ -15,19 +12,7 @@ type document struct{}
 var Document = &document{}
 
 func (c *document) Upload(ctx context.Context, req *dto.UploadDocumentReq) (*dto.UploadDocumentRes, error) {
-	if req.File == nil {
-		return nil, gerror.New("请选择文件")
-	}
-	f, err := req.File.Open()
-	if err != nil {
-		return nil, err
-	}
-	defer func() { _ = f.Close() }()
-	data, err := io.ReadAll(f)
-	if err != nil {
-		return nil, err
-	}
-	doc, err := service.DocumentService.Upload(ctx, req.DatasetId, req.File.Filename, data)
+	doc, err := service.DocumentService.Upload(ctx, req.DatasetId, req.File)
 	if err != nil {
 		return nil, err
 	}