mlogclub
53 lines
1.6 KiB
Go
53 lines
1.6 KiB
Go
package rag
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"cs-agent/internal/models"
|
|
"cs-agent/internal/pkg/enums"
|
|
"cs-agent/internal/repositories"
|
|
|
|
"github.com/mlogclub/simple/sqls"
|
|
)
|
|
|
|
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 document, nil
|
|
}
|
|
|
|
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 faq, nil
|
|
}
|
|
|
|
func (s *index) loadDocumentKnowledgeBase(document *models.KnowledgeDocument) (*models.KnowledgeBase, error) {
|
|
if document == nil {
|
|
return nil, fmt.Errorf("document is nil")
|
|
}
|
|
knowledgeBase := repositories.KnowledgeBaseRepository.Get(sqls.DB(), document.KnowledgeBaseID)
|
|
if knowledgeBase == nil {
|
|
return nil, fmt.Errorf("knowledge base not found: %d", document.KnowledgeBaseID)
|
|
}
|
|
return knowledgeBase, nil
|
|
}
|
|
|
|
func (s *index) loadFAQKnowledgeBase(faq *models.KnowledgeFAQ) (*models.KnowledgeBase, error) {
|
|
if faq == nil {
|
|
return nil, fmt.Errorf("faq is nil")
|
|
}
|
|
knowledgeBase := repositories.KnowledgeBaseRepository.Get(sqls.DB(), faq.KnowledgeBaseID)
|
|
if knowledgeBase == nil {
|
|
return nil, 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 knowledgeBase, nil
|
|
}
|