TokenBudgetExtensions

NexusLabs.Needlr.AgentFramework.Workflows

NexusLabs.Needlr.AgentFramework.Workflows.Budget

TokenBudgetExtensions Class

Extension methods for wiring token-budget enforcement into the agent framework.

public static class TokenBudgetExtensions

Inheritance System.Object 🡒 TokenBudgetExtensions

Methods

TokenBudgetExtensions.UsingTokenBudget(this AgentFrameworkSyringe) Method

Wraps the configured Microsoft.Extensions.AI.IChatClient with TokenBudgetChatMiddleware, enabling per-pipeline token budgets via NexusLabs.Needlr.AgentFramework.Budget.ITokenBudgetTracker. Automatically includes UsingTokenTracking(this AgentFrameworkSyringe) for token recording.

public static NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe UsingTokenBudget(this NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe syringe);

Parameters

syringe NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe

Returns

NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe

TokenBudgetExtensions.UsingTokenTracking(this AgentFrameworkSyringe) Method

Wires TokenUsageRecordingMiddleware to record token usage from every LLM call into NexusLabs.Needlr.AgentFramework.Budget.ITokenBudgetTracker. This enables NexusLabs.Needlr.AgentFramework.Budget.ITokenBudgetTracker.CurrentTokens without enforcing any budget.

public static NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe UsingTokenTracking(this NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe syringe);

Parameters

syringe NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe

Returns

NexusLabs.Needlr.AgentFramework.AgentFrameworkSyringe

Remarks

Idempotent — calling this multiple times (or via both UsingTokenBudget() and UsingDiagnostics()) wires the recording middleware exactly once.