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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

use chrono::{SecondsFormat, Utc};
use serde_json::{json, Value};

use crate::prompt::templates;

use super::options::{BuildSystemPromptOptions, BuiltSystemPrompt};
use super::section::PromptSection;
use super::system_builder::SystemPromptBuilder;

pub fn build_system_prompt(original_system_prompt: impl Into<String>) -> String {
    build_system_prompt_bundle(original_system_prompt).prompt
}

pub fn build_system_prompt_with_options(
    original_system_prompt: impl Into<String>,
    options: BuildSystemPromptOptions,
) -> String {
    build_system_prompt_bundle_with_options(original_system_prompt, options).prompt
}

pub fn build_system_prompt_sections(original_system_prompt: impl Into<String>) -> Vec<Value> {
    build_system_prompt_bundle(original_system_prompt).sections
}

pub fn build_system_prompt_sections_with_options(
    original_system_prompt: impl Into<String>,
    options: BuildSystemPromptOptions,
) -> Vec<Value> {
    build_system_prompt_bundle_with_options(original_system_prompt, options).sections
}

pub fn build_system_prompt_bundle(original_system_prompt: impl Into<String>) -> BuiltSystemPrompt {
    build_system_prompt_bundle_with_options(
        original_system_prompt,
        BuildSystemPromptOptions::default(),
    )
}

pub fn build_system_prompt_bundle_with_options(
    original_system_prompt: impl Into<String>,
    options: BuildSystemPromptOptions,
) -> BuiltSystemPrompt {
    create_system_prompt_builder(original_system_prompt, options).build_result()
}

pub fn create_system_prompt_builder(
    original_system_prompt: impl Into<String>,
    options: BuildSystemPromptOptions,
) -> SystemPromptBuilder {
    let language = options.language;
    let mut builder = SystemPromptBuilder::default();
    builder.add_section(PromptSection::constant(
        "agent_definition",
        format!(
            "<Agent Definition>\n{}\n</Agent Definition>",
            original_system_prompt.into()
        ),
        true,
    ));

    if !options.session_memory_context.trim().is_empty() {
        builder.add_section(PromptSection::constant(
            "session_memory",
            options.session_memory_context,
            false,
        ));
    }

    if options.agent_type.as_deref() == Some("computer") {
        builder.add_section(PromptSection::constant(
            "environment",
            format!(
                "<Environment>\n{}\n</Environment>",
                templates::computer_agent_env_prompt(&language)
            ),
            true,
        ));
    }

    let mut tools_lines = Vec::new();
    if options.allow_interruption {
        tools_lines.push(templates::ask_user_prompt(&language).to_string());
    }
    if options.use_workspace {
        tools_lines.push(templates::render_workspace_tools(&language));
        tools_lines.push(templates::tool_priority_prompt(&language).to_string());
    }
    if options.enable_todo_management {
        tools_lines.push(templates::todo_prompt(&language).to_string());
    }
    if !options.available_sub_agents.is_empty() {
        tools_lines.push(templates::render_sub_agents(
            &language,
            &options.available_sub_agents,
        ));
    }
    if let Some(available_skills) = options.available_skills.as_ref() {
        tools_lines.push(templates::render_available_skills(
            &language,
            available_skills,
            options.workspace.as_deref(),
        ));
    }
    tools_lines.push(templates::task_finish_prompt(&language).to_string());
    builder.add_section(PromptSection::constant(
        "tools",
        format!("<Tools>\n{}\n</Tools>", tools_lines.join("\n\n")),
        true,
    ));

    let current_time = options.current_time_utc.unwrap_or_else(current_utc_text);
    builder.add_section(PromptSection::constant(
        "current_time",
        format!(
            "<Current Time>\n{}\n{}\n</Current Time>",
            templates::current_time_prompt(&language),
            current_time
        ),
        false,
    ));
    builder
}

pub fn build_raw_system_prompt_sections(system_prompt: impl Into<String>) -> Vec<Value> {
    let text = system_prompt.into().trim().to_string();
    if text.is_empty() {
        return Vec::new();
    }
    vec![json!({
        "id": "raw_system_prompt",
        "text": text,
        "stable": true,
    })]
}

fn current_utc_text() -> String {
    Utc::now().to_rfc3339_opts(SecondsFormat::Secs, true)
}