cerebro 1.1.8

A blazing-fast AI memory layer that enables teams of specialized agents to collaborate through a shared cognitive architecture.
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

//! # Swarm Immunology
//!
//! A specialized supervisor that monitors failing agents and dynamically rewrites
//! their system prompts ("vaccinates" them) to prevent future failures.

use super::agent::{AgentConfig, ChatMessage, LlmProvider, Role};
use super::llm::LlmClient;
use crate::traits::Result;

/// The Immunology Supervisor monitors agents and "vaccinates" them (rewrites their prompt) if they fail.
pub struct ImmunologySupervisor {
    pub enabled: bool,
    pub doctor_provider: Option<LlmProvider>,
}

impl ImmunologySupervisor {
    /// Create a new supervisor. Requires an LLM provider to act as the "Doctor".
    pub fn new(doctor_provider: Option<LlmProvider>) -> Self {
        Self {
            enabled: doctor_provider.is_some(),
            doctor_provider,
        }
    }

    /// If an agent fails repeatedly, pass the error and its config here to get a patched config.
    pub async fn heal_agent(
        &self,
        llm: &LlmClient,
        mut failed_agent: AgentConfig,
        error_context: &str,
    ) -> Result<AgentConfig> {
        let provider = match &self.doctor_provider {
            Some(p) => p,
            None => return Ok(failed_agent),
        };

        let system_prompt =
            "You are the Swarm Immunology Agent. Another agent in the swarm has failed. \
        You will be provided with its current system prompt and the error it encountered. \
        Your job is to REWRITE its system prompt to patch the blind spot so it doesn't fail again. \
        Output ONLY the new system prompt as plain text, no markdown blocks, no explanations.";

        let user_prompt = format!(
            "AGENT NAME: {}\n\nCURRENT PROMPT:\n{}\n\nERROR ENCOUNTERED:\n{}",
            failed_agent.name, failed_agent.system_prompt, error_context
        );

        let messages = vec![
            ChatMessage::new(Role::System, system_prompt),
            ChatMessage::new(Role::User, user_prompt),
        ];

        let response = llm.chat(provider, &messages).await?;

        let new_prompt = response
            .content
            .trim()
            .trim_start_matches("```text")
            .trim_end_matches("```")
            .trim()
            .to_string();

        if !new_prompt.is_empty() {
            println!(
                "💉 Immunology Agent vaccinated [{}] with a patched prompt.",
                failed_agent.id
            );
            failed_agent.system_prompt = new_prompt;
        }

        Ok(failed_agent)
    }
}