atman-runtime 1.9.1

atman flow execution runtime: evaluator, tool dispatch, provider dispatch, executor, memory stores
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

mod common;

use std::sync::Arc;

use atman_runtime::event::EventSink;
use atman_runtime::providers::mock::MockProvider;
use atman_runtime::session_meta::{NameSource, SessionMeta};
use atman_runtime::{Executor, Session, Value};

#[tokio::test]
async fn built_in_flow_generates_and_persists_session_name() {
    let _registry = common::ModelRegistryGuard::mock("cheap").await;
    let tmp = tempfile::tempdir().unwrap();
    let session = Arc::new(Session::open(tmp.path()).unwrap());
    let executor = Executor::with_events(EventSink::new());
    executor.providers.register(Arc::new(
        MockProvider::new("cheap").with_fallback(Value::Str("Fix session switching".into())),
    ));

    let written = atman_runtime::session_naming::maybe_generate_session_name(&executor, &session)
        .await
        .unwrap();

    assert!(written);
    let meta = SessionMeta::load(session.dir()).unwrap();
    assert_eq!(meta.title.as_deref(), Some("Fix session switching"));
    assert_eq!(meta.name_source, NameSource::Auto);
    assert!(executor.events.snapshot().is_empty());
}

#[tokio::test]
async fn scheduled_auto_name_does_not_replace_user_name() {
    let tmp = tempfile::tempdir().unwrap();
    let session = Arc::new(Session::open(tmp.path()).unwrap());
    SessionMeta::set_title(session.dir(), Some("Pinned by user".into())).unwrap();
    let executor = Executor::with_events(EventSink::new());

    let written = atman_runtime::session_naming::maybe_generate_session_name(&executor, &session)
        .await
        .unwrap();

    assert!(!written);
    let meta = SessionMeta::load(session.dir()).unwrap();
    assert_eq!(meta.title.as_deref(), Some("Pinned by user"));
    assert_eq!(meta.name_source, NameSource::User);
}

#[tokio::test]
async fn user_triggered_auto_name_replaces_user_name_and_source() {
    let tmp = tempfile::tempdir().unwrap();
    let session = Arc::new(Session::open(tmp.path()).unwrap());
    SessionMeta::set_title(session.dir(), Some("Pinned by user".into())).unwrap();
    let executor = Executor::with_events(EventSink::new());
    executor.providers.register(Arc::new(
        MockProvider::new("cheap").with_fallback(Value::Str("Replacement".into())),
    ));

    let written = atman_runtime::session_naming::force_generate_session_name(&executor, &session)
        .await
        .unwrap();

    assert!(written);
    let meta = SessionMeta::load(session.dir()).unwrap();
    assert_eq!(meta.title.as_deref(), Some("Replacement"));
    assert_eq!(meta.name_source, NameSource::Auto);
    assert_eq!(session.successful_flow_count(), 0);
}