OpenAIKit is a Swift package used to communicate with the OpenAI API.
Add the dependency to Package.swift:
dependencies: [
...
.package(url: "https://github.com/dylanshine/openai-kit.git", from: "1.0.0")
],
targets: [
.target(name: "App", dependencies: [
.product(name: "OpenAIKit", package: "openai-kit"),
]),
It is encouraged to use environment variables to inject the OpenAI API key, instead of hardcoding it in the source code.
# .env
OPENAI_API_KEY="YOUR-API-KEY"
OPENAI_ORGANIZATION="YOUR-ORGANIZATION"
Create a OpenAIKit.Client
using a httpClient and configuration.
var apiKey: String {
ProcessInfo.processInfo.environment["OPENAI_API_KEY"]!
}
var organization: String {
ProcessInfo.processInfo.environment["OPENAI_ORGANIZATION"]!
}
...
let httpClient = HTTPClient(eventLoopGroupProvider: .createNew)
let configuration = Configuration(apiKey: apiKey, organization: organization)
let openAIClient = OpenAIKit.Client(httpClient: httpClient, configuration: configuration)
The OpenAIKit.Client implements a handful of methods to interact with the OpenAI API:
import OpenAIKit
let completion = try await openAIClient.completions.create(
model: Model.GPT3.davinci,
prompts: ["Write a haiku"]
)
If the request to the API failed for any reason an OpenAIKit.APIErrorResponse
is thrown
.
Simply ensure you catch errors thrown like any other throwing function
do {
...
} catch let error as APIErrorResponse {
print(error)
}