Method Flush

internal/toolcall/parser.go:106–137

Flush 在流结束时调用,把 buffer 中剩余的、未闭合的 也尝试解析一次。 典型场景:模型在末尾输出了 {...} 但忘了写结束标签。

()

// Flush 在流结束时调用,把 buffer 中剩余的、未闭合的 <tool_call> 也尝试解析一次。
// 典型场景:模型在末尾输出了 <tool_call>{...} 但忘了写结束标签。
func (p *Parser) Flush() (textDelta string, toolCalls []official.ToolCall) {
	remaining := p.buffer
	p.buffer = ""
	if remaining == "" {
		return "", nil
	}
	if p.inside {
		// 还在工具块里 —— 尝试解析剩余内容
		if tc := buildToolCallFromRaw(remaining); tc != nil {
			toolCalls = append(toolCalls, *tc)
			p.emittedCount++
			return "", toolCalls
		}
		// 解析不出来 —— 把 <tool_call> 标签也作为文本回吐,避免吞掉
		if p.emittedCount == 0 {
			return StartTag + remaining, nil
		}
		return "", nil
	}
	// 不在工具块里 —— 看剩余文本是否整体就是一个 JSON
	if p.emittedCount == 0 {
		if tc := buildToolCallFromRaw(remaining); tc != nil {
			toolCalls = append(toolCalls, *tc)
			p.emittedCount++
			return "", toolCalls
		}
		if !p.emittedText {
			return remaining, nil
		}
	}
	return "", nil
}

// normalize 把模型常见的标签变体(<tool_calls>、<tool call> 等)统一为
// <tool_call> / </tool_call>。