vv-agent 0.6.0

VectorVein agent runtime, SDK, CLI, tools, and workspace backends
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
73
74
75
76
77
78
79
80
81
82
83

use serde_json::Value;

use crate::tools::{ToolHandler, ToolSpec};
use crate::types::ToolExecutionResult;

pub trait Tool: Send + Sync {
    fn name(&self) -> &str;
    fn description(&self) -> &str;
    fn parameters_schema(&self) -> &Value;

    fn strict_schema(&self) -> bool {
        true
    }

    fn as_tool_spec(&self) -> ToolSpec;
}

#[derive(Clone)]
pub struct StaticTool {
    name: String,
    description: String,
    parameters_schema: Value,
    handler: ToolHandler,
}

impl StaticTool {
    pub fn new(
        name: impl Into<String>,
        description: impl Into<String>,
        parameters_schema: Value,
        handler: ToolHandler,
    ) -> Self {
        Self {
            name: name.into(),
            description: description.into(),
            parameters_schema,
            handler,
        }
    }
}

impl Tool for StaticTool {
    fn name(&self) -> &str {
        &self.name
    }

    fn description(&self) -> &str {
        &self.description
    }

    fn parameters_schema(&self) -> &Value {
        &self.parameters_schema
    }

    fn as_tool_spec(&self) -> ToolSpec {
        let mut spec = ToolSpec::new(
            self.name.clone(),
            self.description.clone(),
            self.handler.clone(),
        );
        spec.schema = serde_json::json!({
            "type": "function",
            "function": {
                "name": self.name,
                "description": self.description,
                "parameters": self.parameters_schema,
            }
        });
        spec
    }
}

impl<F> From<(String, String, Value, F)> for StaticTool
where
    F: Fn(&mut crate::tools::ToolContext, &crate::types::ToolArguments) -> ToolExecutionResult
        + Send
        + Sync
        + 'static,
{
    fn from(value: (String, String, Value, F)) -> Self {
        Self::new(value.0, value.1, value.2, std::sync::Arc::new(value.3))
    }
}