Function CombineConsecutiveSameRoleMessages

pkg/aiusechat/usechat-utils.go:13–64

CombineConsecutiveSameRoleMessages combines consecutive UIMessages with the same role by appending their Parts together. This is useful for APIs like OpenAI that may split assistant messages into separate messages (e.g., one for text and one for tool calls).

(uiChat *uctypes.UIChat)

// by appending their Parts together. This is useful for APIs like OpenAI that may split
// assistant messages into separate messages (e.g., one for text and one for tool calls).
func CombineConsecutiveSameRoleMessages(uiChat *uctypes.UIChat) *uctypes.UIChat {
	if uiChat == nil || len(uiChat.Messages) == 0 {
		return uiChat
	}

	combined := make([]uctypes.UIMessage, 0, len(uiChat.Messages))
	var current *uctypes.UIMessage

	for i := range uiChat.Messages {
		msg := &uiChat.Messages[i]

		if current == nil {
			// First message - start a new combined message
			current = &uctypes.UIMessage{
				ID:       msg.ID,
				Role:     msg.Role,
				Metadata: msg.Metadata,
				Parts:    make([]uctypes.UIMessagePart, len(msg.Parts)),
			}
			copy(current.Parts, msg.Parts)
			continue
		}

		if current.Role == msg.Role {
			// Same role - append parts to current message
			current.Parts = append(current.Parts, msg.Parts...)
		} else {
			// Different role - save current and start new
			combined = append(combined, *current)
			current = &uctypes.UIMessage{
				ID:       msg.ID,
				Role:     msg.Role,
				Metadata: msg.Metadata,
				Parts:    make([]uctypes.UIMessagePart, len(msg.Parts)),
			}
			copy(current.Parts, msg.Parts)
		}
	}

	// Don't forget the last message
	if current != nil {
		combined = append(combined, *current)
	}

	return &uctypes.UIChat{
		ChatId:     uiChat.ChatId,
		APIType:    uiChat.APIType,
		Model:      uiChat.Model,
		APIVersion: uiChat.APIVersion,
		Messages:   combined,
	}
}


// ConvertAIChatToUIChat converts an AIChat to a UIChat by routing to the appropriate