Remove all closeTo tests

Author: DellaBitta

packages/ai/integration/chat.test.ts

@@ -24,7 +24,7 @@ import {
   SafetySetting,
   getGenerativeModel
 } from '../src';
-import { testConfigs, TOKEN_COUNT_DELTA } from './constants';
+import { testConfigs } from './constants';
 
 describe('Chat Session', () => {
   testConfigs.forEach(testConfig => {
@@ -98,31 +98,19 @@ describe('Chat Session', () => {
 
         if (model.model.includes('gemini-2.5-flash')) {
           // Token counts can vary slightly in chat context
-          expect(response1.usageMetadata!.promptTokenCount).to.be.closeTo(
-            17, // "What is the capital of France?" + system instruction
-            TOKEN_COUNT_DELTA + 2 // More variance for chat context
-          );
+          expect(response1.usageMetadata!.promptTokenCount).to.not.equal(0);
           expect(response1.usageMetadata!.candidatesTokenCount).to.not.equal(0);
           expect(response1.usageMetadata!.totalTokenCount).to.not.equal(0);
-          expect(response2.usageMetadata!.promptTokenCount).to.be.closeTo(
-            32, // History + "And what about Italy?" + system instruction
-            TOKEN_COUNT_DELTA + 5 // More variance for chat context with history
-          );
+          expect(response2.usageMetadata!.promptTokenCount).to.not.equal(0);
           expect(response2.usageMetadata!.candidatesTokenCount).to.not.equal(0);
           expect(response2.usageMetadata!.totalTokenCount).to.not.equal(0);
         } else if (model.model.includes('gemini-2.0-flash')) {
           expect(response1.usageMetadata).to.not.be.null;
           // Token counts can vary slightly in chat context
-          expect(response1.usageMetadata!.promptTokenCount).to.be.closeTo(
-            15, // "What is the capital of France?" + system instruction
-            TOKEN_COUNT_DELTA + 2 // More variance for chat context
-          );
+          expect(response1.usageMetadata!.promptTokenCount).to.not.equal(0);
           expect(response1.usageMetadata!.candidatesTokenCount).to.not.equal(0);
           expect(response1.usageMetadata!.totalTokenCount).to.not.equal(0);
-          expect(response2.usageMetadata!.promptTokenCount).to.be.closeTo(
-            28, // History + "And what about Italy?" + system instruction
-            TOKEN_COUNT_DELTA + 5 // More variance for chat context with history
-          );
+          expect(response2.usageMetadata!.promptTokenCount).to.not.equal(0);
           expect(response2.usageMetadata!.candidatesTokenCount).to.not.equal(0);
           expect(response2.usageMetadata!.totalTokenCount).to.not.equal(0);
         }

packages/ai/integration/generate-content.test.ts

@@ -29,7 +29,7 @@ import {
   URLRetrievalStatus,
   getGenerativeModel
 } from '../src';
-import { testConfigs, TOKEN_COUNT_DELTA } from './constants';
+import { testConfigs } from './constants';
 
 describe('Generate Content', function () {
   this.timeout(20_000);
@@ -88,10 +88,7 @@ describe('Generate Content', function () {
         expect(response.usageMetadata).to.not.be.null;
 
         if (model.model.includes('gemini-2.5-flash')) {
-          expect(response.usageMetadata!.promptTokenCount).to.be.closeTo(
-            22,
-            TOKEN_COUNT_DELTA
-          );
+          expect(response.usageMetadata!.promptTokenCount).to.not.equal(0);
           expect(response.usageMetadata!.candidatesTokenCount).to.not.equal(0);
           expect(response.usageMetadata!.thoughtsTokenCount).to.not.equal(0);
           expect(response.usageMetadata!.totalTokenCount).to.not.equal(0);
@@ -104,14 +101,11 @@ describe('Generate Content', function () {
           ).to.equal(Modality.TEXT);
           expect(
             response.usageMetadata!.promptTokensDetails![0].tokenCount
-          ).to.closeTo(22, TOKEN_COUNT_DELTA);
+          ).to.not.equal(0);
 
           // candidatesTokenDetails comes back about half the time, so let's just not test it.
         } else if (model.model.includes('gemini-2.0-flash')) {
-          expect(response.usageMetadata!.promptTokenCount).to.be.closeTo(
-            21,
-            TOKEN_COUNT_DELTA
-          );
+          expect(response.usageMetadata!.promptTokenCount).to.not.equal(0);
           expect(response.usageMetadata!.candidatesTokenCount).to.not.equal(0);
           expect(response.usageMetadata!.totalTokenCount).to.not.equal(0);
           expect(response.usageMetadata!.promptTokensDetails).to.not.be.null;