Skip to content

Commit 15d8704

Browse files
committed
Remove seed from ImageGenerationParameters
1 parent 1229559 commit 15d8704

File tree

4 files changed

+0
-11
lines changed

4 files changed

+0
-11
lines changed

FirebaseVertexAI/Sources/Types/Internal/Imagen/ImageGenerationParameters.swift

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
struct ImageGenerationParameters {
1717
let sampleCount: Int?
1818
let storageURI: String?
19-
let seed: Int32?
2019
let negativePrompt: String?
2120
let aspectRatio: String?
2221
let safetyFilterLevel: String?
@@ -36,7 +35,6 @@ extension ImageGenerationParameters: Encodable {
3635
enum CodingKeys: String, CodingKey {
3736
case sampleCount
3837
case storageURI = "storageUri"
39-
case seed
4038
case negativePrompt
4139
case aspectRatio
4240
case safetyFilterLevel = "safetySetting"
@@ -50,7 +48,6 @@ extension ImageGenerationParameters: Encodable {
5048
var container = encoder.container(keyedBy: CodingKeys.self)
5149
try container.encodeIfPresent(sampleCount, forKey: .sampleCount)
5250
try container.encodeIfPresent(storageURI, forKey: .storageURI)
53-
try container.encodeIfPresent(seed, forKey: .seed)
5451
try container.encodeIfPresent(negativePrompt, forKey: .negativePrompt)
5552
try container.encodeIfPresent(aspectRatio, forKey: .aspectRatio)
5653
try container.encodeIfPresent(safetyFilterLevel, forKey: .safetyFilterLevel)

FirebaseVertexAI/Sources/Types/Public/Imagen/ImagenModel.swift

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,11 +86,9 @@ public final class ImagenModel {
8686
func imageGenerationParameters(storageURI: String?,
8787
generationConfig: ImagenGenerationConfig? = nil)
8888
-> ImageGenerationParameters {
89-
// TODO(#14221): Add support for configuring remaining parameters.
9089
return ImageGenerationParameters(
9190
sampleCount: generationConfig?.numberOfImages ?? 1,
9291
storageURI: storageURI,
93-
seed: nil,
9492
negativePrompt: generationConfig?.negativePrompt,
9593
aspectRatio: generationConfig?.aspectRatio?.rawValue,
9694
safetyFilterLevel: safetySettings?.safetyFilterLevel?.rawValue,

FirebaseVertexAI/Tests/Unit/Types/Imagen/ImageGenerationParametersTests.swift

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ final class ImageGenerationParametersTests: XCTestCase {
2929
func testEncodeParameters_allSpecified() throws {
3030
let sampleCount = 4
3131
let storageURI = "gs://bucket/folder"
32-
let seed: Int32 = 1_076_107_968
3332
let negativePrompt = "test-negative-prompt"
3433
let aspectRatio = "16:9"
3534
let safetyFilterLevel = "block_low_and_above"
@@ -41,7 +40,6 @@ final class ImageGenerationParametersTests: XCTestCase {
4140
let parameters = ImageGenerationParameters(
4241
sampleCount: sampleCount,
4342
storageURI: storageURI,
44-
seed: seed,
4543
negativePrompt: negativePrompt,
4644
aspectRatio: aspectRatio,
4745
safetyFilterLevel: safetyFilterLevel,
@@ -66,7 +64,6 @@ final class ImageGenerationParametersTests: XCTestCase {
6664
"personGeneration" : "\(personGeneration)",
6765
"safetySetting" : "\(safetyFilterLevel)",
6866
"sampleCount" : \(sampleCount),
69-
"seed" : \(seed),
7067
"storageUri" : "\(storageURI)"
7168
}
7269
""")
@@ -80,7 +77,6 @@ final class ImageGenerationParametersTests: XCTestCase {
8077
let parameters = ImageGenerationParameters(
8178
sampleCount: sampleCount,
8279
storageURI: nil,
83-
seed: nil,
8480
negativePrompt: nil,
8581
aspectRatio: aspectRatio,
8682
safetyFilterLevel: safetyFilterLevel,
@@ -107,7 +103,6 @@ final class ImageGenerationParametersTests: XCTestCase {
107103
let parameters = ImageGenerationParameters(
108104
sampleCount: nil,
109105
storageURI: nil,
110-
seed: nil,
111106
negativePrompt: nil,
112107
aspectRatio: nil,
113108
safetyFilterLevel: nil,

FirebaseVertexAI/Tests/Unit/Types/Imagen/ImageGenerationRequestTests.swift

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@ final class ImageGenerationRequestTests: XCTestCase {
2828
lazy var parameters = ImageGenerationParameters(
2929
sampleCount: sampleCount,
3030
storageURI: nil,
31-
seed: nil,
3231
negativePrompt: nil,
3332
aspectRatio: aspectRatio,
3433
safetyFilterLevel: safetyFilterLevel,

0 commit comments

Comments
 (0)