diff --git a/skynet/brain.py b/skynet/brain.py index 1d8d8a7..6ec745c 100644 --- a/skynet/brain.py +++ b/skynet/brain.py @@ -406,8 +406,8 @@ async def run_skynet( own_cert_string=tls_cert_data) with ( - pynng.Rep0() as rpc_sock, - pynng.Bus0() as dgpu_bus + pynng.Rep0(recv_max_size=0) as rpc_sock, + pynng.Bus0(recv_max_size=0) as dgpu_bus ): async with open_database_connection( db_user, db_pass, db_host) as db_pool: diff --git a/skynet/dgpu.py b/skynet/dgpu.py index d002d59..1186630 100644 --- a/skynet/dgpu.py +++ b/skynet/dgpu.py @@ -177,7 +177,7 @@ async def open_dgpu_node( ca_string=skynet_cert_data) logging.info(f'connecting to {dgpu_address}') - with pynng.Bus0() as dgpu_sock: + with pynng.Bus0(recv_max_size=0) as dgpu_sock: dgpu_sock.tls_config = tls_config dgpu_sock.dial(dgpu_address) diff --git a/skynet/frontend/__init__.py b/skynet/frontend/__init__.py index f238186..06a9623 100644 --- a/skynet/frontend/__init__.py +++ b/skynet/frontend/__init__.py @@ -70,7 +70,7 @@ async def open_skynet_rpc( own_cert_string=tls_cert_data, ca_string=skynet_cert_data) - with pynng.Req0() as sock: + with pynng.Req0(recv_max_size=0) as sock: if security: sock.tls_config = tls_config