test(agent): cover tool call streak argument changes

Author: RhoninSeiei

tests/test_tool_loop_agent_runner.py

@@ -261,9 +261,10 @@ async def text_chat(self, **kwargs) -> LLMResponse:
 
 
 class SequentialToolProvider(MockProvider):
-    def __init__(self, tool_sequence: list[str]):
+    def __init__(self, tool_sequence: list[str], *, same_args: bool = False):
         super().__init__()
         self.tool_sequence = tool_sequence
+        self.same_args = same_args
 
     async def text_chat(self, **kwargs) -> LLMResponse:
         self.call_count += 1
@@ -276,11 +277,16 @@ async def text_chat(self, **kwargs) -> LLMResponse:
             )
 
         tool_name = self.tool_sequence[self.call_count - 1]
+        tool_args = (
+            {"query": "same"}
+            if self.same_args
+            else {"query": f"step-{self.call_count}"}
+        )
         return LLMResponse(
             role="assistant",
             completion_text="",
             tools_call_name=[tool_name],
-            tools_call_args=[{"query": f"step-{self.call_count}"}],
+            tools_call_args=[tool_args],
             tools_call_ids=[f"call_{self.call_count}"],
             usage=TokenUsage(input_other=10, output=5),
         )
@@ -734,7 +740,7 @@ async def test_same_tool_consecutive_results_include_escalating_guidance(
 ):
     runner_cls = type(runner)
     total_calls = runner_cls.REPEATED_TOOL_NOTICE_L3_THRESHOLD
-    provider = SequentialToolProvider(["test_tool"] * total_calls)
+    provider = SequentialToolProvider(["test_tool"] * total_calls, same_args=True)
     tool = FunctionTool(
         name="test_tool",
         description="测试工具",
@@ -797,14 +803,69 @@ async def test_same_tool_consecutive_results_include_escalating_guidance(
             assert level_3_notice in content
 
 
+@pytest.mark.asyncio
+async def test_same_tool_streak_resets_when_arguments_change(
+    runner, mock_tool_executor, mock_hooks
+):
+    runner_cls = type(runner)
+    total_calls = runner_cls.REPEATED_TOOL_NOTICE_L3_THRESHOLD
+    provider = SequentialToolProvider(["test_tool"] * total_calls)
+    tool = FunctionTool(
+        name="test_tool",
+        description="测试工具",
+        parameters={"type": "object", "properties": {"query": {"type": "string"}}},
+        handler=AsyncMock(),
+    )
+    request = ProviderRequest(
+        prompt="请连续执行不同参数的工具",
+        func_tool=ToolSet(tools=[tool]),
+        contexts=[],
+    )
+
+    await runner.reset(
+        provider=provider,
+        request=request,
+        run_context=ContextWrapper(context=None),
+        tool_executor=mock_tool_executor,
+        agent_hooks=mock_hooks,
+        streaming=False,
+    )
+
+    async for _ in runner.step_until_done(total_calls + 1):
+        pass
+
+    tool_messages = [
+        m for m in runner.run_context.messages if getattr(m, "role", None) == "tool"
+    ]
+    assert len(tool_messages) == total_calls
+
+    tool_contents = [str(message.content) for message in tool_messages]
+    notices = [
+        runner_cls.REPEATED_TOOL_NOTICE_L1_TEMPLATE.format(
+            tool_name="test_tool",
+            streak=runner_cls.REPEATED_TOOL_NOTICE_L1_THRESHOLD,
+        ),
+        runner_cls.REPEATED_TOOL_NOTICE_L2_TEMPLATE.format(
+            tool_name="test_tool",
+            streak=runner_cls.REPEATED_TOOL_NOTICE_L2_THRESHOLD,
+        ),
+        runner_cls.REPEATED_TOOL_NOTICE_L3_TEMPLATE.format(
+            tool_name="test_tool",
+            streak=runner_cls.REPEATED_TOOL_NOTICE_L3_THRESHOLD,
+        ),
+    ]
+    assert all(notice not in content for content in tool_contents for notice in notices)
+
+
 @pytest.mark.asyncio
 async def test_same_tool_streak_resets_after_switching_tools(
     runner, mock_tool_executor, mock_hooks
 ):
     runner_cls = type(runner)
     repeated_after_reset = runner_cls.REPEATED_TOOL_NOTICE_L1_THRESHOLD
     provider = SequentialToolProvider(
-        ["test_tool", "other_tool", *(["test_tool"] * repeated_after_reset)]
+        ["test_tool", "other_tool", *(["test_tool"] * repeated_after_reset)],
+        same_args=True,
     )
     tool_a = FunctionTool(
         name="test_tool",