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.
39 lines
1012 B
Go
39 lines
1012 B
Go
package rag
|
|
|
|
import (
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"cs-ai-agent/internal/models"
|
|
)
|
|
|
|
func newRetrieveTrace() *RetrieveTrace {
|
|
return &RetrieveTrace{}
|
|
}
|
|
|
|
func (s *retrieve) prepareRetrievableKnowledgeBases(req RetrieveRequest, trace *RetrieveTrace) ([]models.KnowledgeBase, []int64, bool) {
|
|
if req.Query == "" {
|
|
return nil, nil, false
|
|
}
|
|
knowledgeBaseIDs := normalizeKnowledgeBaseIDs(req.KnowledgeBaseIDs)
|
|
if len(knowledgeBaseIDs) == 0 {
|
|
return nil, nil, false
|
|
}
|
|
|
|
retrievableKnowledgeBases := s.loadRetrievableKnowledgeBases(knowledgeBaseIDs)
|
|
if len(retrievableKnowledgeBases) == 0 {
|
|
slog.Info("Skip retrieve for non-enabled knowledge bases",
|
|
"knowledge_base_ids", fmt.Sprint(knowledgeBaseIDs))
|
|
return nil, knowledgeBaseIDs, false
|
|
}
|
|
return retrievableKnowledgeBases, knowledgeBaseIDs, true
|
|
}
|
|
|
|
func applySearchTrace(target *RetrieveTrace, source *RetrieveTrace) {
|
|
if target == nil || source == nil {
|
|
return
|
|
}
|
|
target.EmbeddingMs = source.EmbeddingMs
|
|
target.VectorSearchMs = source.VectorSearchMs
|
|
}
|