96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
package toolemulation
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestLooksLikeMissedToolUseDetectsLocalToolAvoidance(t *testing.T) {
|
|
cases := []string{
|
|
"我需要使用终端工具来查看内存。",
|
|
"由于当前环境限制，请手动运行 top。",
|
|
"I need to read the file first.",
|
|
"Let me use the web search tool.",
|
|
"现在我需要切换到计划模式。",
|
|
}
|
|
for _, tc := range cases {
|
|
if !LooksLikeMissedToolUse(tc) {
|
|
t.Fatalf("LooksLikeMissedToolUse(%q) = false", tc)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLooksLikeMissedToolUseIgnoresFinalAnswers(t *testing.T) {
|
|
text := "这个文件负责 HTTP API 路由和 OpenAI 兼容响应。"
|
|
if LooksLikeMissedToolUse(text) {
|
|
t.Fatalf("LooksLikeMissedToolUse(%q) = true", text)
|
|
}
|
|
}
|
|
|
|
func TestInjectToolingIncludesAutoToolGuidance(t *testing.T) {
|
|
prompt := InjectTooling("", []ToolDef{{
|
|
Name: "read_file",
|
|
Description: "Read a text file.",
|
|
InputSchema: map[string]any{
|
|
"properties": map[string]any{
|
|
"path": map[string]any{"type": "string"},
|
|
},
|
|
"required": []any{"path"},
|
|
},
|
|
}}, ToolChoice{Mode: "auto"}, nil)
|
|
if prompt == "" {
|
|
t.Fatal("empty prompt")
|
|
}
|
|
for _, want := range []string{
|
|
"tool_choice=auto means you must decide",
|
|
"inspect a local file path",
|
|
"Core tool examples",
|
|
"NEVER ask the user to run a command",
|
|
} {
|
|
if !strings.Contains(prompt, want) {
|
|
t.Fatalf("prompt missing %q:\n%s", want, prompt)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseActionBlocksMapsCommonToolAliases(t *testing.T) {
|
|
text := "```json action\n{\"tool\":\"Bash\",\"parameters\":{\"command\":\"pwd\",\"extra\":true}}\n```"
|
|
calls, clean, err := ParseActionBlocks(text, []ToolDef{{
|
|
Name: "terminal",
|
|
InputSchema: map[string]any{
|
|
"properties": map[string]any{
|
|
"command": map[string]any{"type": "string"},
|
|
},
|
|
},
|
|
}}, Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if clean != "" {
|
|
t.Fatalf("clean = %q", clean)
|
|
}
|
|
if len(calls) != 1 {
|
|
t.Fatalf("call count = %d", len(calls))
|
|
}
|
|
if calls[0].Name != "terminal" {
|
|
t.Fatalf("tool name = %q", calls[0].Name)
|
|
}
|
|
if _, ok := calls[0].Arguments["command"]; !ok {
|
|
t.Fatalf("missing command arg: %+v", calls[0].Arguments)
|
|
}
|
|
if _, ok := calls[0].Arguments["extra"]; ok {
|
|
t.Fatalf("unexpected extra arg: %+v", calls[0].Arguments)
|
|
}
|
|
}
|
|
|
|
func TestParseActionBlocksMapsReadAlias(t *testing.T) {
|
|
text := "```json action\n{\"name\":\"Read\",\"arguments\":{\"path\":\"/tmp/a.txt\"}}\n```"
|
|
calls, _, err := ParseActionBlocks(text, []ToolDef{{Name: "read_file"}}, Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(calls) != 1 || calls[0].Name != "read_file" {
|
|
t.Fatalf("calls = %+v", calls)
|
|
}
|
|
}
|