opencrabs 0.3.58

The autonomous, self-improving AI agent. Single Rust binary. Every channel. Install with: cargo install opencrabs
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

use crate::db::models::Message;
use crate::tui::app::state::*;
use uuid::Uuid;

#[test]
fn test_display_message_from_db_message() {
    let msg = Message {
        id: Uuid::new_v4(),
        session_id: Uuid::new_v4(),
        role: "user".to_string(),
        content: "Hello".to_string(),
        sequence: 1,
        created_at: chrono::Utc::now(),
        token_count: Some(10),
        cost: Some(0.001),
        input_tokens: None,
        cache_creation_tokens: None,
        cache_read_tokens: None,
        thinking: None,
    };

    let display_msg: DisplayMessage = msg.into();
    assert_eq!(display_msg.role, "user");
    assert_eq!(display_msg.content, "Hello");
    assert!(display_msg.details.is_none());
}

#[test]
fn test_display_message_thinking_from_db() {
    let msg = Message {
        id: Uuid::new_v4(),
        session_id: Uuid::new_v4(),
        role: "assistant".to_string(),
        content: "Here is the answer.".to_string(),
        sequence: 2,
        created_at: chrono::Utc::now(),
        token_count: Some(50),
        cost: Some(0.005),
        input_tokens: Some(200),
        cache_creation_tokens: None,
        cache_read_tokens: None,
        thinking: Some("I need to analyze this carefully...".to_string()),
    };

    let display_msg: DisplayMessage = msg.into();
    assert_eq!(display_msg.role, "assistant");
    assert_eq!(display_msg.content, "Here is the answer.");
    assert_eq!(
        display_msg.details,
        Some("I need to analyze this carefully...".to_string())
    );
}