chore: Changed token_count to only use tokenCountCallback

[jsumners-nr]

This PR resolves #2069.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 97.18%. Comparing base (64b4ca2) to head (19e0a2b).

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #2070   +/-   ##
=======================================
  Coverage   97.18%   97.18%           
=======================================
  Files         248      248           
  Lines       41936    41938    +2     
=======================================
+ Hits        40754    40756    +2     
  Misses       1182     1182           

Flag	Coverage Δ
integration-tests-16.x	78.24% <ø> (+0.01%)	⬆️
integration-tests-18.x	78.21% <ø> (ø)
integration-tests-20.x	78.23% <ø> (ø)
unit-tests-16.x	90.49% <100.00%> (+<0.01%)	⬆️
unit-tests-18.x	90.47% <100.00%> (+<0.01%)	⬆️
unit-tests-20.x	90.47% <100.00%> (+<0.01%)	⬆️
versioned-tests-16.x	74.48% <72.72%> (-0.03%)	⬇️
versioned-tests-18.x	75.60% <72.72%> (-0.03%)	⬇️
versioned-tests-20.x	75.61% <72.72%> (-0.03%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.