DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-05-06 18:07:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
lpf
2026-02-13 13:50:09 +08:00
parent f88a78ef8b
commit 085c265319
15 changed files with 1485 additions and 179 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pkg/tools/web.go
View File

@@ -13,7 +13,8 @@ import (
)
const (
userAgent = "Mozilla/5.0 (compatible; clawgo/1.0)"
userAgent = "Mozilla/5.0 (compatible; clawgo/1.0)"
maxFetchResponseBytes = 8 * 1024 * 1024
)
type WebSearchTool struct {
@@ -93,10 +94,14 @@ func (t *WebSearchTool) Execute(ctx context.Context, args map[string]interface{}
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
limitedReader := io.LimitReader(resp.Body, maxFetchResponseBytes+1)
body, err := io.ReadAll(limitedReader)
if err != nil {
return "", fmt.Errorf("failed to read response: %w", err)
}
if len(body) > maxFetchResponseBytes {
return "", fmt.Errorf("response body too large (>%d bytes)", maxFetchResponseBytes)
}
var searchResp struct {
Web struct {