e06284d0b0
后端： - ♻️ 将 `taskquery` 模块迁移至 `agent2`，并完成与 `agent2` 业务链路及整体结构的正式接入 前端： - 🧱 已完成基础框架搭建，并完成了登录、注册、主页等页面并对接了对应接口；但整体功能实现仍在完善中
82 lines
3.2 KiB
Go
82 lines
3.2 KiB
Go
package agentnode
|
||
|
||
import (
|
||
"strings"
|
||
"testing"
|
||
|
||
agentllm "github.com/LoveLosita/smartflow/backend/agent2/llm"
|
||
agentmodel "github.com/LoveLosita/smartflow/backend/agent2/model"
|
||
)
|
||
|
||
// TestExtractExplicitLimitFromUser_Number 验证阿拉伯数字条数可以被正确提取。
|
||
func TestExtractExplicitLimitFromUser_Number(t *testing.T) {
|
||
limit, ok := extractExplicitLimitFromUser("给我3个优先级低的任务")
|
||
if !ok {
|
||
t.Fatalf("期望识别到显式数量")
|
||
}
|
||
if limit != 3 {
|
||
t.Fatalf("数量识别错误，期望=3 实际=%d", limit)
|
||
}
|
||
}
|
||
|
||
// TestExtractExplicitLimitFromUser_ChineseNumber 验证中文数字也可以被正确提取。
|
||
func TestExtractExplicitLimitFromUser_ChineseNumber(t *testing.T) {
|
||
limit, ok := extractExplicitLimitFromUser("前五个简单任务给我看看")
|
||
if !ok {
|
||
t.Fatalf("期望识别到中文数量")
|
||
}
|
||
if limit != 5 {
|
||
t.Fatalf("数量识别错误，期望=5 实际=%d", limit)
|
||
}
|
||
}
|
||
|
||
// TestExtractExplicitLimitFromUser_LaiYiGe 验证“来一个”这类口语表达也能命中数量提取。
|
||
func TestExtractExplicitLimitFromUser_LaiYiGe(t *testing.T) {
|
||
limit, ok := extractExplicitLimitFromUser("来一个我的简单任务")
|
||
if !ok {
|
||
t.Fatalf("期望识别到‘来一个’的显式数量")
|
||
}
|
||
if limit != 1 {
|
||
t.Fatalf("数量识别错误，期望=1 实际=%d", limit)
|
||
}
|
||
}
|
||
|
||
// TestBuildTaskQueryFinalReply_RespectsLimit 验证最终回复严格遵守 plan.limit。
|
||
func TestBuildTaskQueryFinalReply_RespectsLimit(t *testing.T) {
|
||
items := []agentmodel.TaskQueryItem{
|
||
{ID: 1, Title: "任务1", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-16 10:00"},
|
||
{ID: 2, Title: "任务2", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-17 10:00"},
|
||
{ID: 3, Title: "任务3", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-18 10:00"},
|
||
}
|
||
reply := buildTaskQueryFinalReply(items, agentmodel.TaskQueryPlan{Limit: 2}, "好的")
|
||
if !strings.Contains(reply, "整理了 2 条任务") {
|
||
t.Fatalf("回复未体现 limit=2，reply=%s", reply)
|
||
}
|
||
if strings.Contains(reply, "3. ") {
|
||
t.Fatalf("回复中不应出现第 3 条，reply=%s", reply)
|
||
}
|
||
}
|
||
|
||
// TestBuildTaskQueryFinalReply_NoDuplicateList 验证 llmReply 自带列表时不会与后端列表重复拼接。
|
||
func TestBuildTaskQueryFinalReply_NoDuplicateList(t *testing.T) {
|
||
items := []agentmodel.TaskQueryItem{{ID: 1, Title: "任务1", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-16 10:00"}}
|
||
llmReply := "以下是你的任务：\n#1 任务1"
|
||
reply := buildTaskQueryFinalReply(items, agentmodel.TaskQueryPlan{Limit: 1}, llmReply)
|
||
if strings.Contains(reply, "以下是你的任务") {
|
||
t.Fatalf("不应保留 llm 列表头，reply=%s", reply)
|
||
}
|
||
if !strings.Contains(reply, "整理了 1 条任务") {
|
||
t.Fatalf("应保留后端确定性列表头，reply=%s", reply)
|
||
}
|
||
}
|
||
|
||
// TestApplyRetryPatch_RespectExplicitLimit 验证显式数量存在时，反思补丁不能覆盖 limit。
|
||
func TestApplyRetryPatch_RespectExplicitLimit(t *testing.T) {
|
||
plan := agentmodel.TaskQueryPlan{Limit: 1, SortBy: "deadline", Order: "asc"}
|
||
limit := 10
|
||
next := applyRetryPatch(plan, agentllm.TaskQueryRetryPatch{Limit: &limit}, 1)
|
||
if next.Limit != 1 {
|
||
t.Fatalf("显式数量锁应生效，期望=1 实际=%d", next.Limit)
|
||
}
|
||
}
|