kaxil commented on code in PR #74404:
URL: https://github.com/apache/airflow/pull/74404#discussion_r4263978344
##########
providers/common/ai/tests/unit/common/ai/tools/test_strands.py:
##########
@@ -188,6 +222,25 @@ def lookup(key: str) -> str:
assert str(agent("go")).startswith("error:")
assert str(agent("go again")).startswith("error:")
+ def test_resuming_from_a_checkpoint_keeps_the_retry_budget(self):
+ """With checkpointing every cycle is its own invocation; the budget
must still span the run."""
+
+ def lookup(key: str) -> str:
+ """Look a key up."""
Review Comment:
I'd keep it: the same tool in
`test_each_run_of_a_reused_agent_starts_with_a_fresh_retry_budget` just above
has the same docstring.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
