feat: make read timeout configurable

Author: mpetazzoni

wherobots/db/connection.py

@@ -49,8 +49,13 @@ class Connection:
     corresponding query state. Queries are tracked by their unique execution ID.
     """
 
-    def __init__(self, ws: websockets.sync.client.ClientConnection):
+    def __init__(
+        self,
+        ws: websockets.sync.client.ClientConnection,
+        read_timeout: float = DEFAULT_READ_TIMEOUT_SECONDS,
+    ):
         self.__ws = ws
+        self.__read_timeout = read_timeout
         self.__queries: dict[str, Query] = {}
         self.__thread = threading.Thread(
             target=self.__main_loop, daemon=True, name="wherobots-connection"
@@ -168,7 +173,7 @@ def __send(self, message: dict[str, Any]) -> None:
         self.__ws.send(json.dumps(message))
 
     def __recv(self) -> dict[str, Any]:
-        frame = self.__ws.recv(timeout=DEFAULT_READ_TIMEOUT_SECONDS)
+        frame = self.__ws.recv(timeout=self.__read_timeout)
         if isinstance(frame, str):
             message = json.loads(frame)
         elif isinstance(frame, bytes):

wherobots/db/constants.py

@@ -4,13 +4,13 @@
 from .runtime import Runtime
 
 
-DEFAULT_ENDPOINT = "api.wherobots.services"  # "api.cloud.wherobots.com"
-STAGING_ENDPOINT = "api.staging.wherobots.services"  # "api.staging.wherobots.com"
-DEFAULT_RUNTIME = Runtime.SEDONA
-DEFAULT_REGION = Region.AWS_US_WEST_2
-DEFAULT_READ_TIMEOUT_SECONDS = 0.5
-DEFAULT_SESSION_WAIT_TIMEOUT_SECONDS = 300
-MAX_MESSAGE_SIZE = 100 * 2**20  # 100MiB
+DEFAULT_ENDPOINT: str = "api.wherobots.services"  # "api.cloud.wherobots.com"
+STAGING_ENDPOINT: str = "api.staging.wherobots.services"  # "api.staging.wherobots.com"
+DEFAULT_RUNTIME: Runtime = Runtime.SEDONA
+DEFAULT_REGION: Region = Region.AWS_US_WEST_2
+DEFAULT_READ_TIMEOUT_SECONDS: float = 0.25
+DEFAULT_SESSION_WAIT_TIMEOUT_SECONDS: float = 300
+MAX_MESSAGE_SIZE: int = 100 * 2**20  # 100MiB
 
 
 class ExecutionState(StrEnum):