63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
package ai
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
|
|
"gorm.io/gorm"
|
|
"senlinai-agent/backend/internal/models"
|
|
)
|
|
|
|
// ExpertFilter 是本地专家库的查询条件。
|
|
type ExpertFilter struct {
|
|
Category string
|
|
Query string
|
|
}
|
|
|
|
func (s *SessionService) ListExperts(filter ExpertFilter) ([]models.SenlinAgentAIExpertItem, error) {
|
|
query := models.DBService.Model(&models.SenlinAgentAIExpertItem{}).Where("enabled = ?", true)
|
|
if category := strings.TrimSpace(filter.Category); category != "" {
|
|
query = query.Where("category = ?", category)
|
|
}
|
|
if keyword := strings.ToLower(strings.TrimSpace(filter.Query)); keyword != "" {
|
|
like := "%" + keyword + "%"
|
|
query = query.Where(
|
|
"LOWER(name) LIKE ? OR LOWER(description) LIKE ? OR LOWER(category_name) LIKE ?",
|
|
like, like, like,
|
|
)
|
|
}
|
|
var experts []models.SenlinAgentAIExpertItem
|
|
if err := query.Order("category asc, id asc").Find(&experts).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
if experts == nil {
|
|
experts = []models.SenlinAgentAIExpertItem{}
|
|
}
|
|
return experts, nil
|
|
}
|
|
|
|
func (s *SessionService) GetExpert(identity string) (*models.SenlinAgentAIExpertItem, error) {
|
|
var expert models.SenlinAgentAIExpertItem
|
|
if err := models.DBService.Where("identity = ? AND enabled = ?", strings.TrimSpace(identity), true).First(&expert).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, ErrExpertNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
return &expert, nil
|
|
}
|
|
|
|
func findExpertByIdentity(tx *gorm.DB, identity string) (*models.SenlinAgentAIExpertItem, error) {
|
|
if strings.TrimSpace(identity) == "" {
|
|
return nil, nil
|
|
}
|
|
var expert models.SenlinAgentAIExpertItem
|
|
if err := tx.Where("identity = ? AND enabled = ?", identity, true).First(&expert).Error; err != nil {
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil, ErrExpertNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
return &expert, nil
|
|
}
|