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

refactor: update document index removal methods for consistency and clarity

Browse Source
This commit is contained in:
mlogclub
2026-06-02 11:29:06 +08:00
parent abedafe911
commit 7a2f673a02
2 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/ai/rag/index.go
+3 -7
View File
@@ -119,15 +119,11 @@ func (s *index) IndexFAQByID(ctx context.Context, faqID int64) error {
} }
func (s *index) RemoveDocumentIndex(ctx context.Context, documentID int64) error { func (s *index) RemoveDocumentIndex(ctx context.Context, documentID int64) error {
chunks := repositories.KnowledgeChunkRepository.Find(sqls.DB(), sqls.NewCnd().Eq("document_id", documentID)) chunks := repositories.KnowledgeChunkRepository.FindByDocumentID(sqls.DB(), documentID)
return s.removeDocumentIndexByChunks(ctx, documentID, chunks) return s.RemoveDocumentIndexByChunks(ctx, documentID, chunks)
} }
func (s *index) RemoveDocumentIndexByChunkModels(ctx context.Context, documentID int64, chunks []models.KnowledgeChunk) error { func (s *index) RemoveDocumentIndexByChunks(ctx context.Context, documentID int64, chunks []models.KnowledgeChunk) error {
return s.removeDocumentIndexByChunks(ctx, documentID, chunks)
}
func (s *index) removeDocumentIndexByChunks(ctx context.Context, documentID int64, chunks []models.KnowledgeChunk) error {
if len(chunks) == 0 { if len(chunks) == 0 {
return nil return nil
} }
internal/services/knowledge_document_service.go
+1 -1
View File
@@ -178,7 +178,7 @@ func (s *knowledgeDocumentService) DeleteKnowledgeDocument(id int64) error {
}); err != nil { }); err != nil {
return err return err
} }
return rag.Index.RemoveDocumentIndexByChunkModels(context.Background(), id, chunks) return rag.Index.RemoveDocumentIndexByChunks(context.Background(), id, chunks)
} }
func (s *knowledgeDocumentService) buildKnowledgeDocumentModel(req request.CreateKnowledgeDocumentRequest) (*models.KnowledgeDocument, error) { func (s *knowledgeDocumentService) buildKnowledgeDocumentModel(req request.CreateKnowledgeDocumentRequest) (*models.KnowledgeDocument, error) {