Skip to content

Commit ec5df60

Browse files
committed
Update GenerationConfig unit test to include responseModalities
1 parent 8326319 commit ec5df60

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

FirebaseVertexAI/Tests/Unit/GenerationConfigTests.swift

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,8 @@ final class GenerationConfigTests: XCTestCase {
6161
frequencyPenalty: frequencyPenalty,
6262
stopSequences: stopSequences,
6363
responseMIMEType: responseMIMEType,
64-
responseSchema: .array(items: .string())
64+
responseSchema: .array(items: .string()),
65+
responseModalities: [.text, .image]
6566
)
6667

6768
let jsonData = try encoder.encode(generationConfig)
@@ -74,6 +75,10 @@ final class GenerationConfigTests: XCTestCase {
7475
"maxOutputTokens" : \(maxOutputTokens),
7576
"presencePenalty" : \(presencePenalty),
7677
"responseMimeType" : "\(responseMIMEType)",
78+
"responseModalities" : [
79+
"TEXT",
80+
"IMAGE"
81+
],
7782
"responseSchema" : {
7883
"items" : {
7984
"nullable" : false,

0 commit comments

Comments
 (0)