(callback_context: CallbackContext)
| 334 | ) |
| 335 | |
| 336 | async def after_main_agent(callback_context: CallbackContext): |
| 337 | # Artifact 2 should be available after the tool agent. |
| 338 | artifact_2 = await callback_context.load_artifact('artifact_2') |
| 339 | await callback_context.save_artifact( |
| 340 | 'artifact_3', Part.from_text(text=artifact_2.text + ' 3') |
| 341 | ) |
| 342 | |
| 343 | mock_model = testing_utils.MockModel.create( |
| 344 | responses=[function_call_no_schema, 'response2'] |
nothing calls this directly
no test coverage detected