From 7c6841782d36faacc2aa3616dc3ed1b9403fa26e Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 2 Nov 2023 10:45:11 +0800
Subject: refactor: improve code quanity (#197)

- move model_info.rs/message.rs to clients/
- rename SharedConfig to GlobalConfig
---
 src/client/message.rs | 52 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 src/client/message.rs

(limited to 'src/client/message.rs')

diff --git a/src/client/message.rs b/src/client/message.rs
new file mode 100644
index 0000000..55b2663
--- /dev/null
+++ b/src/client/message.rs
@@ -0,0 +1,52 @@
+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\"}"
+        );
+    }
+}
-- 
cgit v1.2.3