diff options
| author | sigoden <sigoden@gmail.com> | 2024-11-14 08:14:32 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-11-14 08:14:32 +0800 |
| commit | 80684ec84047bda0299f8fd845b1c97f6fac5c99 (patch) | |
| tree | f37e377accd664d31a2bfe977521cfc2157a9cd1 /src/client/model.rs | |
| parent | cfa9217422dfa6cf10bed6e6e3fab0722f58588d (diff) | |
| download | aichat-80684ec84047bda0299f8fd845b1c97f6fac5c99.tar.gz | |
refactor: improve tool calls (#995)
- rename MessageContent:ToolResults to MessageContent:ToolCalls
- rename ToolResults to MessageContentToolCalls
- persist tool_calls to messages.md
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() |