collet 0.1.0

Relentless agentic coding orchestrator with zero-drop agent loops
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

//! Auto-evolution trigger — mirrors the Soul reflection pattern.
//!
//! After each agent task completes, if `[evolution] enabled = true` in config,
//! one evolution cycle runs asynchronously in the background.
//!
//! Workspace layout (same split as Soul):
//! - global (`"collet"`): `~/.collet/workspace/`
//! - per-agent:           `~/.collet/agents/{name}/workspace/`

use std::path::{Path, PathBuf};

use anyhow::Result;
use tokio_util::sync::CancellationToken;

use crate::api::provider::OpenAiCompatibleProvider;
use crate::config::Config;
use crate::evolution::{
    adapter::ColletEvolvable, benchmarks::NullBenchmark, config::EvolveConfig,
    engines::SkillforgeEngine, r#loop::EvolutionLoop, trial::TrialRunner,
};

/// The special agent name whose workspace lives at `~/.collet/workspace/`
/// (analogous to `soul::GLOBAL_SOUL`).
pub const GLOBAL_AGENT: &str = "collet";

/// Resolve the evolution workspace path for an agent.
///
/// - `"collet"` → `<collet_home>/workspace/`
/// - other      → `<collet_home>/agents/<name>/workspace/`
pub fn workspace_path(collet_home: &Path, agent_name: &str) -> PathBuf {
    if agent_name == GLOBAL_AGENT {
        collet_home.join("workspace")
    } else {
        collet_home
            .join("agents")
            .join(agent_name)
            .join("workspace")
    }
}

/// Returns true when auto-evolution should fire after an agent task completes.
pub fn is_auto_enabled(config: &Config) -> bool {
    config.evolution_enabled
}

/// Run one auto-evolution cycle for the given agent (async, non-blocking caller).
///
/// Called via `tokio::spawn` from the `AgentEvent::Done` handler — mirrors
/// `soul::reflect_simple`.
pub async fn run_auto(
    client: OpenAiCompatibleProvider,
    config: Config,
    collet_home: PathBuf,
    agent_name: String,
    working_dir: String,
    cancel: CancellationToken,
) -> Result<()> {
    let workspace_root = workspace_path(&collet_home, &agent_name);
    std::fs::create_dir_all(&workspace_root)?;

    let cycles = config.evolution_cycles;
    let evolver_model = config
        .evolution_model
        .clone()
        .unwrap_or_else(|| config.model.clone());

    let evolve_config = EvolveConfig {
        max_cycles: cycles,
        batch_size: 1,
        evolver_model,
        ..Default::default()
    };

    let evolvable = ColletEvolvable::new(
        client.clone(),
        config.clone(),
        workspace_root.clone(),
        working_dir,
    );

    let benchmark = Box::new(NullBenchmark);
    let trial = TrialRunner::new(Box::new(evolvable), benchmark);

    // Discard events — auto mode is silent (no TUI channel needed)
    let (evo_tx, _evo_rx) = tokio::sync::mpsc::unbounded_channel();

    let mut evo_loop = EvolutionLoop::new(workspace_root, trial, evolve_config.clone(), evo_tx)?;

    // Build a dedicated evolver client when the evolver model differs from the main model.
    let evolver_client = if evolve_config.evolver_model != config.model {
        crate::api::provider::OpenAiCompatibleProvider::new(
            client.base_url().to_string(),
            client.api_key().to_string(),
            evolve_config.evolver_model.clone(),
            config.context_max_tokens,
        )
        .unwrap_or_else(|_| client.clone())
    } else {
        client
    };
    let mut engine = SkillforgeEngine::new(evolve_config, evolver_client);
    evo_loop.run(&mut engine, cancel).await?;

    Ok(())
}