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

refactor(runtime): update function signatures to use value receivers for models

Browse Source
This commit is contained in:
mlogclub
2026-04-19 11:38:05 +08:00
parent 680793f527
commit 3e98e9379c
9 changed files with 107 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/ai/rag/index_flow.go
+13 -19
View File
@@ -10,43 +10,37 @@ import (
"github.com/mlogclub/simple/sqls"
)
func (s *index) loadDocumentByID(documentID int64) (*models.KnowledgeDocument, error) {
func (s *index) loadDocumentByID(documentID int64) (models.KnowledgeDocument, error) {
document := repositories.KnowledgeDocumentRepository.Get(sqls.DB(), documentID)
if document == nil {
return nil, fmt.Errorf("document not found: %d", documentID)
return models.KnowledgeDocument{}, fmt.Errorf("document not found: %d", documentID)
}
return document, nil
return *document, nil
}
func (s *index) loadFAQByID(faqID int64) (*models.KnowledgeFAQ, error) {
func (s *index) loadFAQByID(faqID int64) (models.KnowledgeFAQ, error) {
faq := repositories.KnowledgeFAQRepository.Get(sqls.DB(), faqID)
if faq == nil {
return nil, fmt.Errorf("faq not found: %d", faqID)
return models.KnowledgeFAQ{}, fmt.Errorf("faq not found: %d", faqID)
}
return faq, nil
return *faq, nil
}
func (s *index) loadDocumentKnowledgeBase(document *models.KnowledgeDocument) (*models.KnowledgeBase, error) {
if document == nil {
return nil, fmt.Errorf("document is nil")
}
func (s *index) loadDocumentKnowledgeBase(document models.KnowledgeDocument) (models.KnowledgeBase, error) {
knowledgeBase := repositories.KnowledgeBaseRepository.Get(sqls.DB(), document.KnowledgeBaseID)
if knowledgeBase == nil {
return nil, fmt.Errorf("knowledge base not found: %d", document.KnowledgeBaseID)
return models.KnowledgeBase{}, fmt.Errorf("knowledge base not found: %d", document.KnowledgeBaseID)
}
return knowledgeBase, nil
return *knowledgeBase, nil
}
func (s *index) loadFAQKnowledgeBase(faq *models.KnowledgeFAQ) (*models.KnowledgeBase, error) {
if faq == nil {
return nil, fmt.Errorf("faq is nil")
}
func (s *index) loadFAQKnowledgeBase(faq models.KnowledgeFAQ) (models.KnowledgeBase, error) {
knowledgeBase := repositories.KnowledgeBaseRepository.Get(sqls.DB(), faq.KnowledgeBaseID)
if knowledgeBase == nil {
return nil, fmt.Errorf("knowledge base not found: %d", faq.KnowledgeBaseID)
return models.KnowledgeBase{}, fmt.Errorf("knowledge base not found: %d", faq.KnowledgeBaseID)
}
if knowledgeBase.KnowledgeType != string(enums.KnowledgeBaseTypeFAQ) {
return nil, fmt.Errorf("knowledge base %d is not faq type", knowledgeBase.ID)
return models.KnowledgeBase{}, fmt.Errorf("knowledge base %d is not faq type", knowledgeBase.ID)
}
return knowledgeBase, nil
return *knowledgeBase, nil
}