ben/coze-studio
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

optimize(knowledge): Optimize the index logic of the knowledge base (#841)

Browse Source
This commit is contained in:
liuyunchao-1998
2025-08-21 17:44:26 +08:00
committed by GitHub
parent 3d53aaa785
commit 09d00c26cb
3 changed files with 454 additions and 165 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/domain/knowledge/internal/dal/dao/knowledge_document_slice.go
View File

@@ -25,6 +25,7 @@ import (
"golang.org/x/sync/errgroup"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"github.com/coze-dev/coze-studio/backend/domain/knowledge/entity"
"github.com/coze-dev/coze-studio/backend/domain/knowledge/internal/dal/model"
@@ -52,7 +53,7 @@ func (dao *KnowledgeDocumentSliceDAO) Update(ctx context.Context, slice *model.K
}
func (dao *KnowledgeDocumentSliceDAO) BatchCreate(ctx context.Context, slices []*model.KnowledgeDocumentSlice) error {
return dao.Query.KnowledgeDocumentSlice.WithContext(ctx).CreateInBatches(slices, 100)
return dao.Query.KnowledgeDocumentSlice.WithContext(ctx).Clauses(clause.OnConflict{UpdateAll: true}).CreateInBatches(slices, 100)
}
func (dao *KnowledgeDocumentSliceDAO) BatchSetStatus(ctx context.Context, ids []int64, status int32, reason string) error {