Cassadra - Connection closed exception - datastax driver core exceptions transportexception connection has been closed

Hi Experts,

We are facing an issue in org, when we are querying for a small dataset it’s working but for one item there is a large json being pulled. Then we are facing a connection closed exception.

400 records - working fine
2000 records - we are facing an issue

We are deploying it on a private cloud. Any configuration changes do we need to do to get this working?

caused by com datastax driver core exceptions transportexception connection has been closed

It is due go live soon.