Skip to content

Commit b92df27

Browse files
committed
Make HTTPClient.createRequest non-async
1 parent d2159da commit b92df27

File tree

3 files changed

+20
-16
lines changed

3 files changed

+20
-16
lines changed

Sources/HuggingFace/Hub/HubClient+Files.swift

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ public extension HubClient {
4747
message: String? = nil
4848
) async throws -> (path: String, commit: String?) {
4949
let urlPath = "/api/\(kind.pluralized)/\(repo)/upload/\(branch)"
50-
var request = try await httpClient.createRequest(.post, urlPath)
50+
var request = try httpClient.createRequest(.post, urlPath)
5151

5252
let boundary = "----hf-\(UUID().uuidString)"
5353
request.setValue(
@@ -184,7 +184,7 @@ public extension HubClient {
184184
) async throws -> Data {
185185
let endpoint = useRaw ? "raw" : "resolve"
186186
let urlPath = "/\(repo)/\(endpoint)/\(revision)/\(repoPath)"
187-
var request = try await httpClient.createRequest(.get, urlPath)
187+
var request = try httpClient.createRequest(.get, urlPath)
188188
request.cachePolicy = cachePolicy
189189

190190
let (data, response) = try await session.data(for: request)
@@ -216,7 +216,7 @@ public extension HubClient {
216216
) async throws -> URL {
217217
let endpoint = useRaw ? "raw" : "resolve"
218218
let urlPath = "/\(repo)/\(endpoint)/\(revision)/\(repoPath)"
219-
var request = try await httpClient.createRequest(.get, urlPath)
219+
var request = try httpClient.createRequest(.get, urlPath)
220220
request.cachePolicy = cachePolicy
221221

222222
let (tempURL, response) = try await session.download(
@@ -421,7 +421,7 @@ public extension HubClient {
421421
revision: String = "main"
422422
) async throws -> File {
423423
let urlPath = "/\(repo)/resolve/\(revision)/\(repoPath)"
424-
var request = try await httpClient.createRequest(.head, urlPath)
424+
var request = try httpClient.createRequest(.head, urlPath)
425425
request.setValue("bytes=0-0", forHTTPHeaderField: "Range")
426426

427427
do {

Sources/HuggingFace/Shared/HTTPClient.swift

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ final class HTTPClient: @unchecked Sendable {
7171
params: [String: Value]? = nil,
7272
headers: [String: String]? = nil
7373
) async throws -> T {
74-
let request = try await createRequest(method, path, params: params, headers: headers)
74+
let request = try createRequest(method, path, params: params, headers: headers)
7575
let (data, response) = try await session.data(for: request)
7676
let httpResponse = try validateResponse(response, data: data)
7777

@@ -98,7 +98,7 @@ final class HTTPClient: @unchecked Sendable {
9898
params: [String: Value]? = nil,
9999
headers: [String: String]? = nil
100100
) async throws -> PaginatedResponse<T> {
101-
let request = try await createRequest(method, path, params: params, headers: headers)
101+
let request = try createRequest(method, path, params: params, headers: headers)
102102
let (data, response) = try await session.data(for: request)
103103
let httpResponse = try validateResponse(response, data: data)
104104

@@ -123,7 +123,7 @@ final class HTTPClient: @unchecked Sendable {
123123
AsyncThrowingStream { @Sendable continuation in
124124
let task = Task {
125125
do {
126-
let request = try await createRequest(method, path, params: params, headers: headers)
126+
let request = try createRequest(method, path, params: params, headers: headers)
127127
let (bytes, response) = try await session.bytes(for: request)
128128
let httpResponse = try validateResponse(response)
129129

@@ -176,7 +176,7 @@ final class HTTPClient: @unchecked Sendable {
176176
params: [String: Value]? = nil,
177177
headers: [String: String]? = nil
178178
) async throws -> Data {
179-
let request = try await createRequest(method, path, params: params, headers: headers)
179+
let request = try createRequest(method, path, params: params, headers: headers)
180180
let (data, response) = try await session.data(for: request)
181181
let _ = try validateResponse(response, data: data)
182182

@@ -188,7 +188,7 @@ final class HTTPClient: @unchecked Sendable {
188188
_ path: String,
189189
params: [String: Value]? = nil,
190190
headers: [String: String]? = nil
191-
) async throws -> URLRequest {
191+
) throws -> URLRequest {
192192
var urlComponents = URLComponents(url: host, resolvingAgainstBaseURL: true)
193193
urlComponents?.path = path
194194

@@ -230,7 +230,7 @@ final class HTTPClient: @unchecked Sendable {
230230
}
231231

232232
// Get authentication token from provider
233-
if let token = try await tokenProvider.getToken() {
233+
if let token = try tokenProvider.getToken() {
234234
request.addValue("Bearer \(token)", forHTTPHeaderField: "Authorization")
235235
}
236236

Sources/HuggingFace/Shared/TokenProvider.swift

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -177,27 +177,31 @@ public indirect enum TokenProvider: Sendable {
177177
/// let token = try await provider.getToken()
178178
/// // Returns: "hf_abc123"
179179
/// ```
180-
public func getToken() async throws -> String? {
180+
public func getToken() throws -> String? {
181181
switch self {
182182
case .fixed(let token):
183183
return token
184184

185185
case .environment:
186-
return try await getTokenFromEnvironment()
186+
return try getTokenFromEnvironment()
187187

188188
case .oauth(let manager):
189-
return try await manager.getValidToken()
189+
fatalError(
190+
"OAuth token provider requires async context. Use getToken() in an async context or switch to a synchronous provider."
191+
)
190192

191193
case .composite(let providers):
192194
for provider in providers {
193-
if let token = try await provider.getToken() {
195+
if let token = try provider.getToken() {
194196
return token
195197
}
196198
}
197199
return nil
198200

199201
case .custom(let implementation):
200-
return try await implementation()
202+
fatalError(
203+
"Custom async token provider requires async context. Use getToken() in an async context or switch to a synchronous provider."
204+
)
201205

202206
case .none:
203207
return nil
@@ -267,7 +271,7 @@ private func readTokenFromPath(_ path: String, fileManager: FileManager = .defau
267271
private func getTokenFromEnvironment(
268272
_ env: [String: String] = ProcessInfo.processInfo.environment,
269273
fileManager: FileManager = .default
270-
) async throws -> String? {
274+
) throws -> String? {
271275
let tokenSources: [() -> String?] = [
272276
{ env["HF_TOKEN"] },
273277
{ env["HUGGING_FACE_HUB_TOKEN"] },

0 commit comments

Comments
 (0)