From 3a2c9fe21db3e068f9c954b99c220988e59471e9 Mon Sep 17 00:00:00 2001 From: Victor Skvortsov Date: Thu, 29 May 2025 14:09:03 +0500 Subject: [PATCH] Increase upload_code client timeout --- src/dstack/_internal/server/services/runner/client.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/dstack/_internal/server/services/runner/client.py b/src/dstack/_internal/server/services/runner/client.py index d8aae611c..e205990aa 100644 --- a/src/dstack/_internal/server/services/runner/client.py +++ b/src/dstack/_internal/server/services/runner/client.py @@ -32,6 +32,7 @@ from dstack._internal.utils.logging import get_logger REQUEST_TIMEOUT = 9 +UPLOAD_CODE_REQUEST_TIMEOUT = 60 logger = get_logger(__name__) @@ -109,7 +110,9 @@ def submit_job( resp.raise_for_status() def upload_code(self, file: Union[BinaryIO, bytes]): - resp = requests.post(self._url("/api/upload_code"), data=file, timeout=REQUEST_TIMEOUT) + resp = requests.post( + self._url("/api/upload_code"), data=file, timeout=UPLOAD_CODE_REQUEST_TIMEOUT + ) resp.raise_for_status() def run_job(self):