turboclaudeagent 0.3.0

Interactive Agent SDK for TurboClaude - Use Claude agents in your Rust applications
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
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143

//! Session state management
//!
//! Provides structures and operations for tracking session state including
//! connection status, model settings, permission modes, and conversation history.

use turboclaude_protocol::{Message, PermissionMode};

/// Current state of the agent session
#[derive(Debug, Clone)]
pub struct SessionState {
    /// Whether the session is connected to CLI
    pub is_connected: bool,

    /// Current model
    pub current_model: String,

    /// Current permission mode
    pub current_permission_mode: PermissionMode,

    /// Number of active queries
    pub active_queries: u32,

    /// Conversation history (for fork support)
    pub(crate) conversation_history: Vec<Message>,
}

impl SessionState {
    /// Create a new session state
    pub(crate) fn new(model: String, permission_mode: PermissionMode) -> Self {
        Self {
            is_connected: true,
            current_model: model,
            current_permission_mode: permission_mode,
            active_queries: 0,
            conversation_history: Vec::new(),
        }
    }

    /// Add a message to the conversation history
    pub(crate) fn add_to_history(&mut self, message: Message) {
        self.conversation_history.push(message);
    }

    /// Get a clone of the conversation history
    pub(crate) fn get_history(&self) -> Vec<Message> {
        self.conversation_history.clone()
    }

    /// Clear conversation history
    #[allow(dead_code)]
    pub(crate) fn clear_history(&mut self) {
        self.conversation_history.clear();
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_session_state_new() {
        let state = SessionState::new("claude-3-5-sonnet".to_string(), PermissionMode::Default);

        assert!(state.is_connected);
        assert_eq!(state.current_model, "claude-3-5-sonnet");
        assert_eq!(state.current_permission_mode, PermissionMode::Default);
        assert_eq!(state.active_queries, 0);
        assert!(state.conversation_history.is_empty());
    }

    #[test]
    fn test_conversation_history() {
        use turboclaude_protocol::{
            message::MessageRole,
            types::{CacheUsage, StopReason, Usage},
        };

        let mut state = SessionState::new("claude-3-5-sonnet".to_string(), PermissionMode::Default);

        let msg1 = Message {
            id: "msg_1".to_string(),
            message_type: "message".to_string(),
            role: MessageRole::User,
            content: vec![],
            model: "claude-3-5-sonnet".to_string(),
            stop_reason: StopReason::EndTurn,
            stop_sequence: None,
            usage: Usage {
                input_tokens: 0,
                output_tokens: 0,
            },
            cache_usage: CacheUsage {
                cache_read_input_tokens: 0,
                cache_creation_input_tokens: 0,
            },
            created_at: String::new(),
        };
        let msg2 = Message {
            id: "msg_2".to_string(),
            message_type: "message".to_string(),
            role: MessageRole::Assistant,
            content: vec![],
            model: "claude-3-5-sonnet".to_string(),
            stop_reason: StopReason::EndTurn,
            stop_sequence: None,
            usage: Usage {
                input_tokens: 0,
                output_tokens: 0,
            },
            cache_usage: CacheUsage {
                cache_read_input_tokens: 0,
                cache_creation_input_tokens: 0,
            },
            created_at: String::new(),
        };

        state.add_to_history(msg1.clone());
        state.add_to_history(msg2.clone());

        let history = state.get_history();
        assert_eq!(history.len(), 2);
        assert_eq!(history[0].role, MessageRole::User);
        assert_eq!(history[1].role, MessageRole::Assistant);

        state.clear_history();
        assert!(state.conversation_history.is_empty());
    }

    #[test]
    fn test_session_state_clone() {
        let state = SessionState {
            is_connected: true,
            current_model: "claude-3-5-sonnet".to_string(),
            current_permission_mode: PermissionMode::Default,
            active_queries: 0,
            conversation_history: Vec::new(),
        };

        let state2 = state.clone();
        assert_eq!(state.is_connected, state2.is_connected);
        assert_eq!(state.current_model, state2.current_model);
    }
}