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/vertexai.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/vertexai.rs')
| -rw-r--r-- | src/client/vertexai.rs | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs index 452eb09..07ff2a4 100644 --- a/src/client/vertexai.rs +++ b/src/client/vertexai.rs @@ -342,8 +342,7 @@ pub fn gemini_build_chat_completions_body( .collect(); vec![json!({ "role": role, "parts": parts })] }, - MessageContent::ToolResults(results) => { - let tool_results = results.tool_results; + MessageContent::ToolCalls(MessageContentToolCalls { tool_results, .. }) => { let model_parts: Vec<Value> = tool_results.iter().map(|tool_result| { json!({ "functionCall": { |