diff options
Diffstat (limited to 'packages/console')
| -rw-r--r-- | packages/console/app/src/routes/zen/util/handler.ts | 7 | ||||
| -rw-r--r-- | packages/console/core/src/model.ts | 3 |
2 files changed, 6 insertions, 4 deletions
diff --git a/packages/console/app/src/routes/zen/util/handler.ts b/packages/console/app/src/routes/zen/util/handler.ts index d2bcaa851..a8e275ba9 100644 --- a/packages/console/app/src/routes/zen/util/handler.ts +++ b/packages/console/app/src/routes/zen/util/handler.ts @@ -110,6 +110,7 @@ export async function handler( providerInfo.modifyBody({ ...createBodyConverter(opts.format, providerInfo.format)(body), model: providerInfo.model, + ...(providerInfo.payloadModifier ?? {}), }), ) logger.debug("REQUEST URL: " + reqUrl) @@ -274,8 +275,8 @@ export async function handler( part = part.trim() usageParser.parse(part) - if (providerInfo.bodyModifier) { - for (const [k, v] of Object.entries(providerInfo.bodyModifier)) { + if (providerInfo.responseModifier) { + for (const [k, v] of Object.entries(providerInfo.responseModifier)) { part = part.replace(k, v) } c.enqueue(encoder.encode(part + "\n\n")) @@ -285,7 +286,7 @@ export async function handler( } } - if (!providerInfo.bodyModifier && providerInfo.format === opts.format) { + if (!providerInfo.responseModifier && providerInfo.format === opts.format) { c.enqueue(value) } diff --git a/packages/console/core/src/model.ts b/packages/console/core/src/model.ts index 9a2908e32..da2677844 100644 --- a/packages/console/core/src/model.ts +++ b/packages/console/core/src/model.ts @@ -62,12 +62,13 @@ export namespace ZenData { apiKey: z.string(), format: FormatSchema.optional(), headerMappings: z.record(z.string(), z.string()).optional(), + payloadModifier: z.record(z.string(), z.any()).optional(), family: z.string().optional(), }) const ProviderFamilySchema = z.object({ headers: z.record(z.string(), z.string()).optional(), - bodyModifier: z.record(z.string(), z.string()).optional(), + responseModifier: z.record(z.string(), z.string()).optional(), }) const ModelsSchema = z.object({ |