From 7c6841782d36faacc2aa3616dc3ed1b9403fa26e Mon Sep 17 00:00:00 2001 From: sigoden 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/config/message.rs | 52 --------------------------------------------------- 1 file changed, 52 deletions(-) delete mode 100644 src/config/message.rs (limited to 'src/config/message.rs') 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\"}" - ); - } -} -- cgit v1.2.3