feat: add token counting from API calls

[MuggleJinx]

Description

Close #1623.

Checklist

Go over all the following points, and put an x in all the boxes that apply.

• [x] I have read the CONTRIBUTION guide (required)
• [x] I have linked this PR to an issue using the Development section on the right sidebar or by adding Fixes #issue-number in the PR description (required)
• [x] I have checked if any dependencies need to be added or updated in pyproject.toml and poetry.lock
• [x] I have updated the tests accordingly (required for a bug fix or a new feature)
• [x] I have updated the documentation if needed

If you are unsure about any of these, don't hesitate to ask. We are here to help!

[Wendong-Fan]

thanks @MuggleJinx , there's some conflict need to be resolved

[MuggleJinx]

thanks @MuggleJinx , there's some conflict need to be resolved

Fixed! @Wendong-Fan