remove unneeded tool calls

This commit is contained in:
Michael Yang 2024-07-16 13:48:38 -07:00
parent a8388beb94
commit 5a83f79afd
3 changed files with 1 additions and 15 deletions

View file

@ -115,8 +115,6 @@ type Message struct {
}
type ToolCall struct {
ID string `json:"id"`
Type string `json:"type"`
Function struct {
Name string `json:"name"`
Arguments map[string]any `json:"arguments"`

View file

@ -16,7 +16,6 @@ import (
"strings"
"text/template/parse"
"github.com/google/uuid"
"github.com/ollama/ollama/api"
"github.com/ollama/ollama/convert"
"github.com/ollama/ollama/llm"
@ -363,11 +362,7 @@ func (m *Model) parseToolCalls(s string) ([]api.ToolCall, bool) {
var toolCalls []api.ToolCall
for _, kv := range objs {
call := api.ToolCall{
ID: uuid.New().String(),
Type: "function",
}
var call api.ToolCall
for k, v := range kv {
switch k {
case name:

View file

@ -181,7 +181,6 @@ The temperature in San Francisco, CA is 70°F and in Toronto, Canada is 20°C.`,
calls := []api.ToolCall{
{
Type: "function",
Function: function{
Name: "get_current_weather",
Arguments: map[string]any{
@ -191,7 +190,6 @@ The temperature in San Francisco, CA is 70°F and in Toronto, Canada is 20°C.`,
},
},
{
Type: "function",
Function: function{
Name: "get_current_weather",
Arguments: map[string]any{
@ -228,11 +226,6 @@ The temperature in San Francisco, CA is 70°F and in Toronto, Canada is 20°C.`,
}
if tt.ok {
for i := range actual {
// ID is randomly generated so clear it for comparison
actual[i].ID = ""
}
if diff := cmp.Diff(actual, calls); diff != "" {
t.Errorf("mismatch (-got +want):\n%s", diff)
}