wlw/ai-agent
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: streamline document and FAQ index removal methods and enhance knowledge base deletion logic

Browse Source
This commit is contained in:
mlogclub
2026-06-02 11:25:30 +08:00
parent 5c4d486df0
commit abedafe911
13 changed files with 215 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/repositories/knowledge_chunk_repository.go
+8 -4
View File
@@ -96,12 +96,16 @@ func (r *knowledgeChunkRepository) Delete(db *gorm.DB, id int64) {
db.Delete(&models.KnowledgeChunk{}, "id = ?", id)
}
func (r *knowledgeChunkRepository) DeleteByDocumentID(db *gorm.DB, documentID int64) {
db.Delete(&models.KnowledgeChunk{}, "document_id = ?", documentID)
func (r *knowledgeChunkRepository) DeleteByDocumentID(db *gorm.DB, documentID int64) error {
return db.Delete(&models.KnowledgeChunk{}, "document_id = ?", documentID).Error
}
func (r *knowledgeChunkRepository) DeleteByFaqID(db *gorm.DB, faqID int64) {
db.Delete(&models.KnowledgeChunk{}, "faq_id = ?", faqID)
func (r *knowledgeChunkRepository) DeleteByFaqID(db *gorm.DB, faqID int64) error {
return db.Delete(&models.KnowledgeChunk{}, "faq_id = ?", faqID).Error
}
func (r *knowledgeChunkRepository) DeleteByKnowledgeBaseID(db *gorm.DB, knowledgeBaseID int64) error {
return db.Delete(&models.KnowledgeChunk{}, "knowledge_base_id = ?", knowledgeBaseID).Error
}
func (r *knowledgeChunkRepository) FindByDocumentID(db *gorm.DB, documentID int64) (list []models.KnowledgeChunk) {