Function createContentGenerator

packages/core/src/core/contentGenerator.ts:121–184

(
  config: ContentGeneratorConfig,
  gcConfig: Config,
  sessionId?: string,
)

}

export async function createContentGenerator(
  config: ContentGeneratorConfig,
  gcConfig: Config,
  sessionId?: string,
): Promise<ContentGenerator> {
  const version = process.env['CLI_VERSION'] || process.version;
  const userAgent = `AnusCLI/${version} (${process.platform}; ${process.arch})`;
  const baseHeaders: Record<string, string> = {
    'User-Agent': userAgent,
  };

  if (
    config.authType === AuthType.LOGIN_WITH_GOOGLE ||
    config.authType === AuthType.CLOUD_SHELL
  ) {
    const httpOptions = { headers: baseHeaders };
    return new LoggingContentGenerator(
      await createCodeAssistContentGenerator(
        httpOptions,
        config.authType,
        gcConfig,
        sessionId,
      ),
      gcConfig,
    );
  }

  // Provider registry: route to GrokContentGenerator if provider is GROK
  // or if a Grok-compatible key is present (implicit provider selection).
  // When createContentGeneratorConfig detects a Grok key, it sets provider=GROK and apiKey accordingly.
  const isGrokSelected =
    config.provider === Provider.GROK || config.authType === AuthType.USE_GROK;
  if (isGrokSelected) {
    const { GrokContentGenerator } = await import(
      '../providers/grok/contentGenerator.js'
    );
    return new LoggingContentGenerator(
      new GrokContentGenerator(config.model, config.apiKey),
      gcConfig,
    );
  }

  if (config.authType === AuthType.USE_VERTEX_AI) {
    let headers: Record<string, string> = { ...baseHeaders };
    if (gcConfig?.getUsageStatisticsEnabled()) {
      const installationId = getInstallationId();
      headers = {
        ...headers,
        'x-anus-api-privileged-user-id': `${installationId}`,
      };
    }
    const httpOptions = { headers };

    const googleGenAI = new GoogleGenAI({
      apiKey: config.apiKey === '' ? undefined : config.apiKey,
      vertexai: config.vertexai,
      httpOptions,
    });