Allow for a simpler way to set system instructions (google-gemini#149)

Signed-off-by: Peter Friese <[email protected]>
SsomsakTH · Jun 21, 2024 · 637a612 · 637a612
1 parent 7090497
commit 637a612
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/Tests/GoogleAITests/GoogleAITests.swift b/Tests/GoogleAITests/GoogleAITests.swift
@@ -50,6 +50,12 @@ final class GoogleGenerativeAITests: XCTestCase {
       systemInstruction: systemInstruction
     )
 
+    let _ = GenerativeModel(
+      name: "gemini-1.5-pro-latest",
+      apiKey: "API_KEY",
+      systemInstruction: "Talk like a pirate"
+    )
+
     // All arguments passed.
     let genAI = GenerativeModel(name: "gemini-1.5-pro-latest",
                                 apiKey: "API_KEY",