Skip to content

Commit 0ff780b

Browse files
andrewheardG.Dev.Ssomsak
authored and
G.Dev.Ssomsak
committed
Add system instruction support (google-gemini#129)
1 parent d529c48 commit 0ff780b

File tree

2 files changed

+10
-1
lines changed

2 files changed

+10
-1
lines changed

Sources/GoogleAI/GenerateContentRequest.swift

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ struct GenerateContentRequest {
2323
let safetySettings: [SafetySetting]?
2424
let tools: [Tool]?
2525
let toolConfig: ToolConfig?
26+
let systemInstruction: ModelContent?
2627
let isStreaming: Bool
2728
let options: RequestOptions
2829
}
@@ -35,6 +36,7 @@ extension GenerateContentRequest: Encodable {
3536
case safetySettings
3637
case tools
3738
case toolConfig
39+
case systemInstruction
3840
}
3941
}
4042

Tests/GoogleAITests/GoogleAITests.swift

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,17 +30,24 @@ final class GoogleGenerativeAITests: XCTestCase {
3030
maxOutputTokens: 256,
3131
stopSequences: ["..."])
3232
let filters = [SafetySetting(harmCategory: .dangerousContent, threshold: .blockOnlyHigh)]
33+
let systemInstruction = ModelContent(role: "system", parts: [.text("Talk like a pirate.")])
3334

3435
// Permutations without optional arguments.
3536
let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY")
3637
let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", safetySettings: filters)
3738
let _ = GenerativeModel(name: "gemini-1.0-pro", apiKey: "API_KEY", generationConfig: config)
39+
let _ = GenerativeModel(
40+
name: "gemini-1.0-pro",
41+
apiKey: "API_KEY",
42+
systemInstruction: systemInstruction
43+
)
3844

3945
// All arguments passed.
4046
let genAI = GenerativeModel(name: "gemini-1.0-pro",
4147
apiKey: "API_KEY",
4248
generationConfig: config, // Optional
43-
safetySettings: filters // Optional
49+
safetySettings: filters, // Optional
50+
systemInstruction: systemInstruction // Optional
4451
)
4552
// Full Typed Usage
4653
let pngData = Data() // ....

0 commit comments

Comments
 (0)