vv_agent/cli/
task.rs

1use chrono::Utc;
2use serde_json::Value;
3
4use crate::config::{apply_resolved_model_limits, ResolvedModelConfig};
5use crate::prompt::{build_system_prompt_bundle_with_options, BuildSystemPromptOptions};
6use crate::types::AgentTask;
7
8use super::args::CliArgs;
9
10pub fn build_cli_task(
11    args: &CliArgs,
12    model_id: impl Into<String>,
13    task_id: impl Into<String>,
14) -> Result<AgentTask, String> {
15    let prompt_bundle = build_system_prompt_bundle_with_options(
16        "You are Vector Vein agent runtime demo. Execute tasks with reliable tool usage and clear final outputs.",
17        BuildSystemPromptOptions {
18            language: args.language.clone(),
19            allow_interruption: true,
20            use_workspace: true,
21            enable_todo_management: true,
22            agent_type: args.agent_type.clone(),
23            workspace: Some(args.workspace.clone()),
24            ..BuildSystemPromptOptions::default()
25        },
26    );
27    let mut task = AgentTask::new(task_id, model_id, prompt_bundle, args.prompt.clone());
28    task.max_cycles = args.max_cycles.max(1);
29    task.agent_type = args.agent_type.clone();
30    task.model_settings = args.model_settings.clone();
31    task.metadata
32        .insert("language".to_string(), Value::String(args.language.clone()));
33    Ok(task)
34}
35
36pub fn build_cli_task_from_resolved(
37    args: &CliArgs,
38    resolved: &ResolvedModelConfig,
39    task_id: impl Into<String>,
40) -> Result<AgentTask, String> {
41    let mut task = build_cli_task(args, resolved.model_id.clone(), task_id)?;
42    apply_resolved_model_limits(&mut task, resolved);
43    Ok(task)
44}
45
46pub(super) fn generate_task_id() -> String {
47    format!("task_{}", Utc::now().format("%Y%m%d%H%M%S%3f"))
48}
vv_agent/cli/task.rs

vv_agent/cli/
task.rs
