add headers to provider

server/routers/aiProvider/updateAiProvider.ts

@@ -15,14 +15,16 @@ import { toPublicAiProvider } from "@server/routers/aiProvider/types";
 import {
     aiAuthTypeSchema,
     aiCapabilitiesSchema,
+    aiProviderHeadersSchema,
     aiProviderTypeSchema,
     aiRoutingModeSchema,
     refineProviderUpstreamFields
 } from "@server/routers/aiProvider/validation";
-import type {
-    AiProviderAuthType,
-    AiProviderRoutingMode,
-    AiProviderType
+import {
+    serializeAiProviderHeaders,
+    type AiProviderAuthType,
+    type AiProviderRoutingMode,
+    type AiProviderType
 } from "@server/lib/aiProviderDefaults";
 import {
     parseCapabilities,
@@ -40,6 +42,7 @@ const bodySchema = z.strictObject({
     authType: aiAuthTypeSchema.optional(),
     routingMode: aiRoutingModeSchema.optional(),
     capabilities: aiCapabilitiesSchema.optional(),
+    headers: aiProviderHeadersSchema,
     skipTlsVerification: z.boolean().optional(),
     enabled: z.boolean().optional()
 });
@@ -202,6 +205,11 @@ export async function updateAiProvider(
             updateData.apiKeyLastChars = body.apiKey.slice(-4);
         }
 
+        if (body.headers !== undefined) {
+            const key = config.getRawConfig().server.secret!;
+            updateData.headers = serializeAiProviderHeaders(body.headers, key);
+        }
+
         const [provider] = await db
             .update(aiProviders)
             .set(updateData)