75 lines
2.4 KiB
Go
75 lines
2.4 KiB
Go
package agent
|
|
|
|
import "testing"
|
|
|
|
// TestParseDecisionFinalAnswer 测试 is_final_answer=true 时能正确解析 final_answer
|
|
func TestParseDecisionFinalAnswer(t *testing.T) {
|
|
raw := `{"thought":"直接回答","action":"none","action_input":"","is_final_answer":true,"final_answer":"你好！"}`
|
|
got, err := parseDecision(raw)
|
|
if err != nil {
|
|
t.Fatalf("parseDecision error: %v", err)
|
|
}
|
|
if !got.IsFinalAnswer {
|
|
t.Fatal("expected is_final_answer=true")
|
|
}
|
|
if got.FinalAnswer == nil || *got.FinalAnswer != "你好！" {
|
|
t.Fatalf("unexpected final_answer: %v", got.FinalAnswer)
|
|
}
|
|
}
|
|
|
|
// TestParseDecisionToolCall 测试需要调工具时的解析
|
|
func TestParseDecisionToolCall(t *testing.T) {
|
|
raw := `{"thought":"需要搜索","action":"web_search","action_input":"NVIDIA stock price","is_final_answer":false,"final_answer":null}`
|
|
got, err := parseDecision(raw)
|
|
if err != nil {
|
|
t.Fatalf("parseDecision error: %v", err)
|
|
}
|
|
if got.IsFinalAnswer {
|
|
t.Fatal("expected is_final_answer=false")
|
|
}
|
|
if got.Action != "web_search" {
|
|
t.Fatalf("expected action=web_search, got %s", got.Action)
|
|
}
|
|
input := got.GetActionInputString()
|
|
if input != "NVIDIA stock price" {
|
|
t.Fatalf("expected action_input string, got %q", input)
|
|
}
|
|
}
|
|
|
|
// TestParseDecisionStructuredActionInput 测试 action_input 为结构化对象时的解析
|
|
func TestParseDecisionStructuredActionInput(t *testing.T) {
|
|
raw := `{"thought":"搜索","action":"web_search","action_input":{"query":"test","context":"dev"},"is_final_answer":false,"final_answer":null}`
|
|
got, err := parseDecision(raw)
|
|
if err != nil {
|
|
t.Fatalf("parseDecision error: %v", err)
|
|
}
|
|
input := got.GetActionInputString()
|
|
if input == "" {
|
|
t.Fatal("expected non-empty action_input")
|
|
}
|
|
// 结构化对象应保留 JSON 原文
|
|
if input[0] != '{' {
|
|
t.Fatalf("expected JSON object string, got %q", input)
|
|
}
|
|
}
|
|
|
|
// TestParseDecisionCodeFence 测试被 markdown code fence 包裹的 JSON
|
|
func TestParseDecisionCodeFence(t *testing.T) {
|
|
raw := "```json\n{\"thought\":\"t\",\"action\":\"shell\",\"action_input\":\"ls\",\"is_final_answer\":false,\"final_answer\":null}\n```"
|
|
got, err := parseDecision(raw)
|
|
if err != nil {
|
|
t.Fatalf("parseDecision error: %v", err)
|
|
}
|
|
if got.Action != "shell" {
|
|
t.Fatalf("unexpected action: %s", got.Action)
|
|
}
|
|
}
|
|
|
|
// TestParseDecisionInvalid 测试非 JSON 输入时返回错误
|
|
func TestParseDecisionInvalid(t *testing.T) {
|
|
_, err := parseDecision("not json")
|
|
if err == nil {
|
|
t.Fatal("expected parse error")
|
|
}
|
|
}
|