Function RecoverFromText

internal/toolcall/recover.go:61–98

RecoverFromText 兜底:从已输出的纯文本里扫描 JSON 对象,抽出 (a) {"name":..., "arguments":...} 或 (b) {"cmd": [...]} / {"cmd": "..."} 这两种 chatgpt 沙箱形状 然后转成 ToolCall。返回的列表已去重。

(text string, tools []official.Tool)

// (b) {"cmd": [...]} / {"cmd": "..."} 这两种 chatgpt 沙箱形状
// 然后转成 ToolCall。返回的列表已去重。
func RecoverFromText(text string, tools []official.Tool) []official.ToolCall {
	if !strings.Contains(text, "{") {
		return nil
	}
	shellName, shellParam := ResolveShellTool(tools)
	seen := make(map[string]bool)
	var out []official.ToolCall
	for _, obj := range iterJSONObjects(text) {
		var tc *official.ToolCall
		if name := pickString(obj, "name", "tool", "tool_name", "function"); name != "" {
			tc = buildToolCallFromObject(obj)
		} else if shellName != "" {
			if cmd, ok := obj["cmd"]; ok {
				if s := cmdToString(cmd); s != "" {
					raw, _ := json.Marshal(map[string]string{shellParam: s})
					tc = &official.ToolCall{
						ID:   generateCallID(),
						Type: "function",
						Function: official.ToolCallFunc{
							Name:      shellName,
							Arguments: string(raw),
						},
					}
				}
			}
		}
		if tc == nil {
			continue
		}
		key := tc.Function.Name + "\x00" + tc.Function.Arguments
		if seen[key] {
			continue
		}
		seen[key] = true
		out = append(out, *tc)
	}
	return out
}

// iterJSONObjects 扫描 text 中所有花括号平衡的 JSON 对象。
// 按出现顺序返回(只对顶层对象计数,字符串里的 {} 不算)。