axonerai 0.1.0

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

use crate::provider::ToolCall;
use crate::tool::ToolRegistry;
use anyhow::{anyhow, Result};

/// Executes tool calls and returns results
pub struct ToolExecutor<'a> {
    registry: &'a ToolRegistry,
}

impl<'a> ToolExecutor<'a> {
    pub fn new(registry: &'a ToolRegistry) -> Self {
        Self { registry }
    }

    /// Execute a single tool call
    pub async fn execute(&self, tool_call: &ToolCall) -> Result<ToolResult> {
        println!("  🔧 Executing tool: {}", tool_call.name);
        
        let tool = self
            .registry
            .get(&tool_call.name)
            .ok_or_else(|| anyhow!("Tool not found: {}", tool_call.name))?;

        let result = tool.execute(tool_call.input.clone()).await?;

        Ok(ToolResult {
            tool_call_id: tool_call.id.clone(),
            tool_name: tool_call.name.clone(),
            result,
        })
    }

    /// Execute multiple tool calls
    
    pub async fn execute_all(&self, tool_calls: &[ToolCall]) -> Result<Vec<ToolResult>> {
        let mut results = Vec::new();
        for call in tool_calls {
            results.push(self.execute(call).await?);
        }
        Ok(results)
    }
}

/// Result from executing a tool
#[derive(Debug, Clone)]
pub struct ToolResult {
    pub tool_call_id: String,
    pub tool_name: String,
    pub result: String,
}