t
2bbf42b741
Remove Agent Desk users, roles, login sessions, tokens, and local permission persistence. Expose the backend as an embeddable ai-agent module with host-provided subject lookup and operation authorization callbacks, and complete the frontend/backend repository split.
92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package rag
|
|
|
|
import (
|
|
"time"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/models"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/enums"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/repositories"
|
|
|
|
"github.com/mlogclub/simple/sqls"
|
|
)
|
|
|
|
func (s *index) markDocumentIndexPending(documentID int64) error {
|
|
return repositories.KnowledgeDocumentRepository.Updates(sqls.DB(), documentID, map[string]any{
|
|
"index_status": enums.KnowledgeDocumentIndexStatusPending,
|
|
"indexed_at": nil,
|
|
"index_error": "",
|
|
"updated_at": time.Now(),
|
|
})
|
|
}
|
|
|
|
func (s *index) markDocumentIndexIndexed(documentID int64) error {
|
|
now := time.Now()
|
|
return repositories.KnowledgeDocumentRepository.Updates(sqls.DB(), documentID, map[string]any{
|
|
"index_status": enums.KnowledgeDocumentIndexStatusIndexed,
|
|
"indexed_at": &now,
|
|
"index_error": "",
|
|
"updated_at": now,
|
|
})
|
|
}
|
|
|
|
func (s *index) markDocumentIndexFailed(documentID int64, err error) error {
|
|
return repositories.KnowledgeDocumentRepository.Updates(sqls.DB(), documentID, map[string]any{
|
|
"index_status": enums.KnowledgeDocumentIndexStatusFailed,
|
|
"index_error": truncateIndexError(err),
|
|
"updated_at": time.Now(),
|
|
})
|
|
}
|
|
|
|
func (s *index) markKnowledgeBaseDocumentsIndexPending(knowledgeBaseID int64, documentIDs []int64) error {
|
|
if len(documentIDs) == 0 {
|
|
return nil
|
|
}
|
|
return sqls.DB().Model(&models.KnowledgeDocument{}).
|
|
Where("knowledge_base_id = ?", knowledgeBaseID).
|
|
Where("id IN ?", documentIDs).
|
|
Updates(map[string]any{
|
|
"index_status": enums.KnowledgeDocumentIndexStatusPending,
|
|
"indexed_at": nil,
|
|
"index_error": "",
|
|
"updated_at": time.Now(),
|
|
}).Error
|
|
}
|
|
|
|
func (s *index) markFAQIndexPending(faqID int64) error {
|
|
return repositories.KnowledgeFAQRepository.Updates(sqls.DB(), faqID, map[string]any{
|
|
"index_status": enums.KnowledgeDocumentIndexStatusPending,
|
|
"indexed_at": nil,
|
|
"index_error": "",
|
|
"updated_at": time.Now(),
|
|
})
|
|
}
|
|
|
|
func (s *index) markFAQIndexIndexed(faqID int64) error {
|
|
now := time.Now()
|
|
return repositories.KnowledgeFAQRepository.Updates(sqls.DB(), faqID, map[string]any{
|
|
"index_status": enums.KnowledgeDocumentIndexStatusIndexed,
|
|
"indexed_at": &now,
|
|
"index_error": "",
|
|
"updated_at": now,
|
|
})
|
|
}
|
|
|
|
func (s *index) markFAQIndexFailed(faqID int64, err error) error {
|
|
return repositories.KnowledgeFAQRepository.Updates(sqls.DB(), faqID, map[string]any{
|
|
"index_status": enums.KnowledgeDocumentIndexStatusFailed,
|
|
"index_error": truncateIndexError(err),
|
|
"updated_at": time.Now(),
|
|
})
|
|
}
|
|
|
|
func truncateIndexError(err error) string {
|
|
if err == nil {
|
|
return ""
|
|
}
|
|
message := err.Error()
|
|
if len(message) <= 1000 {
|
|
return message
|
|
}
|
|
return message[:1000]
|
|
}
|