diff options
| author | sigoden <sigoden@gmail.com> | 2025-02-03 22:35:01 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-02-03 22:35:01 +0800 |
| commit | 700d8a3245f1133c37e035039e511a1e5dce1a5d (patch) | |
| tree | 7b1b2cf2e816adb0af399c9ca34051d3fcbef47f /src/client/message.rs | |
| parent | 07e58aaacd8c7f486a0d2a7c8d35a195f27d487e (diff) | |
| download | aichat-700d8a3245f1133c37e035039e511a1e5dce1a5d.tar.gz | |
feat: strip reasoning contents (#1141)
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)] |