Test message statistics functionality
(mock_llm_client, temp_workspace)
| 142 | |
| 143 | |
| 144 | def test_message_statistics(mock_llm_client, temp_workspace): |
| 145 | """Test message statistics functionality""" |
| 146 | agent = Agent( |
| 147 | llm_client=mock_llm_client, |
| 148 | system_prompt="System", |
| 149 | tools=[], |
| 150 | workspace_dir=temp_workspace, |
| 151 | ) |
| 152 | |
| 153 | # Add different types of messages |
| 154 | agent.add_user_message("User message 1") |
| 155 | agent.messages.append(Message(role="assistant", content="Assistant response 1")) |
| 156 | agent.add_user_message("User message 2") |
| 157 | agent.messages.append( |
| 158 | Message( |
| 159 | role="tool", content="Tool result", tool_call_id="123", name="test_tool" |
| 160 | ) |
| 161 | ) |
| 162 | |
| 163 | # Count different types of messages |
| 164 | user_msgs = sum(1 for m in agent.messages if m.role == "user") |
| 165 | assistant_msgs = sum(1 for m in agent.messages if m.role == "assistant") |
| 166 | tool_msgs = sum(1 for m in agent.messages if m.role == "tool") |
| 167 | |
| 168 | assert user_msgs == 2 |
| 169 | assert assistant_msgs == 1 |
| 170 | assert tool_msgs == 1 |
| 171 | assert len(agent.messages) == 5 # 1 system + 2 user + 1 assistant + 1 tool |
nothing calls this directly
no test coverage detected