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

refactor: replace document and FAQ chunk handling with repository methods for improved clarity and maintainability

Browse Source
This commit is contained in:
mlogclub
2026-06-02 12:01:01 +08:00
parent 99d5a951be
commit c09ee0e596
3 changed files with 26 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/ai/rag/index_run_helpers.go
+2 -2
View File
@@ -39,7 +39,7 @@ func (s *index) runDocumentIndex(ctx context.Context, document models.KnowledgeD
if err := provider.UpsertVectors(ctx, collectionName, vectors); err != nil {
return nil, 0, fmt.Errorf("failed to upsert vectors: %w", err)
}
if err := s.replaceDocumentChunks(document.ID, chunkModels); err != nil {
if err := repositories.KnowledgeChunkRepository.ReplaceByDocumentID(sqls.DB(), document.ID, chunkModels); err != nil {
return nil, 0, fmt.Errorf("failed to save chunks: %w", err)
}
if staleVectorIDs := s.collectStaleVectorIDs(existingChunks, vectors); len(staleVectorIDs) > 0 {
@@ -77,7 +77,7 @@ func (s *index) runFAQIndex(ctx context.Context, faq models.KnowledgeFAQ, knowle
if err := provider.UpsertVectors(ctx, collectionName, []vectordb.Vector{vector}); err != nil {
return fmt.Errorf("failed to upsert vectors: %w", err)
}
if err := s.replaceFAQChunk(faq.ID, &chunkModel); err != nil {
if err := repositories.KnowledgeChunkRepository.ReplaceByFaqID(sqls.DB(), faq.ID, &chunkModel); err != nil {
return fmt.Errorf("failed to save faq chunk: %w", err)
}
if staleVectorIDs := s.collectStaleVectorIDs(existingChunks, []vectordb.Vector{vector}); len(staleVectorIDs) > 0 {