| 186 | |
| 187 | |
| 188 | def test_no_schema(): |
| 189 | mock_model = testing_utils.MockModel.create( |
| 190 | responses=[ |
| 191 | function_call_no_schema, |
| 192 | 'response1', |
| 193 | 'response2', |
| 194 | ] |
| 195 | ) |
| 196 | |
| 197 | tool_agent = Agent( |
| 198 | name='tool_agent', |
| 199 | model=mock_model, |
| 200 | ) |
| 201 | |
| 202 | root_agent = Agent( |
| 203 | name='root_agent', |
| 204 | model=mock_model, |
| 205 | tools=[AgentTool(agent=tool_agent)], |
| 206 | ) |
| 207 | |
| 208 | runner = testing_utils.InMemoryRunner(root_agent) |
| 209 | |
| 210 | assert testing_utils.simplify_events(runner.run('test1')) == [ |
| 211 | ('root_agent', function_call_no_schema), |
| 212 | ('root_agent', function_response_no_schema), |
| 213 | ('root_agent', 'response2'), |
| 214 | ] |
| 215 | |
| 216 | |
| 217 | def test_use_plugins(): |