summaryrefslogtreecommitdiffstats
path: root/src/client/ernie.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-11-14 08:14:32 +0800
committerGitHub <noreply@github.com>2024-11-14 08:14:32 +0800
commit80684ec84047bda0299f8fd845b1c97f6fac5c99 (patch)
treef37e377accd664d31a2bfe977521cfc2157a9cd1 /src/client/ernie.rs
parentcfa9217422dfa6cf10bed6e6e3fab0722f58588d (diff)
downloadaichat-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/ernie.rs')
-rw-r--r--src/client/ernie.rs6
1 files changed, 4 insertions, 2 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs
index 0c325e5..7e1f8e7 100644
--- a/src/client/ernie.rs
+++ b/src/client/ernie.rs
@@ -231,9 +231,11 @@ fn build_chat_completions_body(data: ChatCompletionsData, model: &Model) -> Valu
.flat_map(|message| {
let Message { role, content } = message;
match content {
- MessageContent::ToolResults(results) => {
+ MessageContent::ToolCalls(MessageContentToolCalls {
+ tool_results, ..
+ }) => {
let mut list = vec![];
- for tool_result in results.tool_results {
+ for tool_result in tool_results {
list.push(json!({
"role": "assistant",
"content": format!("Action: {}\nAction Input: {}", tool_result.call.name, tool_result.call.arguments)