mlogclub
3c0abaaedc
- Moved the skill routing logic from matcher.go to a new router.go file for better organization. - Implemented replyCommitService to handle sending AI replies and managing reply rounds. - Added replyInterruptService to manage conversation interrupts and resume handling. - Created replyRunLogService to log AI reply actions and their outcomes. - Introduced helper functions for building conversation interrupts and resolving prompts. - Added unit tests for the new services and functions to ensure correctness. - Removed unused code and optimized imports in matcher.go.
49 lines
1.5 KiB
Go
49 lines
1.5 KiB
Go
package skills
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"cs-agent/internal/models"
|
|
)
|
|
|
|
func TestParseSkillExamples(t *testing.T) {
|
|
examples := parseSkillExamples(`[" 退款进度 ","","发票补开","修改收货地址","多余示例"]`)
|
|
if len(examples) != 3 {
|
|
t.Fatalf("expected 3 examples, got %d", len(examples))
|
|
}
|
|
if examples[0] != "退款进度" || examples[1] != "发票补开" || examples[2] != "修改收货地址" {
|
|
t.Fatalf("unexpected examples: %#v", examples)
|
|
}
|
|
}
|
|
|
|
func TestNormalizeRouteDecision(t *testing.T) {
|
|
if got := normalizeRouteDecision("```refund_skill```\n补充说明"); got != "refund_skill" {
|
|
t.Fatalf("unexpected normalized decision: %q", got)
|
|
}
|
|
if got := normalizeRouteDecision(" none "); got != "NONE" {
|
|
t.Fatalf("expected NONE, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestBuildSkillRoutePrompt(t *testing.T) {
|
|
prompt := buildSkillRoutePrompt("我要申请退款", []models.SkillDefinition{
|
|
{
|
|
Code: "refund_skill",
|
|
Name: "退款处理",
|
|
Description: "负责退款和退货相关问题",
|
|
Examples: `["退款进度","退货运费"]`,
|
|
},
|
|
})
|
|
|
|
if !strings.Contains(prompt, "skillCode=refund_skill") {
|
|
t.Fatalf("expected prompt to include skill code, got %q", prompt)
|
|
}
|
|
if !strings.Contains(prompt, "examples=退款进度 | 退货运费") {
|
|
t.Fatalf("expected prompt to include examples, got %q", prompt)
|
|
}
|
|
if !strings.Contains(prompt, "请只输出一个 skillCode 或 NONE。") {
|
|
t.Fatalf("expected prompt to include output constraint, got %q", prompt)
|
|
}
|
|
}
|