Skip to content

Commit 22dce81

Browse files
authored
[Vertex AI] Add Imagen warning for invalid JPEG compression quality (#14440)
1 parent 1fe5229 commit 22dce81

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,12 @@ public struct ImagenImageFormat {
4545
/// compression (lowest image quality, smallest file size) and `100` is the lowest level of
4646
/// compression (highest image quality, largest file size); defaults to `75`.
4747
public static func jpeg(compressionQuality: Int? = nil) -> ImagenImageFormat {
48+
if let compressionQuality, compressionQuality < 0 || compressionQuality > 100 {
49+
VertexLog.warning(code: .imagenInvalidJPEGCompressionQuality, """
50+
Invalid JPEG compression quality of \(compressionQuality) specified; the supported range is \
51+
[0, 100].
52+
""")
53+
}
4854
return ImagenImageFormat(mimeType: "image/jpeg", compressionQuality: compressionQuality)
4955
}
5056
}

FirebaseVertexAI/Sources/VertexLog.swift

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,12 @@ enum VertexLog {
3131
// API Enablement Errors
3232
case vertexAIInFirebaseAPIDisabled = 200
3333

34-
// Model Configuration
34+
// Generative Model Configuration
3535
case generativeModelInitialized = 1000
3636

37+
// Imagen Model Configuration
38+
case imagenInvalidJPEGCompressionQuality = 1201
39+
3740
// Network Errors
3841
case generativeAIServiceNonHTTPResponse = 2000
3942
case loadRequestResponseError = 2001

0 commit comments

Comments
 (0)