Adding initial formatting error counting and token counter

tests/agent_test.py

@@ -4,11 +4,13 @@ from unittest.mock import patch
 
 import pytest
 from langchain.tools import tool
+from langchain_core.exceptions import OutputParserException
 from langchain_openai import ChatOpenAI
 
 from crewai import Agent, Crew, Task
 from crewai.agents.cache import CacheHandler
 from crewai.agents.executor import CrewAgentExecutor
+from crewai.agents.parser import CrewAgentParser
 from crewai.tools.tool_calling import InstructorToolCalling
 from crewai.tools.tool_usage import ToolUsage
 from crewai.utilities import RPMController
@@ -576,3 +578,39 @@ def test_agent_function_calling_llm():
 
         crew.kickoff()
         private_mock.assert_called()
+
+
+def test_agent_count_formatting_error():
+    from unittest.mock import patch
+
+    agent1 = Agent(
+        role="test role",
+        goal="test goal",
+        backstory="test backstory",
+        verbose=True,
+    )
+
+    parser = CrewAgentParser()
+    parser.agent = agent1
+
+    with patch.object(Agent, "count_formatting_errors") as mock_count_errors:
+        test_text = "This text does not match expected formats."
+        with pytest.raises(OutputParserException):
+            parser.parse(test_text)
+        mock_count_errors.assert_called_once()
+
+
+def test_agent_llm_uses_token_calc_handler_with_llm_has_model_name():
+    agent1 = Agent(
+        role="test role",
+        goal="test goal",
+        backstory="test backstory",
+        verbose=True,
+    )
+
+    assert len(agent1.llm.callbacks) == 1
+    assert agent1.llm.callbacks[0].__class__.__name__ == "TokenCalcHandler"
+    assert agent1.llm.callbacks[0].model == "gpt-4"
+    assert (
+        agent1.llm.callbacks[0].token_cost_process.__class__.__name__ == "TokenProcess"
+    )