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/ernie.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/ernie.rs')
| -rw-r--r-- | src/client/ernie.rs | 6 |
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) |
