Function CollectContextPartsFromMessages

coderd/x/chatd/contextparts.go:50–78

CollectContextPartsFromMessages unmarshals chat message content and collects the context-file and skill parts it contains. When keepEmptyContextFiles is false, empty context-file parts are skipped. When it is true, empty context-file parts are included in the result.

(
	ctx context.Context,
	logger slog.Logger,
	messages []database.ChatMessage,
	keepEmptyContextFiles bool,
)

// keepEmptyContextFiles is false, empty context-file parts are skipped.
// When it is true, empty context-file parts are included in the result.
func CollectContextPartsFromMessages(
	ctx context.Context,
	logger slog.Logger,
	messages []database.ChatMessage,
	keepEmptyContextFiles bool,
) ([]codersdk.ChatMessagePart, error) {
	var collected []codersdk.ChatMessagePart
	for _, msg := range messages {
		if !msg.Content.Valid {
			continue
		}

		var parts []codersdk.ChatMessagePart
		if err := json.Unmarshal(msg.Content.RawMessage, &parts); err != nil {
			logger.Warn(ctx, "skipping malformed chat context message",
				slog.F("chat_message_id", msg.ID),
				slog.Error(err),
			)
			continue
		}

		collected = append(
			collected,
			FilterContextParts(parts, keepEmptyContextFiles)...,
		)
	}

	return collected, nil
}

func latestContextAgentIDFromParts(parts []codersdk.ChatMessagePart) (uuid.UUID, bool) {
	var lastID uuid.UUID