From a6682b8ba916ca41443ca0465d98d13000bde342 Mon Sep 17 00:00:00 2001 From: shikihane Date: Tue, 3 Mar 2026 13:02:21 +0800 Subject: [PATCH] fix(tools): adapt Bocha search to upstream NewWebSearchTool signature change NewWebSearchTool now returns (*WebSearchTool, error). Update Bocha provider initialization and test call sites to handle the new signature. Co-Authored-By: Claude Opus 4.6 --- pkg/tools/web.go | 2 +- pkg/tools/web_test.go | 10 ++++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/pkg/tools/web.go b/pkg/tools/web.go index 77e1d3a9a..735e3f5e9 100644 --- a/pkg/tools/web.go +++ b/pkg/tools/web.go @@ -564,7 +564,7 @@ func NewWebSearchTool(opts WebSearchToolOptions) (*WebSearchTool, error) { } else if opts.BochaEnabled && opts.BochaAPIKey != "" { bochaClient, err := createHTTPClient(opts.Proxy, 15*time.Second) if err != nil { - return nil + return nil, err } provider = &BochaSearchProvider{ apiKey: opts.BochaAPIKey, diff --git a/pkg/tools/web_test.go b/pkg/tools/web_test.go index 09b5e5a51..4fa7503df 100644 --- a/pkg/tools/web_test.go +++ b/pkg/tools/web_test.go @@ -731,12 +731,15 @@ func TestWebTool_BochaSearch_Success(t *testing.T) { })) defer server.Close() - tool := NewWebSearchTool(WebSearchToolOptions{ + tool, err := NewWebSearchTool(WebSearchToolOptions{ BochaEnabled: true, BochaAPIKey: "test-bocha-key", BochaBaseURL: server.URL, BochaMaxResults: 5, }) + if err != nil { + t.Fatalf("NewWebSearchTool failed: %v", err) + } ctx := context.Background() result := tool.Execute(ctx, map[string]any{"query": "test query"}) @@ -775,11 +778,14 @@ func TestWebTool_BochaSearch_APIError(t *testing.T) { })) defer server.Close() - tool := NewWebSearchTool(WebSearchToolOptions{ + tool, err := NewWebSearchTool(WebSearchToolOptions{ BochaEnabled: true, BochaAPIKey: "bad-key", BochaBaseURL: server.URL, }) + if err != nil { + t.Fatalf("NewWebSearchTool failed: %v", err) + } ctx := context.Background() result := tool.Execute(ctx, map[string]any{"query": "test"})