mlogclub
5d7c10aeab
- Updated runtime configuration to use __CS_AI_AGENT_WIDGET_CONFIG__ instead of __CS_AGENT_WIDGET_CONFIG__. - Changed message types in support host bridge from "cs-agent" to "cs-ai-agent". - Minified SDK script updated to reflect new AI agent naming conventions. - Adjusted scrollbar styles in main.scss to use .cs-ai-agent-scrollbar instead of .cs-agent-scrollbar.
92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package rag
|
|
|
|
import (
|
|
"time"
|
|
|
|
"cs-ai-agent/internal/models"
|
|
"cs-ai-agent/internal/pkg/enums"
|
|
"cs-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]
|
|
}
|