highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-11-10 22:37:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

tools: parse tool calls that don't conform to ("name": name, "arguments": args} (#12738)

Browse Source
This commit is contained in:
frob
2025-10-22 20:34:27 +02:00
committed by GitHub
parent e0ead1adee
commit 0334e67ffd
2 changed files with 39 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tools/tools.go
View File

@@ -125,7 +125,7 @@ func (p *Parser) parseToolCall() *api.ToolCall {
}
var args map[string]any
if found, i := findArguments(p.buffer); found == nil {
if found, i := findArguments(tool, p.buffer); found == nil {
return nil
} else {
args = found
@@ -219,7 +219,7 @@ func findTool(tools []api.Tool, buf []byte) (*api.Tool, int) {
// objects for functions that have all-optional parameters
// e.g. `{"name": "get_conditions", "arguments": {}}` will work but
// `{"name": "get_conditions"}` will not currently work
func findArguments(buffer []byte) (map[string]any, int) {
func findArguments(tool *api.Tool, buffer []byte) (map[string]any, int) {
if len(buffer) == 0 {
return nil, 0
}
@@ -269,27 +269,30 @@ func findArguments(buffer []byte) (map[string]any, int) {
var findObject func(obj map[string]any) (map[string]any, bool)
findObject = func(obj map[string]any) (map[string]any, bool) {
findMap := func(name string, obj map[string]any) (map[string]any, bool) {
if args, ok := obj[name].(map[string]any); ok {
return args, true
}
if argsStr, ok := obj[name].(string); ok {
var argsData map[string]interface{}
if err := json.Unmarshal([]byte(argsStr), &argsData); err == nil {
return argsData, ok
}
}
return nil, false
}
if _, hasName := obj["name"]; hasName {
if args, ok := obj["arguments"].(map[string]any); ok {
if args, ok := findMap("arguments", obj); ok {
return args, true
}
if argsStr, ok := obj["arguments"].(string); ok {
var argsData map[string]interface{}
if err := json.Unmarshal([]byte(argsStr), &argsData); err == nil {
return argsData, ok
}
}
if args, ok := obj["parameters"].(map[string]any); ok {
if args, ok := findMap("parameters", obj); ok {
return args, true
}
if argsStr, ok := obj["parameters"].(string); ok {
var argsData map[string]interface{}
if err := json.Unmarshal([]byte(argsStr), &argsData); err == nil {
return argsData, ok
}
}
return nil, true
}
if args, ok := findMap(tool.Function.Name, obj); ok {
return args, true
}
for _, v := range obj {
switch child := v.(type) {