diff options
Diffstat (limited to 'src/client/model.rs')
| -rw-r--r-- | src/client/model.rs | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/src/client/model.rs b/src/client/model.rs index af864e2..5d496d2 100644 --- a/src/client/model.rs +++ b/src/client/model.rs @@ -1,7 +1,7 @@ use super::{ list_chat_models, list_embedding_models, list_reranker_models, message::{Message, MessageContent, MessageContentPart}, - ToolResults, + MessageContentToolCalls, }; use crate::config::Config; @@ -237,10 +237,9 @@ impl Model { MessageContentPart::ImageUrl { .. } => 0, }) .sum(), - MessageContent::ToolResults(results) => { - let ToolResults { - tool_results, text, .. - } = results; + MessageContent::ToolCalls(MessageContentToolCalls { + tool_results, text, .. + }) => { estimate_token_length(text) + tool_results .iter() |