hippox 0.1.0

The most reliable AI agent.🦛
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

use anyhow::Result;
use serde_json::Value;
use std::collections::HashMap;

use super::common;
use crate::executors::types::Skill;

#[derive(Debug)]
pub struct PowerSkill;

#[async_trait::async_trait]
impl Skill for PowerSkill {
    fn name(&self) -> &str {
        "math_power"
    }

    fn description(&self) -> &str {
        "Calculate power or root. Parameters: base (required), exponent (required) for power, or use sqrt for square root"
    }

    async fn execute(&self, parameters: &HashMap<String, Value>) -> Result<String> {
        if let Some(value) = parameters.get("sqrt").and_then(|v| v.as_str()) {
            let num = common::validate_number(value)?;
            if num < 0.0 {
                anyhow::bail!("Cannot calculate square root of negative number: {}", num);
            }
            let result = num.sqrt();
            return Ok(format!("√{} = {}", num, result));
        }
        if let Some(value) = parameters.get("cbrt").and_then(|v| v.as_str()) {
            let num = common::validate_number(value)?;
            let result = num.cbrt();
            return Ok(format!("∛{} = {}", num, result));
        }
        let base = parameters
            .get("base")
            .and_then(|v| v.as_str())
            .ok_or_else(|| anyhow::anyhow!("Missing 'base' parameter"))?;
        let exponent = parameters
            .get("exponent")
            .and_then(|v| v.as_str())
            .ok_or_else(|| anyhow::anyhow!("Missing 'exponent' parameter"))?;
        let base_num = common::validate_number(base)?;
        let exp_num = common::validate_number(exponent)?;
        let result = base_num.powf(exp_num);
        let precision = parameters
            .get("precision")
            .and_then(|v| v.as_u64())
            .unwrap_or(2);
        Ok(format!(
            "{} ^ {} = {}",
            base_num,
            exp_num,
            common::format_number(result, precision as usize)
        ))
    }

    fn validate(&self, parameters: &HashMap<String, Value>) -> Result<()> {
        // Either (base + exponent) OR sqrt OR cbrt
        let has_power = parameters.contains_key("base") && parameters.contains_key("exponent");
        let has_sqrt = parameters.contains_key("sqrt");
        let has_cbrt = parameters.contains_key("cbrt");

        if !has_power && !has_sqrt && !has_cbrt {
            anyhow::bail!("Missing parameters: provide (base + exponent) or (sqrt) or (cbrt)");
        }
        Ok(())
    }
}