diff options
| author | sigoden <sigoden@gmail.com> | 2023-12-24 16:04:18 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-24 16:04:18 +0800 |
| commit | 1c9ca1b002badb511ff21d58b3f20986741ab31f (patch) | |
| tree | 171edf24341f96ac4a93e00b315a3ce8b90455fc /src/config/session.rs | |
| parent | 89fefb4d1a3239496e90daf74be8b08decc603d2 (diff) | |
| download | aichat-1c9ca1b002badb511ff21d58b3f20986741ab31f.tar.gz | |
feat: custom REPL prompt (#283)
Diffstat (limited to 'src/config/session.rs')
| -rw-r--r-- | src/config/session.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/config/session.rs b/src/config/session.rs index e135c58..cbfe1c2 100644 --- a/src/config/session.rs +++ b/src/config/session.rs @@ -78,6 +78,10 @@ impl Session { self.model.total_tokens(&self.messages) } + pub fn user_messages_len(&self) -> usize { + self.messages.iter().filter(|v| v.role.is_user()).count() + } + pub fn export(&self) -> Result<String> { self.guard_save()?; let (tokens, percent) = self.tokens_and_percent(); |