fix(bench): rewrite bench functions

[lwaekfjlk]

Closes #

📑 Description

✅ Checks

• My pull request adheres to the code style of this project
• My code requires changes to the documentation
• I have updated the documentation as required
• All the tests have passed
• Branch name follows type/descript (e.g. feature/add-llm-agents)
• Ready for code review

ℹ Additional Information

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 87.59%. Comparing base (aaebb50) to head (836e0f0).
Report is 1 commits behind head on main.

@@           Coverage Diff           @@
##             main     #748   +/-   ##
=======================================
  Coverage   87.59%   87.59%           
=======================================
  Files          60       60           
  Lines        2588     2588           
=======================================
  Hits         2267     2267           
  Misses        321      321