agent

type AssignmentHook func(next func(*livekit.JobAssignment) error) func(*livekit.JobAssignment) error

type Client interface {
	// LaunchJob starts a room or participant job on an agent.
	// it will launch a job once for each worker in each namespace
	LaunchJob(ctx context.Context, desc *JobRequest) *serverutils.IncrementalDispatcher[*livekit.Job]
	TerminateJob(ctx context.Context, jobID string, reason rpc.JobTerminateReason) (*livekit.JobState, error)
	Stop() error
}

type Config struct {
	EnableUserDataRecording bool    `yaml:"enable_user_data_recording"`
	EnableUserDataRedaction bool    `yaml:"enable_user_data_redaction"`
	TargetLoad              float32 `yaml:"target_load,omitempty"`

	// agent HTTP endpoints data plane; zero values take the endpoint package
	// defaults
	Endpoints EndpointsConfig `yaml:"endpoints,omitempty"`
}

type EndpointsConfig struct {
	// Disabled turns off the /agents/{agent_name}/{deployment}/... front and
	// rejects registrations that declare endpoints.
	Disabled bool `yaml:"disabled,omitempty"`

	// MaxStreams is the soft per-session concurrent-stream cap used for capacity
	// weighting; 0 takes the endpoint package default.
	MaxStreams uint32 `yaml:"max_streams,omitempty"`

	// MaxTunnels caps one worker's concurrent WebSocket tunnels, reserving the
	// rest of its streams for HTTP routes; 0 takes half of MaxStreams.
	MaxTunnels uint32 `yaml:"max_tunnels,omitempty"`

	// TunnelDrainTimeout is how long a WebSocket tunnel outlives its worker or
	// this node starting to drain; 0 takes the endpoint package default.
	TunnelDrainTimeout time.Duration `yaml:"tunnel_drain_timeout,omitempty"`
}

type JobRequest struct {
	DispatchId string
	JobType    livekit.JobType
	Room       *livekit.Room
	// only set for participant jobs
	Participant *livekit.ParticipantInfo
	Metadata    string
	AgentName   string
	Deployment  string
	Attributes  map[string]string
}

type SignalConn interface {
	WriteServerMessage(msg *livekit.ServerMessage) (int, error)
	ReadWorkerMessage() (*livekit.WorkerMessage, int, error)
	SetReadDeadline(time.Time) error
	Close() error
	CloseWithReason(reason string) error
}

type UnimplementedWorkerSignalHandler struct{}

type Worker struct {
	WorkerPingHandler
	WorkerRegistration
	// contains filtered or unexported fields
}

type WorkerPingHandler struct {
	UnimplementedWorkerSignalHandler
	// contains filtered or unexported fields
}

type WorkerProtocolVersion int

type WorkerRegisterHandler func(req *livekit.RegisterWorkerRequest, res *livekit.RegisterWorkerResponse, reg *WorkerRegistration) error

type WorkerRegisterer struct {
	WorkerPingHandler
	// contains filtered or unexported fields
}

type WorkerRegistration struct {
	Protocol    WorkerProtocolVersion
	ID          string
	Version     string
	AgentID     string
	AgentName   string
	Namespace   string
	JobType     livekit.JobType
	Permissions *livekit.ParticipantPermission
	ClientIP    string
	Deployment  string

	// agent HTTP endpoints data plane
	Endpoints        []*livekit.AgentHttp_AgentEndpoint
	InstanceID       string
	EndpointSettings *livekit.AgentHttp_AgentEndpointSettings

	// KindDetails, when set by the server, are stamped onto the participant
	// join token minted for every job assigned to this worker (e.g. marking
	// hosted/cloud agents so they can be distinguished from self-hosted ones in
	// observability). This is server-controlled and never populated from the
	// worker's register request.
	KindDetails []livekit.ParticipantInfo_KindDetail
}

type WorkerSignalHandler interface {
	HandleRegister(*livekit.RegisterWorkerRequest) error
	HandleAvailability(*livekit.AvailabilityResponse) error
	HandleUpdateJob(*livekit.UpdateJobStatus) error
	HandleSimulateJob(*livekit.SimulateJobRequest) error
	HandlePing(*livekit.WorkerPing) error
	HandleUpdateWorker(*livekit.UpdateWorkerStatus) error
	HandleMigrateJob(*livekit.MigrateJobRequest) error
}