From 0238c8734ecadfc6172fdded0fbcf83949b198fa Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 1 Nov 2023 22:48:49 +0800
Subject: chore: spellcheck typos

---
 src/config/mod.rs        | 2 +-
 src/config/model_info.rs | 4 ++--
 src/config/session.rs    | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/config')

diff --git a/src/config/mod.rs b/src/config/mod.rs
index eff9bcf..958400c 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -582,7 +582,7 @@ impl Config {
     pub fn maybe_print_send_tokens(&self, input: &str) {
         if self.dry_run {
             if let Ok(messages) = self.build_messages(input) {
-                let tokens = self.model_info.totatl_tokens(&messages);
+                let tokens = self.model_info.total_tokens(&messages);
                 println!(">>> This message consumes {tokens} tokens. <<<");
             }
         }
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")
diff --git a/src/config/session.rs b/src/config/session.rs
index d07e4cf..62c1b09 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -69,7 +69,7 @@ impl Session {
     }
 
     pub fn tokens(&self) -> usize {
-        self.model_info.totatl_tokens(&self.messages)
+        self.model_info.total_tokens(&self.messages)
     }
 
     pub fn export(&self) -> Result<String> {
-- 
cgit v1.2.3