package toolbox

import (
	"context"
	"time"

	"github.com/tmc/langchaingo/llms"
)

type Context interface {
	context.Context

	WithCancel() (Context, func())
	WithTimeout(time.Duration) (Context, func())
	WithMessages([]llms.MessageContent) Context
	GetMessages() []llms.MessageContent
}