1

2026-08-20 12:05:53 +08:00
parent c98fee1bca
commit b13e10d28a
46 changed files with 3026 additions and 87 deletions
@@ -4,7 +4,6 @@ import (
 	"context"

 	"rag-local/kb/model/dto"
-	"rag-local/kb/model/entity"
 	"rag-local/kb/service"
 )

@@ -21,19 +20,7 @@ func (c *dataset) List(ctx context.Context, _ *dto.ListDatasetReq) (*dto.ListDat
 }

 func (c *dataset) Save(ctx context.Context, req *dto.SaveDatasetReq) (*dto.SaveDatasetRes, error) {
-	id, err := service.DatasetService.Save(ctx, &entity.Dataset{
-		Id:             req.Id,
-		Name:           req.Name,
-		Description:    req.Description,
-		EmbeddingCfgId: req.EmbeddingCfgId,
-		ChunkSize:      req.ChunkSize,
-		ChunkOverlap:   req.ChunkOverlap,
-		ReactRounds:    req.ReactRounds,
-		VecTopK:        req.VecTopK,
-		FtsTopK:        req.FtsTopK,
-		RerankTopK:     req.RerankTopK,
-		RecallTopK:     req.RecallTopK,
-	})
+	id, err := service.DatasetService.Save(ctx, req)
 	if err != nil {
 		return nil, err
 	}