71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package ctxmgr
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestConversationTaskState_Lifecycle(t *testing.T) {
|
|
ts := NewConversationTaskState()
|
|
|
|
// Turn 1: 用户表达立项意图与器械类型
|
|
ts.UpdateFromTurn(
|
|
"我们打算立项研发一款高频超声手术刀系统",
|
|
"SUPPLEMENT_INFO",
|
|
map[string]string{"device_type": "高频超声手术刀系统"},
|
|
"NORMAL",
|
|
"V0.1_triage",
|
|
)
|
|
|
|
if ts.ActiveTask != "SUPPLEMENT_INFO" {
|
|
t.Errorf("expected ActiveTask SUPPLEMENT_INFO, got %s", ts.ActiveTask)
|
|
}
|
|
if ts.TopicAnchor != "我们打算立项研发一款高频超声手术刀系统" {
|
|
t.Errorf("unexpected TopicAnchor: %s", ts.TopicAnchor)
|
|
}
|
|
if ts.CollectedSlots["device_type"] != "高频超声手术刀系统" {
|
|
t.Errorf("expected device_type collected, got %v", ts.CollectedSlots)
|
|
}
|
|
// 缺失槽位应有 target_market 和 budget
|
|
hasMarket := false
|
|
hasBudget := false
|
|
for _, m := range ts.MissingSlots {
|
|
if m == "target_market" {
|
|
hasMarket = true
|
|
}
|
|
if m == "budget" {
|
|
hasBudget = true
|
|
}
|
|
}
|
|
if !hasMarket || !hasBudget {
|
|
t.Errorf("expected missing target_market and budget, got %v", ts.MissingSlots)
|
|
}
|
|
|
|
// Turn 2: 补充目标市场与预算
|
|
ts.UpdateFromTurn(
|
|
"目标市场主要为国内和欧洲，初期预算约为200万元",
|
|
"SUPPLEMENT_INFO",
|
|
map[string]string{
|
|
"target_market": "国内和欧洲",
|
|
"budget": "200万元",
|
|
},
|
|
"NORMAL",
|
|
"V0.1_triage",
|
|
)
|
|
|
|
if len(ts.MissingSlots) != 0 {
|
|
t.Errorf("expected 0 missing slots after turn 2, got %v", ts.MissingSlots)
|
|
}
|
|
|
|
snippet := ts.RenderPromptSnippet()
|
|
if !strings.Contains(snippet, "【本轮任务状态 (Working Memory)】") {
|
|
t.Errorf("missing header in prompt snippet: %s", snippet)
|
|
}
|
|
if !strings.Contains(snippet, "device_type=高频超声手术刀系统") {
|
|
t.Errorf("missing device_type in prompt snippet: %s", snippet)
|
|
}
|
|
if !strings.Contains(snippet, "target_market=国内和欧洲") {
|
|
t.Errorf("missing target_market in prompt snippet: %s", snippet)
|
|
}
|
|
}
|