Class PackAgent

src/runtime/agent.ts:420–982

// ---------------------------------------------------------------------------

export class PackAgent implements IPackAgent {
  private options: PackAgentOptions;
  private channels = new Map<string, ChannelSession>();
  private pendingSessionCreations = new Map<string, Promise<ChannelSession>>();
  private pendingAbortChannels = new Set<string>();
  private schedulerRef: { current: SchedulerAdapter | null } = {
    current: null,
  };
  private authStorage: AuthStorage;
  private readonly delegatedCustomToolClient = new DelegatedCustomToolClient();
  private readonly hostIpcClient = new HostIpcClient();

  constructor(options: PackAgentOptions) {
    this.options = options;

    // Use ConfigFileAuthBackend to persist OAuth credentials in config.json._auth
    const configPath = path.resolve(options.rootDir, "data", "config.json");
    const backend = new ConfigFileAuthBackend(configPath);
    this.authStorage = AuthStorage.fromStorage(backend);

    // For API Key providers, set runtime key (not persisted to _auth)
    const providerMeta = SUPPORTED_PROVIDERS[options.provider];
    if (providerMeta?.authType === "api_key" && options.apiKey) {
      this.authStorage.setRuntimeApiKey(options.provider, options.apiKey);
    }
  }

  /** Get the shared AuthStorage instance (used by OAuth API endpoints) */
  getAuthStorage(): AuthStorage {
    return this.authStorage;
  }

  /** Update runtime auth when provider/apiKey changes */
  updateAuth(provider: string, apiKey?: string): void {
    // Remove old runtime key
    this.authStorage.removeRuntimeApiKey(this.options.provider);
    this.options.provider = provider;
    this.options.apiKey = apiKey ?? "";
    if (apiKey) {
      this.authStorage.setRuntimeApiKey(provider, apiKey);
    }
  }

  /**
   * Inject scheduler reference (called by server.ts after adapter init).
   */
  setScheduler(scheduler: SchedulerAdapter): void {
    this.schedulerRef.current = scheduler;
  }

  private async createCustomTools(
    adapter: RuntimePlatform,
    channelId: string,
    fileOutputCallbackRef: { current: FileOutputCallback | null },
    delegatedToolRunContextRef: DelegatedToolRunContextRef,
  ): Promise<any[]> {
    const delegatedDefinitions =
      await this.delegatedCustomToolClient.listDefinitions();