opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(tools): adapt Bocha search to upstream NewWebSearchTool signature change

Browse source 
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 <noreply@anthropic.com>
This commit is contained in:
shikihane 2026-03-03 13:02:21 +08:00
parent 295c0186f6
commit a6682b8ba9
2 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pkg/tools/web.go
View file

@ -564,7 +564,7 @@ func NewWebSearchTool(opts WebSearchToolOptions) (*WebSearchTool, error) {
} else if opts.BochaEnabled && opts.BochaAPIKey != "" { } else if opts.BochaEnabled && opts.BochaAPIKey != "" {
bochaClient, err := createHTTPClient(opts.Proxy, 15*time.Second) bochaClient, err := createHTTPClient(opts.Proxy, 15*time.Second)
if err != nil { if err != nil {
return nil return nil, err
} }
provider = &BochaSearchProvider{ provider = &BochaSearchProvider{
apiKey: opts.BochaAPIKey, apiKey: opts.BochaAPIKey,

10
pkg/tools/web_test.go
View file

@ -731,12 +731,15 @@ func TestWebTool_BochaSearch_Success(t *testing.T) {
})) }))
defer server.Close() defer server.Close()
tool := NewWebSearchTool(WebSearchToolOptions{ tool, err := NewWebSearchTool(WebSearchToolOptions{
BochaEnabled: true, BochaEnabled: true,
BochaAPIKey: "test-bocha-key", BochaAPIKey: "test-bocha-key",
BochaBaseURL: server.URL, BochaBaseURL: server.URL,
BochaMaxResults: 5, BochaMaxResults: 5,
}) })
if err != nil {
t.Fatalf("NewWebSearchTool failed: %v", err)
}
ctx := context.Background() ctx := context.Background()
result := tool.Execute(ctx, map[string]any{"query": "test query"}) result := tool.Execute(ctx, map[string]any{"query": "test query"})
@ -775,11 +778,14 @@ func TestWebTool_BochaSearch_APIError(t *testing.T) {
})) }))
defer server.Close() defer server.Close()
tool := NewWebSearchTool(WebSearchToolOptions{ tool, err := NewWebSearchTool(WebSearchToolOptions{
BochaEnabled: true, BochaEnabled: true,
BochaAPIKey: "bad-key", BochaAPIKey: "bad-key",
BochaBaseURL: server.URL, BochaBaseURL: server.URL,
}) })
if err != nil {
t.Fatalf("NewWebSearchTool failed: %v", err)
}
ctx := context.Background() ctx := context.Background()
result := tool.Execute(ctx, map[string]any{"query": "test"}) result := tool.Execute(ctx, map[string]any{"query": "test"})