axonerai 0.1.1

A type-safe, blazing fast agentic AI framework in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use crate::tool::Tool;
use anyhow::{anyhow, Ok, Result};
use async_trait::async_trait;
use serde::{Deserialize, Serialize};
use serde_json::{json, Value};
use reqwest;
pub struct WebSearch;

#[derive(Debug, Deserialize, Serialize)]
struct WebSearchInput {
    search_term: String
}
#[async_trait]
impl Tool for WebSearch {


    fn name(&self) -> String{ "WebSearch".to_string()}

    fn description(&self) -> String{ "This tool uses Google Search and returns some links".to_string()}

    fn input_schema(&self) -> Value {
        json!({
            "type": "object",
            "properties": {
                "search_term": {
                    "type": "string",
                    "description": "query text"
                }
            },
            "required": ["search_term"]
        })
    }

   async fn execute(&self, input: Value) -> Result<String> {
        let input: WebSearchInput = serde_json::from_value(input)
            .map_err(|e| anyhow!("Invalid calculator input: {}", e))?;

        let search_api_key = env::var("SEARCH_API_KEY")?;
        let cx = env::var("CX_ENGINE")?;
        let url = "https://www.googleapis.com/customsearch/v1?".to_string();
        let response = reqwest::Client::new()
            .get(url)
            .query(&[
                ("key", search_api_key),
                ("cx", cx),
                ("q", input.search_term),
                ("num", "3".to_string())
            ])
            .send()
            .await?;

        let data: Value = response.json().await?;

        Ok(process_json_data(&data)?)

    }
}

fn process_json_data(data: &Value) -> Result<String> {

    let items: &Vec<Value> = data["items"].as_array().ok_or(anyhow!("Missing items"))?;
    let mut search_results:String = "title, link, description_preview".to_string();
    //filter(|item| item.key.contains("title") || item.key.contains("link") || item.key.contains("snippet") )
    for item in items {

        search_results.push_str("\n");
        search_results.push_str(&format!("{},{},{}", &item["title"], &item["link"], &item["snippet"]));

        println!("Matched item: Title: {}", &item["title"]);
    }
    Ok(search_results)
}