From 5e172b3888df0991404e9b65eea86feb9b4b2760 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Thu, 23 Apr 2026 02:16:29 +0100
Subject: [PATCH] fix(qa): preserve image parity plugin allowlist

---
 extensions/qa-lab/src/providers/image-generation.test.ts | 1 -
 extensions/qa-lab/src/suite-runtime-agent-media.test.ts  | 1 -
 qa/scenarios/config/config-restart-capability-flip.md    | 6 +++++-
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/extensions/qa-lab/src/providers/image-generation.test.ts b/extensions/qa-lab/src/providers/image-generation.test.ts
index 928aa76ed68..a775e26713a 100644
--- a/extensions/qa-lab/src/providers/image-generation.test.ts
+++ b/extensions/qa-lab/src/providers/image-generation.test.ts
@@ -30,7 +30,6 @@ describe("QA provider image generation config", () => {
       "qa-channel",
     ]);
   });
-
   it("uses the selected mock provider for AIMock image generation", () => {
     const patch = buildQaImageGenerationConfigPatch({
       providerMode: "aimock",
diff --git a/extensions/qa-lab/src/suite-runtime-agent-media.test.ts b/extensions/qa-lab/src/suite-runtime-agent-media.test.ts
index da1e5fa6879..31789cdc963 100644
--- a/extensions/qa-lab/src/suite-runtime-agent-media.test.ts
+++ b/extensions/qa-lab/src/suite-runtime-agent-media.test.ts
@@ -108,7 +108,6 @@ describe("qa suite runtime agent media helpers", () => {
     expect(waitForGatewayHealthyMock).toHaveBeenCalled();
     expect(waitForTransportReadyMock).toHaveBeenCalledWith(expect.anything(), 60_000);
   });
-
   it("preserves plugins already allowed by the gateway when configuring media", async () => {
     readConfigSnapshotMock.mockResolvedValue({
       hash: "hash",
diff --git a/qa/scenarios/config/config-restart-capability-flip.md b/qa/scenarios/config/config-restart-capability-flip.md
index 0b180b5955a..ea2a47a4935 100644
--- a/qa/scenarios/config/config-restart-capability-flip.md
+++ b/qa/scenarios/config/config-restart-capability-flip.md
@@ -50,6 +50,9 @@ steps:
       - set: originalToolsDeny
         value:
           expr: "originalTools ? (Object.prototype.hasOwnProperty.call(originalTools, 'deny') ? structuredClone(originalTools.deny) : undefined) : undefined"
+      - set: originalImageGenerationModelPrimary
+        value:
+          expr: "original.config.agents?.defaults?.imageGenerationModel?.primary ?? null"
       - set: denied
         value:
           expr: "Array.isArray(originalToolsDeny) ? originalToolsDeny.map((entry) => String(entry)) : []"
@@ -112,7 +115,8 @@ steps:
                     agents:
                       defaults:
                         imageGenerationModel:
-                          primary: openai/gpt-image-1
+                          primary:
+                            ref: originalImageGenerationModelPrimary
                   sessionKey:
                     ref: sessionKey
                   note: