fix: preserve kie market model ids

This commit is contained in:
wauputr4
2026-05-07 00:08:32 +07:00
parent f1cd77472c
commit fb0361fc8c
3 changed files with 6 additions and 9 deletions

View File

@@ -392,8 +392,6 @@ async function handleKieImageGeneration({
if (isMarket) {
// Unified Market API endpoint
baseUrl = `${providerConfig.baseUrl.replace(/\/$/, "")}/api/v1/jobs/createTask`;
// Strip category prefix (e.g., "gpt/gpt-image-2" -> "gpt-image-2")
const marketModelId = model.includes("/") ? model.split("/").pop() : model;
const input: Record<string, unknown> = {
prompt,
aspect_ratio: mapImageSize(size, "1:1"),
@@ -402,7 +400,7 @@ async function handleKieImageGeneration({
input.image_url = imageUrl;
}
payload = {
model: marketModelId,
model,
input,
};
} else {

View File

@@ -266,7 +266,7 @@ async function handleKieMusicGeneration({
if (isMarket) {
url = `${baseUrl}/api/v1/jobs/createTask`;
payload = {
model: model.includes("/") ? model.split("/").pop() : model,
model,
callBackUrl: getKieCallbackUrl(body),
input: {
prompt,
@@ -310,7 +310,9 @@ async function handleKieMusicGeneration({
const statusUrl = isMarket
? `${baseUrl}/api/v1/jobs/recordInfo`
: providerConfig.statusUrl || `${baseUrl}/api/v1/generate/record-info`;
: providerConfig.statusUrl && !providerConfig.statusUrl.includes("jobs/recordInfo")
? providerConfig.statusUrl
: `${baseUrl}/api/v1/generate/record-info`;
const { data: recordData, state } = await kieExecutor.pollTask({
statusUrl,

View File

@@ -328,11 +328,8 @@ async function handleKieVideoGeneration({
return { success: false, status: 401, error: "KIE API key is required" };
}
// Strip category prefix (e.g., "veo/veo-3-1" -> "veo-3-1")
const marketModelId = model.includes("/") ? model.split("/").pop() : model;
const payload = {
model: marketModelId,
model,
input: {
prompt,
duration: body.duration ? String(body.duration) : "5",