Method parseStreamChunk

pkg/provider/gemini.go:431–503

parseStreamChunk 解析单个流式 chunk

(chunk map[string]any)

// parseStreamChunk 解析单个流式 chunk
func (p *GeminiProvider) parseStreamChunk(chunk map[string]any) []StreamChunk {
	result := make([]StreamChunk, 0)

	// 获取 candidates
	candidates, ok := chunk["candidates"].([]any)
	if !ok || len(candidates) == 0 {
		return result
	}

	candidate := candidates[0].(map[string]any)

	// 获取 content
	content, ok := candidate["content"].(map[string]any)
	if !ok {
		// 检查是否完成
		if finishReason, ok := candidate["finishReason"].(string); ok {
			result = append(result, StreamChunk{
				Type:         string(ChunkTypeDone),
				FinishReason: strings.ToLower(finishReason),
			})
		}
		return result
	}

	// 获取 parts
	parts, ok := content["parts"].([]any)
	if !ok {
		return result
	}

	// 解析每个 part
	for _, partData := range parts {
		part := partData.(map[string]any)

		// 文本内容
		if text, ok := part["text"].(string); ok && text != "" {
			result = append(result, StreamChunk{
				Type:      string(ChunkTypeText),
				TextDelta: text,
				Delta:     text,
			})
		}

		// 函数调用
		if functionCall, ok := part["functionCall"].(map[string]any); ok {
			name := functionCall["name"].(string)
			args := functionCall["args"].(map[string]any)

			// 序列化参数
			argsJSON, _ := json.Marshal(args)

			result = append(result, StreamChunk{
				Type: string(ChunkTypeToolCall),
				ToolCall: &ToolCallDelta{
					Index:          0,
					Name:           name,
					ArgumentsDelta: string(argsJSON),
				},