changed partition session to use partition proxy instead of connection