Files
ai-agent/internal/services/knowledge_retrieve_log_service.go
T
mlogclub 101577f163 Refactor internal services to use agent-desk package structure
- Updated import paths in multiple service files to reflect the new agent-desk module.
- Added a new configuration file for agent-desk in the Docker setup.
2026-05-31 18:43:48 +08:00

48 lines
1.2 KiB
Go

package services
import (
"agent-desk/internal/models"
"agent-desk/internal/pkg/httpx/params"
"github.com/mlogclub/simple/sqls"
)
var KnowledgeRetrieveLogService = newKnowledgeRetrieveLogService()
func newKnowledgeRetrieveLogService() *knowledgeRetrieveLogService {
return &knowledgeRetrieveLogService{}
}
type knowledgeRetrieveLogService struct {
}
func (s *knowledgeRetrieveLogService) Get(id int64) *models.KnowledgeRetrieveLog {
ret := &models.KnowledgeRetrieveLog{}
if err := sqls.DB().First(ret, "id = ?", id).Error; err != nil {
return nil
}
return ret
}
func (s *knowledgeRetrieveLogService) FindPageByParams(params *params.QueryParams) (list []models.KnowledgeRetrieveLog, paging *sqls.Paging) {
cnd := &params.Cnd
cnd.Find(sqls.DB(), &list)
count := cnd.Count(sqls.DB(), &models.KnowledgeRetrieveLog{})
paging = &sqls.Paging{
Page: cnd.Paging.Page,
Limit: cnd.Paging.Limit,
Total: count,
}
return
}
func (s *knowledgeRetrieveLogService) FindHitsByRetrieveLogID(retrieveLogID int64) []models.KnowledgeRetrieveHit {
if retrieveLogID <= 0 {
return nil
}
var list []models.KnowledgeRetrieveHit
sqls.DB().Where("retrieve_log_id = ?", retrieveLogID).Order("rank_no asc, id asc").Find(&list)
return list
}