138 lines
3.8 KiB
Go
138 lines
3.8 KiB
Go
package chat_test
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"chat-agent-system/internal/bizdev"
|
|
"chat-agent-system/internal/chat"
|
|
"chat-agent-system/internal/config"
|
|
"chat-agent-system/internal/coordinator"
|
|
"chat-agent-system/internal/ctxmgr"
|
|
"chat-agent-system/internal/intent"
|
|
"chat-agent-system/internal/squad"
|
|
)
|
|
|
|
func TestChatSession_AssistantMetadataTurnIntentAndReplySource(t *testing.T) {
|
|
rec := intent.NewRecognizer()
|
|
engine := coordinator.NewCoordinatorEngine(nil)
|
|
|
|
userCtx := &coordinator.UserProjectContext{
|
|
SnapshotID: "sess_ctx_turn_intent",
|
|
User: coordinator.UserInfo{
|
|
ID: "u1",
|
|
Name: "张工",
|
|
Role: "engineer",
|
|
ProjectRole: "composer",
|
|
},
|
|
Project: coordinator.ProjectInfo{
|
|
Name: "数字化超声吸引系统",
|
|
CurrentStage: bizdev.StageV01_Triage,
|
|
},
|
|
LatestAdvisorBrief: &squad.AdvisorBriefing{
|
|
MaturityScore: 80,
|
|
MaturityLevel: "V0.1",
|
|
MaturityHeadline: "已就绪",
|
|
},
|
|
}
|
|
|
|
cfg := &config.Config{DataDir: t.TempDir()}
|
|
svc := bizdev.MustNewService(t, cfg)
|
|
mgr := bizdev.Snapshot(svc).Ctx
|
|
|
|
sess := chat.NewSession("sess_test_turn_intent", userCtx, rec, engine)
|
|
sess.CtxMgr = mgr
|
|
|
|
ctx := context.Background()
|
|
_, err := sess.SendMessage(ctx, "你好，请问当前项目进展如何？")
|
|
if err != nil {
|
|
t.Fatalf("SendMessage failed: %v", err)
|
|
}
|
|
|
|
// 校验通过 ContextManager 记录的消息
|
|
history, err := mgr.GetChatHistory(ctx, "sess_test_turn_intent", 10, 0)
|
|
if err != nil {
|
|
t.Fatalf("GetChatHistory failed: %v", err)
|
|
}
|
|
|
|
if len(history) < 2 {
|
|
t.Fatalf("expected at least 2 messages in history, got %d", len(history))
|
|
}
|
|
|
|
// 助手消息在后
|
|
var asstMsg *ctxmgr.ChatMessageRecord
|
|
for i := range history {
|
|
if history[i].Role == "assistant" {
|
|
asstMsg = &history[i]
|
|
break
|
|
}
|
|
}
|
|
|
|
if asstMsg == nil {
|
|
t.Fatalf("assistant message not found in history")
|
|
}
|
|
|
|
if asstMsg.Metadata == nil {
|
|
t.Fatalf("expected assistant message metadata to be non-nil")
|
|
}
|
|
|
|
turnIntent, ok := asstMsg.Metadata["turn_intent"].(string)
|
|
if !ok || turnIntent == "" {
|
|
t.Errorf("expected turn_intent to be non-empty string in assistant metadata, got %v", asstMsg.Metadata["turn_intent"])
|
|
}
|
|
|
|
replySource, ok := asstMsg.Metadata["reply_source"].(string)
|
|
if !ok || replySource == "" {
|
|
t.Errorf("expected reply_source to be non-empty string in assistant metadata, got %v", asstMsg.Metadata["reply_source"])
|
|
}
|
|
}
|
|
|
|
func TestChatSession_MaterialFallbackReasonPersisted(t *testing.T) {
|
|
rec := intent.NewRecognizer()
|
|
engine := coordinator.NewCoordinatorEngine(nil)
|
|
userCtx := &coordinator.UserProjectContext{
|
|
SnapshotID: "sess_ctx_material",
|
|
Project: coordinator.ProjectInfo{
|
|
Name: "全自动冲击波治疗仪1",
|
|
CurrentStage: bizdev.StageV01_Triage,
|
|
},
|
|
}
|
|
cfg := &config.Config{DataDir: t.TempDir()}
|
|
svc := bizdev.MustNewService(t, cfg)
|
|
mgr := bizdev.Snapshot(svc).Ctx
|
|
|
|
sess := chat.NewSession("sess_test_material_fallback", userCtx, rec, engine)
|
|
sess.CtxMgr = mgr
|
|
|
|
ctx := context.Background()
|
|
_, err := sess.SendMessage(ctx, "上一代产品说明书")
|
|
if err != nil {
|
|
t.Fatalf("SendMessage failed: %v", err)
|
|
}
|
|
history, err := mgr.GetChatHistory(ctx, "sess_test_material_fallback", 10, 0)
|
|
if err != nil {
|
|
t.Fatalf("GetChatHistory failed: %v", err)
|
|
}
|
|
var asst *ctxmgr.ChatMessageRecord
|
|
for i := range history {
|
|
if history[i].Role == "assistant" {
|
|
asst = &history[i]
|
|
break
|
|
}
|
|
}
|
|
if asst == nil {
|
|
t.Fatal("assistant message missing")
|
|
}
|
|
if strings.Contains(asst.Content, "掂量") {
|
|
t.Fatalf("persisted template sales pitch: %s", asst.Content)
|
|
}
|
|
reason, _ := asst.Metadata["fallback_reason"].(string)
|
|
if reason == "" {
|
|
t.Fatalf("fallback_reason missing in metadata: %#v", asst.Metadata)
|
|
}
|
|
if src, _ := asst.Metadata["reply_source"].(string); src != coordinator.ReplySourceTemplate {
|
|
t.Fatalf("reply_source=%q", src)
|
|
}
|
|
}
|