synwire-core 0.1.0

Core traits and types for the Synwire AI framework
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

//! Trait for types convertible to a [`Message`].

use crate::messages::Message;

/// Trait for types that can be converted to a [`Message`].
///
/// This provides a uniform interface for accepting various types as messages,
/// such as raw strings, tuples of (role, content), or `Message` values directly.
///
/// # Examples
///
/// ```
/// use synwire_core::messages::{Message, MessageLike};
///
/// let msg: Message = "Hello".to_message();
/// assert_eq!(msg.message_type(), "human");
///
/// let msg: Message = ("ai", "Hi there").to_message();
/// assert_eq!(msg.message_type(), "ai");
/// ```
pub trait MessageLike {
    /// Convert to a [`Message`].
    fn to_message(&self) -> Message;
}

impl MessageLike for Message {
    fn to_message(&self) -> Message {
        self.clone()
    }
}

impl MessageLike for &str {
    fn to_message(&self) -> Message {
        Message::human(*self)
    }
}

impl MessageLike for String {
    fn to_message(&self) -> Message {
        Message::human(self)
    }
}

impl MessageLike for (&str, &str) {
    /// First element is role (`"human"`, `"user"`, `"ai"`, `"assistant"`, `"system"`),
    /// second is content. Unrecognised roles default to human.
    fn to_message(&self) -> Message {
        match self.0 {
            "ai" | "assistant" => Message::ai(self.1),
            "system" => Message::system(self.1),
            // "human", "user", or any unrecognised role defaults to human.
            _ => Message::human(self.1),
        }
    }
}

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

    #[test]
    fn test_message_like_from_str() {
        let msg = "Hello".to_message();
        assert_eq!(msg.message_type(), "human");
        assert_eq!(msg.content().as_text(), "Hello");
    }

    #[test]
    fn test_message_like_from_string() {
        let s = String::from("Hello");
        let msg = s.to_message();
        assert_eq!(msg.message_type(), "human");
    }

    #[test]
    fn test_message_like_from_message() {
        let original = Message::ai("Hi");
        let msg = original.to_message();
        assert_eq!(msg.message_type(), "ai");
    }

    #[test]
    fn test_message_like_from_tuple_human() {
        let msg = ("human", "Hello").to_message();
        assert_eq!(msg.message_type(), "human");
    }

    #[test]
    fn test_message_like_from_tuple_user() {
        let msg = ("user", "Hello").to_message();
        assert_eq!(msg.message_type(), "human");
    }

    #[test]
    fn test_message_like_from_tuple_ai() {
        let msg = ("ai", "Hi").to_message();
        assert_eq!(msg.message_type(), "ai");
    }

    #[test]
    fn test_message_like_from_tuple_assistant() {
        let msg = ("assistant", "Hi").to_message();
        assert_eq!(msg.message_type(), "ai");
    }

    #[test]
    fn test_message_like_from_tuple_system() {
        let msg = ("system", "Be helpful").to_message();
        assert_eq!(msg.message_type(), "system");
    }

    #[test]
    fn test_message_like_from_tuple_unknown_defaults_to_human() {
        let msg = ("unknown_role", "Hello").to_message();
        assert_eq!(msg.message_type(), "human");
    }
}