Quellcode durchsuchen

zen: handle cache key

Frank vor 1 Monat
Ursprung
Commit
df8464f89c

+ 5 - 0
packages/console/app/src/routes/zen/util/handler.ts

@@ -136,6 +136,11 @@ export async function handler(
             ...createBodyConverter(opts.format, providerInfo.format)(body),
             model: providerInfo.model,
             ...(providerInfo.payloadModifier ?? {}),
+            ...Object.fromEntries(
+              Object.entries(providerInfo.payloadMappings ?? {})
+                .map(([k, v]) => [k, input.request.headers.get(v)])
+                .filter(([_k, v]) => !!v),
+            ),
           },
           authInfo?.workspaceID,
         ),

+ 1 - 0
packages/console/core/src/model.ts

@@ -47,6 +47,7 @@ export namespace ZenData {
     format: FormatSchema.optional(),
     headerMappings: z.record(z.string(), z.string()).optional(),
     payloadModifier: z.record(z.string(), z.any()).optional(),
+    payloadMappings: z.record(z.string(), z.string()).optional(),
   })
 
   const ModelsSchema = z.object({