web_search.go

 1package tools
 2
 3import (
 4	"context"
 5	_ "embed"
 6	"log/slog"
 7	"net/http"
 8	"time"
 9
10	"charm.land/fantasy"
11)
12
13//go:embed web_search.md
14var webSearchToolDescription []byte
15
16// NewWebSearchTool creates a web search tool for sub-agents (no permissions needed).
17func NewWebSearchTool(client *http.Client) fantasy.AgentTool {
18	if client == nil {
19		client = &http.Client{
20			Timeout: 30 * time.Second,
21			Transport: &http.Transport{
22				MaxIdleConns:        100,
23				MaxIdleConnsPerHost: 10,
24				IdleConnTimeout:     90 * time.Second,
25			},
26		}
27	}
28
29	return fantasy.NewParallelAgentTool(
30		WebSearchToolName,
31		string(webSearchToolDescription),
32		func(ctx context.Context, params WebSearchParams, call fantasy.ToolCall) (fantasy.ToolResponse, error) {
33			if params.Query == "" {
34				return fantasy.NewTextErrorResponse("query is required"), nil
35			}
36
37			maxResults := params.MaxResults
38			if maxResults <= 0 {
39				maxResults = 10
40			}
41			if maxResults > 20 {
42				maxResults = 20
43			}
44
45			maybeDelaySearch()
46			results, err := searchDuckDuckGo(ctx, client, params.Query, maxResults)
47			slog.Debug("Web search completed", "query", params.Query, "results", len(results), "err", err)
48			if err != nil {
49				return fantasy.NewTextErrorResponse("Failed to search: " + err.Error()), nil
50			}
51
52			return fantasy.NewTextResponse(formatSearchResults(results)), nil
53		})
54}