harn-vm 0.10.170

Async bytecode virtual machine for the Harn programming language
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

//! Native trace classification for interpreter operations.

use crate::{
    tracing::SpanKind,
    vm::{ScopeSpan, VmSpanKind},
};

pub(super) fn start_span(kind: VmSpanKind, name: String) -> ScopeSpan {
    let kind = match kind {
        VmSpanKind::Pipeline => SpanKind::Pipeline,
        VmSpanKind::Import => SpanKind::Import,
        VmSpanKind::Parallel => SpanKind::Parallel,
        VmSpanKind::Spawn => SpanKind::Spawn,
    };
    begin(kind, name)
}

pub(super) fn builtin_span(name: &str) -> Option<ScopeSpan> {
    // Both public capability paths and ambient names use the live registry.
    let resolved = crate::stdlib::builtin_for_harness_path(name).unwrap_or(name);
    let kind = match resolved.strip_prefix("__cap_").unwrap_or(resolved) {
        "llm_call" | "llm_stream" | "llm_stream_call" | "agent_loop" => SpanKind::LlmCall,
        "mcp_call" => SpanKind::ToolCall,
        _ => return None,
    };
    if !crate::tracing::is_tracing_enabled() {
        return None;
    }
    Some(begin(kind, name.to_string()))
}

fn begin(kind: SpanKind, name: String) -> ScopeSpan {
    ScopeSpan::new(crate::tracing::span_start(kind, name), end)
}

fn end(id: u64) {
    crate::tracing::span_end(id);
}

#[cfg(test)]
mod tests {
    use crate::value::VmValue;

    #[tokio::test(flavor = "current_thread")]
    async fn builtin_observation_is_inert_when_disabled_and_classifies_both_public_paths() {
        crate::orchestration::scope_fresh_tracing_runtime(async {
            let mut vm = crate::new_vm();
            vm.register_builtin("llm_call", |_, _| Ok(VmValue::Int(7)));
            assert!(vm.runtime().builtin_span("llm_call").is_none());
            assert_eq!(
                vm.call_named_builtin("llm_call", vec![VmValue::String("fixture".into())])
                    .await
                    .unwrap()
                    .display(),
                "7"
            );
            assert!(crate::tracing::take_spans().is_empty());

            crate::tracing::set_tracing_enabled(true);
            assert_eq!(
                vm.call_named_builtin("llm_call", vec![VmValue::String("fixture".into())])
                    .await
                    .unwrap()
                    .display(),
                "7"
            );
            drop(
                vm.runtime()
                    .builtin_span("harness.llm.call")
                    .expect("live capability registry resolves the public path"),
            );
            assert!(vm.runtime().builtin_span("harness.stdio.println").is_none());
            let spans = crate::tracing::take_spans();
            assert_eq!(
                spans.len(),
                2,
                "both enabled observations reached the collector"
            );
            assert!(spans
                .iter()
                .all(|span| span.kind == crate::tracing::SpanKind::LlmCall));
            assert_eq!(spans[0].name, "llm_call");
            assert_eq!(spans[1].name, "harness.llm.call");
        })
        .await;
    }
}