diff --git a/apps/api/internal/clients/universal.go b/apps/api/internal/clients/universal.go index 976a20e..4d1b4da 100644 --- a/apps/api/internal/clients/universal.go +++ b/apps/api/internal/clients/universal.go @@ -242,9 +242,13 @@ func universalScriptContext(request Request, modelType string, payload map[strin return selectedBase + "/" + strings.TrimLeft(path, "/") } context["creatRequestURL"] = context["createRequestURL"] - context["processedParams"] = cloneBody(request.Body) + processedParams := cloneBody(request.Body) + processedParams["model"] = upstreamModelName(request.Candidate) + context["processedParams"] = processedParams context["preProcessParams"] = func(params map[string]any, _ ...string) map[string]any { - return cloneMapAny(params) + processed := cloneMapAny(params) + processed["model"] = upstreamModelName(request.Candidate) + return processed } context["resolveGetTaskURL"] = func(taskID string) string { return resolveUniversalTaskURL(request.Candidate.PlatformConfig, taskID) diff --git a/apps/api/internal/clients/universal_test.go b/apps/api/internal/clients/universal_test.go index f08d084..42579cf 100644 --- a/apps/api/internal/clients/universal_test.go +++ b/apps/api/internal/clients/universal_test.go @@ -102,7 +102,7 @@ func TestUniversalClientSupportsServerMainScriptContextAndNestedResult(t *testin "customGetParamsScript": map[string]any{ "image_generate": `async function getParams(params, context) { const processed = await context.preProcessParams(params, context.type); - return { prompt: processed.prompt + "-" + context.processedParams.prompt }; + return { prompt: processed.prompt + "-" + processed.model + "-" + context.processedParams.model }; }`, }, "customSubmitScript": map[string]any{ @@ -128,7 +128,7 @@ func TestUniversalClientSupportsServerMainScriptContextAndNestedResult(t *testin t.Fatalf("unexpected nested result: %#v", response.Result) } image, ok := data[0].(map[string]any) - if !ok || image["url"] != "https://cdn.example/hello-hello.png" { + if !ok || image["url"] != "https://cdn.example/hello-provider-model-provider-model.png" { t.Fatalf("unexpected image result: %#v", response.Result) } }