Skip to content

Commit 71210ea

Browse files
Apply suggestions from code review
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 48eda93 commit 71210ea

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

FirebaseAI/Tests/Unit/GenerativeModelGoogleAITests.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -374,13 +374,13 @@ final class GenerativeModelGoogleAITests: XCTestCase {
374374
let urlContextMetadata = try XCTUnwrap(candidate.urlContextMetadata)
375375
XCTAssertEqual(urlContextMetadata.urlMetadata.count, 3)
376376

377-
let paywallURLMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata[0])
377+
let paywallURLMetadata = urlContextMetadata.urlMetadata[0]
378378
XCTAssertEqual(paywallURLMetadata.retrievalStatus, .error)
379379

380-
let successURLMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata[1])
380+
let successURLMetadata = urlContextMetadata.urlMetadata[1]
381381
XCTAssertEqual(successURLMetadata.retrievalStatus, .success)
382382

383-
let errorURLMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata[2])
383+
let errorURLMetadata = urlContextMetadata.urlMetadata[2]
384384
XCTAssertEqual(errorURLMetadata.retrievalStatus, .error)
385385
}
386386

FirebaseAI/Tests/Unit/GenerativeModelVertexAITests.swift

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -529,13 +529,13 @@ final class GenerativeModelVertexAITests: XCTestCase {
529529
let urlContextMetadata = try XCTUnwrap(candidate.urlContextMetadata)
530530
XCTAssertEqual(urlContextMetadata.urlMetadata.count, 3)
531531

532-
let paywallURLMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata[0])
532+
let paywallURLMetadata = urlContextMetadata.urlMetadata[0]
533533
XCTAssertEqual(paywallURLMetadata.retrievalStatus, .error)
534534

535-
let successURLMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata[1])
535+
let successURLMetadata = urlContextMetadata.urlMetadata[1]
536536
XCTAssertEqual(successURLMetadata.retrievalStatus, .success)
537537

538-
let errorURLMetadata = try XCTUnwrap(urlContextMetadata.urlMetadata[2])
538+
let errorURLMetadata = urlContextMetadata.urlMetadata[2]
539539
XCTAssertEqual(errorURLMetadata.retrievalStatus, .error)
540540
}
541541

0 commit comments

Comments
 (0)