objectiveai-api 2.2.7

ObjectiveAI API Server
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

use super::super::{ContinuationItem, UpstreamClient};
use super::State;

fn make_client() -> super::Client {
    // objectiveai_dir is unused by response_continuation (no runner is
    // spawned in these pure-logic tests), so a dummy path is fine.
    super::Client::new(String::new(), true, 0, 180, 1, std::path::PathBuf::new())
}

#[test]
fn test_no_continuation_no_request_continuation() {
    let client = make_client();
    let result = client.response_continuation(
        None,
        &[],
        None,
        ""
    );
    assert_eq!(result, objectiveai_sdk::agent::claude_agent_sdk::Continuation {
        upstream: objectiveai_sdk::agent::claude_agent_sdk::Upstream::ClaudeAgentSdk,
        agent_instance_hierarchy: String::new(),
        session_id: String::new(),
    });
}

#[test]
fn test_session_id_from_continuation_state() {
    let client = make_client();
    let continuation = vec![
        ContinuationItem::State(State {
            session_id: "internal-sess".into(),
            message_count: 1,
        }),
    ];
    let result = client.response_continuation(
        None,
        &[],
        Some(&continuation),
        ""
    );
    assert_eq!(result.session_id, "internal-sess");
}

#[test]
fn test_session_id_falls_back_to_request_continuation() {
    let client = make_client();
    let rc = objectiveai_sdk::agent::claude_agent_sdk::Continuation {
        upstream: objectiveai_sdk::agent::claude_agent_sdk::Upstream::ClaudeAgentSdk,
        agent_instance_hierarchy: String::new(),
        session_id: "req-sess-123".into(),
    };
    let result = client.response_continuation(
        Some(&rc),
        &[],
        None,
        ""
    );
    assert_eq!(result.session_id, "req-sess-123");
}

#[test]
fn test_internal_session_id_takes_precedence() {
    let client = make_client();
    let continuation = vec![
        ContinuationItem::State(State {
            session_id: "internal-sess".into(),
            message_count: 1,
        }),
    ];
    let rc = objectiveai_sdk::agent::claude_agent_sdk::Continuation {
        upstream: objectiveai_sdk::agent::claude_agent_sdk::Upstream::ClaudeAgentSdk,
        agent_instance_hierarchy: String::new(),
        session_id: "req-sess-456".into(),
    };
    let result = client.response_continuation(
        Some(&rc),
        &[],
        Some(&continuation),
        ""
    );
    assert_eq!(result.session_id, "internal-sess");
}

#[test]
fn test_empty_internal_session_falls_back_to_request() {
    let client = make_client();
    let continuation = vec![
        ContinuationItem::State(State {
            session_id: String::new(),
            message_count: 0,
        }),
    ];
    let rc = objectiveai_sdk::agent::claude_agent_sdk::Continuation {
        upstream: objectiveai_sdk::agent::claude_agent_sdk::Upstream::ClaudeAgentSdk,
        agent_instance_hierarchy: String::new(),
        session_id: "req-sess-fallback".into(),
    };
    let result = client.response_continuation(
        Some(&rc),
        &[],
        Some(&continuation),
        ""
    );
    assert_eq!(result.session_id, "req-sess-fallback");
}