mlogclub
541e4c5874
- Updated skill handling to use skill IDs instead of skill codes in various components, services, and models. - Modified tests to reflect changes in skill identification. - Removed references to skill codes in favor of skill IDs for consistency and clarity. - Updated localization files to remove skill code references and adjust error messages accordingly.
71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
package skills
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"agent-desk/internal/models"
|
|
"agent-desk/internal/repositories"
|
|
|
|
"github.com/mlogclub/simple/sqls"
|
|
)
|
|
|
|
func newRunLogService() *RunLogService {
|
|
return &RunLogService{}
|
|
}
|
|
|
|
type RunLogService struct{}
|
|
|
|
// Build 根据执行计划与运行结果构建 Skill 运行日志。
|
|
func (s *RunLogService) Build(ctx RuntimeContext, plan *ExecutionPlan, trace *ExecutionTrace, err error) *models.SkillRunLog {
|
|
log := &models.SkillRunLog{
|
|
ConversationID: ctx.ConversationID,
|
|
AIAgentID: ctx.AIAgent.ID,
|
|
ManualSkillID: ctx.ManualSkillDefinitionID,
|
|
UserMessage: ctx.UserMessage,
|
|
TraceData: s.buildTraceData(trace),
|
|
CreatedAt: time.Now(),
|
|
}
|
|
if plan != nil {
|
|
log.AIConfigID = plan.AIConfig.ID
|
|
log.UsedModel = plan.AIConfig.ModelName
|
|
log.UsedProvider = plan.AIConfig.Provider
|
|
|
|
if plan.Skill != nil {
|
|
log.SkillDefinitionID = plan.Skill.ID
|
|
log.Matched = true
|
|
log.FinalSelected = true
|
|
log.MatchReason = plan.MatchReason
|
|
}
|
|
}
|
|
if err != nil {
|
|
log.ErrorMessage = err.Error()
|
|
} else if !log.Matched {
|
|
if plan != nil && plan.MatchReason != "" {
|
|
log.MatchReason = plan.MatchReason
|
|
} else {
|
|
log.MatchReason = "not_matched"
|
|
}
|
|
}
|
|
return log
|
|
}
|
|
|
|
// Write 写入 Skill 路由日志。
|
|
func (s *RunLogService) Write(log *models.SkillRunLog) error {
|
|
if log == nil {
|
|
return nil
|
|
}
|
|
return repositories.SkillRunLogRepository.Create(sqls.DB(), log)
|
|
}
|
|
|
|
func (s *RunLogService) buildTraceData(trace *ExecutionTrace) string {
|
|
if trace == nil {
|
|
return ""
|
|
}
|
|
data, err := json.Marshal(trace)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
return string(data)
|
|
}
|