Skip to content

Commit 8be1e92

Browse files
committed
chore: fix test
1 parent 4590479 commit 8be1e92

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

tests/test_tool_loop_agent_runner.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -600,17 +600,17 @@ async def test_same_tool_consecutive_results_include_escalating_guidance(
600600

601601
tool_contents = [str(message.content) for message in tool_messages]
602602
runner_cls = type(runner)
603-
level_1_notice = runner_cls.SAME_TOOL_NOTICE_LEVEL_1_TEMPLATE.format(
603+
level_1_notice = runner_cls.REPEATED_TOOL_NOTICE_L1_TEMPLATE.format(
604604
tool_name="test_tool",
605-
streak=runner_cls.SAME_TOOL_NOTICE_LEVEL_1_THRESHOLD,
605+
streak=runner_cls.REPEATED_TOOL_NOTICE_L1_THRESHOLD,
606606
)
607-
level_2_notice = runner_cls.SAME_TOOL_NOTICE_LEVEL_2_TEMPLATE.format(
607+
level_2_notice = runner_cls.REPEATED_TOOL_NOTICE_L2_TEMPLATE.format(
608608
tool_name="test_tool",
609-
streak=runner_cls.SAME_TOOL_NOTICE_LEVEL_2_THRESHOLD,
609+
streak=runner_cls.REPEATED_TOOL_NOTICE_L2_THRESHOLD,
610610
)
611-
level_3_notice = runner_cls.SAME_TOOL_NOTICE_LEVEL_3_TEMPLATE.format(
611+
level_3_notice = runner_cls.REPEATED_TOOL_NOTICE_L3_TEMPLATE.format(
612612
tool_name="test_tool",
613-
streak=runner_cls.SAME_TOOL_NOTICE_LEVEL_3_THRESHOLD,
613+
streak=runner_cls.REPEATED_TOOL_NOTICE_L3_THRESHOLD,
614614
)
615615

616616
assert level_1_notice not in tool_contents[0]
@@ -664,13 +664,13 @@ async def test_same_tool_streak_resets_after_switching_tools(
664664

665665
tool_contents = [str(message.content) for message in tool_messages]
666666
runner_cls = type(runner)
667-
level_1_notice = runner_cls.SAME_TOOL_NOTICE_LEVEL_1_TEMPLATE.format(
667+
level_1_notice = runner_cls.REPEATED_TOOL_NOTICE_L1_TEMPLATE.format(
668668
tool_name="test_tool",
669-
streak=runner_cls.SAME_TOOL_NOTICE_LEVEL_1_THRESHOLD,
669+
streak=runner_cls.REPEATED_TOOL_NOTICE_L1_THRESHOLD,
670670
)
671-
level_2_notice = runner_cls.SAME_TOOL_NOTICE_LEVEL_2_TEMPLATE.format(
671+
level_2_notice = runner_cls.REPEATED_TOOL_NOTICE_L2_TEMPLATE.format(
672672
tool_name="test_tool",
673-
streak=runner_cls.SAME_TOOL_NOTICE_LEVEL_2_THRESHOLD,
673+
streak=runner_cls.REPEATED_TOOL_NOTICE_L2_THRESHOLD,
674674
)
675675

676676
assert level_1_notice not in tool_contents[0]

0 commit comments

Comments
 (0)