adolf/extensions/google/thinking.test.ts

// Google tests cover thinking plugin behavior.
import { describe, expect, it } from "vitest";
import {
  resolveGoogleGemini3ThinkingLevel,
  sanitizeGoogleThinkingPayload,
} from "./thinking-api.js";

describe("google thinking policy", () => {
  it.each([
    ["off", "LOW"],
    ["minimal", "LOW"],
    ["low", "LOW"],
    ["medium", "HIGH"],
    ["adaptive", undefined],
    ["high", "HIGH"],
    ["xhigh", "HIGH"],
  ] as const)("maps Gemini 3 Pro thinking level %s to %s", (thinkingLevel, expected) => {
    expect(
      resolveGoogleGemini3ThinkingLevel({
        modelId: "gemini-3.1-pro-preview",
        thinkingLevel,
      }),
    ).toBe(expected);
  });

  it.each([
    [0, "LOW"],
    [2048, "LOW"],
    [2049, "HIGH"],
  ] as const)("maps Gemini 3 Pro budget %s to %s", (thinkingBudget, expected) => {
    expect(
      resolveGoogleGemini3ThinkingLevel({
        modelId: "gemini-pro-latest",
        thinkingBudget,
      }),
    ).toBe(expected);
  });

  it.each([
    ["off", "MINIMAL"],
    ["minimal", "MINIMAL"],
    ["low", "LOW"],
    ["medium", "MEDIUM"],
    ["adaptive", undefined],
    ["high", "HIGH"],
    ["xhigh", "HIGH"],
  ] as const)("maps Gemini 3 Flash thinking level %s to %s", (thinkingLevel, expected) => {
    expect(
      resolveGoogleGemini3ThinkingLevel({
        modelId: "gemini-flash-latest",
        thinkingLevel,
      }),
    ).toBe(expected);
  });

  it.each([
    [-1, undefined],
    [0, "MINIMAL"],
    [2048, "LOW"],
    [8192, "MEDIUM"],
    [8193, "HIGH"],
  ] as const)("maps Gemini 3 Flash budget %s to %s", (thinkingBudget, expected) => {
    expect(
      resolveGoogleGemini3ThinkingLevel({
        modelId: "gemini-3.1-flash-lite",
        thinkingBudget,
      }),
    ).toBe(expected);
  });

  it("removes thinkingBudget=0 for Gemini 2.5 Pro", () => {
    const payload = {
      config: {
        thinkingConfig: { thinkingBudget: 0 },
      },
    };

    sanitizeGoogleThinkingPayload({ payload, modelId: "google/gemini-2.5-pro-preview" });

    expect(payload.config).not.toHaveProperty("thinkingConfig");
  });

  it("rewrites Gemini 3 thinking budgets to thinkingLevel", () => {
    const payload = {
      generationConfig: {
        thinkingConfig: { thinkingBudget: 8193, includeThoughts: true },
      },
    };

    sanitizeGoogleThinkingPayload({
      payload,
      modelId: "gemini-3-flash-preview",
      thinkingLevel: "medium",
    });

    expect(payload.generationConfig.thinkingConfig).toEqual({
      includeThoughts: true,
      thinkingLevel: "MEDIUM",
    });
  });

  it("keeps Gemini 3 adaptive thinking provider-dynamic instead of forcing a fixed level", () => {
    const payload = {
      generationConfig: {
        thinkingConfig: { thinkingBudget: 8192, includeThoughts: true },
      },
    };

    sanitizeGoogleThinkingPayload({
      payload,
      modelId: "gemini-3-flash-preview",
      thinkingLevel: "adaptive",
    });

    expect(payload.generationConfig.thinkingConfig).toEqual({
      includeThoughts: true,
    });
  });

  it("maps Gemini 2.5 adaptive thinking to dynamic thinkingBudget", () => {
    const payload = {
      config: {
        thinkingConfig: { thinkingBudget: 8192, includeThoughts: true },
      },
    };

    sanitizeGoogleThinkingPayload({
      payload,
      modelId: "gemini-2.5-flash",
      thinkingLevel: "adaptive",
    });

    expect(payload.config.thinkingConfig).toEqual({
      includeThoughts: true,
      thinkingBudget: -1,
    });
  });

  it("maps Gemma 4 thinking mode without sending thinkingBudget", () => {
    const payload = {
      config: {
        thinkingConfig: { thinkingBudget: 4096 },
      },
    };

    sanitizeGoogleThinkingPayload({
      payload,
      modelId: "gemma-4-26b-a4b-it",
      thinkingLevel: "high",
    });

    expect(payload.config.thinkingConfig).toEqual({ thinkingLevel: "HIGH" });
  });
});