unified-agent-api 0.2.0

Agent-agnostic facade and registry for wrapper backends
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

use super::*;

#[tokio::test]
async fn spawn_failures_surface_via_run_handle() {
    let adapter = std::sync::Arc::new(ToyAdapter { fail_spawn: true });
    let request = crate::AgentWrapperRunRequest {
        prompt: "hello".to_string(),
        ..Default::default()
    };

    let mut handle = run_harnessed_backend(adapter, BackendDefaults::default(), request)
        .await
        .expect("run should return a handle before startup resolves");

    let event = handle
        .events
        .next()
        .await
        .expect("spawn failure should surface as an error event");
    assert_eq!(event.kind, AgentWrapperEventKind::Error);
    assert_eq!(
        event.message.as_deref(),
        Some("toy backend error (redacted): phase=spawn")
    );

    assert!(
        handle.events.next().await.is_none(),
        "error event should be terminal for spawn failure"
    );

    let err = handle
        .completion
        .await
        .expect_err("completion should surface the spawn failure");
    assert!(matches!(
        err,
        crate::AgentWrapperError::Backend { ref message }
            if message == "toy backend error (redacted): phase=spawn"
    ));
}

#[tokio::test]
async fn spawn_failures_surface_via_control_handle_after_return() {
    let adapter = std::sync::Arc::new(ToyAdapter { fail_spawn: true });
    let request = crate::AgentWrapperRunRequest {
        prompt: "hello".to_string(),
        ..Default::default()
    };

    let crate::AgentWrapperRunControl {
        mut handle,
        cancel: _,
    } = run_harnessed_backend_control(adapter, BackendDefaults::default(), request, None)
        .await
        .expect("control entrypoint should return before startup resolves");

    let event = handle
        .events
        .next()
        .await
        .expect("spawn failure should surface as an error event");
    assert_eq!(event.kind, AgentWrapperEventKind::Error);
    assert_eq!(
        event.message.as_deref(),
        Some("toy backend error (redacted): phase=spawn")
    );

    assert!(
        handle.events.next().await.is_none(),
        "error event should be terminal for spawn failure"
    );

    let err = handle
        .completion
        .await
        .expect_err("completion should be backend error");
    assert!(matches!(
        err,
        crate::AgentWrapperError::Backend { ref message }
        if message == "toy backend error (redacted): phase=spawn"
    ));
}