From 80684ec84047bda0299f8fd845b1c97f6fac5c99 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 14 Nov 2024 08:14:32 +0800
Subject: refactor: improve tool calls (#995)

- rename MessageContent:ToolResults to MessageContent:ToolCalls
- rename ToolResults to MessageContentToolCalls
- persist tool_calls to messages.md
---
 src/client/model.rs | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

(limited to 'src/client/model.rs')

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()
-- 
cgit v1.2.3