feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

backend/domain/knowledge/internal/mock/dal/dao/knowledge.go

@@ -0,0 +1,176 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: knowledge.go
+//
+// Generated by this command:
+//
+//	mockgen -destination ../../mock/dal/dao/knowledge.go --package dao -source knowledge.go
+//
+
+// Package dao is a generated GoMock package.
+package dao
+
+import (
+	context "context"
+	reflect "reflect"
+
+	gomock "go.uber.org/mock/gomock"
+	gorm "gorm.io/gorm"
+
+	"github.com/coze-dev/coze-studio/backend/domain/knowledge/entity"
+	model "github.com/coze-dev/coze-studio/backend/domain/knowledge/internal/dal/model"
+)
+
+// MockKnowledgeRepo is a mock of KnowledgeRepo interface.
+type MockKnowledgeRepo struct {
+	ctrl     *gomock.Controller
+	recorder *MockKnowledgeRepoMockRecorder
+}
+
+// MockKnowledgeRepoMockRecorder is the mock recorder for MockKnowledgeRepo.
+type MockKnowledgeRepoMockRecorder struct {
+	mock *MockKnowledgeRepo
+}
+
+// NewMockKnowledgeRepo creates a new mock instance.
+func NewMockKnowledgeRepo(ctrl *gomock.Controller) *MockKnowledgeRepo {
+	mock := &MockKnowledgeRepo{ctrl: ctrl}
+	mock.recorder = &MockKnowledgeRepoMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockKnowledgeRepo) EXPECT() *MockKnowledgeRepoMockRecorder {
+	return m.recorder
+}
+
+// Create mocks base method.
+func (m *MockKnowledgeRepo) Create(ctx context.Context, knowledge *model.Knowledge) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Create", ctx, knowledge)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Create indicates an expected call of Create.
+func (mr *MockKnowledgeRepoMockRecorder) Create(ctx, knowledge any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockKnowledgeRepo)(nil).Create), ctx, knowledge)
+}
+
+// Delete mocks base method.
+func (m *MockKnowledgeRepo) Delete(ctx context.Context, id int64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Delete", ctx, id)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Delete indicates an expected call of Delete.
+func (mr *MockKnowledgeRepoMockRecorder) Delete(ctx, id any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockKnowledgeRepo)(nil).Delete), ctx, id)
+}
+
+// FilterEnableKnowledge mocks base method.
+func (m *MockKnowledgeRepo) FilterEnableKnowledge(ctx context.Context, ids []int64) ([]*model.Knowledge, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "FilterEnableKnowledge", ctx, ids)
+	ret0, _ := ret[0].([]*model.Knowledge)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// FilterEnableKnowledge indicates an expected call of FilterEnableKnowledge.
+func (mr *MockKnowledgeRepoMockRecorder) FilterEnableKnowledge(ctx, ids any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FilterEnableKnowledge", reflect.TypeOf((*MockKnowledgeRepo)(nil).FilterEnableKnowledge), ctx, ids)
+}
+
+// FindKnowledgeByCondition mocks base method.
+func (m *MockKnowledgeRepo) FindKnowledgeByCondition(ctx context.Context, opts *entity.WhereKnowledgeOption) ([]*model.Knowledge, int64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "FindKnowledgeByCondition", ctx, opts)
+	ret0, _ := ret[0].([]*model.Knowledge)
+	ret1, _ := ret[1].(int64)
+	ret2, _ := ret[2].(error)
+	return ret0, ret1, ret2
+}
+
+// FindKnowledgeByCondition indicates an expected call of FindKnowledgeByCondition.
+func (mr *MockKnowledgeRepoMockRecorder) FindKnowledgeByCondition(ctx, opts any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindKnowledgeByCondition", reflect.TypeOf((*MockKnowledgeRepo)(nil).FindKnowledgeByCondition), ctx, opts)
+}
+
+// GetByID mocks base method.
+func (m *MockKnowledgeRepo) GetByID(ctx context.Context, id int64) (*model.Knowledge, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetByID", ctx, id)
+	ret0, _ := ret[0].(*model.Knowledge)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetByID indicates an expected call of GetByID.
+func (mr *MockKnowledgeRepoMockRecorder) GetByID(ctx, id any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByID", reflect.TypeOf((*MockKnowledgeRepo)(nil).GetByID), ctx, id)
+}
+
+// InitTx mocks base method.
+func (m *MockKnowledgeRepo) InitTx() (*gorm.DB, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "InitTx")
+	ret0, _ := ret[0].(*gorm.DB)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// InitTx indicates an expected call of InitTx.
+func (mr *MockKnowledgeRepoMockRecorder) InitTx() *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InitTx", reflect.TypeOf((*MockKnowledgeRepo)(nil).InitTx))
+}
+
+// MGetByID mocks base method.
+func (m *MockKnowledgeRepo) MGetByID(ctx context.Context, ids []int64) ([]*model.Knowledge, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "MGetByID", ctx, ids)
+	ret0, _ := ret[0].([]*model.Knowledge)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// MGetByID indicates an expected call of MGetByID.
+func (mr *MockKnowledgeRepoMockRecorder) MGetByID(ctx, ids any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByID", reflect.TypeOf((*MockKnowledgeRepo)(nil).MGetByID), ctx, ids)
+}
+
+// Update mocks base method.
+func (m *MockKnowledgeRepo) Update(ctx context.Context, knowledge *model.Knowledge) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Update", ctx, knowledge)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Update indicates an expected call of Update.
+func (mr *MockKnowledgeRepoMockRecorder) Update(ctx, knowledge any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockKnowledgeRepo)(nil).Update), ctx, knowledge)
+}
+
+// UpdateWithTx mocks base method.
+func (m *MockKnowledgeRepo) UpdateWithTx(ctx context.Context, tx *gorm.DB, knowledgeID int64, updateMap map[string]any) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "UpdateWithTx", ctx, tx, knowledgeID, updateMap)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// UpdateWithTx indicates an expected call of UpdateWithTx.
+func (mr *MockKnowledgeRepoMockRecorder) UpdateWithTx(ctx, tx, knowledgeID, updateMap any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWithTx", reflect.TypeOf((*MockKnowledgeRepo)(nil).UpdateWithTx), ctx, tx, knowledgeID, updateMap)
+}

backend/domain/knowledge/internal/mock/dal/dao/knowledge_document.go

@@ -0,0 +1,206 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: knowledge_document.go
+//
+// Generated by this command:
+//
+//	mockgen -destination ../../mock/dal/dao/knowledge_document.go --package dao -source knowledge_document.go
+//
+
+// Package dao is a generated GoMock package.
+package dao
+
+import (
+	context "context"
+	reflect "reflect"
+
+	gomock "go.uber.org/mock/gomock"
+	gorm "gorm.io/gorm"
+
+	"github.com/coze-dev/coze-studio/backend/domain/knowledge/entity"
+	model "github.com/coze-dev/coze-studio/backend/domain/knowledge/internal/dal/model"
+)
+
+// MockKnowledgeDocumentRepo is a mock of KnowledgeDocumentRepo interface.
+type MockKnowledgeDocumentRepo struct {
+	ctrl     *gomock.Controller
+	recorder *MockKnowledgeDocumentRepoMockRecorder
+	isgomock struct{}
+}
+
+// MockKnowledgeDocumentRepoMockRecorder is the mock recorder for MockKnowledgeDocumentRepo.
+type MockKnowledgeDocumentRepoMockRecorder struct {
+	mock *MockKnowledgeDocumentRepo
+}
+
+// NewMockKnowledgeDocumentRepo creates a new mock instance.
+func NewMockKnowledgeDocumentRepo(ctrl *gomock.Controller) *MockKnowledgeDocumentRepo {
+	mock := &MockKnowledgeDocumentRepo{ctrl: ctrl}
+	mock.recorder = &MockKnowledgeDocumentRepoMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockKnowledgeDocumentRepo) EXPECT() *MockKnowledgeDocumentRepoMockRecorder {
+	return m.recorder
+}
+
+// Create mocks base method.
+func (m *MockKnowledgeDocumentRepo) Create(ctx context.Context, document *model.KnowledgeDocument) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Create", ctx, document)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Create indicates an expected call of Create.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) Create(ctx, document any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).Create), ctx, document)
+}
+
+// CreateWithTx mocks base method.
+func (m *MockKnowledgeDocumentRepo) CreateWithTx(ctx context.Context, tx *gorm.DB, document []*model.KnowledgeDocument) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "CreateWithTx", ctx, tx, document)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// CreateWithTx indicates an expected call of CreateWithTx.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) CreateWithTx(ctx, tx, document any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateWithTx", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).CreateWithTx), ctx, tx, document)
+}
+
+// Delete mocks base method.
+func (m *MockKnowledgeDocumentRepo) Delete(ctx context.Context, id int64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Delete", ctx, id)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Delete indicates an expected call of Delete.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) Delete(ctx, id any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).Delete), ctx, id)
+}
+
+// DeleteDocuments mocks base method.
+func (m *MockKnowledgeDocumentRepo) DeleteDocuments(ctx context.Context, ids []int64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "DeleteDocuments", ctx, ids)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// DeleteDocuments indicates an expected call of DeleteDocuments.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) DeleteDocuments(ctx, ids any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteDocuments", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).DeleteDocuments), ctx, ids)
+}
+
+// FindDocumentByCondition mocks base method.
+func (m *MockKnowledgeDocumentRepo) FindDocumentByCondition(ctx context.Context, opts *entity.WhereDocumentOpt) ([]*model.KnowledgeDocument, int64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "FindDocumentByCondition", ctx, opts)
+	ret0, _ := ret[0].([]*model.KnowledgeDocument)
+	ret1, _ := ret[1].(int64)
+	ret2, _ := ret[2].(error)
+	return ret0, ret1, ret2
+}
+
+// FindDocumentByCondition indicates an expected call of FindDocumentByCondition.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) FindDocumentByCondition(ctx, opts any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindDocumentByCondition", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).FindDocumentByCondition), ctx, opts)
+}
+
+// GetByID mocks base method.
+func (m *MockKnowledgeDocumentRepo) GetByID(ctx context.Context, id int64) (*model.KnowledgeDocument, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetByID", ctx, id)
+	ret0, _ := ret[0].(*model.KnowledgeDocument)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetByID indicates an expected call of GetByID.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) GetByID(ctx, id any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetByID", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).GetByID), ctx, id)
+}
+
+// List mocks base method.
+func (m *MockKnowledgeDocumentRepo) List(ctx context.Context, knowledgeID int64, name *string, limit int, cursor *string) ([]*model.KnowledgeDocument, *string, bool, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "List", ctx, knowledgeID, name, limit, cursor)
+	ret0, _ := ret[0].([]*model.KnowledgeDocument)
+	ret1, _ := ret[1].(*string)
+	ret2, _ := ret[2].(bool)
+	ret3, _ := ret[3].(error)
+	return ret0, ret1, ret2, ret3
+}
+
+// List indicates an expected call of List.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) List(ctx, knowledgeID, name, limit, cursor any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).List), ctx, knowledgeID, name, limit, cursor)
+}
+
+// MGetByID mocks base method.
+func (m *MockKnowledgeDocumentRepo) MGetByID(ctx context.Context, ids []int64) ([]*model.KnowledgeDocument, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "MGetByID", ctx, ids)
+	ret0, _ := ret[0].([]*model.KnowledgeDocument)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// MGetByID indicates an expected call of MGetByID.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) MGetByID(ctx, ids any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetByID", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).MGetByID), ctx, ids)
+}
+
+// SetStatus mocks base method.
+func (m *MockKnowledgeDocumentRepo) SetStatus(ctx context.Context, documentID int64, status int32, reason string) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "SetStatus", ctx, documentID, status, reason)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// SetStatus indicates an expected call of SetStatus.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) SetStatus(ctx, documentID, status, reason any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetStatus", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).SetStatus), ctx, documentID, status, reason)
+}
+
+// Update mocks base method.
+func (m *MockKnowledgeDocumentRepo) Update(ctx context.Context, document *model.KnowledgeDocument) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Update", ctx, document)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Update indicates an expected call of Update.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) Update(ctx, document any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).Update), ctx, document)
+}
+
+// UpdateDocumentSliceInfo mocks base method.
+func (m *MockKnowledgeDocumentRepo) UpdateDocumentSliceInfo(ctx context.Context, documentID int64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "UpdateDocumentSliceInfo", ctx, documentID)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// UpdateDocumentSliceInfo indicates an expected call of UpdateDocumentSliceInfo.
+func (mr *MockKnowledgeDocumentRepoMockRecorder) UpdateDocumentSliceInfo(ctx, documentID any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateDocumentSliceInfo", reflect.TypeOf((*MockKnowledgeDocumentRepo)(nil).UpdateDocumentSliceInfo), ctx, documentID)
+}

backend/domain/knowledge/internal/mock/dal/dao/knowledge_document_slice.go

@@ -0,0 +1,222 @@
+// Code generated by MockGen. DO NOT EDIT.
+// Source: knowledge_document_slice.go
+//
+// Generated by this command:
+//
+//	mockgen -destination ../../mock/dal/dao/knowledge_document_slice.go --package dao -source knowledge_document_slice.go
+//
+
+// Package dao is a generated GoMock package.
+package dao
+
+import (
+	context "context"
+	reflect "reflect"
+
+	gomock "go.uber.org/mock/gomock"
+
+	"github.com/coze-dev/coze-studio/backend/domain/knowledge/entity"
+	model "github.com/coze-dev/coze-studio/backend/domain/knowledge/internal/dal/model"
+)
+
+// MockKnowledgeDocumentSliceRepo is a mock of KnowledgeDocumentSliceRepo interface.
+type MockKnowledgeDocumentSliceRepo struct {
+	ctrl     *gomock.Controller
+	recorder *MockKnowledgeDocumentSliceRepoMockRecorder
+}
+
+// MockKnowledgeDocumentSliceRepoMockRecorder is the mock recorder for MockKnowledgeDocumentSliceRepo.
+type MockKnowledgeDocumentSliceRepoMockRecorder struct {
+	mock *MockKnowledgeDocumentSliceRepo
+}
+
+// NewMockKnowledgeDocumentSliceRepo creates a new mock instance.
+func NewMockKnowledgeDocumentSliceRepo(ctrl *gomock.Controller) *MockKnowledgeDocumentSliceRepo {
+	mock := &MockKnowledgeDocumentSliceRepo{ctrl: ctrl}
+	mock.recorder = &MockKnowledgeDocumentSliceRepoMockRecorder{mock}
+	return mock
+}
+
+// EXPECT returns an object that allows the caller to indicate expected use.
+func (m *MockKnowledgeDocumentSliceRepo) EXPECT() *MockKnowledgeDocumentSliceRepoMockRecorder {
+	return m.recorder
+}
+
+// BatchCreate mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) BatchCreate(ctx context.Context, slices []*model.KnowledgeDocumentSlice) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "BatchCreate", ctx, slices)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// BatchCreate indicates an expected call of BatchCreate.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) BatchCreate(ctx, slices any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchCreate", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).BatchCreate), ctx, slices)
+}
+
+// BatchSetStatus mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) BatchSetStatus(ctx context.Context, ids []int64, status int32, reason string) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "BatchSetStatus", ctx, ids, status, reason)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// BatchSetStatus indicates an expected call of BatchSetStatus.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) BatchSetStatus(ctx, ids, status, reason any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchSetStatus", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).BatchSetStatus), ctx, ids, status, reason)
+}
+
+// Create mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) Create(ctx context.Context, slice *model.KnowledgeDocumentSlice) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Create", ctx, slice)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Create indicates an expected call of Create.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) Create(ctx, slice any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).Create), ctx, slice)
+}
+
+// Delete mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) Delete(ctx context.Context, slice *model.KnowledgeDocumentSlice) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Delete", ctx, slice)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Delete indicates an expected call of Delete.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) Delete(ctx, slice any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).Delete), ctx, slice)
+}
+
+// DeleteByDocument mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) DeleteByDocument(ctx context.Context, documentID int64) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "DeleteByDocument", ctx, documentID)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// DeleteByDocument indicates an expected call of DeleteByDocument.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) DeleteByDocument(ctx, documentID any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteByDocument", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).DeleteByDocument), ctx, documentID)
+}
+
+// FindSliceByCondition mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) FindSliceByCondition(ctx context.Context, opts *entity.WhereSliceOpt) ([]*model.KnowledgeDocumentSlice, int64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "FindSliceByCondition", ctx, opts)
+	ret0, _ := ret[0].([]*model.KnowledgeDocumentSlice)
+	ret1, _ := ret[1].(int64)
+	ret2, _ := ret[2].(error)
+	return ret0, ret1, ret2
+}
+
+// FindSliceByCondition indicates an expected call of FindSliceByCondition.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) FindSliceByCondition(ctx, opts any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FindSliceByCondition", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).FindSliceByCondition), ctx, opts)
+}
+
+// GetDocumentSliceIDs mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) GetDocumentSliceIDs(ctx context.Context, docIDs []int64) ([]int64, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetDocumentSliceIDs", ctx, docIDs)
+	ret0, _ := ret[0].([]int64)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetDocumentSliceIDs indicates an expected call of GetDocumentSliceIDs.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) GetDocumentSliceIDs(ctx, docIDs any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDocumentSliceIDs", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).GetDocumentSliceIDs), ctx, docIDs)
+}
+
+// GetSliceBySequence mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) GetSliceBySequence(ctx context.Context, documentID, sequence int64) ([]*model.KnowledgeDocumentSlice, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "GetSliceBySequence", ctx, documentID, sequence)
+	ret0, _ := ret[0].([]*model.KnowledgeDocumentSlice)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// GetSliceBySequence indicates an expected call of GetSliceBySequence.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) GetSliceBySequence(ctx, documentID, sequence any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSliceBySequence", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).GetSliceBySequence), ctx, documentID, sequence)
+}
+
+// List mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) List(ctx context.Context, knowledgeID, documentID int64, limit int, cursor *string) ([]*model.KnowledgeDocumentSlice, *string, bool, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "List", ctx, knowledgeID, documentID, limit, cursor)
+	ret0, _ := ret[0].([]*model.KnowledgeDocumentSlice)
+	ret1, _ := ret[1].(*string)
+	ret2, _ := ret[2].(bool)
+	ret3, _ := ret[3].(error)
+	return ret0, ret1, ret2, ret3
+}
+
+// List indicates an expected call of List.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) List(ctx, knowledgeID, documentID, limit, cursor any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).List), ctx, knowledgeID, documentID, limit, cursor)
+}
+
+// ListStatus mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) ListStatus(ctx context.Context, documentID int64, limit int, cursor *string) ([]*model.SliceProgress, *string, bool, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "ListStatus", ctx, documentID, limit, cursor)
+	ret0, _ := ret[0].([]*model.SliceProgress)
+	ret1, _ := ret[1].(*string)
+	ret2, _ := ret[2].(bool)
+	ret3, _ := ret[3].(error)
+	return ret0, ret1, ret2, ret3
+}
+
+// ListStatus indicates an expected call of ListStatus.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) ListStatus(ctx, documentID, limit, cursor any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListStatus", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).ListStatus), ctx, documentID, limit, cursor)
+}
+
+// MGetSlices mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) MGetSlices(ctx context.Context, sliceIDs []int64) ([]*model.KnowledgeDocumentSlice, error) {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "MGetSlices", ctx, sliceIDs)
+	ret0, _ := ret[0].([]*model.KnowledgeDocumentSlice)
+	ret1, _ := ret[1].(error)
+	return ret0, ret1
+}
+
+// MGetSlices indicates an expected call of MGetSlices.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) MGetSlices(ctx, sliceIDs any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MGetSlices", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).MGetSlices), ctx, sliceIDs)
+}
+
+// Update mocks base method.
+func (m *MockKnowledgeDocumentSliceRepo) Update(ctx context.Context, slice *model.KnowledgeDocumentSlice) error {
+	m.ctrl.T.Helper()
+	ret := m.ctrl.Call(m, "Update", ctx, slice)
+	ret0, _ := ret[0].(error)
+	return ret0
+}
+
+// Update indicates an expected call of Update.
+func (mr *MockKnowledgeDocumentSliceRepoMockRecorder) Update(ctx, slice any) *gomock.Call {
+	mr.mock.ctrl.T.Helper()
+	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockKnowledgeDocumentSliceRepo)(nil).Update), ctx, slice)
+}