fix / verify count_tokens count for function calls

Function call count_tokens counts might not be correct, this needs to be tested

Tests should already be counting tokens. We need to make a function_completion which takes a function call and count the tokens for it.

https://community.openai.com/t/how-to-calculate-the-tokens-when-using-function-call/266573