brainwires-agents 0.8.0

Agent orchestration, coordination, and lifecycle management for the Brainwires Agent Framework
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
139
140
141
142
143
144
145
146
147
148
149
150

//! Status Operations
//!
//! Task status update operations: start, complete, fail, skip, block.

use anyhow::{Context, Result};

use super::TaskManager;
use brainwires_core::TaskStatus;

impl TaskManager {
    /// Update task status
    #[tracing::instrument(name = "agent.task.update_status", skip(self, summary), fields(status = ?status))]
    pub async fn update_status(
        &self,
        task_id: &str,
        status: TaskStatus,
        summary: Option<String>,
    ) -> Result<()> {
        let parent_id = {
            let mut tasks = self.tasks.write().await;
            let task = tasks
                .get_mut(task_id)
                .context(format!("Task '{}' not found", task_id))?;

            task.status = status.clone();
            if let Some(s) = summary {
                task.summary = Some(s);
            }
            task.updated_at = chrono::Utc::now().timestamp();

            task.parent_id.clone()
        };

        // If completing a task, unblock dependents and check parent
        if status == TaskStatus::Completed {
            self.unblock_dependents(task_id).await?;
            if let Some(ref parent_id) = parent_id {
                self.check_parent_completion(parent_id).await?;
            }
        }

        Ok(())
    }

    /// Mark a task as started
    pub async fn start_task(&self, task_id: &str) -> Result<()> {
        self.update_status(task_id, TaskStatus::InProgress, None)
            .await
    }

    /// Mark a task as completed
    pub async fn complete_task(&self, task_id: &str, summary: String) -> Result<()> {
        self.update_status(task_id, TaskStatus::Completed, Some(summary))
            .await
    }

    /// Mark a task as failed
    pub async fn fail_task(&self, task_id: &str, error: String) -> Result<()> {
        self.update_status(task_id, TaskStatus::Failed, Some(error))
            .await
    }

    /// Mark a task as skipped
    pub async fn skip_task(&self, task_id: &str, reason: Option<String>) -> Result<()> {
        let parent_id = {
            let mut tasks = self.tasks.write().await;
            let task = tasks
                .get_mut(task_id)
                .context(format!("Task '{}' not found", task_id))?;

            let now = chrono::Utc::now().timestamp();
            task.status = TaskStatus::Skipped;
            task.completed_at = Some(now);
            task.updated_at = now;

            if let Some(r) = reason {
                task.summary = Some(r);
            }

            task.parent_id.clone()
        };

        // Check if parent can be updated (lock released above)
        if let Some(ref pid) = parent_id {
            self.check_parent_completion(pid).await?;
        }

        // Unblock tasks that depend on this one
        self.unblock_dependents(task_id).await?;

        Ok(())
    }

    /// Mark a task as blocked with optional reason
    pub async fn block_task(&self, task_id: &str, reason: Option<String>) -> Result<()> {
        let mut tasks = self.tasks.write().await;
        let task = tasks
            .get_mut(task_id)
            .context(format!("Task '{}' not found", task_id))?;

        task.status = TaskStatus::Blocked;
        task.updated_at = chrono::Utc::now().timestamp();

        if let Some(r) = reason {
            task.summary = Some(r);
        }

        Ok(())
    }

    /// Check if parent task should be auto-completed when all children are done
    pub(crate) async fn check_parent_completion(&self, parent_id: &str) -> Result<()> {
        let tasks = self.tasks.read().await;

        if let Some(parent) = tasks.get(parent_id) {
            // Check if all children are complete
            let all_complete = parent.children.iter().all(|child_id| {
                tasks
                    .get(child_id)
                    .map(|t| t.status == TaskStatus::Completed)
                    .unwrap_or(false)
            });

            if all_complete && !parent.children.is_empty() {
                let grandparent_id = parent.parent_id.clone();
                drop(tasks);

                // Auto-complete the parent task
                let mut tasks = self.tasks.write().await;
                if let Some(parent) = tasks.get_mut(parent_id)
                    && (parent.status == TaskStatus::InProgress
                        || parent.status == TaskStatus::Pending)
                {
                    parent.status = TaskStatus::Completed;
                    parent.summary =
                        Some(format!("All {} subtasks completed", parent.children.len()));
                    parent.updated_at = chrono::Utc::now().timestamp();
                }
                drop(tasks);

                // Recursively check grandparent
                if let Some(gp_id) = grandparent_id {
                    Box::pin(self.check_parent_completion(&gp_id)).await?;
                }
            }
        }

        Ok(())
    }
}