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}