diff options
Diffstat (limited to 'src/client/model.rs')
| -rw-r--r-- | src/client/model.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/client/model.rs b/src/client/model.rs index d0d86e4..1af3e59 100644 --- a/src/client/model.rs +++ b/src/client/model.rs @@ -1,6 +1,6 @@ use super::message::{Message, MessageContent}; -use crate::utils::{count_tokens, format_option_value}; +use crate::utils::{estimate_token_length, format_option_value}; use anyhow::{bail, Result}; use serde::Deserialize; @@ -181,7 +181,7 @@ impl Model { .iter() .map(|v| { match &v.content { - MessageContent::Text(text) => count_tokens(text), + MessageContent::Text(text) => estimate_token_length(text), MessageContent::Array(_) => 0, // TODO } }) |