Skip to content

Commit aa3b77e

Browse files
committed
Add public API tests for GenerateContentResponse in Vertex AI (#12791)
1 parent 8bb8243 commit aa3b77e

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

FirebaseVertexAI/Tests/Unit/VertexAIAPITests.swift

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,24 @@ final class VertexAIAPITests: XCTestCase {
167167
_ = genAI.startChat(history: [ModelContent(parts: "abc")])
168168
}
169169

170+
// Public API tests for GenerateContentResponse.
171+
func generateContentResponseAPI() {
172+
let response = GenerateContentResponse(candidates: [])
173+
174+
let _: [CandidateResponse] = response.candidates
175+
let _: PromptFeedback? = response.promptFeedback
176+
177+
// Usage Metadata
178+
guard let usageMetadata = response.usageMetadata else { fatalError() }
179+
let _: Int = usageMetadata.promptTokenCount
180+
let _: Int = usageMetadata.candidatesTokenCount
181+
let _: Int = usageMetadata.totalTokenCount
182+
183+
// Computed Properties
184+
let _: String? = response.text
185+
let _: [FunctionCall] = response.functionCalls
186+
}
187+
170188
// Result builder alternative
171189

172190
/*

0 commit comments

Comments
 (0)