Function clearAgentChatContext

coderd/workspaceagents.go:2791–2864

(
	ctx context.Context,
	db database.Store,
	chatID uuid.UUID,
	agentID uuid.UUID,
	workspaceOwnerID uuid.UUID,
)

}

func clearAgentChatContext(
	ctx context.Context,
	db database.Store,
	chatID uuid.UUID,
	agentID uuid.UUID,
	workspaceOwnerID uuid.UUID,
) error {
	return db.InTx(func(tx database.Store) error {
		locked, err := tx.GetChatByIDForUpdate(ctx, chatID)
		if err != nil {
			return xerrors.Errorf("lock chat: %w", err)
		}
		if !isActiveAgentChat(locked) {
			return errChatNotActive
		}
		if !locked.AgentID.Valid || locked.AgentID.UUID != agentID {
			return errChatDoesNotBelongToAgent
		}
		if locked.OwnerID != workspaceOwnerID {
			return errChatDoesNotBelongToWorkspaceOwner
		}
		messages, err := tx.GetChatMessagesByChatID(ctx, database.GetChatMessagesByChatIDParams{
			ChatID:  chatID,
			AfterID: 0,
		})
		if err != nil {
			return xerrors.Errorf("get chat messages: %w", err)
		}
		hadInjectedContext := locked.LastInjectedContext.Valid
		var skillOnlyMessageIDs []int64
		for _, msg := range messages {
			if !msg.Content.Valid {
				continue
			}
			hasContextFile := messageHasPartTypes(msg.Content.RawMessage, codersdk.ChatMessagePartTypeContextFile)
			hasSkill := messageHasPartTypes(msg.Content.RawMessage, codersdk.ChatMessagePartTypeSkill)
			if hasContextFile || hasSkill {
				hadInjectedContext = true
			}
			if hasSkill && !hasContextFile {
				skillOnlyMessageIDs = append(skillOnlyMessageIDs, msg.ID)
			}
		}
		if !hadInjectedContext {
			return nil
		}
		if err := tx.SoftDeleteContextFileMessages(ctx, chatID); err != nil {
			return xerrors.Errorf("soft delete context-file messages: %w", err)
		}
		for _, messageID := range skillOnlyMessageIDs {
			if err := tx.SoftDeleteChatMessageByID(ctx, messageID); err != nil {
				return xerrors.Errorf("soft delete context message %d: %w", messageID, err)
			}
		}
		// Reset provider-side Responses chaining so the next turn replays
		// the post-clear history instead of inheriting cleared context.
		if err := tx.ClearChatMessageProviderResponseIDsByChatID(ctx, chatID); err != nil {
			return xerrors.Errorf("clear provider response chain: %w", err)