🎯 feat(config): Custom Endpoint Request Headers (danny-avila#1588)

aitok-ai · Jan 19, 2024 · fc362f6 · fc362f6
1 parent dcfd403
commit fc362f6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/server/services/Endpoints/custom/initializeClient.js b/server/services/Endpoints/custom/initializeClient.js
@@ -22,6 +22,13 @@ const initializeClient = async ({ req, res, endpointOption }) => {
   const CUSTOM_API_KEY = extractEnvVariable(endpointConfig.apiKey);
   const CUSTOM_BASE_URL = extractEnvVariable(endpointConfig.baseURL);
 
+  let resolvedHeaders = {};
+  if (endpointConfig.headers && typeof endpointConfig.headers === 'object') {
+    Object.keys(endpointConfig.headers).forEach((key) => {
+      resolvedHeaders[key] = extractEnvVariable(endpointConfig.headers[key]);
+    });
+  }
+
   if (CUSTOM_API_KEY.match(envVarRegex)) {
     throw new Error(`Missing API Key for ${endpoint}.`);
   }
@@ -31,6 +38,7 @@ const initializeClient = async ({ req, res, endpointOption }) => {
   }
 
   const customOptions = {
+    headers: resolvedHeaders,
     addParams: endpointConfig.addParams,
     dropParams: endpointConfig.dropParams,
     titleConvo: endpointConfig.titleConvo,