Function updateAgentChatLastInjectedContextFromMessages

coderd/workspaceagents.go:2914–2947

updateAgentChatLastInjectedContextFromMessages rebuilds the injected-context cache from all persisted context-file and skill parts.

(
	ctx context.Context,
	logger slog.Logger,
	db database.Store,
	chatID uuid.UUID,
)

// updateAgentChatLastInjectedContextFromMessages rebuilds the
// injected-context cache from all persisted context-file and skill parts.
func updateAgentChatLastInjectedContextFromMessages(
	ctx context.Context,
	logger slog.Logger,
	db database.Store,
	chatID uuid.UUID,
) error {
	messages, err := db.GetChatMessagesByChatID(ctx, database.GetChatMessagesByChatIDParams{
		ChatID:  chatID,
		AfterID: 0,
	})
	if err != nil {
		return xerrors.Errorf("load context messages for injected context: %w", err)
	}

	sortChatMessagesByCreatedAtAndID(messages)

	parts, err := chatd.CollectContextPartsFromMessages(ctx, logger, messages, true)
	if err != nil {
		return xerrors.Errorf("collect injected context parts: %w", err)
	}
	parts = chatd.FilterContextPartsToLatestAgent(parts)

	param, err := chatd.BuildLastInjectedContext(parts)
	if err != nil {
		return xerrors.Errorf("update injected context: %w", err)
	}
	if _, err := db.UpdateChatLastInjectedContext(ctx, database.UpdateChatLastInjectedContextParams{
		ID:                  chatID,
		LastInjectedContext: param,
	}); err != nil {
		return xerrors.Errorf("update injected context: %w", err)
	}
	return nil
}

func messageHasPartTypes(raw []byte, types ...codersdk.ChatMessagePartType) bool {
	var parts []codersdk.ChatMessagePart