Gate/AI for iOS

import GateAI

#if targetEnvironment(simulator)
let devToken = ProcessInfo.processInfo.environment["GATEAI_DEV_TOKEN"]
#else
let devToken: String? = nil
#endif

let configuration = try GateAIConfiguration(
    baseURLString: "https://your-team.us01.gate-ai.net",
    teamIdentifier: "ABCDE12345",
    developmentToken: devToken,
    logLevel: .info
)

let gateAIClient = GateAIClient(configuration: configuration)
gateAIClient.userStatus = "premium"

struct ChatRequest: Encodable {
    let model: String
    let messages: [Message]
}

struct Message: Encodable {
    let role: String
    let content: String
}

let request = ChatRequest(
    model: "gpt-4o",
    messages: [
        Message(role: "user", content: "Say hello from Gate/AI")
    ]
)

let body = try JSONEncoder().encode(request)

let (data, response) = try await gateAIClient.performProxyRequest(
    path: "openai/chat/completions",
    method: .post,
    body: body,
    additionalHeaders: ["Content-Type": "application/json"]
)

guard response.statusCode == 200 else {
    throw MyAppError.unexpectedStatus(response.statusCode)
}

var request = URLRequest(url: configuration.baseURL.appendingPathComponent("anthropic/v1/messages"))
request.httpMethod = "POST"
request.httpBody = anthropicPayload
request.setValue("application/json", forHTTPHeaderField: "Content-Type")
request.setValue("2023-06-01", forHTTPHeaderField: "anthropic-version")

do {
    let headers = try await gateAIClient.authorizationHeaders(
        for: request.url!,
        method: .post
    )

    headers.forEach { key, value in
        request.setValue(value, forHTTPHeaderField: key)
    }

    let (data, response) = try await URLSession.shared.data(for: request)
    // Handle response...
} catch {
    if let nonce = gateAIClient.extractDPoPNonce(from: error) {
        let retryHeaders = try await gateAIClient.authorizationHeaders(
            for: request.url!,
            method: .post,
            nonce: nonce
        )

        retryHeaders.forEach { key, value in
            request.setValue(value, forHTTPHeaderField: key)
        }

        _ = try await URLSession.shared.data(for: request)
    } else {
        throw error
    }
}