diff --git a/tests/package/wp7-02-controlled-executor.test.mjs b/tests/package/wp7-02-controlled-executor.test.mjs index 3c6a8a7..5e67074 100644 --- a/tests/package/wp7-02-controlled-executor.test.mjs +++ b/tests/package/wp7-02-controlled-executor.test.mjs @@ -81,7 +81,11 @@ test("TDD-WP7-EXT-001 builds protocol-specific requests without auth in argument const gemini = buildProviderRequest({ modelConfig: models[0], prompt: "controlled fixture prompt", ratio: "3:4", reference }); assert.equal(gemini.method, "POST"); assert.equal(gemini.body.contents[0].parts.some((part) => part.inlineData?.data), true); - assert.equal(gemini.body.generationConfig.imageConfig.aspectRatio, "3:4"); + assert.deepEqual(gemini.body.generationConfig.responseModalities, ["IMAGE"]); + assert.deepEqual(gemini.body.generationConfig.responseFormat, { + image: { aspectRatio: "3:4", imageSize: "1K" }, + }); + assert.equal("imageConfig" in gemini.body.generationConfig, false); assert.equal("authorization" in gemini.headers, false); const openai = buildProviderRequest({ modelConfig: models[2], prompt: "controlled fixture prompt", ratio: "9:16" });