diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/client/common.rs | 4 | ||||
| -rw-r--r-- | src/config/mod.rs | 2 | ||||
| -rw-r--r-- | src/config/model_info.rs | 4 | ||||
| -rw-r--r-- | src/config/session.rs | 2 |
4 files changed, 6 insertions, 6 deletions
diff --git a/src/client/common.rs b/src/client/common.rs index c77f89b..7963c76 100644 --- a/src/client/common.rs +++ b/src/client/common.rs @@ -197,7 +197,7 @@ pub trait Client { let data = global_config.read().prepare_send_data(content, false)?; self.send_message_inner(&client, data) .await - .with_context(|| "Failed to get awswer") + .with_context(|| "Failed to get answer") }) } @@ -233,7 +233,7 @@ pub trait Client { self.send_message_streaming_inner(&client, handler, data).await } => { handler.done()?; - ret.with_context(|| "Failed to get awswer") + ret.with_context(|| "Failed to get answer") } _ = watch_abort(abort.clone()) => { handler.done()?; 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> { |