feat: add custom headers support

fal-ai · Feb 17, 2025 · a7db4cb · a7db4cb
1 parent 4a589b9
commit a7db4cb
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/Sources/FalClient/Client+Config.swift b/Sources/FalClient/Client+Config.swift
@@ -32,9 +32,11 @@ public enum ClientCredentials: CustomStringConvertible {
 public struct ClientConfig {
     public let credentials: ClientCredentials
     public let requestProxy: String?
+    public let customHeaders: [String: String]?
 
-    init(credentials: ClientCredentials = .fromEnv, requestProxy: String? = nil) {
+    init(credentials: ClientCredentials = .fromEnv, requestProxy: String? = nil, customHeaders: [String: String]? = nil) {
         self.credentials = credentials
         self.requestProxy = requestProxy
+        self.customHeaders = customHeaders
     }
 }
diff --git a/Sources/FalClient/Client+Request.swift b/Sources/FalClient/Client+Request.swift
@@ -33,6 +33,11 @@ extension Client {
         }
 
         var request = URLRequest(url: url)
+        if let header = config.customHeaders {
+            for (key, value) in header {
+                request.setValue(value, forHTTPHeaderField: key)
+            }
+        }
         request.httpMethod = options.httpMethod.rawValue.uppercased()
         request.setValue("application/json", forHTTPHeaderField: "accept")
         request.setValue("application/json", forHTTPHeaderField: "content-type")

diff --git a/Sources/FalClient/FalClient.swift b/Sources/FalClient/FalClient.swift
@@ -76,8 +76,8 @@ public struct FalClient: Client {
 }
 
 public extension FalClient {
-    static func withProxy(_ url: String) -> Client {
-        FalClient(config: ClientConfig(requestProxy: url))
+    static func withProxy(_ url: String, headers: [String: String]? = nil) -> Client {
+        FalClient(config: ClientConfig(requestProxy: url, customHeaders: headers))
     }
 
     static func withCredentials(_ credentials: ClientCredentials) -> Client {