Add responseMIMEType to GenerationConfig (google-gemini#161)

SsomsakTH · Jun 21, 2024 · 4b25d81 · 4b25d81
1 parent 49adf92
commit 4b25d81
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Tests/GoogleAITests/GoogleAITests.swift b/Tests/GoogleAITests/GoogleAITests.swift
@@ -28,7 +28,8 @@ final class GoogleGenerativeAITests: XCTestCase {
                                   topK: 16,
                                   candidateCount: 4,
                                   maxOutputTokens: 256,
-                                  stopSequences: ["..."])
+                                  stopSequences: ["..."],
+                                  responseMIMEType: "text/plain")
     let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
     let systemInstruction = ModelContent(role: "system", parts: [.text("Talk like a pirate.")])