diff --git a/vlmrun/client/agent.py b/vlmrun/client/agent.py index 7c68736..856121e 100644 --- a/vlmrun/client/agent.py +++ b/vlmrun/client/agent.py @@ -25,7 +25,7 @@ def __init__(self, client: "VLMRunProtocol") -> None: client: VLM Run API instance """ self._client = client - self._requestor = APIRequestor(client) + self._requestor = APIRequestor(client, timeout=client.timeout) def get( self, diff --git a/vlmrun/client/base_requestor.py b/vlmrun/client/base_requestor.py index 0548e61..eb86ac9 100644 --- a/vlmrun/client/base_requestor.py +++ b/vlmrun/client/base_requestor.py @@ -27,7 +27,7 @@ ) # Constants -DEFAULT_TIMEOUT = 30.0 # seconds +DEFAULT_TIMEOUT = 120.0 # seconds DEFAULT_MAX_RETRIES = 5 INITIAL_RETRY_DELAY = 1 # seconds MAX_RETRY_DELAY = 10 # seconds