summaryrefslogtreecommitdiffstats
path: root/src/config/message.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/config/message.rs')
-rw-r--r--src/config/message.rs52
1 files changed, 0 insertions, 52 deletions
diff --git a/src/config/message.rs b/src/config/message.rs
deleted file mode 100644
index 55b2663..0000000
--- a/src/config/message.rs
+++ /dev/null
@@ -1,52 +0,0 @@
-use serde::{Deserialize, Serialize};
-
-#[derive(Debug, Clone, Deserialize, Serialize)]
-pub struct Message {
- pub role: MessageRole,
- pub content: String,
-}
-
-impl Message {
- pub fn new(content: &str) -> Self {
- Self {
- role: MessageRole::User,
- content: content.to_string(),
- }
- }
-}
-
-#[derive(Debug, Clone, Copy, Deserialize, Serialize)]
-#[serde(rename_all = "snake_case")]
-pub enum MessageRole {
- System,
- Assistant,
- User,
-}
-
-#[allow(dead_code)]
-impl MessageRole {
- pub fn is_system(&self) -> bool {
- matches!(self, MessageRole::System)
- }
-
- pub fn is_user(&self) -> bool {
- matches!(self, MessageRole::User)
- }
-
- pub fn is_assistant(&self) -> bool {
- matches!(self, MessageRole::Assistant)
- }
-}
-
-#[cfg(test)]
-mod tests {
- use super::*;
-
- #[test]
- fn test_serde() {
- assert_eq!(
- serde_json::to_string(&Message::new("Hello World")).unwrap(),
- "{\"role\":\"user\",\"content\":\"Hello World\"}"
- );
- }
-}