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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use crate::llm::reasoning_receipt::{record, scope_request_receipts, RequestReceipts};

fn emit_receipt() {
    record(
        "gemini",
        "receipt-proof",
        "gemini_interactions",
        &crate::llm::api::ThinkingConfig::Effort {
            level: crate::llm::api::ReasoningEffort::XHigh,
        },
        &["generation_config.thinking_level"],
        &serde_json::json!({"generation_config": {"thinking_level": "high"}}),
    );
}

#[tokio::test(flavor = "current_thread")]
async fn provider_thread_handoff_keeps_the_owning_durable_journal() {
    crate::agent_sessions::reset_session_store();
    let root = tempfile::tempdir().unwrap();
    let session_id = "reasoning-journal-thread-handoff";
    let mut options = crate::value::DictMap::new();
    crate::value::VmDictExt::put_str(&mut options, "root", root.path().to_string_lossy());
    let prepared = crate::agent_session_journal::prepare(
        session_id,
        &options,
        "reasoning-handoff-run".into(),
        "reasoning-handoff-turn".into(),
    )
    .await
    .unwrap();
    crate::agent_sessions::open_or_create_for_test(Some(session_id.into()));
    crate::agent_sessions::install_journal(session_id, prepared.state).unwrap();
    let store = crate::agent_sessions::journal_store(session_id).unwrap();
    let mut call_options = crate::llm::api::LlmCallOptions {
        session_id: Some(session_id.into()),
        ..Default::default()
    };
    call_options.set_call_attribution("completion.judge", "review");
    let receipts = RequestReceipts::for_call(&call_options, "handoff-call");
    // A measured same-thread receipt establishes the journal and reporting path.
    scope_request_receipts(receipts.clone(), async {
        emit_receipt();
    })
    .await;
    assert!(receipts.take_journal_error().is_none());
    crate::agent_session_journal::flush(session_id)
        .await
        .unwrap();
    let events = crate::stdlib::session_store::read_all_events(&store, session_id)
        .await
        .unwrap();
    assert_eq!(events.len(), 1);
    let provider_receipts = receipts.clone();
    let runtime = tokio::runtime::Handle::current();
    crate::runtime_stack::spawn(move || {
        runtime.block_on(scope_request_receipts(provider_receipts, async {
            emit_receipt();
        }));
    })
    .join()
    .unwrap();
    assert_eq!(receipts.take_journal_error(), None);
    crate::agent_session_journal::flush(session_id)
        .await
        .unwrap();
    let events = crate::stdlib::session_store::read_all_events(&store, session_id)
        .await
        .unwrap();
    assert_eq!(events.len(), 2);
    assert!(events
        .iter()
        .all(|event| event.kind.discriminator() == "reasoning_receipt"));
    crate::agent_sessions::append_terminal_event_once(
        session_id,
        crate::llm::helpers::transcript_event(
            "agent_run_terminal",
            "system",
            "internal",
            "completed",
            None,
        ),
    )
    .unwrap();
    scope_request_receipts(receipts.clone(), async {
        emit_receipt();
    })
    .await;
    assert_eq!(
        receipts.take_journal_error().as_deref(),
        Some("agent transcript audit writer: journal is closed"),
    );
    crate::agent_session_journal::flush(session_id)
        .await
        .unwrap();
    let events = crate::stdlib::session_store::read_all_events(&store, session_id)
        .await
        .unwrap();
    assert_eq!(
        events.len(),
        3,
        "late provider facts must not follow the terminal"
    );
    assert_eq!(
        events.last().unwrap().kind.discriminator(),
        "agent_run_terminal"
    );
    crate::agent_sessions::reset_session_store();
}