matrixcode_core/tools/
webfetch.rs

1use anyhow::Result;
2use async_trait::async_trait;
3use serde_json::{Value, json};
4
5use super::{Tool, ToolDefinition};
6
7pub struct WebFetchTool;
8
9#[async_trait]
10impl Tool for WebFetchTool {
11    fn definition(&self) -> ToolDefinition {
12        ToolDefinition {
13            name: "webfetch".to_string(),
14            description: "从 URL 获取内容并返回为文本".to_string(),
15            parameters: json!({
16                "type": "object",
17                "properties": {
18                    "url": {
19                        "type": "string",
20                        "description": "要获取的 URL"
21                    },
22                    "max_length": {
23                        "type": "integer",
24                        "description": "最大响应长度（字符数，默认 10000）"
25                    }
26                },
27                "required": ["url"]
28            }),
29        }
30    }
31
32    async fn execute(&self, params: Value) -> Result<String> {
33        let url = params["url"]
34            .as_str()
35            .ok_or_else(|| anyhow::anyhow!("missing 'url'"))?;
36        let max_length = params["max_length"].as_u64().unwrap_or(10000) as usize;
37
38        // Show spinner while fetching - RAII guard ensures cleanup on error
39        // let mut spinner = ToolSpinner::new(&format!("fetching {}", url));
40
41        let response = reqwest::get(url).await?;
42        let status = response.status();
43
44        if !status.is_success() {
45            // spinner.finish_error(&format!("HTTP {}", status));
46            anyhow::bail!("HTTP {} for {}", status, url);
47        }
48
49        let body = response.text().await?;
50
51        let truncated = if body.len() > max_length {
52            // 找到不超过 max_length 的最后一个有效字符边界
53            let end = body.floor_char_boundary(max_length);
54            format!(
55                "{}...\n\n(truncated, {} total bytes)",
56                &body[..end],
57                body.len()
58            )
59        } else {
60            body
61        };
62
63        // spinner.finish_success(&format!("{} bytes", bytes));
64        Ok(truncated)
65    }
66}
matrixcode_core/tools/webfetch.rs

matrixcode_core/tools/
webfetch.rs
