awaken-runtime 0.4.0

Phase-based execution engine, plugin system, and agent loop for Awaken
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

use std::sync::Arc;

use awaken_contract::StateError;
use awaken_contract::model::Phase;

use crate::hooks::{PhaseContext, PhaseHook};
use crate::state::StateCommand;

use crate::agent::state::PendingWorkKey;

use super::manager::BackgroundTaskManager;
use super::state::BackgroundTaskStateKey;

/// Phase hook that syncs background task metadata with the persisted state.
///
/// - `RunStart`: restores persisted metadata and performs orphan detection.
/// - `RunEnd`: no-op (metadata is committed directly by the manager).
/// - `StepEnd`: updates `PendingWorkKey` based on running task status.
pub(crate) struct BackgroundTaskSyncHook {
    pub(crate) manager: Arc<BackgroundTaskManager>,
}

#[async_trait::async_trait]
impl PhaseHook for BackgroundTaskSyncHook {
    async fn run(&self, ctx: &PhaseContext) -> Result<StateCommand, StateError> {
        match ctx.phase {
            Phase::RunStart => {
                let thread_id = &ctx.run_identity.thread_id;
                let snapshot = ctx
                    .state::<BackgroundTaskStateKey>()
                    .cloned()
                    .unwrap_or_default();
                self.manager.restore_for_thread(thread_id, &snapshot).await;
                Ok(StateCommand::new())
            }
            Phase::RunEnd => {
                // Metadata is committed directly by the manager.
                Ok(StateCommand::new())
            }
            Phase::StepEnd => {
                let thread_id = &ctx.run_identity.thread_id;
                let has_running = self.manager.has_running(thread_id).await;
                let mut cmd = StateCommand::new();
                cmd.update::<PendingWorkKey>(has_running);
                Ok(cmd)
            }
            _ => Ok(StateCommand::new()),
        }
    }
}