lash-core 0.1.0-alpha.84

Sans-IO turn machine and runtime kernel for the lash agent runtime.
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

use super::*;

impl RuntimeTurnDriver<'_> {
    pub(super) fn trace_context(&self, protocol_iteration: usize) -> lash_trace::TraceContext {
        lash_trace::TraceContext::default()
            .for_session(self.session_id.clone())
            .for_turn_index(self.turn_index)
            .for_protocol_iteration(protocol_iteration)
            .for_turn(self.turn_id.clone())
    }

    pub(super) fn llm_call_id(&mut self, protocol_iteration: usize) -> String {
        let ordinal = self.next_llm_ordinal;
        self.next_llm_ordinal += 1;
        format!(
            "{}:{}:{}:{}",
            self.session_id, self.turn_index, protocol_iteration, ordinal
        )
    }

    pub(super) fn emit_trace(&self, protocol_iteration: usize, event: lash_trace::TraceEvent) {
        crate::trace::emit_trace(
            &self.host.core.tracing.trace_sink,
            &self.host.core.tracing.trace_context,
            self.trace_context(protocol_iteration),
            event,
            self.host.core.clock.as_ref(),
        );
    }

    /// Build the per-tool trace handle threaded into a code/tool execution
    /// context so tool trace events are emitted from the shared seam. `None`
    /// when the host installed no trace sink, keeping emission a no-op.
    pub(super) fn execution_tracing(
        &self,
        protocol_iteration: usize,
    ) -> Option<crate::RuntimeExecutionTracing> {
        let tracing = &self.host.core.tracing;
        tracing.trace_sink.as_ref().map(|sink| {
            crate::RuntimeExecutionTracing::new(
                std::sync::Arc::clone(sink),
                tracing.trace_context.clone(),
                self.trace_context(protocol_iteration),
            )
        })
    }

    pub(super) fn mark_phase_begin(&self, phase: RuntimeTurnPhase) {
        if let Some(probe) = self.turn_phase_probe.as_ref() {
            probe.begin(phase);
        }
    }

    pub(super) fn emit_protocol_diagnostic_trace(
        &self,
        protocol_iteration: usize,
        phase: &str,
        payload: serde_json::Value,
    ) {
        let protocol_event = crate::ProtocolEvent::typed(
            "runtime",
            serde_json::json!({
                "diagnostic": {
                    "phase": phase,
                    "payload": payload,
                }
            }),
        )
        .expect("protocol diagnostic event serializes");
        self.emit_trace(
            protocol_iteration,
            protocol_step_trace_event(&protocol_event),
        );
    }

    pub(super) fn mark_phase_end(&self, phase: RuntimeTurnPhase) {
        if let Some(probe) = self.turn_phase_probe.as_ref() {
            probe.end(phase);
        }
    }
}

pub(in crate::runtime) fn protocol_step_trace_event(
    protocol_event: &crate::ProtocolEvent,
) -> lash_trace::TraceEvent {
    lash_trace::TraceEvent::ProtocolStep {
        plugin_id: protocol_event.plugin_id.clone(),
        payload: protocol_event.payload.clone(),
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn protocol_step_trace_event_preserves_protocol_payload() {
        let protocol_event = crate::ProtocolEvent::typed(
            "custom",
            serde_json::json!({
                "code": "print \"hi\"",
                "final_output": "done"
            }),
        )
        .expect("protocol event");

        let lash_trace::TraceEvent::ProtocolStep { plugin_id, payload } =
            protocol_step_trace_event(&protocol_event)
        else {
            panic!("expected protocol step trace event");
        };

        assert_eq!(plugin_id, "custom");
        assert_eq!(
            payload.get("code").and_then(serde_json::Value::as_str),
            Some("print \"hi\"")
        );
        assert_eq!(
            payload.get("final_output"),
            Some(&serde_json::json!("done"))
        );
    }
}