Add getitem support for QueryChain