Path prefix to proxy config added (#59)

dylanshine · Jun 15, 2023 · bf4e816 · bf4e816
1 parent b26111a
commit bf4e816
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/Sources/OpenAIKit/API.swift b/Sources/OpenAIKit/API.swift
@@ -3,13 +3,16 @@ import Foundation
 public struct API {
     public let scheme: Scheme
     public let host: String
+    public let path: String?
 
     public init(
         scheme: API.Scheme,
-        host: String
+        host: String,
+        pathPrefix path: String? = nil
     ) {
         self.scheme = scheme
         self.host = host
+        self.path = path
     }
 }
 

diff --git a/Sources/OpenAIKit/RequestHandler/RequestHandler.swift b/Sources/OpenAIKit/RequestHandler/RequestHandler.swift
@@ -12,7 +12,9 @@ extension RequestHandler {
         var components = URLComponents()
         components.scheme = configuration.api?.scheme.value ?? request.scheme.value
         components.host = configuration.api?.host ?? request.host
-        components.path = request.path
+        components.path = [configuration.api?.path, request.path]
+            .compactMap { $0 }
+            .joined()
 
         guard let url = components.url else {
             throw RequestHandlerError.invalidURLGenerated