[Vertex AI] Fix testCountTokens_jsonSchema integration test

andrewheard · andrewheard · commit 255eb147a33a · 2024-11-07T16:45:36.000-05:00
diff --git a/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift b/FirebaseVertexAI/Tests/TestApp/Tests/Integration/IntegrationTests.swift
@@ -217,8 +217,8 @@ final class IntegrationTests: XCTestCase {
 
     let response = try await model.countTokens(prompt)
 
-    XCTAssertEqual(response.totalTokens, 34)
-    XCTAssertEqual(response.totalBillableCharacters, 59)
+    XCTAssertEqual(response.totalTokens, 58)
+    XCTAssertEqual(response.totalBillableCharacters, 160)
   }
 
   func testCountTokens_appCheckNotConfigured_shouldFail() async throws {

Original file line number	Diff line number	Diff line change
`@@ -217,8 +217,8 @@ final class IntegrationTests: XCTestCase {`
`217`	`217`
`218`	`218`	`let response = try await model.countTokens(prompt)`
`219`	`219`
`220`		`- XCTAssertEqual(response.totalTokens, 34)`
`221`		`- XCTAssertEqual(response.totalBillableCharacters, 59)`
	`220`	`+ XCTAssertEqual(response.totalTokens, 58)`
	`221`	`+ XCTAssertEqual(response.totalBillableCharacters, 160)`
`222`	`222`	`}`
`223`	`223`
`224`	`224`	`func testCountTokens_appCheckNotConfigured_shouldFail() async throws {`