fix(context): scale replay cap with context window

2026-06-29 14:23:55 +08:00
parent dacc699293
commit 40282e3b74
5 changed files with 83 additions and 11 deletions
@@ -236,10 +236,12 @@ class TestModifyRestricted:

    @pytest.mark.asyncio
    async def test_modify_context_window_valid(self):
-        tool = _make_tool()
+        loop = _make_mock_loop(_sync_replay_max_messages=MagicMock())
+        tool = _make_tool(runtime_state=loop)
        result = await tool.execute(action="set", key="context_window_tokens", value=131072)
        assert "Set context_window_tokens" in result
-        assert tool._runtime_state.context_window_tokens == 131072
+        assert loop.context_window_tokens == 131072
+        loop._sync_replay_max_messages.assert_called_once_with()

    @pytest.mark.asyncio
    async def test_modify_none_value_for_restricted_int(self):