Struct TaskScheduler 

pub struct TaskScheduler { /* private fields */ }

Shared actor handle. One instance belongs to each Agent.

Implementations

impl TaskScheduler

pub fn new(config: TaskSchedulerConfig) -> Result<Self, TaskSchedulerError>

Start a scheduler on the current Tokio runtime.

pub async fn acquire( &self, priority: TaskPriority, label: impl Into<String>, cancellation: &CancellationToken, ) -> Result<TaskLease, TaskSchedulerError>

Wait until this task owns one global execution slot.

pub async fn acquire_with_identity( &self, priority: TaskPriority, label: impl Into<String>, identity: Option<ExecutionIdentityV1>, cancellation: &CancellationToken, ) -> Result<TaskLease, TaskSchedulerError>

Wait until this task owns one global execution slot and carry its semantic execution identity through the admission boundary.

The identity is optional for backwards compatibility with callers that only need capacity. When present it is validated before anything is queued, and the resulting lease retains it for tracing and downstream adapters.

pub async fn acquire_quota( &self, priority: TaskPriority, label: impl Into<String>, quota: &TaskSchedulerQuota, cancellation: &CancellationToken, ) -> Result<TaskLease, TaskSchedulerError>

Wait for one or more quota dimensions without consuming another global execution slot.

This is used at leaf resource boundaries (for example, one model generation inside an already-admitted session run). The request still enters the same priority queue and actor as global admissions, so a provider limit cannot be bypassed with a local semaphore and a max-active=1 session does not deadlock while a nested model call waits.

pub async fn acquire_quotas( &self, priority: TaskPriority, label: impl Into<String>, quotas: &[TaskSchedulerQuota], cancellation: &CancellationToken, ) -> Result<TaskLease, TaskSchedulerError>

Multi-dimensional quota-only counterpart of Self::acquire_quota.

pub async fn acquire_with_quota( &self, priority: TaskPriority, label: impl Into<String>, quota: &TaskSchedulerQuota, identity: Option<ExecutionIdentityV1>, cancellation: &CancellationToken, ) -> Result<TaskLease, TaskSchedulerError>

Wait until this task owns a global execution slot subject to a capacity quota. The quota reservation is made in the same scheduler actor as global admission, so a caller cannot bypass it by creating a fresh local semaphore or executor handle.

pub async fn acquire_with_quotas( &self, priority: TaskPriority, label: impl Into<String>, quotas: &[TaskSchedulerQuota], identity: Option<ExecutionIdentityV1>, cancellation: &CancellationToken, ) -> Result<TaskLease, TaskSchedulerError>

Wait until this task owns a global execution slot subject to multiple immutable quota dimensions. All dimensions are evaluated by the same scheduler actor, so a caller cannot bypass one limit by splitting the request across independent local gates.

pub async fn quota_snapshot( &self, quota: &TaskSchedulerQuota, ) -> Result<TaskSchedulerQuotaSnapshot, TaskSchedulerError>

Return the live occupancy projection for one owner quota.

pub async fn quota_health( &self, quota: &TaskSchedulerQuota, ) -> Result<TaskSchedulerQuotaHealthSnapshot, TaskSchedulerError>

Return bounded cumulative health for one quota identity.

The scheduler keeps a fixed number of recent idle quota epochs so a host can inspect a completed provider generation without turning the actor into an unbounded metrics store. A descriptor that has never been admitted returns observed = false and zero counters.

pub async fn stats(&self) -> Result<TaskSchedulerStats, TaskSchedulerError>

Return a consistent actor-owned occupancy snapshot.

pub async fn health( &self, ) -> Result<TaskSchedulerHealthSnapshot, TaskSchedulerError>

Return occupancy plus bounded cumulative admission/fairness counters.

This is intentionally a separate method from Self::stats so the long-lived counters can be added without changing the established occupancy wire shape consumed by older SDKs.

pub async fn shutdown(&self)

Reject pending work and wait for already-admitted leases to finish.

Trait Implementations

impl Debug for TaskScheduler

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.