mlogclub
84 lines
2.4 KiB
Go
84 lines
2.4 KiB
Go
package skills
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"cs-agent/internal/models"
|
|
"cs-agent/internal/pkg/enums"
|
|
"cs-agent/internal/pkg/errorsx"
|
|
|
|
"github.com/mlogclub/simple/common/strs"
|
|
)
|
|
|
|
type intentTriggerConfig struct {
|
|
Intents []string `json:"intents"`
|
|
}
|
|
|
|
// MatchSkill 对单个 SkillDefinition 执行命中判断。
|
|
func MatchSkill(execCtx context.Context, ctx RuntimeContext, aiAgent *models.AIAgent, aiConfig *models.AIConfig) (*models.SkillDefinition, string, *RouteTrace, error) {
|
|
loader := newCandidateLoader()
|
|
if strs.IsNotBlank(ctx.ManualSkillCode) {
|
|
skill := loader.findManualSkillDefinition(ctx.ManualSkillCode)
|
|
if skill == nil || skill.Status != enums.StatusOk {
|
|
return nil, "", nil, errorsx.InvalidParam("Skill 不存在或未启用")
|
|
}
|
|
return skill, "manual_skill_code", &RouteTrace{
|
|
Status: "manual_selected",
|
|
SelectedSkillCode: skill.Code,
|
|
}, nil
|
|
}
|
|
|
|
candidates := loader.loadCandidateSkills(aiAgent)
|
|
trace := &RouteTrace{
|
|
Status: "started",
|
|
CandidateSkillCodes: make([]string, 0, len(candidates)),
|
|
}
|
|
for _, item := range candidates {
|
|
trace.CandidateSkillCodes = append(trace.CandidateSkillCodes, item.Code)
|
|
}
|
|
if len(candidates) == 0 {
|
|
trace.Status = "no_candidate"
|
|
return nil, "no_enabled_skill_bound", trace, nil
|
|
}
|
|
|
|
intentCode := strings.TrimSpace(ctx.IntentCode)
|
|
if intentCode != "" {
|
|
for _, item := range candidates {
|
|
if strings.EqualFold(strings.TrimSpace(item.Code), intentCode) {
|
|
trace.Status = "intent_selected"
|
|
trace.SelectedSkillCode = item.Code
|
|
return &item, "intent_code", trace, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(candidates) == 1 {
|
|
trace.Status = "single_candidate"
|
|
trace.SelectedSkillCode = candidates[0].Code
|
|
return &candidates[0], "single_candidate", trace, nil
|
|
}
|
|
|
|
selected, routeTrace, err := routeSkillWithLLM(execCtx, aiConfig, ctx.UserMessage, candidates)
|
|
if routeTrace != nil {
|
|
trace.Status = routeTrace.Status
|
|
trace.SelectedSkillCode = routeTrace.SelectedSkillCode
|
|
trace.RawDecision = routeTrace.RawDecision
|
|
trace.LatencyMs = routeTrace.LatencyMs
|
|
trace.Error = routeTrace.Error
|
|
}
|
|
if err != nil {
|
|
if trace.Error == "" {
|
|
trace.Error = err.Error()
|
|
}
|
|
return nil, "route_error", trace, err
|
|
}
|
|
if selected == nil {
|
|
if trace.Status == "started" {
|
|
trace.Status = "not_matched"
|
|
}
|
|
return nil, "route_none", trace, nil
|
|
}
|
|
return selected, "llm_route", trace, nil
|
|
}
|