@@ -35,11 +35,12 @@ private let deviceCheckWarning = """
35
35
extension Endpoint {
36
36
37
37
private func urlComponents(
38
- serviceURL: String ? ,
38
+ serviceURL: String ,
39
+ path: String ,
39
40
queryItems: [ URLQueryItem ] )
40
41
-> URLComponents
41
42
{
42
- var components = URLComponents ( string: serviceURL ?? " https://api.aiproxy.pro " ) !
43
+ var components = URLComponents ( string: serviceURL) !
43
44
components. path = components. path. appending ( path)
44
45
if !queryItems. isEmpty {
45
46
components. queryItems = queryItems
@@ -49,16 +50,17 @@ extension Endpoint {
49
50
50
51
func request(
51
52
aiproxyPartialKey: String ,
52
- serviceURL: String ? ,
53
53
clientID: String ? ,
54
54
organizationID: String ? ,
55
+ openAIEnvironment: OpenAIEnvironment ,
55
56
method: HTTPMethod ,
56
57
params: Encodable ? = nil ,
57
58
queryItems: [ URLQueryItem ] = [ ] ,
58
59
betaHeaderField: String ? = nil )
59
60
async throws -> URLRequest
60
61
{
61
- var request = URLRequest ( url: urlComponents ( serviceURL: serviceURL, queryItems: queryItems) . url!)
62
+ let finalPath = path ( in: openAIEnvironment)
63
+ var request = URLRequest ( url: urlComponents ( serviceURL: openAIEnvironment. baseURL, path: finalPath, queryItems: queryItems) . url!)
62
64
request. addValue ( " application/json " , forHTTPHeaderField: " Content-Type " )
63
65
request. addValue ( aiproxyPartialKey, forHTTPHeaderField: " aiproxy-partial-key " )
64
66
if let organizationID {
@@ -87,7 +89,7 @@ extension Endpoint {
87
89
88
90
func multiPartRequest(
89
91
aiproxyPartialKey: String ,
90
- serviceURL : String ? ,
92
+ openAIEnvironment : OpenAIEnvironment ,
91
93
clientID: String ? ,
92
94
organizationID: String ? ,
93
95
method: HTTPMethod ,
@@ -96,7 +98,8 @@ extension Endpoint {
96
98
)
97
99
async throws -> URLRequest
98
100
{
99
- var request = URLRequest ( url: urlComponents ( serviceURL: serviceURL, queryItems: queryItems) . url!)
101
+ let finalPath = path ( in: openAIEnvironment)
102
+ var request = URLRequest ( url: urlComponents ( serviceURL: openAIEnvironment. baseURL, path: finalPath, queryItems: queryItems) . url!)
100
103
request. httpMethod = method. rawValue
101
104
request. addValue ( aiproxyPartialKey, forHTTPHeaderField: " aiproxy-partial-key " )
102
105
if let organizationID {
0 commit comments