starweaver-stream 0.0.1

Display and replay stream contracts 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
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

//! Run lifecycle display projection.

use serde_json::json;
use starweaver_core::RunId;
use starweaver_runtime::{AgentExecutionNode, AgentStreamRecord};

use super::super::{DisplayMessage, DisplayMessageKind, DisplayProjectionContext};

pub(super) fn project_run_started(
    context: &DisplayProjectionContext,
    sequence: &AgentStreamRecord,
    run_id: RunId,
    conversation_id: &str,
) -> DisplayMessage {
    DisplayMessage::new(
        sequence.sequence,
        context.session_id.clone(),
        run_id,
        DisplayMessageKind::RunStarted,
    )
    .with_payload(json!({"conversation_id": conversation_id}))
    .with_preview("run started")
}

pub(super) fn project_run_failed(
    context: &DisplayProjectionContext,
    sequence: usize,
    run_id: RunId,
    error_kind: &str,
    message: &str,
) -> DisplayMessage {
    DisplayMessage::new(
        sequence,
        context.session_id.clone(),
        run_id,
        DisplayMessageKind::RunFailed,
    )
    .with_payload(json!({"error_kind": error_kind, "error": message}))
    .with_preview(message.to_string())
}

pub(super) fn project_checkpoint(
    context: &DisplayProjectionContext,
    sequence: usize,
    run_id: RunId,
    node: AgentExecutionNode,
    step: usize,
) -> DisplayMessage {
    DisplayMessage::new(
        sequence,
        context.session_id.clone(),
        run_id,
        DisplayMessageKind::Checkpoint,
    )
    .with_payload(json!({"node": node, "step": step}))
    .with_preview(format!("checkpoint {node:?}"))
}

pub(super) fn project_steering_guard(
    context: &DisplayProjectionContext,
    sequence: usize,
    run_id: RunId,
    step: usize,
    prompt: &str,
) -> DisplayMessage {
    DisplayMessage::new(
        sequence,
        context.session_id.clone(),
        run_id,
        DisplayMessageKind::Checkpoint,
    )
    .with_payload(json!({"step": step, "kind": "steering_guard", "prompt": prompt}))
    .with_preview("steering update pending")
}

pub(super) fn project_run_completed(
    context: &DisplayProjectionContext,
    sequence: usize,
    run_id: RunId,
    output: &str,
) -> DisplayMessage {
    DisplayMessage::new(
        sequence,
        context.session_id.clone(),
        run_id,
        DisplayMessageKind::RunCompleted,
    )
    .with_payload(json!({"output": output}))
    .with_preview(output.to_string())
}

pub(super) fn project_run_cancelled(
    context: &DisplayProjectionContext,
    sequence: usize,
    run_id: RunId,
    node: AgentExecutionNode,
    reason: &str,
) -> DisplayMessage {
    DisplayMessage::new(
        sequence,
        context.session_id.clone(),
        run_id,
        DisplayMessageKind::RunCancelled,
    )
    .with_payload(json!({"node": node, "reason": reason}))
    .with_preview(reason.to_string())
}