63 lines
1.3 KiB
TypeScript
63 lines
1.3 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { resolveExtraParams } from "./pi-embedded-runner.js";
|
|
|
|
describe("resolveExtraParams", () => {
|
|
it("returns undefined with no model config", () => {
|
|
const result = resolveExtraParams({
|
|
cfg: undefined,
|
|
provider: "zai",
|
|
modelId: "glm-4.7",
|
|
});
|
|
|
|
expect(result).toBeUndefined();
|
|
});
|
|
|
|
it("returns params for exact provider/model key", () => {
|
|
const result = resolveExtraParams({
|
|
cfg: {
|
|
agents: {
|
|
defaults: {
|
|
models: {
|
|
"openai/gpt-4": {
|
|
params: {
|
|
temperature: 0.7,
|
|
maxTokens: 2048,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
provider: "openai",
|
|
modelId: "gpt-4",
|
|
});
|
|
|
|
expect(result).toEqual({
|
|
temperature: 0.7,
|
|
maxTokens: 2048,
|
|
});
|
|
});
|
|
|
|
it("ignores unrelated model entries", () => {
|
|
const result = resolveExtraParams({
|
|
cfg: {
|
|
agents: {
|
|
defaults: {
|
|
models: {
|
|
"openai/gpt-4": {
|
|
params: {
|
|
temperature: 0.7,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
provider: "openai",
|
|
modelId: "gpt-4.1-mini",
|
|
});
|
|
|
|
expect(result).toBeUndefined();
|
|
});
|
|
});
|