diff options
| author | sigoden <sigoden@gmail.com> | 2023-11-01 22:48:49 +0800 |
|---|---|---|
| committer | sigoden <sigoden@gmail.com> | 2023-11-01 22:48:49 +0800 |
| commit | 0238c8734ecadfc6172fdded0fbcf83949b198fa (patch) | |
| tree | 707e3799eb18f197654a4fb77d21d52cea19cc9d /src/config/model_info.rs | |
| parent | f6da06dad9b2a76016209a7d58ad923c2c72f150 (diff) | |
| download | aichat-0238c8734ecadfc6172fdded0fbcf83949b198fa.tar.gz | |
chore: spellcheck typos
Diffstat (limited to 'src/config/model_info.rs')
| -rw-r--r-- | src/config/model_info.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/config/model_info.rs b/src/config/model_info.rs index fa51b91..c747d82 100644 --- a/src/config/model_info.rs +++ b/src/config/model_info.rs @@ -54,7 +54,7 @@ impl ModelInfo { messages.iter().map(|v| count_tokens(&v.content)).sum() } - pub fn totatl_tokens(&self, messages: &[Message]) -> usize { + pub fn total_tokens(&self, messages: &[Message]) -> usize { if messages.is_empty() { return 0; } @@ -68,7 +68,7 @@ impl ModelInfo { } pub fn max_tokens_limit(&self, messages: &[Message]) -> Result<()> { - let total_tokens = self.totatl_tokens(messages) + self.bias_tokens; + let total_tokens = self.total_tokens(messages) + self.bias_tokens; if let Some(max_tokens) = self.max_tokens { if total_tokens >= max_tokens { bail!("Exceed max tokens limit") |
