vv-agent 0.4.7

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

use std::sync::Arc;

use serde_json::{json, Value};

use super::ToolContext;
use crate::types::{ToolArguments, ToolExecutionResult};

pub type ToolHandler =
    Arc<dyn Fn(&mut ToolContext, &ToolArguments) -> ToolExecutionResult + Send + Sync + 'static>;
pub type SubTaskRunner = Arc<
    dyn Fn(crate::types::SubTaskRequest) -> crate::types::SubTaskOutcome + Send + Sync + 'static,
>;

#[derive(Clone)]
pub struct ToolSpec {
    pub name: String,
    pub handler: ToolHandler,
    pub description: String,
    pub schema: Value,
    pub kind: ToolSpecKind,
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ToolSpecKind {
    Function,
    Agent,
    BackgroundAgent,
    Handoff,
}

impl ToolSpec {
    pub fn new(
        name: impl Into<String>,
        description: impl Into<String>,
        handler: ToolHandler,
    ) -> Self {
        let name = name.into();
        let fallback_description = description.into();
        let schema = crate::tools::schemas::schema_for(&name).unwrap_or_else(|| {
            json!({
                "type": "function",
                "function": {
                    "name": name,
                    "description": fallback_description,
                    "parameters": {"type": "object", "properties": {}, "required": []},
                }
            })
        });
        let description = schema["function"]["description"]
            .as_str()
            .unwrap_or(&fallback_description)
            .to_string();
        Self {
            schema,
            name,
            handler,
            description,
            kind: ToolSpecKind::Function,
        }
    }
}

#[derive(Debug, Clone, thiserror::Error)]
#[error("tool not found: {0}")]
pub struct ToolNotFoundError(pub String);