starweaver-runtime 0.1.0

Agent-loop graph and runtime executor primitives for Starweaver
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

//! Executor checkpoint helper.

use starweaver_context::AgentContext;

use crate::{
    agent::{Agent, AgentError},
    executor::{AgentCheckpoint, AgentExecutionDecision, AgentExecutionNode},
    run::AgentRunState,
    trace::{SpanSpec, SpanStatus},
};

impl Agent {
    pub(in crate::agent) async fn checkpoint(
        &self,
        node: AgentExecutionNode,
        state: &AgentRunState,
        context: &AgentContext,
        stream_cursor: Option<usize>,
    ) -> Result<AgentExecutionDecision, AgentError> {
        let checkpoint_span = self.trace_recorder.start_span(
            SpanSpec::new("starweaver.checkpoint")
                .with_attribute("starweaver.checkpoint.node", serde_json::json!(node)),
            &context.trace_context,
        );
        let mut checkpoint = AgentCheckpoint::new(node, state);
        if let Some(stream_cursor) = stream_cursor {
            checkpoint = checkpoint.with_stream_cursor(stream_cursor);
        }
        checkpoint.resume.trace_context = checkpoint_span.context().clone();
        checkpoint.metadata.insert(
            "trace_id".to_string(),
            serde_json::json!(checkpoint_span.context().trace_id),
        );
        checkpoint.metadata.insert(
            "span_id".to_string(),
            serde_json::json!(checkpoint_span.context().span_id),
        );
        for capability in &self.ordered_capabilities()? {
            capability
                .on_checkpoint_with_context(state, context, &checkpoint)
                .await
                .map_err(Self::capability_error)?;
        }
        let decision = self.executor.checkpoint(checkpoint).await?;
        self.trace_recorder
            .close_span(&checkpoint_span, SpanStatus::Ok);
        Ok(decision)
    }
}