#[derive(Deserialize, Serialize)]
struct SearchArgs {
    pub query_string: String,
}

#[derive(Deserialize, Serialize)]
struct SearchResult {
    pub title: String,
    pub url: String,
}

#[derive(Debug, thiserror::Error)]
#[error("Search error")]
struct SearchError;

#[derive(Deserialize, Serialize)]
struct SearchTool;

impl Tool for SearchTool {
    const NAME: &'static str = "search";

    type Error = SearchError;
    type Args = SearchArgs;
    type Output = SearchResult;

    async fn definition(&self, _prompt: String) -> ToolDefinition {
        serde_json::from_value(json!({
            "name": "search",
            "description": "Search for a website, it will return the title and URL",
            "parameters": {
                "type": "object",
                "properties": {
                    "query_string": {
                        "type": "string",
                        "description": "The query string to search for"
                    },
                }
            }
        }))
        .expect("Tool Definition")
    }

    async fn call(&self, args: Self::Args) -> Result<Self::Output, Self::Error> {
        println!("[tool-call] Searching for: '{}'", args.query_string);

        if args.query_string.to_lowercase().contains("rig") {
            Ok(SearchResult {
                title: "Rig Documentation".to_string(),
                url: "https://docs.rig.ai".to_string(),
            })
        } else {
            Err(SearchError)
        }
    }
}