feat(cli): CLI支持-chat交互式对话REPL与单次自然语言指令调度
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
WeiCHEN
committed by
weichen
weichen
co-authored by
Cursor
Cursor
parent
82bfe41d1f
commit
52c36e6a87
+153
-1
@@ -1,15 +1,20 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"chat-agent-system/internal/bizdev"
|
||||
"chat-agent-system/internal/chat"
|
||||
"chat-agent-system/internal/config"
|
||||
"chat-agent-system/internal/coordinator"
|
||||
"chat-agent-system/internal/intent"
|
||||
"chat-agent-system/internal/squad"
|
||||
)
|
||||
|
||||
@@ -30,6 +35,13 @@ func main() {
|
||||
listTasks := flag.Bool("list-tasks", false, "列出所有受支持的商务对接任务清单")
|
||||
listStages := flag.Bool("list-stages", false, "列出所有受支持的商务成熟度阶段清单")
|
||||
addStep := flag.String("add-step", "", "动态向工作区追加任务步骤 (格式: role:title，配合 -inspect)")
|
||||
|
||||
// 对话协同与意图识别相关命令行参数
|
||||
chatMode := flag.Bool("chat", false, "启动与项目参谋的自然语言交互式对话 (REPL)")
|
||||
chatQuery := flag.String("query", "", "直接在命令行传入单次提问/指令并输出参谋诊断与调度结果 (配合 -chat)")
|
||||
workDirFlag := flag.String("work-dir", "", "关联现有工作区目录 (如 sandbox/intake_xxx)")
|
||||
userName := flag.String("user", "张工", "用户姓名")
|
||||
userRole := flag.String("role", "engineer", "用户角色")
|
||||
flag.Parse()
|
||||
|
||||
cfg, err := config.Load(*envPath)
|
||||
@@ -69,7 +81,25 @@ func main() {
|
||||
return
|
||||
}
|
||||
|
||||
// 3. 执行阶段或常规任务
|
||||
// 3. 交互式对话参谋模式
|
||||
if *chatMode || *chatQuery != "" {
|
||||
handleChatMode(
|
||||
context.Background(),
|
||||
svc,
|
||||
*chatQuery,
|
||||
*workDirFlag,
|
||||
*project,
|
||||
*customer,
|
||||
*device,
|
||||
*use,
|
||||
*market,
|
||||
*userName,
|
||||
*userRole,
|
||||
)
|
||||
return
|
||||
}
|
||||
|
||||
// 4. 执行阶段或常规任务
|
||||
activeBackend := cfg.AgentBackend
|
||||
if *backend != "" {
|
||||
activeBackend = *backend
|
||||
@@ -308,3 +338,125 @@ func handleInspect(ctx context.Context, svc bizdev.BizdevService, workDir string
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func handleChatMode(
|
||||
ctx context.Context,
|
||||
svc bizdev.BizdevService,
|
||||
chatQuery string,
|
||||
workDir string,
|
||||
project string,
|
||||
customer string,
|
||||
device string,
|
||||
use string,
|
||||
market string,
|
||||
userName string,
|
||||
userRole string,
|
||||
) {
|
||||
u := coordinator.UserInfo{
|
||||
ID: "u_local",
|
||||
Name: userName,
|
||||
Role: userRole,
|
||||
ProjectRole: "composer",
|
||||
}
|
||||
|
||||
assembler := coordinator.NewContextAssembler()
|
||||
var userCtx *coordinator.UserProjectContext
|
||||
var err error
|
||||
|
||||
if workDir != "" {
|
||||
userCtx, err = assembler.AssembleFromWorkspace(workDir, u)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "从工作区装载上下文失败 %s: %v，将初始化新上下文\n", workDir, err)
|
||||
}
|
||||
}
|
||||
|
||||
if userCtx == nil {
|
||||
userCtx = &coordinator.UserProjectContext{
|
||||
SnapshotID: fmt.Sprintf("ctx_%d", time.Now().UnixNano()),
|
||||
AssembledAt: time.Now(),
|
||||
User: u,
|
||||
Project: coordinator.ProjectInfo{
|
||||
Name: project,
|
||||
CustomerName: customer,
|
||||
DeviceType: device,
|
||||
IntendedUse: use,
|
||||
TargetMarket: market,
|
||||
CurrentWorkflow: "intake",
|
||||
CurrentStage: bizdev.StageV01_Triage,
|
||||
},
|
||||
BlackboardFacts: make(map[string]string),
|
||||
AvailableActions: coordinator.ComputeAvailableActions(bizdev.StageV01_Triage, u.ProjectRole),
|
||||
}
|
||||
if market != "" {
|
||||
userCtx.BlackboardFacts["target_market"] = market
|
||||
}
|
||||
if device != "" {
|
||||
userCtx.BlackboardFacts["device_type"] = device
|
||||
}
|
||||
}
|
||||
|
||||
rec := intent.NewRecognizer()
|
||||
engine := coordinator.NewCoordinatorEngine(svc)
|
||||
session := chat.NewSession("cli_chat_sess", userCtx, rec, engine)
|
||||
|
||||
// 单次提问模式 (支持脚本与自动化测试)
|
||||
if chatQuery != "" {
|
||||
resp, err := session.SendMessage(ctx, chatQuery)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "对话处理失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
top := resp.IntentResult.TopIntent
|
||||
fmt.Printf("[意图识别: %s (置信度: %.0f%%)] - %s\n", top.Intent, top.Confidence*100, top.Reasoning)
|
||||
fmt.Printf("\n%s\n", resp.ReplyText)
|
||||
if resp.AdvisorBrief != nil && (top.Intent == intent.IntentAdvanceStage || top.Intent == intent.IntentQueryProgress) {
|
||||
printAdvisorBrief(resp.AdvisorBrief)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// 交互式 REPL 模式
|
||||
fmt.Println("==================================================================")
|
||||
fmt.Println(" 🤖 医疗器械商务与研发智能参谋（L1/L2 协作对话终端） ")
|
||||
fmt.Println("==================================================================")
|
||||
fmt.Printf("项目名称: %s | 客户单位: %s\n", userCtx.Project.Name, userCtx.Project.CustomerName)
|
||||
fmt.Printf("当前阶段: %s | 用户身份: %s (%s)\n", userCtx.Project.CurrentStage, userCtx.User.Name, userCtx.User.Role)
|
||||
fmt.Printf("可用动作: %v\n", userCtx.AvailableActions)
|
||||
fmt.Println("------------------------------------------------------------------")
|
||||
fmt.Println("💡 提示：您可以直接输入自然语言，例如：")
|
||||
fmt.Println(" • \"现在这个便携超声立项评估到哪一步了？有啥风险没？\"")
|
||||
fmt.Println(" • \"客户补充说目标市场除了国内还想考虑欧洲 CE 认证\"")
|
||||
fmt.Println(" • \"我看摸底结果可以，进入方案论证阶段吧\"")
|
||||
fmt.Println(" • 输入 'exit' 或 'quit' 退出")
|
||||
fmt.Println("==================================================================")
|
||||
|
||||
scanner := bufio.NewScanner(os.Stdin)
|
||||
for {
|
||||
fmt.Print("\n[您] > ")
|
||||
if !scanner.Scan() {
|
||||
break
|
||||
}
|
||||
input := strings.TrimSpace(scanner.Text())
|
||||
if input == "" {
|
||||
continue
|
||||
}
|
||||
if input == "exit" || input == "quit" {
|
||||
fmt.Println("已退出参谋对话。")
|
||||
break
|
||||
}
|
||||
|
||||
resp, err := session.SendMessage(ctx, input)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "处理异常: %v\n", err)
|
||||
continue
|
||||
}
|
||||
|
||||
top := resp.IntentResult.TopIntent
|
||||
fmt.Printf("\n[智能参谋] (识别意图: %s | 置信度: %.0f%%)\n", top.Intent, top.Confidence*100)
|
||||
fmt.Println(resp.ReplyText)
|
||||
|
||||
if resp.AdvisorBrief != nil && (top.Intent == intent.IntentAdvanceStage || top.Intent == intent.IntentQueryProgress) {
|
||||
printAdvisorBrief(resp.AdvisorBrief)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -149,3 +149,89 @@ func TestBizdevCLIStageExecution(t *testing.T) {
|
||||
t.Errorf("missing customer name in communication kit: %s", outStr)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBizdevCLIChatQueries(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
envContent := "ENV=dev\nDATA_DIR=" + tmpDir + "\nAGENT_BACKEND=stub\n"
|
||||
envPath := filepath.Join(tmpDir, ".env")
|
||||
_ = os.WriteFile(envPath, []byte(envContent), 0o644)
|
||||
|
||||
// 1. 测试闲聊问候
|
||||
cmd1 := exec.Command("go", "run", "./cmd/bizdev-cli",
|
||||
"-env", envPath,
|
||||
"-backend", "stub",
|
||||
"-chat",
|
||||
"-query", "你好，请问你是谁？",
|
||||
"-project", "掌上便携超声",
|
||||
)
|
||||
cmd1.Dir = moduleRoot(t)
|
||||
out1, err := cmd1.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Fatalf("chat query 1 failed: %v, out: %s", err, string(out1))
|
||||
}
|
||||
if !strings.Contains(string(out1), "意图识别: CASUAL_CHAT") {
|
||||
t.Errorf("expected intent CASUAL_CHAT, got: %s", string(out1))
|
||||
}
|
||||
if !strings.Contains(string(out1), "掌上便携超声") {
|
||||
t.Errorf("expected greeting to mention project: %s", string(out1))
|
||||
}
|
||||
|
||||
// 2. 测试进度与风险查询
|
||||
cmd2 := exec.Command("go", "run", "./cmd/bizdev-cli",
|
||||
"-env", envPath,
|
||||
"-backend", "stub",
|
||||
"-chat",
|
||||
"-query", "现在这个便携超声立项评估到哪一步了？有啥风险没？",
|
||||
"-project", "掌上便携超声",
|
||||
)
|
||||
cmd2.Dir = moduleRoot(t)
|
||||
out2, err := cmd2.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Fatalf("chat query 2 failed: %v, out: %s", err, string(out2))
|
||||
}
|
||||
if !strings.Contains(string(out2), "意图识别: QUERY_PROGRESS") {
|
||||
t.Errorf("expected intent QUERY_PROGRESS, got: %s", string(out2))
|
||||
}
|
||||
|
||||
// 3. 测试信息补充
|
||||
cmd3 := exec.Command("go", "run", "./cmd/bizdev-cli",
|
||||
"-env", envPath,
|
||||
"-backend", "stub",
|
||||
"-chat",
|
||||
"-query", "客户刚才补充说，目标市场除了国内还想考虑欧洲 CE 认证",
|
||||
"-project", "掌上便携超声",
|
||||
)
|
||||
cmd3.Dir = moduleRoot(t)
|
||||
out3, err := cmd3.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Fatalf("chat query 3 failed: %v, out: %s", err, string(out3))
|
||||
}
|
||||
if !strings.Contains(string(out3), "意图识别: SUPPLEMENT_INFO") {
|
||||
t.Errorf("expected intent SUPPLEMENT_INFO, got: %s", string(out3))
|
||||
}
|
||||
if !strings.Contains(string(out3), "已收到并为您记录最新信息") {
|
||||
t.Errorf("expected acknowledgement in reply: %s", string(out3))
|
||||
}
|
||||
|
||||
// 4. 测试阶段推进
|
||||
cmd4 := exec.Command("go", "run", "./cmd/bizdev-cli",
|
||||
"-env", envPath,
|
||||
"-backend", "stub",
|
||||
"-chat",
|
||||
"-query", "摸底表我看过了，没问题，可以进入方案论证阶段了",
|
||||
"-project", "掌上便携超声",
|
||||
"-customer", "水木清研",
|
||||
"-device", "二类掌上超声",
|
||||
)
|
||||
cmd4.Dir = moduleRoot(t)
|
||||
out4, err := cmd4.CombinedOutput()
|
||||
if err != nil {
|
||||
t.Fatalf("chat query 4 failed: %v, out: %s", err, string(out4))
|
||||
}
|
||||
if !strings.Contains(string(out4), "意图识别: ADVANCE_STAGE") {
|
||||
t.Errorf("expected intent ADVANCE_STAGE, got: %s", string(out4))
|
||||
}
|
||||
if !strings.Contains(string(out4), "已为您成功推进并完成了【V0.5 方案成型与法规对齐】") {
|
||||
t.Errorf("expected stage advance success, got: %s", string(out4))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user