packages/agent-core-v2/src/features/usage/usageFeature.ts

import { LifecycleScope } from '#/app/scopes';
import { Feature } from '#/features/feature';
import { registerFeature } from '#/features/featureRegistry';
import { ISessionUsageService } from '#/session/usage/sessionUsage';
import { SessionUsageService } from '#/session/usage/sessionUsageService';
import { UsageAgentModelDefinition } from '#/session/usage/usageAgentModel';

export class UsageFeature extends Feature {
  static override readonly name = 'usage';

  constructor() {
    super();
    this.contributeAgentModel(UsageAgentModelDefinition);
    this.contributeService(LifecycleScope.Session, ISessionUsageService, SessionUsageService);
  }
}

registerFeature(UsageFeature);