更新工具链

agents/search/search_test.go

@@ -121,3 +121,42 @@ func TestLoadConfigWritesLegacyProfiles(t *testing.T) {
 		t.Fatal(err)
 	}
 }
+
+func TestToolDefinitionAndExecuteTool(t *testing.T) {
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Query().Get("q") != "golang" {
+			t.Fatalf("query = %s", r.URL.RawQuery)
+		}
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write([]byte(`{"Heading":"Go","Abstract":"Go language","AbstractURL":"https://go.dev"}`))
+	}))
+	defer server.Close()
+	state, err := NewState(&Config{Enabled: true, Profiles: ProfileConfigs{{Name: "ddg", Active: true, Enabled: true, Provider: "duckduckgo", BaseURL: server.URL, Count: 1, Timeout: 1}}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	definition := state.ToolDefinition("custom search")
+	if definition.Function == nil || definition.Function.Name != ToolName || definition.Function.Description != "custom search" {
+		t.Fatalf("unexpected definition: %#v", definition)
+	}
+	text, err := state.ExecuteTool(context.Background(), `{"query":"golang","reason":"测试搜索"}`)
+	if err != nil {
+		t.Fatal(err)
+	}
+	for _, want := range []string{"联网搜索", "golang", "Go", "https://go.dev", "测试搜索"} {
+		if !strings.Contains(text, want) {
+			t.Fatalf("tool result missing %q:\n%s", want, text)
+		}
+	}
+}
+
+func TestExecuteToolRejectsEmptyQuery(t *testing.T) {
+	state, err := NewState(&Config{Enabled: true, Profiles: ProfileConfigs{{Name: "ddg", Active: true, Enabled: true, Provider: "duckduckgo", BaseURL: defaultBaseURL, Count: 1, Timeout: 1}}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	_, err = state.ExecuteTool(context.Background(), `{"query":" "}`)
+	if err == nil || !strings.Contains(err.Error(), "不能为空") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}