mirror of https://github.com/openclaw/openclaw.git
39 lines
1.3 KiB
TypeScript
39 lines
1.3 KiB
TypeScript
import type { Model } from "@mariozechner/pi-ai";
|
|
import { describe, expect, it, vi } from "vitest";
|
|
import { createPiAiStreamSimpleMock } from "./extra-params.pi-ai-mock.js";
|
|
import { runExtraParamsCase } from "./extra-params.test-support.js";
|
|
|
|
vi.mock("@mariozechner/pi-ai", async (importOriginal) =>
|
|
createPiAiStreamSimpleMock(() => importOriginal<typeof import("@mariozechner/pi-ai")>()),
|
|
);
|
|
|
|
describe("extra-params: Google thinking payload compatibility", () => {
|
|
it("strips negative thinking budgets and fills Gemini 3.1 thinkingLevel", () => {
|
|
const payload = runExtraParamsCase({
|
|
applyProvider: "google",
|
|
applyModelId: "gemini-3.1-pro-preview",
|
|
model: {
|
|
api: "google-generative-ai",
|
|
provider: "google",
|
|
id: "gemini-3.1-pro-preview",
|
|
} as unknown as Model<"openai-completions">,
|
|
thinkingLevel: "high",
|
|
payload: {
|
|
contents: [],
|
|
config: {
|
|
thinkingConfig: {
|
|
thinkingBudget: -1,
|
|
},
|
|
},
|
|
},
|
|
}).payload as {
|
|
config?: {
|
|
thinkingConfig?: Record<string, unknown>;
|
|
};
|
|
};
|
|
|
|
expect(payload.config?.thinkingConfig?.thinkingBudget).toBeUndefined();
|
|
expect(payload.config?.thinkingConfig?.thinkingLevel).toBe("HIGH");
|
|
});
|
|
});
|