diff options
Diffstat (limited to 'src/client/message.rs')
| -rw-r--r-- | src/client/message.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/client/message.rs b/src/client/message.rs index 2cfc183..2f7517c 100644 --- a/src/client/message.rs +++ b/src/client/message.rs @@ -65,6 +65,10 @@ impl MessageRole { pub fn is_user(&self) -> bool { matches!(self, MessageRole::User) } + + pub fn is_assistant(&self) -> bool { + matches!(self, MessageRole::Assistant) + } } #[derive(Debug, Clone, Deserialize, Serialize)] |