adding a max size parameter to the connection function

surrealdb/async_surrealdb.py

@@ -69,9 +69,9 @@ async def __aenter__(self):
     async def __aexit__(self, *args):
         await self.close()
 
-    async def connect(self) -> Self:
+    async def connect(self, max_size: int = 1024) -> Self:
         """Connect to SurrealDB."""
-        await self.__connection.connect()
+        await self.__connection.connect(max_size=max_size)
         return self
 
     async def close(self):

surrealdb/connection.py

@@ -120,7 +120,7 @@ async def use(self, namespace: str, database: str) -> None:
         """
         raise NotImplementedError("use method must be implemented")
 
-    async def connect(self) -> None:
+    async def connect(self, max_size: int = 2024) -> None:
         """
         Establish a connection to the server.
         """

surrealdb/connection_http.py

@@ -25,7 +25,7 @@ async def unset(self, key: str):
             if self._request_variables.get(key) is not None:
                 del self._request_variables[key]
 
-    async def connect(self) -> None:
+    async def connect(self, _max_size: int = 1024) -> None:
         if self._base_url is None:
             raise SurrealDbConnectionError("base url not set for http connection")
 

surrealdb/connection_ws.py

@@ -13,10 +13,12 @@ class WebsocketConnection(Connection):
     _ws: ClientConnection
     _receiver_task: Task
 
-    async def connect(self):
+    async def connect(self, max_size: int = 1024) -> None:
         try:
             self._ws = await connect(
-                self._base_url + "/rpc", subprotocols=[Subprotocol("cbor")]
+                self._base_url + "/rpc",
+                subprotocols=[Subprotocol("cbor")],
+                max_size=max_size,
             )
             self._receiver_task = asyncio.create_task(self.listen_to_ws(self._ws))
         except Exception as e:

surrealdb/surrealdb.py

@@ -70,10 +70,12 @@ def __enter__(self):
     def __exit__(self, *args):
         self.close()
 
-    def connect(self) -> Self:
+    def connect(self, max_size: int = 1024) -> Self:
         """Connect to SurrealDB."""
         loop_manager = AsyncioRuntime()
-        loop_manager.loop.run_until_complete(self.__connection.connect())
+        loop_manager.loop.run_until_complete(
+            self.__connection.connect(max_size=max_size)
+        )
 
         return self