import {
invocationParametersToObject,
objectToInvocationParameters,
} from "../fetchPlaygroundPrompt";
type InvocationParameterInput = Parameters<
typeof invocationParametersToObject
>[0][number];
type SupportedParamsType = Parameters<typeof objectToInvocationParameters>[1];
describe("objectToInvocationParameters", () => {
it("should convert simple parameters with matching definitions", () => {
const params = {
temperature: 0.7,
max_tokens: 100,
};
const supportedParams = [
{
__typename: "FloatInvocationParameter",
invocationName: "temperature",
canonicalName: "TEMPERATURE",
invocationInputField: "value_float",
},
{
__typename: "IntInvocationParameter",
invocationName: "max_tokens",
canonicalName: "MAX_COMPLETION_TOKENS",
invocationInputField: "value_int",
},
] satisfies SupportedParamsType;
const result = objectToInvocationParameters(params, supportedParams);
expect(result).toEqual([
{
invocationName: "temperature",
canonicalName: "TEMPERATURE",
valueFloat: 0.7,
},
{
invocationName: "max_tokens",
canonicalName: "MAX_COMPLETION_TOKENS",
valueInt: 100,
},
]);
});
it("should handle parameters without matching definitions", () => {
const params = {
unknownParam: "test",
maxTokens: 100,
};
const supportedParams = [] satisfies SupportedParamsType;
const result = objectToInvocationParameters(params, supportedParams);
expect(result).toEqual([
{
invocationName: "unknownParam",
valueJson: "test",
},
{
invocationName: "maxTokens",
valueJson: 100,
},
]);
});
});
describe("invocationParametersToObject", () => {
it("should convert invocation parameters back to object format", () => {
const params = [
{
invocationName: "temperature",
canonicalName: "TEMPERATURE",
valueFloat: 0.7,
},
{
invocationName: "max_tokens",
canonicalName: "MAX_COMPLETION_TOKENS",
valueInt: 100,
},
] satisfies InvocationParameterInput[];
const supportedParams = [
{
__typename: "FloatInvocationParameter",
invocationName: "temperature",
canonicalName: "TEMPERATURE",
invocationInputField: "value_float",
},
{
__typename: "IntInvocationParameter",
invocationName: "max_tokens",
canonicalName: "MAX_COMPLETION_TOKENS",
invocationInputField: "value_int",
},
] satisfies SupportedParamsType;
const result = invocationParametersToObject(params, supportedParams);
expect(result).toEqual({
temperature: 0.7,
max_tokens: 100,
});
});
it("should handle empty parameters", () => {
const result = invocationParametersToObject([], []);
expect(result).toEqual({});
});
it("should preserve parameters without matching definitions", () => {
const params = [
{
invocationName: "reasoning_effort",
canonicalName: "REASONING_EFFORT",
valueString: "minimal",
},
{
invocationName: "temperature",
canonicalName: "TEMPERATURE",
valueFloat: 0.7,
},
] satisfies InvocationParameterInput[];
const supportedParams = [
{
__typename: "FloatInvocationParameter",
invocationName: "temperature",
canonicalName: "TEMPERATURE",
invocationInputField: "value_float",
},
// Note: reasoning_effort is intentionally not in supported params
// to simulate the case where supported params haven't been fetched yet
] satisfies SupportedParamsType;
const result = invocationParametersToObject(params, supportedParams);
expect(result).toEqual({
temperature: 0.7,
reasoning_effort: "minimal", // Should be preserved even without definition
});
});
it("should preserve valueBoolean parameters without matching definitions", () => {
const params: InvocationParameterInput[] = [
{
invocationName: "stream",
valueBoolean: true,
},
{
invocationName: "legacy_flag",
valueBool: false,
},
{
invocationName: "temperature",
canonicalName: "TEMPERATURE",
valueFloat: 0.5,
},
];
const supportedParams = [
{
__typename: "FloatInvocationParameter",
invocationName: "temperature",
canonicalName: "TEMPERATURE",
invocationInputField: "value_float",
},
// Note: stream and legacy_flag are intentionally not in supported params
// to test that both valueBool and valueBoolean are preserved
] satisfies SupportedParamsType;
const result = invocationParametersToObject(params, supportedParams);
expect(result).toEqual({
temperature: 0.5,
stream: true, // valueBoolean should be preserved
legacy_flag: false, // valueBool should be preserved
});
});
});
