objectiveai-api 2.0.5

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
106
107
108
109
110
111
112
113
114
115
116
117

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

fn make_client() -> super::Client {
    super::Client::new(String::new(), true, 0, 180, 1)
}

#[test]
fn test_no_continuation_no_request_continuation() {
    let client = make_client();
    let result = client.response_continuation(
        indexmap::IndexMap::new(),
        None,
        &[],
        None,
    );
    assert_eq!(result, objectiveai_sdk::agent::claude_agent_sdk::Continuation {
        upstream: objectiveai_sdk::agent::claude_agent_sdk::Upstream::ClaudeAgentSdk,
        session_id: String::new(),
        mcp_sessions: indexmap::IndexMap::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(
        indexmap::IndexMap::new(),
        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,
        session_id: "req-sess-123".into(),
        mcp_sessions: indexmap::IndexMap::new(),
    };
    let result = client.response_continuation(
        indexmap::IndexMap::new(),
        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,
        session_id: "req-sess-456".into(),
        mcp_sessions: indexmap::IndexMap::new(),
    };
    let result = client.response_continuation(
        indexmap::IndexMap::new(),
        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,
        session_id: "req-sess-fallback".into(),
        mcp_sessions: indexmap::IndexMap::new(),
    };
    let result = client.response_continuation(
        indexmap::IndexMap::new(),
        Some(&rc),
        &[],
        Some(&continuation),
    );
    assert_eq!(result.session_id, "req-sess-fallback");
}

#[test]
fn test_mcp_sessions_preserved() {
    let client = make_client();
    let mut mcp_sessions = indexmap::IndexMap::new();
    mcp_sessions.insert("http://mcp.example.com".into(), "sess-abc".into());
    let result = client.response_continuation(
        mcp_sessions.clone(),
        None,
        &[],
        None,
    );
    assert_eq!(result.mcp_sessions, mcp_sessions);
}