vv-agent 0.1.3

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

use std::collections::BTreeMap;

use serde_json::Value;

use super::tool_names::{
    ACTIVATE_SKILL_TOOL_NAME, ASK_USER_TOOL_NAME, COMPRESS_MEMORY_TOOL_NAME, FILE_INFO_TOOL_NAME,
    FILE_STR_REPLACE_TOOL_NAME, LIST_FILES_TOOL_NAME, READ_FILE_TOOL_NAME, TASK_FINISH_TOOL_NAME,
    TODO_WRITE_TOOL_NAME, WORKSPACE_GREP_TOOL_NAME, WRITE_FILE_TOOL_NAME,
};

pub const WORKSPACE_TOOLS: [&str; 8] = [
    LIST_FILES_TOOL_NAME,
    FILE_INFO_TOOL_NAME,
    READ_FILE_TOOL_NAME,
    WRITE_FILE_TOOL_NAME,
    FILE_STR_REPLACE_TOOL_NAME,
    WORKSPACE_GREP_TOOL_NAME,
    COMPRESS_MEMORY_TOOL_NAME,
    TODO_WRITE_TOOL_NAME,
];

pub fn get_default_tool_schemas() -> BTreeMap<String, Value> {
    crate::tools::schemas::default_tool_schemas()
}

pub fn workspace_tools_schemas() -> BTreeMap<String, Value> {
    let mut schemas = get_default_tool_schemas();
    schemas.retain(|name, _| WORKSPACE_TOOLS.contains(&name.as_str()));
    schemas
}

#[allow(non_snake_case)]
pub fn WORKSPACE_TOOLS_SCHEMAS() -> BTreeMap<String, Value> {
    workspace_tools_schemas()
}

pub fn task_finish_tool_schema() -> Value {
    schema_or_null(TASK_FINISH_TOOL_NAME)
}

#[allow(non_snake_case)]
pub fn TASK_FINISH_TOOL_SCHEMA() -> Value {
    task_finish_tool_schema()
}

pub fn ask_user_tool_schema() -> Value {
    schema_or_null(ASK_USER_TOOL_NAME)
}

#[allow(non_snake_case)]
pub fn ASK_USER_TOOL_SCHEMA() -> Value {
    ask_user_tool_schema()
}

pub fn activate_skill_tool_schema() -> Value {
    schema_or_null(ACTIVATE_SKILL_TOOL_NAME)
}

#[allow(non_snake_case)]
pub fn ACTIVATE_SKILL_TOOL_SCHEMA() -> Value {
    activate_skill_tool_schema()
}

fn schema_or_null(name: &str) -> Value {
    crate::tools::schemas::schema_for(name).unwrap_or(Value::Null)
}