gemini-cli/packages/core/src/context/ir/mapper.ts

/**
 * @license
 * Copyright 2026 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

import type { Content } from '@google/genai';
import type { Episode, ConcreteNode } from './types.js';
import { toIr } from './toIr.js';
import { fromIr } from './fromIr.js';
import type { ContextTokenCalculator } from '../utils/contextTokenCalculator.js';

export class IrMapper {
  /**
   * Translates a flat Gemini Content[] array into our rich Episodic Intermediate Representation.
   * Groups adjacent function calls and responses into unified ToolExecution nodes.
   */
  static toIr(
    history: readonly Content[],
    tokenCalculator: ContextTokenCalculator,
  ): Episode[] {
    return toIr(history, tokenCalculator);
  }

  /**
   * Re-serializes a flat array of ConcreteNodes back into a flat Gemini Content[] array.
   */
  static fromIr(ship: readonly ConcreteNode[]): Content[] {
    return fromIr(ship);
  }
}