fix(llama-index): extract token counts for groq when streaming

fix(llama-index): extract token counts for groq when streaming #2672