diff --git a/lib/crewai/src/crewai/llm.py b/lib/crewai/src/crewai/llm.py index 77053deeb..31b6b64bc 100644 --- a/lib/crewai/src/crewai/llm.py +++ b/lib/crewai/src/crewai/llm.py @@ -1185,6 +1185,8 @@ class LLM(BaseLLM): start_time=0, end_time=0, ) + self._track_token_usage_internal(usage_info) + # --- 4) Check for tool calls tool_calls = getattr(response_message, "tool_calls", []) diff --git a/lib/crewai/tests/test_llm.py b/lib/crewai/tests/test_llm.py index 6f3bcd70a..80274020a 100644 --- a/lib/crewai/tests/test_llm.py +++ b/lib/crewai/tests/test_llm.py @@ -72,6 +72,9 @@ def test_llm_call_with_string_input_and_callbacks(): assert len(result.strip()) > 0 assert usage_metrics.successful_requests == 1 + usage_metrics_internal = llm.get_token_usage_summary() + assert usage_metrics_internal == usage_metrics + @pytest.mark.vcr() def test_llm_call_with_message_list():