Skip to content

Commit 01b87b4

Browse files
committed
Added missing initialisers.
1 parent 72e9fb0 commit 01b87b4

File tree

3 files changed

+78
-2
lines changed

3 files changed

+78
-2
lines changed

Sources/OpenAIGPT3/Completions.swift

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,42 @@ extension Completions {
2222
public let bestOf: Percentage?
2323
public let logitBias: [Token: Double]?
2424
public let user: String?
25+
26+
public init(
27+
model: Model.ID,
28+
prompt: Prompt? = nil,
29+
suffix: String? = nil,
30+
maxTokens: Int? = nil,
31+
temperature: Percentage? = nil,
32+
topP: Percentage? = nil,
33+
n: Percentage? = nil,
34+
stream: Bool? = nil,
35+
logprobs: Int? = nil,
36+
echo: Bool? = nil,
37+
stop: [String]? = nil,
38+
presencePenalty: Penalty? = nil,
39+
frequencyPenalty: Penalty? = nil,
40+
bestOf: Percentage? = nil,
41+
logitBias: [Token : Double]? = nil,
42+
user: String? = nil
43+
) {
44+
self.model = model
45+
self.prompt = prompt
46+
self.suffix = suffix
47+
self.maxTokens = maxTokens
48+
self.temperature = temperature
49+
self.topP = topP
50+
self.n = n
51+
self.stream = stream
52+
self.logprobs = logprobs
53+
self.echo = echo
54+
self.stop = stop
55+
self.presencePenalty = presencePenalty
56+
self.frequencyPenalty = frequencyPenalty
57+
self.bestOf = bestOf
58+
self.logitBias = logitBias
59+
self.user = user
60+
}
2561
}
2662
}
2763

@@ -65,6 +101,20 @@ extension Completions {
65101
public let model: Model.ID
66102
public let choices: [Choice]
67103
public let usage: Usage
104+
105+
public init(
106+
id: ID,
107+
created: Date,
108+
model: Model.ID,
109+
choices: [Choice],
110+
usage: Usage
111+
) {
112+
self.id = id
113+
self.created = created
114+
self.model = model
115+
self.choices = choices
116+
self.usage = usage
117+
}
68118
}
69119

70120
/// One of the completion choices.
@@ -80,5 +130,17 @@ extension Completions {
80130

81131
/// The reason for finishing.
82132
public let finishReason: String
133+
134+
public init(
135+
text: String,
136+
index: Int,
137+
logprobs: [String],
138+
finishReason: String
139+
) {
140+
self.text = text
141+
self.index = index
142+
self.logprobs = logprobs
143+
self.finishReason = finishReason
144+
}
83145
}
84146
}

Sources/OpenAIGPT3/Edits.swift

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,14 @@ extension Edits {
1212
public let temperature: Percentage?
1313
public let topP: Percentage?
1414

15-
public init(model: Model.ID, input: String? = nil, instruction: String, n: Int? = nil, temperature: Percentage? = nil, topP: Percentage? = nil) {
15+
public init(
16+
model: Model.ID,
17+
input: String? = nil,
18+
instruction: String,
19+
n: Int? = nil,
20+
temperature: Percentage? = nil,
21+
topP: Percentage? = nil
22+
) {
1623
self.model = model
1724
self.input = input
1825
self.instruction = instruction

Sources/OpenAIGPT3/Model.swift

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,14 @@ public struct Model: Equatable, Codable {
1616
public let root: Model.ID
1717
public let parent: Model.ID?
1818

19-
public init(id: Model.ID, created: Date, ownedBy: String, permissions: [Model.Permission], root: Model.ID, parent: Model.ID?) {
19+
public init(
20+
id: Model.ID,
21+
created: Date,
22+
ownedBy: String,
23+
permissions: [Model.Permission],
24+
root: Model.ID,
25+
parent: Model.ID? = nil
26+
) {
2027
self.id = id
2128
self.created = created
2229
self.ownedBy = ownedBy

0 commit comments

Comments
 (0)