Setup header fields in services

Author: intitni

Tool/Sources/OpenAIService/APIs/OpenAIChatCompletionsService.swift

@@ -259,6 +259,7 @@ actor OpenAIChatCompletionsService: ChatCompletionsStreamAPI, ChatCompletionsAPI
 
         Self.setupAppInformation(&request)
         Self.setupAPIKey(&request, model: model, apiKey: apiKey)
+        Self.setupExtraHeaderFields(&request, model: model)
 
         let (result, response) = try await URLSession.shared.bytes(for: request)
         guard let response = response as? HTTPURLResponse else {
@@ -399,6 +400,12 @@ actor OpenAIChatCompletionsService: ChatCompletionsStreamAPI, ChatCompletionsAPI
             }
         }
     }
+    
+    static func setupExtraHeaderFields(_ request: inout URLRequest, model: ChatModel) {
+        for field in model.info.customHeaderInfo.headers where !field.key.isEmpty {
+            request.setValue(field.value, forHTTPHeaderField: field.key)
+        }
+    }
 }
 
 extension OpenAIChatCompletionsService.ResponseBody {

Tool/Sources/OpenAIService/APIs/OpenAIEmbeddingService.swift

@@ -75,6 +75,7 @@ struct OpenAIEmbeddingService: EmbeddingAPI {
 
         Self.setupAppInformation(&request)
         Self.setupAPIKey(&request, model: model, apiKey: apiKey)
+        Self.setupExtraHeaderFields(&request, model: model)
 
         let (result, response) = try await URLSession.shared.data(for: request)
         guard let response = response as? HTTPURLResponse else {
@@ -143,5 +144,11 @@ struct OpenAIEmbeddingService: EmbeddingAPI {
             }
         }
     }
+    
+    static func setupExtraHeaderFields(_ request: inout URLRequest, model: EmbeddingModel) {
+        for field in model.info.customHeaderInfo.headers where !field.key.isEmpty {
+            request.setValue(field.value, forHTTPHeaderField: field.key)
+        }
+    }
 }