From 667ce4db06ff8b1c10142f2a7dfe4533fe0064d3 Mon Sep 17 00:00:00 2001
From: wauputr4 <103489788+wauputr4@users.noreply.github.com>
Date: Thu, 7 May 2026 01:20:09 +0700
Subject: [PATCH] fix: address kie provider pr review

---
 open-sse/executors/index.ts             |  2 --
 open-sse/handlers/imageGeneration.ts    |  4 ++--
 tests/unit/kie-executor-routing.test.ts | 12 ++++++++++++
 3 files changed, 14 insertions(+), 4 deletions(-)
 create mode 100644 tests/unit/kie-executor-routing.test.ts

diff --git a/open-sse/executors/index.ts b/open-sse/executors/index.ts
index ba3a195676..8d696afac4 100644
--- a/open-sse/executors/index.ts
+++ b/open-sse/executors/index.ts
@@ -14,7 +14,6 @@ import { VertexExecutor } from "./vertex.ts";
 import { CliproxyapiExecutor } from "./cliproxyapi.ts";
 import { PerplexityWebExecutor } from "./perplexity-web.ts";
 import { GrokWebExecutor } from "./grok-web.ts";
-import { KieExecutor } from "./kie.ts";
 import { ChatGptWebExecutor } from "./chatgpt-web.ts";
 import { BlackboxWebExecutor } from "./blackbox-web.ts";
 import { MuseSparkWebExecutor } from "./muse-spark-web.ts";
@@ -53,7 +52,6 @@ const executors = {
   "perplexity-web": new PerplexityWebExecutor(),
   "pplx-web": new PerplexityWebExecutor(), // Alias
   "grok-web": new GrokWebExecutor(),
-  kie: new KieExecutor(),
   "chatgpt-web": new ChatGptWebExecutor(),
   "cgpt-web": new ChatGptWebExecutor(), // Alias
   "blackbox-web": new BlackboxWebExecutor(),
diff --git a/open-sse/handlers/imageGeneration.ts b/open-sse/handlers/imageGeneration.ts
index c58aaf627c..7797bff9b7 100644
--- a/open-sse/handlers/imageGeneration.ts
+++ b/open-sse/handlers/imageGeneration.ts
@@ -466,8 +466,8 @@ async function handleKieImageGeneration({
 
     payload = {
       prompt,
-      image_size: mapImageSize(size, "1:1"),
-      num_images: body.n || 1,
+      size: mapImageSize(size, "1:1"),
+      nVariants: body.n || 1,
     };
   }
 
diff --git a/tests/unit/kie-executor-routing.test.ts b/tests/unit/kie-executor-routing.test.ts
new file mode 100644
index 0000000000..762b34d9a3
--- /dev/null
+++ b/tests/unit/kie-executor-routing.test.ts
@@ -0,0 +1,12 @@
+import test from "node:test";
+import assert from "node:assert/strict";
+
+import { DefaultExecutor } from "../../open-sse/executors/default.ts";
+import { getExecutor, hasSpecializedExecutor } from "../../open-sse/executors/index.ts";
+import { KieExecutor } from "../../open-sse/executors/kie.ts";
+
+test("KIE chat traffic uses the default executor while media keeps its task executor", () => {
+  assert.equal(hasSpecializedExecutor("kie"), false);
+  assert.ok(getExecutor("kie") instanceof DefaultExecutor);
+  assert.equal(typeof KieExecutor, "function");
+});