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

refactor(rag): update loadDocumentByID and loadFAQByID to return pointers and simplify error handling

Browse Source
This commit is contained in:
mlogclub
2026-04-19 11:56:03 +08:00
parent 700ec2a38b
commit 60956488e0
2 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/ai/rag/index.go
+4 -4
View File
@@ -51,7 +51,7 @@ func (s *index) IndexDocumentByID(ctx context.Context, documentID int64) error {
if err != nil {
return err
}
return s.IndexDocument(ctx, document)
return s.IndexDocument(ctx, *document)
}
func (s *index) IndexDocument(ctx context.Context, document models.KnowledgeDocument) error {
@@ -72,7 +72,7 @@ func (s *index) IndexDocument(ctx context.Context, document models.KnowledgeDocu
if err != nil {
return fail(err)
}
vectors, chunkCount, err := s.runDocumentIndex(ctx, document, knowledgeBase)
vectors, chunkCount, err := s.runDocumentIndex(ctx, document, *knowledgeBase)
if err != nil {
return fail(err)
}
@@ -105,11 +105,11 @@ func (s *index) IndexFAQByID(ctx context.Context, faqID int64) error {
}
return err
}
knowledgeBase, err := s.loadFAQKnowledgeBase(faq)
knowledgeBase, err := s.loadFAQKnowledgeBase(*faq)
if err != nil {
return fail(err)
}
if err := s.runFAQIndex(ctx, faq, knowledgeBase); err != nil {
if err := s.runFAQIndex(ctx, *faq, *knowledgeBase); err != nil {
return fail(err)
}
if err := s.markFAQIndexIndexed(faq.ID); err != nil {